一、封装工具类

1、状态码枚举

package com.king.wishuserbase.enums;/*** @author gaoxiaojin*/public enum ResultCodes {/*** 返回code枚举*/OK("20000", "正常"),AUTH_ERROR("30003", "用户访问异常"),QUERY_NULL("QUERY_NULL", "查询为空"),ADD_OR_UPDATE_FAIL("ADD_OR_UPDATE_FAIL", "入库失败"),SYSTEM_ERROR("SYSTEM_ERROR", "系统错误"),RUNTIME_ERROR("RUNTIME_ERROR", "系统错误");/*** 错误码*/private final String code;/*** 描述*/private final String desc;ResultCodes(String code, String desc) {this.code = code;this.desc = desc;}public String getCode() {return code;}public String getDesc() {return desc;}
}

2、返回结果工具类

package com.king.wishuserbase.utils;import com.king.wishuserbase.enums.ResultCodes;
import lombok.Data;/*** @author gaoxiaojin*/
@Data
public class WishResult<T> {private boolean success;private T data;private String code;private String message;private Exception exception;public void initializeSuccess() {this.initialize(ResultCodes.OK);}public void initializeSuccess(T data) {this.initializeSuccess();this.setData(data);}public void initializeFail(String resultCode, String message) {this.code = resultCode;this.message = message;this.success = false;}public void initialize(ResultCodes resultCode) {if (resultCode != null) {this.success = resultCode == ResultCodes.OK;this.code = resultCode.getCode();}}
}

3、异常处理

1）BusinessException工具类

package com.king.wishuserbase.exception;import com.king.wishuserbase.enums.ResultCodes;/*** @author gaoxiaojin*/
public class BusinessException extends RuntimeException{private final Integer status;private final String defineCode;private final String chnDesc;public BusinessException(){this(50000,"SERVER_ERROR","系统暂时无响应,请稍后再试!");}public BusinessException(String chnDesc){this(51000,"SYSTEM_FAIL",chnDesc);}public BusinessException(String defineCode,String chnDesc){this(52000,defineCode,chnDesc);}public BusinessException(Integer status,String defineCode,String chnDesc){super(chnDesc);this.status = status;this.defineCode = defineCode;this.chnDesc = chnDesc;}public BusinessException(ResultCodes businessResultCode){this(52000,businessResultCode.getCode(),businessResultCode.getDesc());}public BusinessException(ResultCodes businessResultCode,String chnDesc){this(52000,businessResultCode.getCode(),chnDesc);}public static BusinessException OPERATOR_NOT_EXIST = new BusinessException(52001,"OPERATOR_NOT_EXIST","操作人不存在!");public static BusinessException PARAM_IS_NULL = new BusinessException(52002,"PARAM_IS_NULL","参数为空!");public static BusinessException DB_RECORD_NO_EXIST = new BusinessException(52300,"DB_RECORD_NO_EXIST","数据库无对应记录!");public static BusinessException DB_RECORD_FIELD_NULL = new BusinessException(52301,"DB_RECORD_FIELD_NULL","数据库记录字段为空!");public static BusinessException DB_RECORD_UPDATE_FAIL = new BusinessException(52310,"DB_RECORD_UPDATE_FAIL","数据库记录更新失败!");public Integer getStatus() {return status;}public String getDefineCode() {return defineCode;}public String getChnDesc() {return chnDesc;}
}

2）异常拦截器

package com.king.wishuserbase.exception;import com.king.wishuserbase.enums.ResultCodes;
import com.king.wishuserbase.utils.WishResult;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.ResponseBody;/*** @author gaoxiaojin*/
@ControllerAdvice
@Slf4j
public class KingBusinessExceptionHandler {@ExceptionHandler(value = BusinessException.class)@ResponseBodypublic WishResult<Void> kingBaseException(BusinessException ex) {log.warn(ex.getChnDesc(), ex);WishResult<Void> result = new WishResult<>();result.setCode(ex.getDefineCode());result.setMessage(ex.getChnDesc());return result;}@ExceptionHandler(value = RuntimeException.class)@ResponseBodypublic WishResult<Void> runtimeException(RuntimeException ex) {log.error(ex.getMessage(), ex);WishResult<Void> result = new WishResult<>();result.setCode(ResultCodes.RUNTIME_ERROR.getCode());// 将异常的消息也打印出来，不用固定的字符替代，方便排查result.setMessage(ex.getMessage());return result;}@ExceptionHandler(value = Exception.class)@ResponseBodypublic WishResult<Void> exception(Exception ex) {log.error(ex.getMessage(), ex);WishResult<Void> result = new WishResult<>();result.setCode(ResultCodes.SYSTEM_ERROR.getCode());// 将异常的消息也打印出来，不用固定的字符替代，方便排查result.setMessage(ex.getMessage());return result;}}

4、访问拦截

1）在application.yml中增加访问放行配置

king:sso:allow-urls: /user

2）配置拦截器

package com.king.wishuserbase.interceptor;import com.google.gson.Gson;
import com.king.wishuserbase.enums.ResultCodes;
import com.king.wishuserbase.utils.WishResult;
import lombok.NonNull;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import org.springframework.web.method.HandlerMethod;
import org.springframework.web.servlet.HandlerInterceptor;import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.PrintWriter;/*** @author gaoxiaojin*/
@Component
@Slf4j
public class AuthInterceptor implements HandlerInterceptor {@Value("${king.sso.allow-urls}")private String[] allowUrls;@Overridepublic boolean preHandle(@NonNull HttpServletRequest request, @NonNull HttpServletResponse response, @NonNull Object handler) {if (!(handler instanceof HandlerMethod)) {return true;}String requestUrl = request.getRequestURI();request.setAttribute("ignore", "no");if (null != allowUrls && allowUrls.length >= 1) {for (String url : allowUrls) {if (requestUrl.contains(url)) {request.setAttribute("ignore", "all");return true;}}}if (request.getAttribute("token") == null) {returnErrorData(response);return false;} else {return true;}}private void returnErrorData(HttpServletResponse response) {WishResult<String> result = new WishResult<>();result.setSuccess(false);result.setCode(ResultCodes.AUTH_ERROR.getCode());result.setMessage(ResultCodes.AUTH_ERROR.getDesc());Gson gson = new Gson();response.setCharacterEncoding("UTF-8");response.setContentType("application/json; charset=utf-8");try (PrintWriter out = response.getWriter()) {out.append(gson.toJson(result));} catch (IOException e) {e.printStackTrace();}}}

3）web拦截配置

package com.king.wishuserbase.config;import com.king.wishuserbase.interceptor.AuthInterceptor;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurerAdapter;/*** @author gaoxiaojin*/
@Configuration
public class WebConfig extends WebMvcConfigurerAdapter {@Beanpublic AuthInterceptor authInterceptor() {return new AuthInterceptor();}@Overridepublic void addInterceptors(InterceptorRegistry registry) {registry.addInterceptor(authInterceptor()).addPathPatterns("/**");super.addInterceptors(registry);}
}

4）测试

package com.king.wishuserbase.controller;import com.king.wishuserbase.utils.WishResult;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;import org.springframework.web.bind.annotation.RestController;/*** <p>* c端用户 前端控制器* </p>** @author gaoxiaojin* @since 2022-02-11*/
@RestController
@RequestMapping("/user/")
public class UserController {@GetMapping("getUserById")public WishResult<Void> getUser() {WishResult<Void> result = new WishResult<>();result.initializeSuccess();return result;;}
}

启动项目访问
没有token访问被拦截

配置不拦截后访问成功

5、代码生成器

package com.king.wishuserbase.utils;import com.baomidou.mybatisplus.core.exceptions.MybatisPlusException;
import com.baomidou.mybatisplus.core.toolkit.StringPool;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.InjectionConfig;
import com.baomidou.mybatisplus.generator.config.*;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;/*** @author gaoxiaojin*/
public class MyBatisPlusGenerator {public static void main(String[] args) {// 代码生成器AutoGenerator mpg = new AutoGenerator();// 全局配置GlobalConfig gc = new GlobalConfig();gc.setBaseColumnList(true);gc.setBaseResultMap(true);String projectPath = System.getProperty("user.dir");gc.setOutputDir(projectPath + "/src/main/java");gc.setAuthor("gaoxiaojin");gc.setOpen(false);mpg.setGlobalConfig(gc);// 数据源配置DataSourceConfig dsc = new DataSourceConfig();dsc.setUrl("jdbc:mysql://121.199.67.189:3306/wish?useUnicode=true&characterEncoding=utf8&useSSL=false&serverTimezone=GMT%2B8");dsc.setDriverName("com.mysql.cj.jdbc.Driver");dsc.setUsername("root");dsc.setPassword("gxj092116");mpg.setDataSource(dsc);// 包配置PackageConfig pc = new PackageConfig();pc.setParent("com.king.wishuserbase").setMapper("dao.mapper").setService("service").setController("controller").setEntity("dao.domain.entity");mpg.setPackageInfo(pc);// 自定义配置InjectionConfig cfg = new InjectionConfig() {@Overridepublic void initMap() {// to do nothing}};List<FileOutConfig> focList = new ArrayList<>();focList.add(new FileOutConfig("/templates/mapper.xml.ftl") {@Overridepublic String outputFile(TableInfo tableInfo) {// 自定义输入文件名称return projectPath + "/src/main/resources/mapper/" // pc//.getModuleName()+ "/"+ tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;}});cfg.setFileOutConfigList(focList);mpg.setCfg(cfg);mpg.setTemplate(new TemplateConfig().setXml(null));// 策略配置StrategyConfig strategy = new StrategyConfig();strategy.setNaming(NamingStrategy.underline_to_camel);strategy.setColumnNaming(NamingStrategy.underline_to_camel);strategy.setRestControllerStyle(true);strategy.setInclude(scanner("表名，多个英文逗号分割").split(","));strategy.setControllerMappingHyphenStyle(true);strategy.setTablePrefix(pc.getModuleName() + "_");mpg.setStrategy(strategy);mpg.setTemplateEngine(new FreemarkerTemplateEngine());mpg.execute();}/*** <p>* 读取控制台内容* </p>*/public static String scanner(String tip) {Scanner scanner = new Scanner(System.in);System.out.println(("请输入" + tip + "："));if (scanner.hasNext()) {String ipt = scanner.next();if (StringUtils.isNotEmpty(ipt)) {return ipt;}}throw new MybatisPlusException("请输入正确的" + tip + "！");}}

封装好后直接运行然后输入表名就可以直接生成controller、service、mapper了