package com.ruoyi.jianguan.mongodb.service.impl;

import com.ruoyi.common.core.utils.StringUtils;
import com.ruoyi.integration.api.model.ReadRealTimeMonitoringData;
import com.ruoyi.integration.api.model.SecurityDetection;
import com.ruoyi.jianguan.constant.IotConstant;
import com.ruoyi.jianguan.mongodb.service.SecurityDetectionService;
import org.springframework.beans.factory.annotation.Autowired;
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.stereotype.Service;

import java.util.Date;
import java.util.List;
import java.util.Objects;


/**
 * 心跳包应答实现类
 **/
@Service
public class SecurityDetectionServiceImpl implements SecurityDetectionService {
    @Autowired
    private MongoTemplate mongoTemplate;

    @Override
    public int create(SecurityDetection securityDetection) {
        mongoTemplate.save(securityDetection);
        return IotConstant.SUCCESS;
    }

    @Override
    public SecurityDetection findById(String id) {
        return mongoTemplate.findById(id, SecurityDetection.class);
    }

    @Override
    public List<SecurityDetection> findAll() {
        return mongoTemplate.findAll(SecurityDetection.class);
    }
    
    /**
     * 根据业务流水号查询数据
     * @param transactionSerialNumber
     * @return
     */
    @Override
    public SecurityDetection getSecurityDetection(String transactionSerialNumber) {
        List<SecurityDetection> securityDetectionList = mongoTemplate.find(new Query().addCriteria(Criteria.where("transaction_serial_number")
                .is(transactionSerialNumber)), SecurityDetection.class);
        return securityDetectionList.size() > 0 ? securityDetectionList.get(0) : null;
    }

    @Override
    public List<SecurityDetection> getRangeTimeData(Date startTime, Date endTime) {
        Query query = new Query();
        // 根据时间范围查询
        if (Objects.nonNull(startTime) && Objects.nonNull(endTime)) {
            query.addCriteria(Criteria.where("create_time").gte(startTime).lte(endTime));
        }
        List<SecurityDetection> list = mongoTemplate.find(
                query.with(Sort.by(Sort.Order.desc("create_time")))
                , SecurityDetection.class);
        return list;
    }
}