kg/src/main/java/com/zglc/kg/service/AircraftService.java

package com.zglc.kg.service;

import com.zglc.kg.base.Result;
import com.zglc.kg.dao.AircraftDao;
import com.zglc.kg.entity.AircraftEntity;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import tk.mybatis.mapper.entity.Example;

import java.util.Iterator;
import java.util.List;

@Service
public class AircraftService {

    @Autowired
    private AircraftDao aircraftDao;

    public List<AircraftEntity> listAll(){
        return aircraftDao.listAll();
    }

    public List<AircraftEntity> getAllRecord() {
        Example example = new Example(AircraftEntity.class);
        return aircraftDao.selectByExample(example);
    }

    public Result<String> add(AircraftEntity data) {
        data.setId(null);
        boolean flag = true;
        String msg = "";
        int index = aircraftDao.insertSelective(data);
        if (index == 0) {
            flag = false;
            msg = "添加信息失败!";
        } else {
            msg = "添加信息成功!";
        }

        return Result.result(flag, msg, msg);
    }

    public Result<String> addAndCheck(AircraftEntity data){
        boolean flag = true;
        String msg = "";
        Example example = new Example(AircraftEntity.class);
        example.createCriteria().andEqualTo("aircraft_type", data.getAircraft_type());
        List<AircraftEntity> lst = aircraftDao.selectByExample(example);
        if (lst.size() > 0) {
            flag = false;
            msg = "机型已存在!";
        } else {
            int index = aircraftDao.insertSelective(data);
            if (index == 0) {
                flag = false;
                msg = "添加机型失败!";
            } else {
                msg = "添加机型成功!";
            }
        }
        return Result.result(flag, msg, msg);
    }



    public Result<String> delete(List<Integer> ids) {
        boolean flag = true;
        String msg = "";
        Iterator iter = ids.iterator();
        while(iter.hasNext()) {
            Integer id = (Integer)iter.next();
            int index = aircraftDao.deleteByPrimaryKey(id);
            if (index == 0) {
                flag = false;
                msg += "信息删除失败!" + id + ";";
            } else {
                msg += "信息删除成功!" + id + ";";
            }
        }
        return Result.result(flag, msg, msg);
    }

    public Result<String> edit(AircraftEntity data) {
        boolean flag = true;
        String msg = "";
        AircraftEntity tmpData = aircraftDao.selectByPrimaryKey(data.getId());
        if (tmpData != null) {
            int index = aircraftDao.updateByPrimaryKeySelective(data);
            if (index == 0) {
                flag = false;
                msg = "信息修改失败";
            } else {
                msg = "信息修改成功";
            }
        } else {
            flag = false;
            msg = "信息不存在!";
        }
        return Result.result(flag, msg, msg);
    }

    public Result<List<AircraftEntity>> findByName(String name){
        if (name==null || "".equals(name)){
            return Result.result(false, "名称不能为空", null);
        }
        List<AircraftEntity> list = aircraftDao.findByName("%" + name + "%");
        return Result.success(list);
    }

    public AircraftEntity getAircraft(Integer id) {
        return aircraftDao.selectByPrimaryKey(id);
    }
}