package com.moon.server.controller.data; import com.moon.server.annotation.SysLog; import com.moon.server.controller.all.BaseController; import com.moon.server.entity.all.ResponseMsg; import com.moon.server.entity.ctrl.PubEntity; import com.moon.server.entity.data.MetaEntity; import com.moon.server.entity.data.PublishEntity; import com.moon.server.entity.sys.UserEntity; import com.moon.server.helper.StringHelper; import com.moon.server.helper.WebHelper; import com.moon.server.service.data.PublishService; import com.moon.server.service.sys.TokenService; import io.swagger.annotations.Api; import io.swagger.annotations.ApiImplicitParam; import io.swagger.annotations.ApiImplicitParams; import io.swagger.annotations.ApiOperation; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.*; import javax.servlet.http.HttpServletRequest; import java.util.List; /** * 数据发布 * @author WWW */ @Api(tags = "数据管理\\发布管理") @RestController @RequestMapping("/publish") public class PublishController extends BaseController { @Autowired PublishService publishService; @Autowired TokenService tokenService; @SysLog() @ApiOperation(value = "分页查询元数据") @ApiImplicitParams({ @ApiImplicitParam(name = "depcode", value = "单位编码", dataType = "String", paramType = "query", example = "00"), @ApiImplicitParam(name = "dircode", value = "目录编码", dataType = "String", paramType = "query", example = "01"), @ApiImplicitParam(name = "verid", value = "版本ID", dataType = "Integer", paramType = "query", example = "0"), @ApiImplicitParam(name = "type", value = "类别", dataType = "String", paramType = "query", example = "DOM"), @ApiImplicitParam(name = "name", value = "名称", dataType = "String", paramType = "query", example = ""), @ApiImplicitParam(name = "pageSize", value = "每页条数", dataType = "Integer", paramType = "query", example = "10"), @ApiImplicitParam(name = "pageIndex", value = "分页数(从1开始)", dataType = "Integer", paramType = "query", example = "1") }) @GetMapping(value = "/selectMetasByPage") public ResponseMsg selectMetasByPage(String depcode, String dircode, Integer verid, String type, String name, Integer pageSize, Integer pageIndex) { try { if (pageSize < 1 || pageIndex < 1) { return fail("每页页数或分页数小于1", null); } if (StringHelper.isEmpty(type)) { return fail("数据类别为空", null); } String types = getType(type); int count = publishService.selectMetasByCount(depcode, dircode, verid, types, name); if (count == 0) { return success(0, null); } List rs = publishService.selectMetasByPage(depcode, dircode, verid, types, name, pageSize, pageSize * (pageIndex - 1)); return success(count, rs); } catch (Exception ex) { return fail(ex.getMessage(), null); } } /** * 获取类型 */ private String getType(String type) throws Exception { switch (type) { case "DOM": return "type in ('tif', 'tiff', 'img')"; case "DEM": return "type in ('tif', 'tiff')"; case "MPT": return "type = 'mpt'"; case "3DML": return "type = '3dml'"; case "BIM": return "type in ('ifc', 'fbx', 'rvt')"; default: throw new Exception("数据类型不匹配"); } } @SysLog() @ApiOperation(value = "分页查询并返回记录数") @ApiImplicitParams({ @ApiImplicitParam(name = "name", value = "名称", dataType = "String", paramType = "query", example = ""), @ApiImplicitParam(name = "pageSize", value = "每页条数", dataType = "Integer", paramType = "query", example = "10"), @ApiImplicitParam(name = "pageIndex", value = "分页数(从1开始)", dataType = "Integer", paramType = "query", example = "1") }) @GetMapping(value = "/selectByPageAndCount") public ResponseMsg> selectByPageAndCount(String name, Integer pageSize, Integer pageIndex) { try { if (pageSize < 1 || pageIndex < 1) { return fail("每页页数或分页数小于1", null); } int count = publishService.selectCount(name); if (count == 0) { return success(0, null); } List rs = publishService.selectByPage(name, pageSize, pageSize * (pageIndex - 1)); return success(count, rs); } catch (Exception ex) { return fail(ex.getMessage(), null); } } @SysLog() @ApiOperation(value = "根据ID查询") @ApiImplicitParams({ @ApiImplicitParam(name = "id", value = "ID", dataType = "int", paramType = "query", example = "1") }) @GetMapping(value = "/selectById") public ResponseMsg selectById(int id) { try { PublishEntity entity = publishService.selectById(id); return success(entity); } catch (Exception ex) { return fail(ex.getMessage(), null); } } @SysLog() @ApiOperation(value = "插入发布数据") @ApiImplicitParams({ @ApiImplicitParam(name = "entity", value = "实体类", dataType = "PubEntity", paramType = "body") }) @PostMapping(value = "/insertForPub", produces = "application/json; charset=UTF-8") public ResponseMsg insertForPub(@RequestBody PubEntity entity, HttpServletRequest req) { try { if (null == entity || null == entity.getIds() || entity.getIds().isEmpty()) { return fail("实体类为空或找不到元数据ID", 0); } if (StringHelper.isEmpty(entity.getType())) { return fail("数据类别为空", null); } UserEntity ue = tokenService.getCurrentUser(req); if (ue != null) { entity.setUserId(ue.getId()); entity.setToken(WebHelper.getToken(req)); } String method = getConvertMethod(entity.getType()); long count = publishService.postForPub(entity, method, req); return success(count); } catch (Exception ex) { return fail(ex.getMessage(), -1); } } /** * 获取转换方法 */ private String getConvertMethod(String type) throws Exception { switch (type) { case "DOM": return "/Convert/ToTiles"; case "DEM": return "/Convert/ToTerra"; case "MPT": case "3DML": return "/Convert/ToSG"; case "BIM": return "/Convert/ToTileset"; default: throw new Exception("数据类型不匹配"); } } @SysLog() @ApiOperation(value = "删除多条") @ApiImplicitParams({ @ApiImplicitParam(name = "ids", value = "ID数组", dataType = "Integer", paramType = "query", allowMultiple = true, example = "1") }) @GetMapping(value = "/deletes") public ResponseMsg deletes(@RequestParam List ids, HttpServletRequest req) { try { if (ids == null || ids.isEmpty()) { return fail("id数组不能为空", -1); } int count = publishService.deletes(ids, req); return success(count); } catch (Exception ex) { return fail(ex.getMessage(), -1); } } @SysLog() @ApiOperation(value = "更新一条") @ApiImplicitParams({ @ApiImplicitParam(name = "entity", value = "实体类", dataType = "PublishEntity", paramType = "body") }) @ResponseBody @PostMapping(value = "/update", produces = "application/json; charset=UTF-8") public ResponseMsg update(@RequestBody PublishEntity entity, HttpServletRequest req) { try { UserEntity ue = tokenService.getCurrentUser(req); if (ue != null) { entity.setUpdateUser(ue.getId()); } int count = publishService.update(entity); return success(count); } catch (Exception ex) { return fail(ex.getMessage(), -1); } } }