现需求：需要批量将1000个数据先查询在更新到mongodb（如果查询不到数据，则添加数据）
1：工具类BathUpdateOptions
1 import org.springframework.data.mongodb.core.query.Query;
2 import org.springframework.data.mongodb.core.query.Update;
3
4 public class BathUpdateOptions {
5
6 private Query query;
7 private Update update;
8 private boolean upsert = true;
9 private boolean multi = false;
10
11 public Query getQuery() {
12 return query;
13 }
14
15 public void setQuery(Query query) {
16 this.query = query;
17 }
18
19 public Update getUpdate() {
20 return update;
21 }
22
23 public void setUpdate(Update update) {
24 this.update = update;
25 }
26
27 public boolean isUpsert() {
28 return upsert;
29 }
30
31 public void setUpsert(boolean upsert) {
32 this.upsert = upsert;
33 }
34
35 public boolean isMulti() {
36 return multi;
37 }
38
39 public void setMulti(boolean multi) {
40 this.multi = multi;
41 }
42
43 }2：需要更新的domain
1 @Document(collection = "video_show_view")
2 public class VideoShowView {
3 //唯一值
4 private String id;
5 //节目id
6 private String cid;
7 //app播放次数
8 private String view;
9 //app虚拟播放次数
10 private String virtualViews;
11 //最后更新时间 时间戳
12 private String updateTime;
13 //get set ......
14 }3：获取BathUpdateOptions 集合
/**
* @author xuyou
* @Package com.migu.live.mao
* @Description:
* @date 2018/6/11 16:13
*/
@Repository
public class VideoShowViewMao {
@Autowired
private MongoTemplate mongoTemplate;
public BathUpdateOptions getBathUpdateOptions(VideoShowView videoShowView){
BathUpdateOptions options = new BathUpdateOptions();
Query query = new Query();
//查询条件
query.addCriteria(Criteria.where("cid").is(videoShowView.getCid()));
query.addCriteria(Criteria.where("types").is(videoShowView.getTypes()));
options.setQuery(query);
//mongodb 默认是false,只更新找到的第一条记录，如果这个参数为true,就把按条件查出来多条记录全部更新。
options.setMulti(true);
Update update = new Update();
//更新内容
update.set("view", videoShowView.getView());
update.set("updateTime", videoShowView.getUpdateTime());
options.setUpdate(update);
return options;
}
public void bathUpdate(List<BathUpdateOptions> bups){
BathUpdateUtil.bathUpdate(mongoTemplate, VideoShowView.class, bups);
}
}4：操作mongodb的工具类BathUpdateUtil
1 import java.util.ArrayList;
2 import java.util.List;
3
4 import org.springframework.dao.InvalidDataAccessApiUsageException;
5 import org.springframework.data.mongodb.core.MongoTemplate;
6 import org.springframework.data.mongodb.core.mapping.Document;
7
8 import com.migu.live.data.BathUpdateOptions;
9 import com.mongodb.BasicDBObject;
10 import com.mongodb.CommandResult;
11 import com.mongodb.DBCollection;
12 import com.mongodb.DBObject;
13
14 public class BathUpdateUtil {
15
16 /**
17 * @description:批量更新mongodb数据
18 * @author: xuyou
19 * @date: 2018年8月14日 上午11:28:29
20 */
21 public static int bathUpdate(MongoTemplate mongoTemplate, Class<?> entityClass,
22 List<BathUpdateOptions> options) {
23 String collectionName = determineCollectionName(entityClass);
24 return doBathUpdate(mongoTemplate.getCollection(collectionName),
25 collectionName, options, true);
26 }
27
28 private static String determineCollectionName(Class<?> entityClass) {
29 if (entityClass == null) {
30 throw new InvalidDataAccessApiUsageException(
31 "No class parameter provided, entity collection can't be determined!");
32 }
33 String collName = entityClass.getSimpleName();
34 if(entityClass.isAnnotationPresent(Document.class)) {
35 Document document = entityClass.getAnnotation(Document.class);
36 collName = document.collection();
37 } else {
38 collName = collName.replaceFirst(collName.substring(0, 1)
39 ,collName.substring(0, 1).toLowerCase()) ;
40 }
41 return collName;
42 }
43
44 private static int doBathUpdate(DBCollection dbCollection, String collName,
45 List<BathUpdateOptions> options, boolean ordered) {
46 DBObject command = new BasicDBObject();
47 command.put("update", collName);
48 List<BasicDBObject> updateList = new ArrayList<BasicDBObject>();
49 for (BathUpdateOptions option : options) {
50 BasicDBObject update = new BasicDBObject();
51 update.put("q", option.getQuery().getQueryObject());
52 update.put("u", option.getUpdate().getUpdateObject());
53 update.put("upsert", option.isUpsert());
54 update.put("multi", option.isMulti());
55 updateList.add(update);
56 }
57 command.put("updates", updateList);
58 command.put("ordered", ordered);
59 CommandResult commandResult = dbCollection.getDB().command(command);
60 return Integer.parseInt(commandResult.get("n").toString());
61 }
62 }5：业务代码 可根据实际需求 进行修改
1 /**
2 * @description:执行更新
3 * @param liveKeys 需要更新的集合
4 * @return void
5 * @author: xuyou
6 * @date: 2018年8月14日 上午11:44:06
7 * @throws
8 */
9 public void bathUpdateMongoDB (Set<String> liveKeys){
10 //将直播播放次数入mongoDB
11 List<BathUpdateOptions> bupsList = new ArrayList<BathUpdateOptions>();
12 for (String id : liveKeys) {
13 VideoShowView videoShowView = new VideoShowView();
14 //设置一些更新条件 此处省略
15 videoShowView.setUpdateTime(new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(new Date()));
16 BathUpdateOptions options = videoShowViewMao.getBathUpdateOptions(videoShowView);
17 bupsList.add(options);
18 if(bupsList.size() >= 1000){
19 logger.info("bupsList : {}",bupsList.size());
20 videoShowViewMao.bathUpdate(bupsList);
21 bupsList = new ArrayList<BathUpdateOptions>();
22 }
23 }
24 //TODO 更新liveList剩余少于1000的数据
25 logger.info("bupsList : {}",bupsList.size());
26 if (bupsList != null && bupsList.size() > 0) {
27 videoShowViewMao.bathUpdate(bupsList);
28 }
29 }6;pom文件
1 <!-- 对mongodb的支持 -->
2 <dependency>
3 <groupId>org.springframework.boot</groupId>
4 <artifactId>spring-boot-starter-data-mongodb</artifactId>
5 <version>2.0.2.RELEASE</version>
6 </dependency>7：配置
spring.data.mongodb.uri=mongodb://10.200.85.97:27017,10.200.85.98:27017,10.200.85.99:27017/data_consumer
