package com.zy.asrs.controller; 
 | 
  
 | 
import com.alibaba.fastjson.JSONArray; 
 | 
import com.alibaba.fastjson.JSONObject; 
 | 
import com.baomidou.mybatisplus.mapper.EntityWrapper; 
 | 
import com.baomidou.mybatisplus.mapper.Wrapper; 
 | 
import com.baomidou.mybatisplus.plugins.Page; 
 | 
import com.core.annotations.ManagerAuth; 
 | 
import com.core.common.BaseRes; 
 | 
import com.core.common.Cools; 
 | 
import com.core.common.DateUtils; 
 | 
import com.core.common.R; 
 | 
import com.zy.asrs.entity.StaDesc; 
 | 
import com.zy.asrs.entity.param.StaDescInitParam; 
 | 
import com.zy.asrs.mapper.StaDescMapper; 
 | 
import com.zy.asrs.service.StaDescService; 
 | 
import com.zy.common.web.BaseController; 
 | 
import org.apache.ibatis.session.ExecutorType; 
 | 
import org.apache.ibatis.session.SqlSession; 
 | 
import org.apache.ibatis.session.SqlSessionFactory; 
 | 
import org.slf4j.Logger; 
 | 
import org.slf4j.LoggerFactory; 
 | 
import org.springframework.beans.factory.annotation.Autowired; 
 | 
import org.springframework.web.bind.annotation.*; 
 | 
  
 | 
import java.util.*; 
 | 
  
 | 
@RestController 
 | 
public class StaDescController extends BaseController { 
 | 
  
 | 
    private static final Logger log = LoggerFactory.getLogger(StaDescController.class); 
 | 
    @Autowired 
 | 
    private StaDescService staDescService; 
 | 
    @Autowired 
 | 
    private SqlSessionFactory sqlSessionFactory; 
 | 
  
 | 
    @RequestMapping(value = "/staDesc/init/auth") 
 | 
    @ManagerAuth(memo = "初始化站点路径") 
 | 
    public R init(StaDescInitParam param) { 
 | 
        try { 
 | 
            if (param.getTypeDesc() == 1) { 
 | 
                staDescService.delete(new EntityWrapper<>()); 
 | 
            } 
 | 
            String[] startStaList = param.getStartStaList().split(";"); 
 | 
            String[] endStaList = param.getEndStaList().split(";"); 
 | 
            List<StaDesc> staDescList = new ArrayList<>(); 
 | 
            SqlSession sqlSession = sqlSessionFactory.openSession(ExecutorType.BATCH, false); 
 | 
            StaDescMapper sqlSessionMapper = sqlSession.getMapper(StaDescMapper.class); 
 | 
            for (String startSta : startStaList) { 
 | 
                for (String endSta : endStaList) { 
 | 
                    for (Integer type : param.getType()) { 
 | 
                        int sameRes = staDescService.selectCount(new EntityWrapper<StaDesc>() 
 | 
                                .eq("type_no", type) 
 | 
                                .eq("stn_no", Integer.parseInt(startSta)) 
 | 
                                .eq("crn_no", param.getCrnNo()) 
 | 
                                .eq("crn_stn", Integer.parseInt(endSta))); 
 | 
                        if (sameRes > 0) { 
 | 
                            continue; 
 | 
                        } 
 | 
                        StaDesc staDesc = new StaDesc(); 
 | 
                        staDesc.setCrnNo(param.getCrnNo()); 
 | 
                        staDesc.setTypeNo(type); 
 | 
                        staDesc.setStnNo(Integer.parseInt(startSta)); 
 | 
                        staDesc.setCrnStn(Integer.parseInt(endSta)); 
 | 
                        staDesc.setModiUser(getUserId()); 
 | 
                        staDesc.setModiTime(new Date()); 
 | 
                        staDesc.setAppeUser(getUserId()); 
 | 
                        staDesc.setAppeTime(new Date()); 
 | 
//                        staDescList.add(staDesc); 
 | 
                        sqlSessionMapper.insert(staDesc); 
 | 
  
 | 
                    } 
 | 
                } 
 | 
            } 
 | 
            try { 
 | 
                sqlSession.commit(); 
 | 
                sqlSession.close(); 
 | 
            } catch (Exception e) { 
 | 
                log.error("初始化站点路径异常===>sql异常:{}", e.getMessage()); 
 | 
            } 
 | 
//            staDescService.insertBatch(staDescList); 
 | 
        } catch (Exception e) { 
 | 
            log.error("初始化站点路径异常:{}", e.getMessage()); 
 | 
            return R.error("初始化站点路径异常:" + e.getMessage()); 
 | 
        } 
 | 
  
 | 
        return R.ok("ok1").add("ok2"); 
 | 
    } 
 | 
  
 | 
    @RequestMapping(value = "/staDesc/{id}/auth") 
 | 
    @ManagerAuth 
 | 
    public R get(@PathVariable("id") String id) { 
 | 
        return R.ok(staDescService.selectById(String.valueOf(id))); 
 | 
    } 
 | 
  
 | 
    @RequestMapping(value = "/staDesc/list/auth") 
 | 
    @ManagerAuth 
 | 
    public R list(@RequestParam(defaultValue = "1") Integer curr, 
 | 
                  @RequestParam(defaultValue = "10") Integer limit, 
 | 
                  @RequestParam(required = false) String orderByField, 
 | 
                  @RequestParam(required = false) String orderByType, 
 | 
                  @RequestParam Map<String, Object> param) { 
 | 
        excludeTrash(param); 
 | 
        EntityWrapper<StaDesc> wrapper = new EntityWrapper<>(); 
 | 
        convert(param, wrapper); 
 | 
        if (!Cools.isEmpty(orderByField)) { 
 | 
            wrapper.orderBy(humpToLine(orderByField), "asc".equals(orderByType)); 
 | 
        } 
 | 
        return R.ok(staDescService.selectPage(new Page<>(curr, limit), wrapper)); 
 | 
    } 
 | 
  
 | 
    private <T> void convert(Map<String, Object> map, EntityWrapper<T> wrapper) { 
 | 
        for (Map.Entry<String, Object> entry : map.entrySet()) { 
 | 
            String val = String.valueOf(entry.getValue()); 
 | 
            if (val.contains(RANGE_TIME_LINK)) { 
 | 
                String[] dates = val.split(RANGE_TIME_LINK); 
 | 
                wrapper.ge(entry.getKey(), DateUtils.convert(dates[0])); 
 | 
                wrapper.le(entry.getKey(), DateUtils.convert(dates[1])); 
 | 
            } else { 
 | 
                wrapper.like(entry.getKey(), val); 
 | 
            } 
 | 
        } 
 | 
    } 
 | 
  
 | 
    @RequestMapping(value = "/staDesc/add/auth") 
 | 
    @ManagerAuth(memo = "站点路径添加") 
 | 
    public R add(StaDesc staDesc) { 
 | 
        int sameRes = staDescService.selectCount(new EntityWrapper<StaDesc>() 
 | 
                .eq("type_no", staDesc.getTypeNo()) 
 | 
                .eq("stn_no", staDesc.getStnNo()) 
 | 
                .eq("crn_no", staDesc.getCrnNo()) 
 | 
                .eq("crn_stn", staDesc.getCrnStn())); 
 | 
        if (sameRes > 0) { 
 | 
            return R.error("路径已存在"); 
 | 
        } 
 | 
        staDesc.setModiUser(getUserId()); 
 | 
        staDesc.setModiTime(new Date()); 
 | 
        staDesc.setAppeUser(getUserId()); 
 | 
        staDesc.setAppeTime(new Date()); 
 | 
        staDescService.insert(staDesc); 
 | 
        return R.ok(); 
 | 
    } 
 | 
  
 | 
    @RequestMapping(value = "/staDesc/update/auth") 
 | 
    @ManagerAuth(memo = "站点路径修改") 
 | 
    public R update(StaDesc staDesc) { 
 | 
        if (Cools.isEmpty(staDesc) || null == staDesc.getTypeId()) { 
 | 
            return R.error(); 
 | 
        } 
 | 
        staDesc.setModiUser(getUserId()); 
 | 
        staDesc.setModiTime(new Date()); 
 | 
        staDescService.updateById(staDesc); 
 | 
        return R.ok(); 
 | 
    } 
 | 
  
 | 
    @RequestMapping(value = "/staDesc/delete/auth") 
 | 
    @ManagerAuth(memo = "站点路径删除") 
 | 
    public R delete(@RequestParam String param) { 
 | 
        List<StaDesc> list = JSONArray.parseArray(param, StaDesc.class); 
 | 
        if (Cools.isEmpty(list)) { 
 | 
            return R.error(); 
 | 
        } 
 | 
        for (StaDesc entity : list) { 
 | 
            staDescService.delete(new EntityWrapper<>(entity)); 
 | 
        } 
 | 
        return R.ok(); 
 | 
    } 
 | 
  
 | 
    @RequestMapping(value = "/staDesc/export/auth") 
 | 
    @ManagerAuth(memo = "站点路径导出") 
 | 
    public R export(@RequestBody JSONObject param) { 
 | 
        List<String> fields = JSONObject.parseArray(param.getJSONArray("fields").toJSONString(), String.class); 
 | 
        EntityWrapper<StaDesc> wrapper = new EntityWrapper<>(); 
 | 
        Map<String, Object> map = excludeTrash(param.getJSONObject("staDesc")); 
 | 
        convert(map, wrapper); 
 | 
        List<StaDesc> list = staDescService.selectList(wrapper); 
 | 
        return R.ok(exportSupport(list, fields)); 
 | 
    } 
 | 
  
 | 
    @RequestMapping(value = "/staDescQuery/auth") 
 | 
    @ManagerAuth 
 | 
    public R query(String condition) { 
 | 
        EntityWrapper<StaDesc> wrapper = new EntityWrapper<>(); 
 | 
        wrapper.like("crn_no", condition); 
 | 
        Page<StaDesc> page = staDescService.selectPage(new Page<>(0, 10), wrapper); 
 | 
        List<Map<String, Object>> result = new ArrayList<>(); 
 | 
        for (StaDesc staDesc : page.getRecords()) { 
 | 
            Map<String, Object> map = new HashMap<>(); 
 | 
            map.put("id", staDesc.getCrnNo()); 
 | 
            map.put("value", staDesc.getCrnNo()); 
 | 
            result.add(map); 
 | 
        } 
 | 
        return R.ok(result); 
 | 
    } 
 | 
  
 | 
    @RequestMapping(value = "/staDesc/check/column/auth") 
 | 
    @ManagerAuth 
 | 
    public R query(@RequestBody JSONObject param) { 
 | 
        Wrapper<StaDesc> wrapper = new EntityWrapper<StaDesc>().eq(humpToLine(String.valueOf(param.get("key"))), param.get("val")); 
 | 
        if (null != staDescService.selectOne(wrapper)) { 
 | 
            return R.parse(BaseRes.REPEAT).add(getComment(StaDesc.class, String.valueOf(param.get("key")))); 
 | 
        } 
 | 
        return R.ok(); 
 | 
    } 
 | 
  
 | 
} 
 |