砂拉豆 发表于 2017-12-15 16:01:30

Scala对MongoDB的增删改查操作

  ===========================================
  原文链接: Scala对MongoDB的增删改查操作 转载请注明出处!
  ===========================================
  依赖环境:jdk1.8、Scala 2.12、idea
  mongodb Driver:3.1.1。注意,mongo for scala的驱动涉及多个jar(如下图),依赖于mongo-java-driver.jar
  这里使用的sbt管理依赖,直接在build.sbt中添加依赖:libraryDependencies += "org.mongodb" %% "casbah" % "3.1.1"(强烈建议使用该方法添加依赖)
  一、创建数据库连接
  A:不需要用户名和密码直接获取MongoDB。
  

//无权限验证连接  def createDatabase(url: String, port: Int, dbName: String): MongoDB = {
  MongoClient(url, port).getDB(dbName)
  }
  

  这里需要注意一下,在导入的jar包存在两个MongoClient类,一个来自于mongo-java-driver.jar的com.mongodb.MongoClient,另一个来自于casbah-core_2.12:3.1.1.jar的com.mongodb.casbah.MongoClient.前者是用于java调用,Scala使用后者!!!
  因此导入包的时候要注意为:import com.mongodb.casbah.MongoClient

  B:通过权限验证进行连接
  

//验证连接权限  def createDatabase(url: String, port: Int, dbName: String, loginName: String, password: String): MongoDB = {
  var server = new ServerAddress(url, port)
  //注意:MongoCredential中有6种创建连接方式,这里使用MONGODB_CR机制进行连接。如果选择错误则会发生权限验证失败
  var credentials = MongoCredential.createCredential(loginName, dbName, password.toCharArray)
  var mongoClient = MongoClient(server, List(credentials))
  mongoClient.getDB(dbName)
  }
  

  这里需要注意的是MongoCredential.createCredential(),在MongoCredential中存在着六种认证机制,这里使用createCredential()进行创建,使用错误则将会验证失败
  com.mongodb.MongoSecurityException: Exception authenticating
  Caused by: com.mongodb.MongoCommandException: Command failed with error 18: 'auth failed' on server localhost:27017. The full response is { "ok" : 0.0, "errmsg" : "auth failed", "code" : 18, "codeName" : "AuthenticationFailed" }
  该方法注释如下:
  

/**  * Creates a MongoCredential instance with an unspecified mechanism.The client will negotiate the best mechanism
  * based on the version of the server that the client is authenticating to.If the server version is 2.8 or higher,
  * the driver will authenticate using the SCRAM-SHA-1 mechanism.Otherwise, the driver will authenticate using the
  * MONGODB_CR mechanism.
  *
  * @param userName the user name
  * @param database the database where the user is defined
  * @param password the user's password
  */
  


  二、数据添加
  

def testInsert(): Unit = {for (i <- 1 to 100)  collection.insert(MongoDBObject(
"name" -> "Jack%d".format(i), "email" -> "jack%d@sina.com".format(i), "age" -> i % 25, "birthDay" -> new SimpleDateFormat("yyyy-MM-dd").parse("2016-03-25")))  }
  

  这里的collection是在下面创建的:
  

var collection= createDatabase("localhost", 27017, "mytest", "user", "123456").getCollection("user")  

  在进行数据插入的时候,如果不存在该collection则会自动创建(这里是user),如果document中不包含“_id”字段则会自动添加该字段。
  DBCollection中存在一个save方法,该save方法区别于insert的地方在于当“_id”存在于集合中,save将会进行更新数据,而insert不会进行任何操作
  根据需要选择适合的数据插入函数

  存储之后数据如下:
  { "_id" : { "$oid" : "592ad4be45aefd09f4867f1e"} , "name" : "Jack86" , "email" : "jack86@sina.com" , "age" : 11 , "birthDay" : { "$date" : "2016-03-24T16:00:00.000Z"}}
  { "_id" : { "$oid" : "592ad4be45aefd09f4867f1f"} , "name" : "Jack87" , "email" : "jack87@sina.com" , "age" : 12 , "birthDay" : { "$date" : "2016-03-24T16:00:00.000Z"}}
  { "_id" : { "$oid" : "592ad4be45aefd09f4867f20"} , "name" : "Jack88" , "email" : "jack88@sina.com" , "age" : 13 , "birthDay" : { "$date" : "2016-03-24T16:00:00.000Z"}}
  { "_id" : { "$oid" : "592ad4be45aefd09f4867f21"} , "name" : "Jack89" , "email" : "jack89@sina.com" , "age" : 14 , "birthDay" : { "$date" : "2016-03-24T16:00:00.000Z"}}
  三、数据更新修改
  A:更新方式一
  

def testUpdate(): Unit = {  var query
= MongoDBObject("name" -> "user1", "email" -> "user1@test.com")  var value
= MongoDBObject("name" -> "user1", "email" -> "user1@test.com123456")  println(
"=========更新之前============")  var query02
= MongoDBObject("name" -> "user1")  collection.find(query02).forEach(x
=> println(x))  

//    query:根据此条件进行查询value:把查询出来结果集的第一条数据设置为value  
    collection.update(query,value)
  println("=========更新之后============")
  collection.find(query02).forEach(x => println(x))
  }
  

  运行结果如下:(注意这里只更新了一条记录,并不是把所有符合结果的数据都进行更新)

  B:更新方式二
  

def testUpdate02(): Unit = {  var query
= MongoDBObject("name" -> "user1", "email" -> "user1@test.com123456")  var value
= new BasicDBObject("$set", new BasicDBObject("email", "user1@test.com"))//   var value = MongoDBObject("$set",MongoDBObject("name" -> "user1", "email" -> "user1@test.com123"))  println("=========更新之前============")
  var query02 = MongoDBObject("name" -> "user1")
  collection.find(query02).forEach(x => println(x))
  collection.update(query, value,true, true)
  println("=========更新之后============")
  collection.find(query02).forEach(x => println(x))
  }
  

  注意该方法:collection.update(query, value,true, true)。
  第三个参数:when true, inserts a document if no document matches the update query criteria
  第四个参数:when true, updates all documents in the collection that match the update query criteria, otherwise only updates one【当该值为true时,则更新所有的结果集数据,不过前提是value必须使用“$XXX”模式进行定义】。源码如下:

  四、数据查询(建议看下这篇文章:MongoDB学习笔记(查询))
  

def testSelect(): Unit ={  println(
"=========查询所有数据===================")  collection.find().forEach(x
=> println(x))  println(
"=========查询name = “user1”同时email=“user1@test.com”===================")  collection.find(MongoDBObject(
"name" -> "user1", "email" -> "user1@test.com")).limit(3).forEach(x => println(x))//    注意此处不能使用put添加其他查询条件,因为put返回的是HashMap,此处应该使用append进行添加查询条件//var query = new BasicDBObject("name",new BasicDBObject("$in",("user145","user155"))).put("qty",new BasicDBObject("$in",(25.0,105.0)))该方法错误//    查询条件为: (name in ("user145","user155")) && (qty in (25.0,105.0))  println("=========查询 (name in (\"user145\",\"user155\")) && (qty in (25.0,105.0))===================")
  var query = new BasicDBObject("name", new BasicDBObject("$in", ("user145", "user155"))).append("qty", new BasicDBObject("$in", (25.0, 105.0)))
  collection.find(query).forEach(x => println(x))
  println("=========查询 start >= 10 && end<= 80 的数据===================")
  var query02 = new BasicDBObject("start", new BasicDBObject("$gte", 10)).append("end", new BasicDBObject("$lte", 80))
  collection.find(query02).forEach(x => println(x))
  }
  

  查询结果如下:

  五、数据删除
  数据删除主要是构建一个查询条件,把符合该条件的所有数据都进行删除
  

def testDelete(): Unit ={  var query
= MongoDBObject("name" -> "user1", "email" -> "user1@test.com")  println(
"=========删除之前============")  collection.find(query).forEach(x
=> println(x))//该参数只是一个查询条件,符合该条件的所有集合都将被删除。重点在于如何构建query  
    collection.remove(query)
  println("=========删除之前============")
  collection.find(query).forEach(x => println(x))
  }
  

  执行结果如下:
  =========删除之前============
  { "_id" : { "$oid" : "592a0c8345aefd1d404c3ed9"} , "name" : "user1" , "email" : "user1@test.com"}
  { "_id" : { "$oid" : "592a0c8345aefd1d404c3eda"} , "name" : "user1" , "email" : "user1@test.com"}
  { "_id" : { "$oid" : "592a0c9845aefd0dfc2a1e21"} , "name" : "user1" , "email" : "user1@test.com"}
  { "_id" : { "$oid" : "592a0c9845aefd0dfc2a1e23"} , "name" : "user1" , "email" : "user1@test.com"}
  { "_id" : { "$oid" : "592a740f434a43d3b1529d0e"} , "name" : "user1" , "email" : "user1@test.com"}
  =========删除之后============
  remove方法有如下四种,根据具体情况选取合适的方法:
  

  

  附测试源码:
  

package tool  

  

import java.text.SimpleDateFormat  

import java.util  

import java.util.Date  

  

import com.mongodb.casbah.commons.MongoDBObject  

import com.mongodb.{BasicDBObject, DBCollection, ServerAddress}  

import com.mongodb.casbah.{MongoClient, MongoCredential, MongoDB}  

import com.mongodb.client.model.Filters  

import org.joda.time.DateTime  

  

/**
  * Created with IntelliJ>  * Description:
  * User: Perkins Zhu
  * Date: 2017-05-28
  * Time: 19:33
*/  
object MongoTool {
  def main(args: Array): Unit
= {  

//      testInsert  

//    testUpdate02  

//    testSelect  
testDelete
  }
  def testDelete(): Unit ={
  var query = MongoDBObject("name" -> "user1", "email" -> "user1@test.com")
  println("=========删除之前============")
  collection.find(query).forEach(x => println(x))
  //该参数只是一个查询条件,符合该条件的所有集合都将被删除
  
    collection.remove(query)
  collection.findAndRemove()
  println("=========删除之前============")
  collection.find(query).forEach(x => println(x))
  }
  

  

  def testUpdate01(): Unit = {
  var query = MongoDBObject("name" -> "user1", "email" -> "user1@test.com")
  var value = MongoDBObject("name" -> "user1", "email" -> "user1@test.com123456")
  println("=========更新之前============")
  var query02 = MongoDBObject("name" -> "user1")
  collection.find(query02).forEach(x => println(x))
  
//    query:根据此条件进行查询value:把查询出来结果集的第一条数据设置为value
  
    collection.update(query,value)
  println("=========更新之后============")
  collection.find(query02).forEach(x => println(x))
  }
  

  def testUpdate02(): Unit = {
  var query = MongoDBObject("name" -> "user1", "email" -> "user1@test.com123456")
  var value = new BasicDBObject("$set", new BasicDBObject("email", "user1@test.com"))
  //   var value = MongoDBObject("$set",MongoDBObject("name" -> "user1", "email" -> "user1@test.com123"))
  println("=========更新之前============")
  var query02 = MongoDBObject("name" -> "user1")
  collection.find(query02).forEach(x => println(x))
  collection.update(query, value,true, true)
  println("=========更新之后============")
  collection.find(query02).forEach(x => println(x))
  }
  

  def testSelect(): Unit ={
  println("=========查询所有数据===================")
  collection.find().forEach(x => println(x))
  println("=========查询name = “user1”同时email=“user1@test.com”===================")
  collection.find(MongoDBObject("name" -> "user1", "email" -> "user1@test.com")).limit(3).forEach(x => println(x))
  //    注意此处不能使用put添加其他查询条件,因为put返回的是HashMap,此处应该使用append进行添加查询条件
  //var query = new BasicDBObject("name",new BasicDBObject("$in",("user145","user155"))).put("qty",new BasicDBObject("$in",(25.0,105.0)))该方法错误
  //    查询条件为: (name in ("user145","user155")) && (qty in (25.0,105.0))
  println("=========查询 (name in (\"user145\",\"user155\")) && (qty in (25.0,105.0))===================")
  var query = new BasicDBObject("name", new BasicDBObject("$in", ("user145", "user155"))).append("qty", new BasicDBObject("$in", (25.0, 105.0)))
  collection.find(query).forEach(x => println(x))
  println("=========查询 start >= 10 && end<= 80 的数据===================")
  var query02 = new BasicDBObject("start", new BasicDBObject("$gte", 10)).append("end", new BasicDBObject("$lte", 80))
  collection.find(query02).forEach(x => println(x))
  }
  

  def testInsert(): Unit = {
  for (i <- 1 to 100)
  
//      注意与saved的区别
  collection.insert(MongoDBObject("name" -> "Jack%d".format(i), "email" -> "jack%d@sina.com".format(i), "age" -> i % 25, "birthDay" -> new SimpleDateFormat("yyyy-MM-dd").parse("2016-03-25")))
  }
  

  var collection = createDatabase("localhost", 27017, "mytest", "user", "123456").getCollection("user")
  

  //验证连接权限
  def createDatabase(url: String, port: Int, dbName: String, loginName: String, password: String): MongoDB = {
  var server = new ServerAddress(url, port)
  //注意:MongoCredential中有6种创建连接方式,这里使用MONGODB_CR机制进行连接。如果选择错误则会发生权限验证失败
  var credentials = MongoCredential.createCredential(loginName, dbName, password.toCharArray)
  var mongoClient = MongoClient(server, List(credentials))
  mongoClient.getDB(dbName)
  }
  

  //无权限验证连接
  def createDatabase(url: String, port: Int, dbName: String): MongoDB = {
  MongoClient(url, port).getDB(dbName)
  }
  
}
  

  --end
页: [1]
查看完整版本: Scala对MongoDB的增删改查操作