package cn.iocoder.byzs.module.iot.dal.mysql.thingmodel;

import cn.iocoder.byzs.framework.common.pojo.PageResult;
import cn.iocoder.byzs.framework.mybatis.core.mapper.BaseMapperX;
import cn.iocoder.byzs.framework.mybatis.core.query.LambdaQueryWrapperX;
import cn.iocoder.byzs.module.iot.controller.admin.thingmodel.vo.IotThingModelListReqVO;
import cn.iocoder.byzs.module.iot.controller.admin.thingmodel.vo.IotThingModelPageReqVO;
import cn.iocoder.byzs.module.iot.dal.dataobject.thingmodel.IotThingModelDO;
import org.apache.ibatis.annotations.Mapper;

import java.time.LocalDateTime;
import java.util.List;

/**
 * IoT 产品物模型 Mapper
 *
 * @author 博雅智算源码
 */
@Mapper
public interface IotThingModelMapper extends BaseMapperX<IotThingModelDO> {

    default PageResult<IotThingModelDO> selectPage(IotThingModelPageReqVO reqVO) {
        return selectPage(reqVO, new LambdaQueryWrapperX<IotThingModelDO>()
                .eqIfPresent(IotThingModelDO::getIdentifier, reqVO.getIdentifier())
                .likeIfPresent(IotThingModelDO::getName, reqVO.getName())
                .eqIfPresent(IotThingModelDO::getType, reqVO.getType())
                .eqIfPresent(IotThingModelDO::getProductId, reqVO.getProductId())
                // TODO @lyb：看看要不要加枚举
                .notIn(IotThingModelDO::getIdentifier, "get", "set", "post")
                .orderByDesc(IotThingModelDO::getId));
    }

    default List<IotThingModelDO> selectList(IotThingModelListReqVO reqVO) {
        return selectList(new LambdaQueryWrapperX<IotThingModelDO>()
                .eqIfPresent(IotThingModelDO::getIdentifier, reqVO.getIdentifier())
                .likeIfPresent(IotThingModelDO::getName, reqVO.getName())
                .eqIfPresent(IotThingModelDO::getType, reqVO.getType())
                .eqIfPresent(IotThingModelDO::getProductId, reqVO.getProductId())
                // TODO @lyb：看看要不要加枚举
                .notIn(IotThingModelDO::getIdentifier, "get", "set", "post")
                .orderByDesc(IotThingModelDO::getId));
    }

    default IotThingModelDO selectByProductIdAndIdentifier(Long productId, String identifier) {
        return selectOne(IotThingModelDO::getProductId, productId,
                IotThingModelDO::getIdentifier, identifier);
    }

    default List<IotThingModelDO> selectListByProductId(Long productId) {
        return selectList(IotThingModelDO::getProductId, productId);
    }

    default List<IotThingModelDO> selectListByProductKey(String productKey) {
        return selectList(IotThingModelDO::getProductKey, productKey);
    }

    default List<IotThingModelDO> selectListByProductIdAndType(Long productId, Integer type) {
        return selectList(IotThingModelDO::getProductId, productId,
                IotThingModelDO::getType, type);
    }

    default List<IotThingModelDO> selectListByProductIdAndIdentifiersAndTypes(Long productId,
                                                                              List<String> identifiers,
                                                                              List<Integer> types) {
        return selectList(new LambdaQueryWrapperX<IotThingModelDO>()
                .eq(IotThingModelDO::getProductId, productId)
                .in(IotThingModelDO::getIdentifier, identifiers)
                .in(IotThingModelDO::getType, types));
    }

    default IotThingModelDO selectByProductIdAndName(Long productId, String name) {
        return selectOne(IotThingModelDO::getProductId, productId,
                IotThingModelDO::getName, name);
    }

    // TODO @super：用不到，删除下；
    /**
     * 统计物模型数量
     *
     * @param createTime 创建时间，如果为空，则统计所有物模型数量
     * @return 物模型数量
     */
    default Long selectCountByCreateTime(LocalDateTime createTime) {
        return selectCount(new LambdaQueryWrapperX<IotThingModelDO>()
                .geIfPresent(IotThingModelDO::getCreateTime, createTime));
    }

}