package com.iplatform.base.cache;

import com.iplatform.base.Constants;
import com.iplatform.base.DeptCacheProvider;
import com.iplatform.base.service.DeptServiceImpl;
import com.iplatform.base.util.DeptUtils;
import com.iplatform.model.po.S_dept;
import com.walker.cache.AbstractCacheProvider;
import com.walker.cache.Cache;
import com.walker.infrastructure.utils.StringUtils;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * 基于本地内存实现的机构缓存提供者。
 * @author 时克英
 * @date 2022-12-05
 */
public class LocalDeptCacheProvider extends AbstractCacheProvider<S_dept> implements DeptCacheProvider {

    private DeptServiceImpl deptService;

    private Map<String, List<String>> deptChildrenIdList = new ConcurrentHashMap<>();

    public void setDeptService(DeptServiceImpl deptService) {
        this.deptService = deptService;
    }

    @Override
    public void setAllowCacheChildren(boolean allow) {

    }

    @Override
    public S_dept getDept(long deptId) {
        if(deptId == 0){
            return DeptUtils.SUPER_VISOR_DEPT;
        }
        return this.getCacheData(String.valueOf(deptId));
    }

    @Override
    public void updateDept(S_dept s_dept) {
        this.updateCacheData(String.valueOf(s_dept.getId()), s_dept);
    }

    @Override
    public void removeDept(long deptId) {
        String key = String.valueOf(deptId);
        String keyChildren = key + DeptUtils.KEY_CHILDREN_PREFIX;
        S_dept current = this.getDept(deptId);
        if(current == null){
            logger.warn("机构缓存已不存在，无需删除");
            return;
        }

        // 删除该机构子机构缓存列表
        this.removeCacheData(keyChildren);
        this.removeCacheData(key);

        // 更新父机构对应的childrenList
//        String parentId = String.valueOf(current.getParent_id());
        List<String> parentChildrenList = this.getChildrenDeptIdOneLevel(current.getParent_id());
        if(parentChildrenList != null){
            parentChildrenList.remove(key);
            // 由于在内存中，所以一旦删除，引用也就变了。2022-12-03
//            this.deptChildrenIdList.put(parentId + DeptUtils.KEY_CHILDREN_PREFIX, parentChildrenList);
        }
    }

    @Override
    public void putDept(S_dept s_dept) {
        String id = String.valueOf(s_dept.getId());
        try{
            this.putCacheData(id, s_dept);
            this.deptChildrenIdList.put(id + DeptUtils.KEY_CHILDREN_PREFIX, new ArrayList<>());

            if(s_dept.getParent_id().longValue() == 0L){
                // 顶级单位，没有父机构了
                logger.debug("顶级机构，不用更新父机构列表, " + id);
                return;
            }
            // 更新父机构列表
            List<String> parentChildrenList = this.getChildrenDeptIdOneLevel(s_dept.getParent_id());
            if(StringUtils.isEmptyList(parentChildrenList)){
                parentChildrenList = new ArrayList<>();
            }
            parentChildrenList.add(id);
            this.deptChildrenIdList.put(String.valueOf(s_dept.getParent_id()) + DeptUtils.KEY_CHILDREN_PREFIX, parentChildrenList);

        } catch (Exception e){
            throw new RuntimeException("添加 [机构] 缓存错误:" + e.getMessage(), e);
        }
    }

    @Override
    public List<String> getChildrenDeptIdOneLevel(long deptId) {
        String key = deptId + DeptUtils.KEY_CHILDREN_PREFIX;
        List<String> childrenIdList = this.deptChildrenIdList.get(key);
        if(childrenIdList == null){
            logger.error("未找到缓存中子机构列表: " + key);
            return null;
        }
        return childrenIdList;
    }

    @Override
    public List<S_dept> getChildrenDeptOneLevel(long deptId) {
        List<String> childrenIdList = this.getChildrenDeptIdOneLevel(deptId);
        if(StringUtils.isEmptyList(childrenIdList)){
            return null;
        }
        List<S_dept> resultList = new ArrayList<>();
        S_dept s_dept = null;
        for(String id : childrenIdList){
            s_dept = this.getDept(Long.parseLong(id));
            resultList.add(s_dept);
        }
        return resultList;
    }

    @Override
    protected int loadDataToCache(Cache cache) {
        List<S_dept> list = this.deptService.queryAllDeptListForCache();
        if(!StringUtils.isEmptyList(list)){
            for(S_dept h : list){
                this.putDept(h);
            }
            return list.size();
        }
        return 0;
    }

    @Override
    public String getProviderName() {
        return Constants.CACHE_NAME_DEPT;
    }

    @Override
    public Class<?> getProviderType() {
        return S_dept.class;
    }
}