package com.walker.jdbc.mongo;

import com.walker.db.Sorts;
import com.walker.db.page.GenericPager;
import com.walker.db.page.PageSearch;
import com.walker.infrastructure.utils.DateUtils;
import com.walker.infrastructure.utils.NumberGenerator;
import com.walker.infrastructure.utils.StringUtils;
import org.junit.Test;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;

import java.util.List;

public class TestMongo {

//    @Test
    public void testDropTable(){
        MongoService mongoService = this.acquireMongoService();
        mongoService.dropCollection("shikeying");
    }

//    @Test
    public void testInsert(){
        String table = "shikeying";
        MongoService mongoService = this.acquireMongoService();

        mongoService.createCollection("shikeying");
        System.out.println("创建了表:" + table);

        mongoService.insert(this.acquireOne(123, "你好!"), table);
        mongoService.insert(this.acquireOne(123, "你叫什么名字？我想问问题"), table);
        mongoService.insert(this.acquireOne(123, "1"), table);
        mongoService.insert(this.acquireOne(123, "2"), table);
        mongoService.insert(this.acquireOne(123, "3"), table);
        mongoService.insert(this.acquireOne(123, "4"), table);
        mongoService.insert(this.acquireOne(123, "5"), table);
        mongoService.insert(this.acquireOne(123, "6"), table);
        mongoService.insert(this.acquireOne(123, "7"), table);
        mongoService.insert(this.acquireOne(789, "8"), table);
        mongoService.insert(this.acquireOne(789, "8"), table);

        PageSearch pageSearch = new PageSearch();
        pageSearch.setPageSize(4);
        pageSearch.setPageIndex(2);
        Query query = new Query(Criteria.where("user_id").is(123));

        GenericPager<ChatRecord> pager = mongoService.queryPageList(query, ChatRecord.class, table, pageSearch, Sorts.DESC().setField("create_time"));
        List<ChatRecord> data = pager.getDatas();
        if(!StringUtils.isEmptyList(data)){
            for(ChatRecord e : data){
                System.out.println(e.toString());
            }
        }
    }

    private ChatRecord acquireOne(long userId, String content){
        ChatRecord record = new ChatRecord();
        record.setId(String.valueOf(NumberGenerator.getLongSequenceNumber()));
        record.setCreateDate(DateUtils.getDateTimeNumber());
        record.setUserId(userId);
        record.setMessage(content);
        return record;
    }

    private MongoService acquireMongoService(){
        MongoProperty property = new MongoProperty();
        property.setIp("116.198.40.76");
        property.setUserName("czt_ops_test");
        property.setPassword("czt_ops_test");
        property.setDatabase("czt_ops_test");

        MongoTemplateFactory factory = new MongoTemplateFactory();
        MongoTemplate mongoTemplate = factory.createTemplate(property);
        MongoService mongoService = new MongoService();
        mongoService.setMongoTemplate(mongoTemplate);
        return mongoService;
    }
}