package com.ld.igds.warn.service;

import com.bstek.bdf2.core.orm.hibernate.HibernateDao;
import com.bstek.dorado.data.provider.Page;
import com.ld.igds.constant.Constant;
import com.ld.igds.models.AiSer;
import com.ld.igds.util.ContextUtil;
import com.ld.igds.util.RedisUtil;
import org.apache.commons.lang3.StringUtils;
import org.hibernate.Query;
import org.hibernate.Session;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import java.util.*;

/**
 * @author
 */
@Component
public class AiSerService extends HibernateDao {

    @Autowired
    private RedisUtil redisUtil;

    public void pageAllAiSer(Page<AiSer> page, Map<String, Object> param) throws Exception {
        String hql = " from " + AiSer.class.getName() + " where companyId =:companyId and deptId =:deptId ";

        Map<String, Object> args = new HashMap<String, Object>();
        args.put("companyId", ContextUtil.getCompanyId());
        args.put("deptId", ContextUtil.subDeptId(null));

        if(param != null){
            String str = (String) param.get("name");
            if (StringUtils.isNotEmpty(str)) {
                hql += " and name like:name";
                args.put("name", "%" + str + "%");
            }

            str = (String) param.get("type");
            if (StringUtils.isNotEmpty(str)) {
                hql += " and type =:type";
                args.put("type", str);
            }

            str = (String) param.get("id");
            if (StringUtils.isNotEmpty(str)) {
                hql += " and id like:id";
                args.put("id","%" + str + "%");
            }
        }

        String countHql = " select count(*) " + hql;
        hql += " order by id";

        this.pagingQuery(page, hql, countHql, args);
    }

    public void updateAiSer(AiSer data) {
        if (null == data.getDeptId()) {
            data.setDeptId(ContextUtil.subDeptId(null));
        }
        if (null == data.getCompanyId()) {
            data.setCompanyId(ContextUtil.getCompanyId());
        }

        Session session = null;
        try {
            session = this.getSessionFactory().openSession();
            if (null == data.getId()) {
                data.setId(ContextUtil.getUUID());
            }
            session.saveOrUpdate(data);
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            session.flush();
            session.close();
        }
    }

    public void delSerById(AiSer ser) {
        Session session = this.getSessionFactory().openSession();
        try {
            session.delete(ser);
        } finally {
            session.flush();
            session.close();
        }
    }

    public void refreshCache(String companyId, String deptId) {
        List<AiSer> list = this.getAllAiSer(companyId, deptId);
        if (null == list || list.isEmpty()) return;

        for (AiSer ser : list) {
            this.setCacheAiSer(ser);
        }
    }

    public List<AiSer> getAllAiSer(String companyId, String deptId) {
        String hql = " from " + AiSer.class.getName() + " where companyId =:companyId and deptId =:deptId order by id";
        Map<String, Object> args = new HashMap<String, Object>();
        args.put("companyId", companyId);
        args.put("deptId", deptId);
        return this.query(hql, args);
    }

    private void setCacheAiSer(AiSer snapSer) {
        if (null == snapSer.getDeptId()){
            snapSer.setDeptId("0000");
        }
        String key = Constant.APP_NAME + ":" + Constant.KEY_AI_SER + ":" + snapSer.getDeptId() + ":" + snapSer.getId();
        redisUtil.set(key, snapSer);
    }

    /**
     * 根据库区编码获取AI配置信息
     * @param deptId
     * @return
     */
    public List<AiSer> getCacheAiSerByDeptId(String deptId) {
        String patten = Constant.APP_NAME + ":" + Constant.KEY_AI_SER + ":" + deptId;
        Set<String> keys = redisUtil.keys(patten);

        if(keys == null || keys.isEmpty()){
            return null;
        }

        List<AiSer> list = new ArrayList<>();
        for (String key : keys) {
            list.add((AiSer) redisUtil.get(key));
        }
        return list;
    }

    /**
     * 根据id获取AI配置信息
     * @param id
     * @return
     */
    public AiSer getCacheAiSerById(String id) {
        String patten = Constant.APP_NAME + ":" + Constant.KEY_AI_SER;

        Set<String> keys = redisUtil.keys(patten);
        AiSer aiSer = null;
        for (String key : keys) {
            if (key.indexOf(id) >= 0) {
                aiSer = (AiSer) redisUtil.get(key);
                break;
            }
        }
        if(aiSer == null){
            List<AiSer> list = this.getAiSerById(id);
            if(null != list){
                aiSer = list.get(0);
            }
        }
        return aiSer;
    }

    public List<AiSer> getAiSerById(String id) {
        String hql = " from " + AiSer.class.getName() + " where id =:id order by id";
        Map<String, Object> args = new HashMap<String, Object>();
        args.put("id", id);
        return this.query(hql, args);
    }

    public int updateAiSerById(String status, String id) {
        Session session = null;
        int i = 0;
        try {
            session = this.getSessionFactory().openSession();

            String hql = " update " + AiSer.class.getName() + " set status =:status,updateTime =:updateTime where id =:id";
            Query query = session.createQuery(hql);
            query.setString("status", status);
            query.setTimestamp("updateTime", new Date());
            query.setString("id", id);
            i = query.executeUpdate();

        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            session.flush();
            session.close();
        }
        return i;
    }

}