Scala对MongoDB的增删改查操作

时间:2021-07-10 16:02:00

===========================================

    原文链接: 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

Scala对MongoDB的增删改查操作

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
*/

Scala对MongoDB的增删改查操作

二、数据添加

  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不会进行任何操作

根据需要选择适合的数据插入函数

Scala对MongoDB的增删改查操作

存储之后数据如下:

{ "_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))
}

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

Scala对MongoDB的增删改查操作

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”模式进行定义】。源码如下:

Scala对MongoDB的增删改查操作

四、数据查询(建议看下这篇文章: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))
}

查询结果如下:

Scala对MongoDB的增删改查操作

五、数据删除

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

  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方法有如下四种,根据具体情况选取合适的方法:

Scala对MongoDB的增删改查操作

附测试源码:

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 IDEA.
* Description:
* User: Perkins Zhu
* Date: 2017-05-28
* Time: 19:33
*/
object MongoTool {
def main(args: Array[String]): 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