wrdep/wrdep-biz/src/main/java/com/jiayue/biz/service/impl/FanModelDataServiceImpl.java

package com.jiayue.biz.service.impl;

import com.jiayue.biz.domain.FanModelData;
import com.jiayue.biz.service.FanModelDataService;
import lombok.AllArgsConstructor;
import org.springframework.beans.factory.annotation.Autowired;
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.List;
import java.util.UUID;

@Service
@AllArgsConstructor
public class FanModelDataServiceImpl implements FanModelDataService {

    private final MongoTemplate mongoTemplate;


    //查询杨总数据（风机计算数据）
    public List<FanModelData> getFanModelDataList() {
        return mongoTemplate.find(new Query(), FanModelData.class, "fan_model_data");
    }
    public List<FanModelData> queryFanModelDataList(String id) {
        if(id.equals("")){
            return mongoTemplate.find(new Query(), FanModelData.class, "fan_model_data");
        }else{
            Query query = new Query();
            Criteria criteria = new Criteria();
            // 设置条件：
            criteria.and("stationId").is(id);
            query.addCriteria(criteria);
            return mongoTemplate.find(query, FanModelData.class,"fan_model_data");
        }
    }
    // 修改
    public void updateFanModelData(FanModelData fanModelData) {
        if(fanModelData.getId().equals("")){
            fanModelData.setId(UUID.randomUUID().toString());
        }
        mongoTemplate.save(fanModelData,"fan_model_data");
    }
    // 删除
    public void deleteFanModelData(String id) {
        Criteria criteria = Criteria.where("_id").is(id);
        Query query = new Query(criteria);
        mongoTemplate.remove(query,"fan_model_data");
    }


}