package com.zy.asrs.common.sys.controller; import com.alibaba.fastjson.JSONArray; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.zy.asrs.common.sys.entity.Host; import com.zy.asrs.common.sys.service.HostService; import com.zy.asrs.common.web.BaseController; import com.zy.asrs.framework.annotations.ManagerAuth; import com.zy.asrs.framework.common.Cools; import com.zy.asrs.framework.common.R; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.PathVariable; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RequestParam; import org.springframework.web.bind.annotation.RestController; import java.sql.Wrapper; import java.util.*; @RestController public class HostController extends BaseController { @Autowired private HostService hostService; @RequestMapping(value = "/host/{id}/auth") @ManagerAuth public R get(@PathVariable("id") Long id) { return R.ok(hostService.getById(String.valueOf(id))); } @RequestMapping(value = "/host/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 param){ LambdaQueryWrapper wrapper = new LambdaQueryWrapper<>(); return R.ok(hostService.page(new Page<>(curr, limit), wrapper)); } @RequestMapping(value = "/host/edit/auth") @ManagerAuth public R edit(Host host) { if (Cools.isEmpty(host)){ return R.error(); } if (null == host.getId()){ hostService.save(host); } else { hostService.updateById(host); } return R.ok(); } @RequestMapping(value = "/host/add/auth") @ManagerAuth public R add(Host host) { host.setStatus(1); host.setCreateTime(new Date()); hostService.save(host); return R.ok(); } @RequestMapping(value = "/host/update/auth") @ManagerAuth public R update(Host host){ if (Cools.isEmpty(host) || null==host.getId()){ return R.error(); } host.setUpdateTime(new Date()); hostService.updateById(host); return R.ok(); } @RequestMapping(value = "/host/delete/auth") @ManagerAuth public R delete(@RequestParam(value = "ids[]") Integer[] ids) { if (Cools.isEmpty(ids)) { return R.error(); } hostService.removeByIds(Arrays.asList(ids)); return R.ok(); } @RequestMapping(value = "/hostQuery/auth") @ManagerAuth public R query(String condition) { LambdaQueryWrapper wrapper = new LambdaQueryWrapper<>(); wrapper.like(Host::getName, condition); Page page = hostService.page(new Page<>(0, 10), wrapper); List> result = new ArrayList<>(); for (Host host : page.getRecords()){ Map map = new HashMap<>(); map.put("id", host.getId()); map.put("value", host.getName()); result.add(map); } return R.ok(result); } @RequestMapping(value = "/host/delete/one/auth") @ManagerAuth public R deleteOne(@RequestParam String param){ if (getUserId() != 9527) { if (getUser().getRoleId() != 2) { return R.error("权限不足"); } } Host host = JSONArray.parseObject(param, Host.class); if (Cools.isEmpty(host)){ return R.error(); } hostService.removeById(host.getId()); return R.ok(); } }