import com.example.yycode.demo.mongo.model.UserInfo;
import kot.bootstarter.kotmongo.KotMongoTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Sort;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.Arrays;
import java.util.Map;

/**
 * @author YangYu
 */
@RestController
public class ExampleController {

    @Autowired
    private KotMongoTemplate kotMongoTemplate;


    @RequestMapping("/all")
    public String all() {
        String collection = "userinfo";
        final long countByCol = kotMongoTemplate.defaultSource().showSql().count(collection);
        System.err.println("count by collection: " + countByCol);

        final long countByBean = kotMongoTemplate.defaultSource().count(new UserInfo());
        System.err.println("count by bean: " + countByBean);

        final Map map = kotMongoTemplate.defaultSource().findOne(collection);
        System.err.println("findOne by collection: " + map);

        final UserInfo one = kotMongoTemplate.defaultSource().findOne(new UserInfo());
        System.err.println("findOne by bean: " + one.toString());

        kotMongoTemplate.defaultSource()
                .between("age", 0, 100) //范围查询
                .lte("score1", 92) // 小于
                .gt("score1", 90) // 大于
                .orderBy("_id").direction(Sort.Direction.DESC)// 排序
                .page(1, 10) // 分页
                .fields(Arrays.asList("userName", "age", "score1", "score2")) // 返回指定字段
                .showSql() // 显示执行命令（类似SQL）
                .findPage(new UserInfo("zhangsan156")).forEach(System.out::println);

        kotMongoTemplate.defaultSource().eq("userName", "zhangsan1").update(UserInfo.builder().userName("zhangsan125").build());

        kotMongoTemplate.defaultSource().eq("userName", "zhangsan2").delete(new UserInfo());


        // 更新文档
        kotMongoTemplate.defaultSource().eq("userName", "zhangsan1").update(UserInfo.builder().userName("zhangsan125").build());
        kotMongoTemplate.defaultSource().eq("userName", "zhangsan1").update(collection);

        // 删除文档
        kotMongoTemplate.defaultSource().eq("userName", "zhangsan2").delete(new UserInfo());
        kotMongoTemplate.defaultSource().eq("userName", "zhangsan2").delete(collection);

        // 删除集合
        kotMongoTemplate.defaultSource().dropCollection(new UserInfo());
        kotMongoTemplate.defaultSource().dropCollection(collection);

        // 多数据源，第二数据源
        final Map<String, Object> userInfo = kotMongoTemplate.secondSource().findOne(collection);
        System.out.println(userInfo);

        return "SUCCESS";
    }
}

import com.mongodb.MongoClientURI;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.data.mongodb.MongoDbFactory;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.SimpleMongoDbFactory;

import java.net.UnknownHostException;

/**
 * @author yangyu
 */
@Configuration
public class MongoConfig {

    @Bean
    @Primary
    public MongoDbFactory mongoDbFactory() throws UnknownHostException {
        return new SimpleMongoDbFactory(new MongoClientURI("mongodb://192.168.117.130:27017/yydb"));
    }

    @Bean(name = "mongoTemplate")
    @Primary
    public MongoTemplate mongoTemplate() throws Exception {
        return new MongoTemplate(mongoDbFactory());
    }

    @Bean("secondMongoDbFactory")
    public MongoDbFactory secondMongoDbFactory() throws UnknownHostException {
        return new SimpleMongoDbFactory(new MongoClientURI("mongodb://192.168.117.131:27017/yydb"));
    }

    @Bean(name = "secondMongoTemplate")
    public MongoTemplate secondMongoTemplate() throws Exception {
        return new MongoTemplate(secondMongoDbFactory());
    }
}