package com.lf.server.controller.show;
|
|
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.core.metadata.OrderItem;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.lf.server.aspect.SysLog;
|
import com.lf.server.controller.all.BaseController;
|
import com.lf.server.entity.all.ResponseMsg;
|
import com.lf.server.entity.ctrl.ShpRecordEntity;
|
import com.lf.server.entity.data.MetaFileEntity;
|
import com.lf.server.entity.sys.UserEntity;
|
import com.lf.server.helper.AesHelper;
|
import com.lf.server.helper.ClassHelper;
|
import com.lf.server.helper.StringHelper;
|
import com.lf.server.mapper.all.GeomBaseMapper;
|
import com.lf.server.service.data.UploaderService;
|
import com.lf.server.service.show.InquiryService;
|
import com.lf.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 javax.servlet.http.HttpServletResponse;
|
import java.util.List;
|
|
/**
|
* 查询
|
* @author WWW
|
*/
|
@Api(tags = "综合展示\\查询")
|
@RestController
|
@RequestMapping("/inquiry")
|
public class InquiryController extends BaseController {
|
private final static int FOUR = 4;
|
|
@Autowired
|
TokenService tokenService;
|
|
@Autowired
|
UploaderService uploaderService;
|
|
@Autowired
|
InquiryService inquiryService;
|
|
@SysLog()
|
@ApiOperation(value = "查询记录数")
|
@ApiImplicitParams({
|
@ApiImplicitParam(name = "name", value = "映射名称", dataType = "String", paramType = "query", example = "dlgAgnp")
|
})
|
@GetMapping({"/selectCount"})
|
public ResponseMsg<Long> selectCount(String name) {
|
try {
|
if (StringHelper.isEmpty(name)) {
|
return fail("名称空间和实体名称不能为空", null);
|
}
|
|
Object obj = ClassHelper.getBean(name.trim() + "Mapper");
|
if (!(obj instanceof BaseMapper)) {
|
return fail("查询对象不存在", null);
|
}
|
|
BaseMapper baseMapper = (BaseMapper) obj;
|
long count = baseMapper.selectCount(null);
|
|
return success(count);
|
} catch (Exception ex) {
|
return fail(ex.getMessage(), null);
|
}
|
}
|
|
@SysLog()
|
@ApiOperation(value = "分页查询")
|
@ApiImplicitParams({
|
@ApiImplicitParam(name = "name", value = "映射名称", dataType = "String", paramType = "query", example = "dlgAgnp"),
|
@ApiImplicitParam(name = "pageIndex", value = "分页数（从1开始）", dataType = "Integer", paramType = "query", example = "1"),
|
@ApiImplicitParam(name = "pageSize", value = "每页条数", dataType = "Integer", paramType = "query", example = "10")
|
})
|
@GetMapping(value = "/selectByPage")
|
public ResponseMsg<List<Object>> selectByPage(String name, Integer pageIndex, Integer pageSize) {
|
try {
|
if (pageSize < 1 || pageIndex < 1) {
|
return fail("每页页数或分页数小于1", null);
|
}
|
if (StringHelper.isEmpty(name)) {
|
return fail("名称空间和实体名称不能为空", null);
|
}
|
|
Object obj = ClassHelper.getBean(name.trim() + "Mapper");
|
if (!(obj instanceof BaseMapper)) {
|
return fail("查询对象不存在", null);
|
}
|
|
BaseMapper baseMapper = (BaseMapper) obj;
|
Page<Object> page = new Page<>(pageIndex, pageSize);
|
page.addOrder(OrderItem.asc("gid"));
|
|
IPage<Object> paged = baseMapper.selectPage(page, null);
|
|
return success(paged.getTotal(), paged.getRecords());
|
} catch (Exception ex) {
|
return fail(ex.getMessage(), null);
|
}
|
}
|
|
@SysLog()
|
@ApiOperation(value = "根据ID查询WKT")
|
@ApiImplicitParams({
|
@ApiImplicitParam(name = "name", value = "映射名称", dataType = "String", paramType = "query", example = "dlgAgnp"),
|
@ApiImplicitParam(name = "gid", value = "GID", dataType = "Integer", paramType = "query", example = "1")
|
})
|
@GetMapping(value = "/selectWktById")
|
public ResponseMsg<String> selectWktById(String name, Integer gid) {
|
try {
|
if (StringHelper.isEmpty(name)) {
|
return fail("名称空间和实体名称不能为空", null);
|
}
|
|
Object obj = ClassHelper.getBean(name.trim() + "Mapper");
|
if (!(obj instanceof GeomBaseMapper)) {
|
return fail("查询对象不存在", null);
|
}
|
|
GeomBaseMapper baseMapper = (GeomBaseMapper) obj;
|
String wkt = baseMapper.selectWktById(gid);
|
if (!StringHelper.isEmpty(wkt)) {
|
wkt = AesHelper.encrypt(wkt);
|
}
|
|
return success(wkt);
|
} catch (Exception ex) {
|
return fail(ex.getMessage(), null);
|
}
|
}
|
|
@SysLog()
|
@ApiOperation(value = "上传Shp文件读取第一条记录的WKT")
|
@ResponseBody
|
@PostMapping(value = "/uploadShp")
|
public ResponseMsg<ShpRecordEntity> uploadShp(HttpServletRequest req, HttpServletResponse res) {
|
try {
|
UserEntity ue = tokenService.getCurrentUser(req);
|
if (ue == null) {
|
return fail("用户未登录", null);
|
}
|
|
List<MetaFileEntity> list = uploaderService.uploadData(null, ue, req, res);
|
if (list.size() < FOUR) {
|
return fail("没有找到已上传的数据或不完整", null);
|
}
|
|
ShpRecordEntity sr = inquiryService.readShpFirstRecord(list);
|
if (sr != null && !StringHelper.isEmpty(sr.getWkt())) {
|
sr.setWkt(AesHelper.encrypt(sr.getWkt()));
|
}
|
|
return success(sr);
|
} catch (Exception ex) {
|
return fail(ex.getMessage(), null);
|
}
|
}
|
}
|
