package com.whyc.service;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.whyc.mapper.EnvironmentMapper;
import com.whyc.pojo.db_fire_robot.Environment;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;

@Service
public class EnvironmentService {

    @Resource
    private EnvironmentMapper mapper;

    public void add(Environment env) {
        mapper.insert(env);
    }

    public void addOrUpdate(Environment env) {
        QueryWrapper<Environment> query = Wrappers.query();
        query.last(" limit 1");
        Environment envInDB = mapper.selectOne(query);
        if (envInDB ==null) {
            mapper.insert(env);
        }else{
            env.setId(envInDB.getId());
            mapper.updateById(env);
        }
    }
}