5.从零搭建WebApi接口开发框架-设计Controller

这里面主要设计以下接口:

  • 用户登录获取accessToken
  • news的增删改查接口

为了保存accessToken,需要引入redis,为了定义返回的json,需要定义公共的JsonResult,下面依次说明

pom.xml添加依赖

<dependency>
     <groupId>org.springframework.boot</groupId>
     <artifactId>spring-boot-starter-data-redis</artifactId>
</dependency>

application.yml增加redis的配置信息

#config Redis
spring:
  redis:
    host: 127.0.0.1
    port: 6379
    database: 0
    password:

定义redis调用相关类

共包括RedisUtil工具类、RedisKey公共值

@Component
public class RedisUtil {
    /**
     * 注入的stringRedisTemplate
     */
    @Autowired
    private StringRedisTemplate stringRedisTemplate;
    /**
     * redisTemplate
     */
    @Autowired
    private RedisTemplate<?, ?> redisTemplate;

    /**
     * 设置key和对象的value
     *
     * @param key key
     * @param value value
     */
    public void set(final String key, Object value) {
        final byte[] vbytes = SerializeUtil.serialize(value);
        redisTemplate.execute(new RedisCallback<Object>() {
            @Override
            public Object doInRedis(RedisConnection connection) throws DataAccessException {
                connection.set(redisTemplate.getStringSerializer().serialize(key), vbytes);
                return null;
            }
        });
    }

    /**
     * 设置key和value,并加上时间(秒)
     *
     * @param key key
     * @param value value
     * @param l 时间
     */
    public void set(final String key, Object value, final long l) {
        final byte[] vbytes = SerializeUtil.serialize(value);
        redisTemplate.execute(new RedisCallback<Object>() {
            @Override
            public Object doInRedis(RedisConnection connection) throws DataAccessException {
                connection.setEx(redisTemplate.getStringSerializer().serialize(key), l, vbytes);
                return null;
            }
        });
    }

    /**
     * 根据key来取值
     *
     * @param key key
     * @param elementType 数据类型
     * @param <T> 泛型
     * @return 获取的对象
     */
    public <T> T get(final String key, Class<T> elementType) {
        return redisTemplate.execute(new RedisCallback<T>() {
            @Override
            public T doInRedis(RedisConnection connection) throws DataAccessException {
                byte[] keybytes = redisTemplate.getStringSerializer().serialize(key);
                if (connection.exists(keybytes)) {
                    byte[] valuebytes = connection.get(keybytes);
                    T value = (T) SerializeUtil.deserialize(valuebytes);
                    return value;
                }
                return null;
            }
        });
    }

    /**
     * 根据key删除该值
     *
     * @param key key
     */
    public void del(final String key) {
        final byte[] keyBytes = redisTemplate.getStringSerializer().serialize(key);
        redisTemplate.execute(new RedisCallback<Object>() {
            @Override
            public Object doInRedis(RedisConnection connection) throws DataAccessException {
                connection.del(keyBytes);
                return null;
            }
        });
    }
}
public class RedisKey {
    //用户登录后的accesstoken
    public static final String ACCESS_TOKEN = "com:balance:accesstoken";
}

定义公共返回的json对象

public class JsonResult {
    /**
     * 是否成功
     */
    private boolean success;
    /**
     * 消息
     */
    private String message;
    /**
     * 返回码
     */
    private String code;

    /**
     * 额外的数据
     */
    private Object data;

    public String getCode() {
        return code;
    }

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

    public boolean isSuccess() {
        return success;
    }

    public void setSuccess(boolean success) {
        this.success = success;
    }

    public String getMessage() {
        return message;
    }

    public void setMessage(String message) {
        this.message = message;
    }

    public Object getData() {
        return data;
    }

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

    public JsonResult(boolean success, String code, Object data) {
        this(success, code);
        this.data = data;
        this.message = ReturnCodeUtil.getMsg(code);
    }

    public JsonResult() {
    }

    public JsonResult(boolean success) {
        this.success = success;
    }

    /**
     * message通过code获取
     *
     * @param success
     * @param code
     */
    public JsonResult(boolean success, String code) {
        this(success);
        this.code = code;
        this.message = ReturnCodeUtil.getMsg(code);
    }

    /**
     * 三个参数,message自己赋值,不通过code获取
     *
     * @param success
     * @param code
     * @param message
     */
    public JsonResult(boolean success, String code, String message) {
        this(success);
        this.code = code;
        this.message = message;
    }

}

这个类用的地方非常多,基本每个controller都会用到,一定要通用、易使用,为了定义公共返回码,还需要定义一个类存储公共的全局返回码,GlobalReturnCode

public class GlobalReturnCode {
    /**
     * 保存成功
     */
    public static final String SAVE_SUCCESS = "10001";
    /**
     * 删除成功
     */
    public static final String DELETE_SUCCESS = "10002";
    /**
     * 操作成功
     */
    public static final String OPERA_SUCCESS = "10003";
    /**
     * 审核成功
     */
    public static final String AUDIT_SUCCESS = "10004";

    /**
     * 操作失败
     */
    public static final String OPERA_FAILURE = "20001";

    /**
     * 无权限
     */
    public static final String NO_AUTH = "30001";
    /**
     * 系统错误
     */
    public static final String SYSTEM_ERROR = "30002";
    /**
     * 参数错误
     */
    public static final String PARAM_ERROR = "30003";

    /**
     * 路径不存在
     */
    public static final String SYSTEM_PATH_NOEXIST = "30004";
}
public class ReturnCodeUtil {
    //定义返回码及文字数据
    private static Map<String, String> returnCodeMap;

    static {
        returnCodeMap.put(GlobalReturnCode.SAVE_SUCCESS, "保存成功");
        returnCodeMap.put(GlobalReturnCode.DELETE_SUCCESS, "删除成功");
        returnCodeMap.put(GlobalReturnCode.OPERA_SUCCESS, "操作成功");
        returnCodeMap.put(GlobalReturnCode.AUDIT_SUCCESS, "审核成功");

        
        returnCodeMap.put(GlobalReturnCode.OPERA_FAILURE, "操作失败");
        returnCodeMap.put("20102", "账号或密码错误");


        returnCodeMap.put(GlobalReturnCode.NO_AUTH, "无权限");
        returnCodeMap.put(GlobalReturnCode.SYSTEM_ERROR, "系统错误");
        returnCodeMap.put(GlobalReturnCode.PARAM_ERROR, "参数错误");
        returnCodeMap.put(GlobalReturnCode.SYSTEM_PATH_NOEXIST, "路径不存在");

    }


    /**
     * 获取返回的中文说明
     *
     * @param resultCode 返回码
     * @return 中文名称
     */
    public static String getMsg(String resultCode) {
        if (returnCodeMap.containsKey(resultCode)) {
            return returnCodeMap.get(resultCode);
        } else {
            return "";
        }
    }
}

ReturnCodeUtil这个类定义类根据返回码获取对应的中文名的方法,实际应用中,可以把固定的返回码存储到数据库中或者配置文件中。

LoginController

@RestController
@RequestMapping("/")
public class LoginControler {
    @Autowired
    private UserService userService;
    @Autowired
    private RedisUtil redisUtil;

    /**
     * 用户登录
     */
    @PostMapping("/checkLogin")
    public JsonResult checkLogin(HttpServletRequest request, @RequestBody LoginDto loginDto) {
        User user = userService.getByUsername(loginDto.getUsername());
        if (user != null && user.getPasssword().equals(loginDto.getPassword())) {
            //登录成功
            String accessToken = UUID.randomUUID().toString();
            //放入redis
            redisUtil.set(RedisKey.ACCESS_TOKEN + accessToken, user, 30 * 60);
            Map<String, String> map = new HashMap<>();
            map.put("accessToken", accessToken);
            JsonResult jsonResult = new JsonResult(true, GlobalReturnCode.OPERA_SUCCESS, accessToken);
            return jsonResult;
        } else {
            JsonResult jsonResult = new JsonResult(false, "20102");
            return jsonResult;
        }
    }
}

这里面是登录后获取accessToken的代码,统一返回JsonResult。
要说明一点,可以利用@PostMapping来指定该请求只接收post,简化了以前的RequestMapping设置。用户账号或密码错误,返回

用户登录成功,返回accessToken


登录成功.png

账号或密码错误.png

NewsController

@RestController
@RequestMapping("/news")
public class NewsController {
    @Autowired
    private NewsService newsService;

    @PostMapping("/add")
    public JsonResult add(HttpServletRequest request, @RequestBody News news) {
        int flag = newsService.add(news);
        if (flag == 1) {
            return new JsonResult(true, GlobalReturnCode.SAVE_SUCCESS);
        } else {
            return new JsonResult(false, GlobalReturnCode.OPERA_FAILURE);
        }
    }

    @PostMapping("/update")
    public JsonResult update(HttpServletRequest request, @RequestBody News news) {
        int flag = newsService.update(news);
        if (flag == 1) {
            return new JsonResult(true, GlobalReturnCode.SAVE_SUCCESS);
        } else {
            return new JsonResult(false, GlobalReturnCode.OPERA_FAILURE);
        }
    }

    @PostMapping("/delete")
    public JsonResult delete(HttpServletRequest request, @RequestBody Map<String, String> map) {
        int flag = newsService.delete(Integer.parseInt(map.get("id")));
        if (flag == 1) {
            return new JsonResult(true, GlobalReturnCode.SAVE_SUCCESS);
        } else {
            return new JsonResult(false, GlobalReturnCode.OPERA_FAILURE);
        }
    }

    @PostMapping("/list")
    public JsonResult list(HttpServletRequest request) {
        List<News> list = newsService.list();
        return new JsonResult(true, GlobalReturnCode.OPERA_SUCCESS, list);
    }

    @PostMapping("/get")
    public JsonResult get(HttpServletRequest request, @RequestBody Map<String, String> map) {
        News news = newsService.get(Integer.parseInt(map.get("id")));
        return new JsonResult(true, GlobalReturnCode.OPERA_SUCCESS, news);
    }
}
新增news.png
根据id获取news.png

这里面比较简单,需要注意的一点就是,传参数即使只包括一个字段,也需要用@RequestBody来接收

整个controller就开发完了,是不是相当简单,后续的就是对接口调用做一些全局控制了。

源码下载

本例子详细源码

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 206,968评论 6 482
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 88,601评论 2 382
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 153,220评论 0 344
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 55,416评论 1 279
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 64,425评论 5 374
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,144评论 1 285
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,432评论 3 401
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,088评论 0 261
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,586评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,028评论 2 325
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,137评论 1 334
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,783评论 4 324
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,343评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,333评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,559评论 1 262
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,595评论 2 355
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,901评论 2 345

推荐阅读更多精彩内容