package com.njcn.rdms.module.project.dal.mysql.status;

import com.njcn.rdms.framework.mybatis.core.mapper.BaseMapperX;
import com.njcn.rdms.framework.mybatis.core.query.LambdaQueryWrapperX;
import com.njcn.rdms.module.project.dal.dataobject.status.ObjectStatusModelDO;
import org.apache.ibatis.annotations.Mapper;

import java.util.List;
import java.util.stream.Collectors;

@Mapper
public interface ObjectStatusModelMapper extends BaseMapperX<ObjectStatusModelDO> {

    default ObjectStatusModelDO selectByObjectTypeAndStatusCode(String objectType, String statusCode) {
        return selectOne(new LambdaQueryWrapperX<ObjectStatusModelDO>()
                .eq(ObjectStatusModelDO::getObjectType, objectType)
                .eq(ObjectStatusModelDO::getStatusCode, statusCode));
    }

    default ObjectStatusModelDO selectByObjectTypeAndStatusCodeEnabled(String objectType, String statusCode) {
        return selectOne(new LambdaQueryWrapperX<ObjectStatusModelDO>()
                .eq(ObjectStatusModelDO::getObjectType, objectType)
                .eq(ObjectStatusModelDO::getStatusCode, statusCode)
                .eq(ObjectStatusModelDO::getStatus, 0));
    }

    default ObjectStatusModelDO selectInitialByObjectTypeEnabled(String objectType) {
        return selectOne(new LambdaQueryWrapperX<ObjectStatusModelDO>()
                .eq(ObjectStatusModelDO::getObjectType, objectType)
                .eq(ObjectStatusModelDO::getInitialFlag, true)
                .eq(ObjectStatusModelDO::getStatus, 0));
    }

    default List<ObjectStatusModelDO> selectListByObjectType(String objectType) {
        return selectList(new LambdaQueryWrapperX<ObjectStatusModelDO>()
                .eq(ObjectStatusModelDO::getObjectType, objectType)
                .orderByAsc(ObjectStatusModelDO::getSort));
    }

    default List<ObjectStatusModelDO> selectListByObjectTypeEnabled(String objectType) {
        return selectList(new LambdaQueryWrapperX<ObjectStatusModelDO>()
                .eq(ObjectStatusModelDO::getObjectType, objectType)
                .eq(ObjectStatusModelDO::getStatus, 0)
                .orderByAsc(ObjectStatusModelDO::getSort));
    }

    /**
     * 查询某对象类型下所有已启用的终态状态码。
     */
    default List<String> selectTerminalStatusCodesByObjectTypeEnabled(String objectType) {
        return selectList(new LambdaQueryWrapperX<ObjectStatusModelDO>()
                .eq(ObjectStatusModelDO::getObjectType, objectType)
                .eq(ObjectStatusModelDO::getStatus, 0)
                .eq(ObjectStatusModelDO::getTerminalFlag, true))
                .stream()
                .map(ObjectStatusModelDO::getStatusCode)
                .collect(Collectors.toList());
    }

}
