package com.walker.jdbc.mongo;

import com.mongodb.client.MongoCollection;
import com.mongodb.client.result.DeleteResult;
import com.mongodb.client.result.UpdateResult;
import com.walker.db.Sorts;
import com.walker.db.page.GenericPager;
import com.walker.db.page.ListPageContext;
import com.walker.db.page.PageSearch;
import com.walker.infrastructure.utils.StringUtils;
import org.bson.Document;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Sort;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.mongodb.core.query.Update;
import org.springframework.lang.Nullable;

import java.util.List;

/**
 * Mongo数据库操作服务基础封装。
 * @author 时克英
 * @date 2023-07-06
 */
public class MongoService {

    public <T> MongoCollection createCollection(Class<T> entity){
        return this.mongoTemplate.createCollection(entity);
    }

    public void createCollection(String name){
        this.mongoTemplate.createCollection(name);
    }

    public <T> void dropCollection(Class<T> entity){
        this.mongoTemplate.dropCollection(entity);
    }
    public void dropCollection(String tableName){
        this.mongoTemplate.dropCollection(tableName);
    }

    public <T> void insert(T data, String tableName){
        this.mongoTemplate.insert(data, tableName);
    }

    public <T> void insert(T data){
        this.mongoTemplate.insert(data);
    }

    public <T> void insertBatch(List<T> data, Class<T> entity){
        this.mongoTemplate.insert(data, entity);
    }

    /**
     * 适合批量写入，表明是动态的，例如：聊天中每个人一张表。
     * @param data 写入记录集合
     * @param tableName 表名
     * @param <T>
     * @date 2023-07-06
     */
    public <T> void insertBatch(List<T> data, String tableName){
        this.mongoTemplate.insert(data, tableName);
    }

    public long deleteById(String id, String tableName){
        Query query = new Query(Criteria.where(KEY_ID_DEFAULT).is(id));
        DeleteResult deleteResult = this.mongoTemplate.remove(query, tableName);
        if(deleteResult != null){
            return deleteResult.getDeletedCount();
        }
        return 0;
    }
    public <T> long deleteById(String id, Class<T> entity){
        Query query = new Query(Criteria.where(KEY_ID_DEFAULT).is(id));
        DeleteResult deleteResult = this.mongoTemplate.remove(query, entity);
        if(deleteResult != null){
            return deleteResult.getDeletedCount();
        }
        return 0;
    }

    /**
     * 更新一条记录
     * @param id 主键，即：对应表中默认字段_id的值
     * @param column 更新字段名称
     * @param value 更新值
     * @param tableName 表名
     * @return
     * @date 2023-07-06
     */
    public UpdateResult updateById(String id, String column, Object value, String tableName){
        Query query = new Query(Criteria.where(KEY_ID_DEFAULT).is(id));
        Update update = new Update();
        update.set(column, value);
        UpdateResult updateResult = this.mongoTemplate.updateFirst(query, update, tableName);
        return updateResult;
    }

    public <T> UpdateResult updateById(String id, String column, Object value, Class<T> entity){
        Query query = new Query(Criteria.where(KEY_ID_DEFAULT).is(id));
        Update update = new Update();
        update.set(column, value);
        UpdateResult updateResult = this.mongoTemplate.updateFirst(query, update, entity);
        return updateResult;
    }

    /**
     * 更新一个完整对象（记录）。
     * @param id
     * @param data
     * @return
     * @param <T>
     */
    public <T> UpdateResult updateEntity(String id, T data){
        Query query = new Query(Criteria.where(KEY_ID_DEFAULT).is(id));
        Document document = (Document)mongoTemplate.getConverter().convertToMongoType(data);
        Update update = Update.fromDocument(document);
        UpdateResult updateResult = this.mongoTemplate.updateFirst(query, update, data.getClass());
        return updateResult;
    }

    public <T> UpdateResult updateEntity(String id, T data, String tableName){
        Query query = new Query(Criteria.where(KEY_ID_DEFAULT).is(id));
        Document document = (Document)mongoTemplate.getConverter().convertToMongoType(data);
        Update update = Update.fromDocument(document);
        UpdateResult updateResult = this.mongoTemplate.updateFirst(query, update, tableName);
        return updateResult;
    }

    public <T> List<T> queryList(Query query, Class<T> entity, @Nullable String tableName, @Nullable Sorts.Sort sorts){
        if(sorts != null){
            if(sorts.isAsc()){
                query.with(Sort.by(Sort.Order.asc(sorts.getField())));
            } else {
                query.with(Sort.by(Sort.Order.desc(sorts.getField())));
            }
        }
        List<T> data = null;
        if(StringUtils.isEmpty(tableName)){
            data = this.mongoTemplate.find(query, entity);
        } else {
            data = this.mongoTemplate.find(query, entity, tableName);
        }
        return data;
    }

    public <T> GenericPager<T> queryPageList(Query query, Class<T> entity, @Nullable String tableName
            , PageSearch pageSearch, @Nullable Sorts.Sort sorts){

        query.with(PageRequest.of(pageSearch.getPageIndex()-1, pageSearch.getPageSize()));
        if(sorts != null){
            if(sorts.isAsc()){
                query.with(Sort.by(Sort.Order.asc(sorts.getField())));
            } else {
                query.with(Sort.by(Sort.Order.desc(sorts.getField())));
            }
        }
        long total = 0;
        List<T> data = null;
        if(StringUtils.isEmpty(tableName)){
            total = this.mongoTemplate.count(query, entity);
            data = this.mongoTemplate.find(query, entity);
        } else {
            total = this.mongoTemplate.count(query, entity, tableName);
            data = this.mongoTemplate.find(query, entity, tableName);
        }

        return ListPageContext.createGenericPager(data, pageSearch.getPageIndex(), pageSearch.getPageSize(), (int)total);
    }

    public MongoTemplate getMongoTemplate() {
        return mongoTemplate;
    }

    public void setMongoTemplate(MongoTemplate mongoTemplate) {
        this.mongoTemplate = mongoTemplate;
    }

    private MongoTemplate mongoTemplate;

    public static final String KEY_ID_DEFAULT = "_id";
}