添加响应管理类

13693261870

2024-09-13 8df2806dfed8ed0fbd326a8bbc83c0f61839103b

添加响应管理类

 src/main/java/com/se/simu/controller/BaseController.java

¶Ô±ÈÐÂÎļþ
@@ -0,0 +1,48 @@
package com.se.simu.controller;

import com.se.simu.domain.R;
import lombok.extern.slf4j.Slf4j;
import org.springframework.http.HttpStatus;

/**
 * åŸºç¡€æŽ§åˆ¶å™¨
 *
 * @author WWW
 * @date   2024-07-30
 */
@Slf4j
public class BaseController {
    public <T> R<T> success(T data) {
        return new R<T>(HttpStatus.OK, data);
    }

    public <T> R<T> success(T data, String msg) {
        return new R<T>(HttpStatus.OK, data, msg);
    }

    public <T> R<T> success(T data, long count) {
        return new R<T>(HttpStatus.OK, data, count);
    }

    public <T> R<T> success(T data, long count, String msg) {
        return new R<T>(HttpStatus.OK, data, count, msg);
    }

    public <T> R<T> fail(T data) {
        return new R<T>(HttpStatus.INTERNAL_SERVER_ERROR, data);
    }

    public <T> R<T> fail(String msg, T data) {
        return new R<T>(HttpStatus.INTERNAL_SERVER_ERROR, data, msg);
    }

    public <T> R<T> fail(Exception ex) {
        log.error(ex.getMessage(), ex);
        return new R<T>(HttpStatus.INTERNAL_SERVER_ERROR, null, ex.getMessage());
    }

    public <T> R<T> fail(Exception ex, T data) {
        log.error(ex.getMessage(), ex);
        return new R<T>(HttpStatus.INTERNAL_SERVER_ERROR, data, ex.getMessage());
    }
}

 src/main/java/com/se/simu/controller/TaskController.java

¶Ô±ÈÐÂÎļþ
@@ -0,0 +1,65 @@
package com.se.simu.controller;

import com.se.simu.domain.R;
import com.se.simu.service.SedbService;
import io.swagger.annotations.*;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.*;

import javax.annotation.Resource;
import java.util.List;

/**
 * ä»»åŠ¡ç®¡ç†
 *
 * @author WWW
 * @date   2024-09-13
 */
@Api(tags = "任务管理")
@Slf4j
@RestController
@RequestMapping("/task")
@SuppressWarnings("ALL")
public class TaskController extends BaseController {
    @Resource
    SedbService sedbService;

    @ApiOperation(value = "创建任务")
    @PostMapping(value = "/createTask", produces = "application/json; charset=UTF-8")
    public R<Object> createTask() {
        try {
            String rs = sedbService.test();

            return success(rs);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }

    @ApiOperation(value = "获取任务")
    @GetMapping("/getTask")
    public R<Object> getTask(@ApiParam("任务视图类") Object vo) {
        try {
            //

            return success(null, 0);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }

    @ApiOperation(value = "删除任务")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "ids", value = "任务ID", dataType = "Integer", paramType = "query", allowMultiple = true, example = "2")
    })
    @GetMapping(value = "/delTask")
    public R<Object> delTask(@RequestParam List<Integer> ids) {
        try {
            //

            return success(0);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }
}

 src/main/java/com/se/simu/controller/WaterController.java

@@ -1,7 +1,6 @@
package com.se.simu.controller;

import com.se.simu.helper.WebHelper;
import com.se.simu.service.SedbService;
import com.se.simu.service.WaterService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
@@ -16,19 +15,16 @@
import javax.servlet.http.HttpServletResponse;

/**
 * å†…涝控制器
 * å†…涝管理
 *
 * @author WWW
 * @date   2024-07-16
 */
@Api(tags = "内涝控制器")
@Api(tags = "内涝管理")
@Slf4j
@RestController
@RequestMapping("/waterlogging")
public class WaterController {
    @Resource
    SedbService sedbService;

    @Resource
    WaterService waterService;

@@ -142,16 +138,5 @@
        }

        return true;
    }

    @ApiOperation(value = "* æµ‹è¯• *")
    @GetMapping("/test")
    public Object test() {
        try {
            return sedbService.test();
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
            return ex.getMessage();
        }
    }
}

 src/main/java/com/se/simu/domain/R.java

¶Ô±ÈÐÂÎļþ
@@ -0,0 +1,102 @@
package com.se.simu.domain;

import io.swagger.annotations.ApiModelProperty;
import org.springframework.http.HttpStatus;

/**
 * å“åº”视图类
 *
 * @author WWW
 * @date 2024-07-30
 */
@SuppressWarnings("ALL")
public class R<T> {
    @ApiModelProperty("状态码：200-正常，400-请求错误，500-服务器错误")
    private int code;

    @ApiModelProperty("消息")
    private String msg;

    @ApiModelProperty("行数")
    private long count;

    @ApiModelProperty("数据")
    private T data;

    @ApiModelProperty("时间")
    private long time;

    public R() {
        this.time = System.currentTimeMillis();
    }

    public R(HttpStatus code, T data) {
        this.data = data;
        this.code = code.value();
        this.time = System.currentTimeMillis();
        this.msg = this.code == 200 ? "成功" : "失败";
    }

    public R(HttpStatus code, T data, String msg) {
        this.msg = msg;
        this.data = data;
        this.code = code.value();
        this.time = System.currentTimeMillis();
    }

    public R(HttpStatus code, T data, long count) {
        this.count = count;
        this.data = data;
        this.code = code.value();
        this.time = System.currentTimeMillis();
        this.msg = this.code == 200 ? "成功" : "失败";
    }

    public R(HttpStatus code, T data, long count, String msg) {
        this.msg = msg;
        this.data = data;
        this.count = count;
        this.code = code.value();
        this.time = System.currentTimeMillis();
    }

    public int getCode() {
        return code;
    }

    public void setCode(int code) {
        this.code = code;
    }

    public String getMsg() {
        return msg;
    }

    public void setMsg(String msg) {
        this.msg = msg;
    }

    public long getCount() {
        return count;
    }

    public void setCount(long count) {
        this.count = count;
    }

    public T getData() {
        return data;
    }

    public void setData(T data) {
        this.data = data;
    }

    public long getTime() {
        return time;
    }

    public void setTime(long time) {
        this.time = time;
    }
}

 src/main/java/com/se/simu/service/SedbService.java

@@ -152,11 +152,13 @@
    }

    public int getCount(String token, SeDb db, SeLayer layer) {
        Map<String, Object> map = new HashMap<>(4);
        Map<String, Object> map = new HashMap<>(6);
        map.put("token", token);
        map.put("dbid", db.getDbid());
        map.put("bbox", db.getBbox());
        map.put("layerid", layer.getId());
        map.put("returnCountOnly", true);
        map.put("inSR", 4326);

        JSONObject obj = restTemplate.postForObject(host + "geo-service/entitydbdata/layer/query", map, JSONObject.class);
        if (null == obj || 200 != obj.getInt("code")) return 0;
@@ -165,7 +167,7 @@
    }

    public JSONArray query(String token, SeDb db, SeLayer layer, int start, int count) {
        Map<String, Object> map = new HashMap<>(8);
        Map<String, Object> map = new HashMap<>(9);
        map.put("token", token);
        map.put("start", start);
        map.put("count", count);
@@ -174,12 +176,13 @@
        map.put("containCount", false);
        map.put("layerid", layer.getId());
        map.put("querytype", layer.getQueryType());
        map.put("inSR", 4326);

        JSONObject obj = restTemplate.postForObject(host + "geo-service/entitydbdata/layer/query", map, JSONObject.class);
        if (null == obj || 200 != obj.getInt("code")) return null;

        JSONObject data = obj.getJSONObject("data");

        return data.getJSONArray("items");
        return data.getJSONArray("features");
    }
}