mongodb的BulkOperations,springboot批量插入数据时若数据库存在数据则更新,同时更新elasticSearch

// mongodb 的批量插入或者更新
            List<Pair<Query, Update>> updateList = new ArrayList<>(userTagDocs.size());

            BulkOperations operations = mongoTemplate.bulkOps(BulkOperations.BulkMode.UNORDERED, TAG_COLLECTION_NAME);

            userTagDocs.forEach(data -> {

                Query query = new Query(new
                        Criteria("userId").is(data.getUserId()).and("code").is(data.getCode()).and("type").is(data.getType()));

                Update update = new Update();

                update.set("value", data.getValue());

                Pair<Query, Update> updatePair = Pair.of(query, update);

                updateList.add(updatePair);

            });

            operations.upsert(updateList);

            BulkWriteResult result = operations.execute();
            
            portraitSearch.upsertTags(userTagBOS);
 /**
     * 批量添加标签时批量索引更新
     */
    public void upsertTags(List<UserTagBO> tagDTOS) {

        // 插入数据
        BulkRequest request = new BulkRequest();
        for (UserTagBO userTagBO :tagDTOS){
            String key = String.format("%s.%s", userTagBO.getType().name(), userTagBO.getCode());

            request.add(new IndexRequest(indexName, TYPE_NAME, userTagBO.getUserId().toString())
                    .source(XContentType.JSON, key, userTagBO.getValue()));

        }

        BulkResponse bulkResponse = null;
        try {
            bulkResponse = client.bulk(request, RequestOptions.DEFAULT);
        } catch (IOException e) {
            log.error("doc index failed, rentUnit: {}", tagDTOS, e);
            throw new RpcException("Es api error", e);
        }

    }
发布了30 篇原创文章 · 获赞 18 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/weixin_40898368/article/details/103485123