sa-token 登录注册

package com.example.auth.app;

import nexus.io.annotation.AComponentScan;
import nexus.io.hotswap.watcher.HotSwapResolver;
import nexus.io.hotswap.wrapper.tio.boot.TioApplicationWrapper;

@AComponentScan
public class AppServer {
  public static void main(String[] args) {
    long start = System.currentTimeMillis();
    // 支持fastjson2
    HotSwapResolver.addHotSwapClassPrefix("com.alibaba.fastjson2.");
    TioApplicationWrapper.run(AppServer.class, args);

    // TioApplication.run(MaLiangPenAiServerApp.class, args);
    long end = System.currentTimeMillis();
    System.out.println((end - start) + "ms");
  }
}

package com.example.auth.app.config;

import com.example.auth.app.db.model._MappingKit;
import com.jfinal.template.Engine;
import com.jfinal.template.source.ClassPathSourceFactory;
import nexus.io.annotation.AConfiguration;
import nexus.io.annotation.Initialization;
import nexus.io.db.activerecord.ActiveRecordPlugin;
import nexus.io.db.activerecord.OrderedFieldContainerFactory;
import nexus.io.tio.boot.server.TioBootServer;
import nexus.io.tio.boot.utils.TioRequestParamUtils;
import nexus.io.tio.utils.environment.EnvUtils;
import com.zaxxer.hikari.HikariConfig;
import com.zaxxer.hikari.HikariDataSource;

import lombok.extern.slf4j.Slf4j;

@AConfiguration
@Slf4j
public class EnoteMysqlDbConfig {

  /*
   *
   * config ActiveRecordPlugin
   */
  @Initialization
  public void config() {
    String jdbcUrl = EnvUtils.get("jdbc.url");
    log.info("jdbcUrl:{}", jdbcUrl);
    String jdbcUser = EnvUtils.get("jdbc.user");
    String jdbcPswd = EnvUtils.get("jdbc.pswd");
    int maximumPoolSize = EnvUtils.getInt("jdbc.MaximumPoolSize", 2);

    HikariConfig config = new HikariConfig();
    // 设定基本参数
    config.setJdbcUrl(jdbcUrl);
    config.setUsername(jdbcUser);
    config.setPassword(jdbcPswd);
    config.setMaximumPoolSize(maximumPoolSize);
    config.setConnectionTimeout(30000);

    HikariDataSource hikariDataSource = new HikariDataSource(config);
    HookCan.me().addDestroyMethod(hikariDataSource::close);

    ActiveRecordPlugin mysqlArp = new ActiveRecordPlugin(hikariDataSource);
    mysqlArp.setContainerFactory(new OrderedFieldContainerFactory());

    Engine engine = mysqlArp.getEngine();
    engine.setSourceFactory(new ClassPathSourceFactory());
    engine.setCompressorOn(' ');
    engine.setCompressorOn('\n');

    if (EnvUtils.isDev()) {
      mysqlArp.setDevMode(true);
      engine.setDevMode(true);
      mysqlArp.setShowSql(true);
    }

    mysqlArp.addSqlTemplate("/sql/all_sqls.sql");
    _MappingKit.mapping(mysqlArp);
    mysqlArp.start();
    HookCan.me().addDestroyMethod(mysqlArp::stop);
    // add
    TioRequestParamUtils.types.add("bigint");
  }
}

package com.example.auth.app.config;

import nexus.io.annotation.AConfiguration;
import nexus.io.annotation.Initialization;
import nexus.io.redis.Redis;
import nexus.io.redis.RedisDb;
import nexus.io.redis.RedisPlugin;
import nexus.io.tio.boot.server.TioBootServer;
import nexus.io.tio.utils.environment.EnvUtils;

@AConfiguration
public class RedisDbConfig {

  @Initialization
  public RedisPlugin redisPlugin() {
    String redisHost = EnvUtils.get("redis.host", "127.0.0.1");
    int redisPort = EnvUtils.getInt("redis.port", 6379);
    String redisPassword = EnvUtils.getStr("redis.password");
    int redistimeout = EnvUtils.getInt("redis.timeout", 60);
    int redisDatabase = EnvUtils.getInt("redis.database", 0);

    // 创建并启动 Redis 插件
    RedisPlugin redisPlugin = new RedisPlugin("main", redisHost, redisPort, redistimeout, redisPassword, redisDatabase);
    redisPlugin.start();
    HookCan.me().addDestroyMethod(redisPlugin::stop);

    // 测试连接
    RedisDb redisDb = Redis.use("main");
    redisDb.getJedis().connect();

    return redisPlugin;
  }
}

package com.example.auth.app.config;

import nexus.io.annotation.AConfiguration;
import nexus.io.annotation.Initialization;
import nexus.io.satoken.SaTokenDaoRedis;
import nexus.io.tio.boot.satoken.SaTokenContextForTio;
import nexus.io.tio.utils.environment.EnvUtils;

import cn.dev33.satoken.SaManager;
import cn.dev33.satoken.config.SaCookieConfig;
import cn.dev33.satoken.config.SaTokenConfig;
import cn.dev33.satoken.context.SaTokenContext;
import cn.dev33.satoken.jwt.StpLogicJwtForSimple;
import cn.dev33.satoken.stp.SaLoginModel;
import cn.dev33.satoken.stp.StpUtil;
import cn.dev33.satoken.util.SaTokenConsts;

@AConfiguration
public class SaTokenConfiguration {

  @Initialization
  public void config() {
    // 初始化 Sa-Token 上下文
    SaTokenContext saTokenContext = new SaTokenContextForTio();

    // 设置 Cookie 配置，例如启用 HttpOnly 属性
    SaCookieConfig saCookieConfig = new SaCookieConfig();
    saCookieConfig.setHttpOnly(true);

    // 初始化和配置 Sa-Token 主配置
    SaTokenConfig saTokenConfig = new SaTokenConfig();
    saTokenConfig.setTokenStyle(SaTokenConsts.TOKEN_STYLE_SIMPLE_UUID);
    saTokenConfig.setActiveTimeout(50 * 60); // 设置活动超时时间为 50 分钟

    saTokenConfig.setIsShare(false);
    saTokenConfig.setTokenName("token"); // 设置 token 的名称
    saTokenConfig.setIsWriteHeader(true); // 将 token 写入响应头
    saTokenConfig.setIsReadHeader(true); // 从请求头中读取 token
    //设置不将cookie信息存入到cookie
    saTokenConfig.setIsReadCookie(false);
    saTokenConfig.setCookie(saCookieConfig);

    String secretKey = EnvUtils.get("sa.token.jwt.secret.key");
    saTokenConfig.setJwtSecretKey(secretKey);
    //saTokenConfig.setTokenPrefix("Bearer");
    StpLogicJwtForSimple stpLogicJwtForSimple = new StpLogicJwtForSimple();
    StpUtil.setStpLogic(stpLogicJwtForSimple);

    // 应用配置到 Sa-Token 管理器
    SaManager.setConfig(saTokenConfig);
    SaManager.setSaTokenContext(saTokenContext);
    SaManager.setSaTokenDao(new SaTokenDaoRedis("main"));

    // 增加一个Api用户设置token永不过期
    SaLoginModel loginModel = new SaLoginModel();
    loginModel.setTimeout(-1);

    String adminToken = EnvUtils.get("sa.token.admin.token");
    loginModel.setToken(adminToken);
    StpUtil.createLoginSession("1", loginModel);
  }
}

package com.example.auth.app.config;

import java.util.HashMap;
import java.util.Map;

import nexus.io.annotation.AConfiguration;
import nexus.io.annotation.Initialization;
import nexus.io.tio.boot.http.interceptor.HttpInteceptorConfigure;
import nexus.io.tio.boot.http.interceptor.HttpInterceptorModel;
import nexus.io.tio.boot.satoken.UserTokenInterceptor;
import nexus.io.tio.boot.server.TioBootServer;

@AConfiguration
public class InterceptorConfiguration {

  @Initialization
  public void config() {
    // 创建 SaToken 拦截器实例
    Map<String, Object> data = new HashMap<>();
    data.put("msg", "please login");
    UserTokenInterceptor saTokenInterceptor = new UserTokenInterceptor(data);
    HttpInterceptorModel model = new HttpInterceptorModel();
    model.setInterceptor(saTokenInterceptor);
    model.addblockeUrl("/**"); // 拦截所有路由

    // 设置例外路由
    model.addAllowUrls("/exception/*", "/status", "/user/register", "/auth/login", "/auth/logout");
    model.addAllowUrls("/sms/send");
    // model.addAlloweUrl("/enote/puma/*");
    //
    model.addAllowUrls("/license/get");

    //
    model.addAllowUrls("/version");

    HttpInteceptorConfigure serverInteceptorConfigure = new HttpInteceptorConfigure();
    serverInteceptorConfigure.add(model);
    // 将拦截器配置添加到 Tio 服务器
    TioBootServer.me().setHttpInteceptorConfigure(serverInteceptorConfigure);
  }
}

import nexus.io.db.activerecord.Db;

public class SysUserInfoDao {

  /**
   * 判断用户是否存在
   * @param username
   * @return
   */
  public boolean existsUsername(String username) {
    //String sql = "select count(1) from sys_user_info where username=?";
    //return Db.exists(sql, username);
    String tableName="sys_user_info";
    return Db.exists(tableName,"username",username);
  }
}

package com.example.auth.app.services.login;

import org.junit.Test;

import com.example.auth.app.config.EnoteMysqlDbConfig;
import com.example.auth.app.dao.SysUserInfoDao;
import nexus.io.jfinal.aop.Aop;
import nexus.io.tio.boot.tesing.TioBootTest;

public class SysUserInfoDaoTest {

  @Test
  public void test() {
    TioBootTest.runWith(EnoteMysqlDbConfig.class);
    boolean existsUsername = Aop.get(SysUserInfoDao.class).existsUsername("litong");
    System.out.println(existsUsername);
  }
}

package com.example.auth.app.services;

import com.example.auth.app.model.LoginVo;
import com.jfinal.kit.Kv;
import nexus.io.db.TableResult;
import nexus.io.db.activerecord.Db;
import nexus.io.db.activerecord.Row;
import nexus.io.model.body.RespBodyVo;

import cn.dev33.satoken.stp.StpUtil;

public class AuthService {

  public TableResult<Kv> doLogin(String username, String password) {
    String sql = "select id from sys_user_info where username=? and password=?";
    Long result = Db.queryLong(sql, username, password);
    if (result != null && result > 0) {
      // 登录
      StpUtil.login(result);
      String tokenValue = StpUtil.getTokenValue();
      long tokenTimeout = StpUtil.getTokenTimeout();
      Kv kv = new Kv();
      kv.set("token", tokenValue);
      kv.set("tokenTimeout", tokenTimeout);
      Row row = Db.findById("sys_user_info", result);
      kv.set(row.toMap());
      return TableResult.ok(kv);
    } else {
      return TableResult.fail("username or password is not correct");
    }
  }

  public RespBodyVo loginByPhone(LoginVo vo) {
    // TODO Auto-generated method stub
    return null;
  }
}

package com.example.auth.app.services;

import com.example.auth.app.constants.ENoteTableNames;
import com.example.auth.app.model.UserRegisterVO;
import com.example.auth.app.utils.UserIdUtils;
import com.jfinal.kit.Kv;
import nexus.io.annotation.Inject;
import nexus.io.db.TableResult;
import nexus.io.db.activerecord.Db;
import nexus.io.db.activerecord.Row;

import cn.hutool.core.util.RandomUtil;
import lombok.extern.slf4j.Slf4j;

@Slf4j
public class UserRegisterService {

  @Inject
  private AuthService authService;

  public TableResult<Kv> index(UserRegisterVO reqVo) {
    Integer appId = reqVo.getAppId();
    String username = reqVo.getUsername();
    String password = reqVo.getPassword();
    Integer userChannel = reqVo.getUserChannel();
    Integer userFrom = reqVo.getUserFrom();
    Integer userLevel = 1;
    Integer status = 1;
    Integer v = reqVo.getV();
    String locale = reqVo.getLocale();

    boolean success = false;
    long userId = UserIdUtils.random();
    while (Db.exists(ENoteTableNames.SYS_USER_INFO, "id", userId)) {
      log.info("user id alread exits,random again:{}", userId);
      userId = UserIdUtils.random();
    }

    try {
      Row row = new Row();
      row.set("id", userId);
      row.set("app_id", appId).set("username", username).set("password", password).set("user_channel", userChannel);
      row.set("user_from", userFrom).set("user_level", userLevel).set("locale", locale);
      row.set("status", status).set("v", v);

      success = Db.tx(() -> {
        boolean save = Db.save(ENoteTableNames.SYS_USER_INFO, row);
        log.info("update result:{}", save);
        return save;
      });
    } catch (Exception e) {
      e.printStackTrace();
      return TableResult.fail(e.getMessage());
    }

    if (!success) {
      return TableResult.fail(-1, "register user fail");
    }

    TableResult<Kv> doLogin = authService.doLogin(username, password);
    return doLogin;

  }

}

package com.example.auth.app.services.login;

import org.junit.Test;

import com.example.auth.app.config.EnoteMysqlDbConfig;
import com.example.auth.app.config.RedisDbConfig;
import com.example.auth.app.config.SaTokenConfiguration;
import com.example.auth.app.model.UserRegisterVO;
import com.example.auth.app.services.UserRegisterService;
import com.jfinal.kit.Kv;
import nexus.io.db.TableResult;
import nexus.io.jfinal.aop.Aop;
import nexus.io.tio.boot.http.TioRequestContext;
import nexus.io.tio.boot.mock.TioMock;
import nexus.io.tio.boot.tesing.TioBootTest;
import nexus.io.tio.http.common.HttpRequest;
import nexus.io.tio.utils.json.JsonUtils;

public class UserRegisterServiceTest {

  @Test
  public void test() {
    //basic config
    TioBootTest.runWith(EnoteMysqlDbConfig.class, RedisDbConfig.class, SaTokenConfiguration.class);
    // mock http request
    HttpRequest request = TioMock.mockGetHttpRequest();
    TioRequestContext.hold(request);

    UserRegisterService userRegisterService = Aop.get(UserRegisterService.class);
    UserRegisterVO reqVo = UserRegisterVO.builder().username("litong").password("Litong2516")
        //
        .locale("zh_CN").userChannel(1).appId(1).userFrom(4).v(1)
        //
        .build();
    TableResult<Kv> result = userRegisterService.index(reqVo);
    System.out.println(JsonUtils.toJson(result));
  }

}

package com.example.auth.app.controller;

import com.example.auth.app.model.UserRegisterVO;
import com.example.auth.app.services.UserRegisterService;
import com.example.auth.app.validator.UserRegisteValidator;
import com.jfinal.kit.Kv;
import nexus.io.annotation.Inject;
import nexus.io.annotation.RequestPath;
import nexus.io.db.TableResult;
import nexus.io.tio.boot.http.TioRequestContext;
import nexus.io.tio.http.common.HttpResponse;
import nexus.io.tio.http.server.util.Resps;

@RequestPath("/user/register")
public class UserRegisterController {

  @Inject
  private UserRegisteValidator userRegisteValidator;
  @Inject
  private UserRegisterService userRegisterService;

  @RequestPath
  public HttpResponse index(UserRegisterVO reqVo) {
    TableResult<Kv> respVo = userRegisteValidator.index(reqVo);
    HttpResponse response = TioRequestContext.getResponse();
    if (respVo != null) {
      return Resps.json(response, respVo);
    }

    return Resps.json(response, userRegisterService.index(reqVo));
  }
}

package com.example.auth.app.validator;

import com.example.auth.app.model.UserRegisterVO;
import com.example.auth.app.dao.SysUserInfoDao;
import com.jfinal.kit.Kv;
import com.jfinal.kit.StrKit;
import nexus.io.annotation.Inject;
import nexus.io.db.TableResult;
import nexus.io.tio.boot.i18n.I18n;
import nexus.io.tio.boot.i18n.I18nLocale;
import nexus.io.tio.boot.i18n.Res;

public class UserRegisteValidator {

  @Inject
  private SysUserInfoDao SysUserInfoDao;

  public TableResult<Kv> index(UserRegisterVO reqVo) {
    String username = reqVo.getUsername();
    String password = reqVo.getPassword();
    String locale = reqVo.getLocale();
    Integer userChannel = reqVo.getUserChannel();
    Integer appId = reqVo.getAppId();
    Integer userFrom = reqVo.getUserFrom();
    Integer v = reqVo.getV();
    Res res = null;
    if (I18nLocale.ZH_CN.equals(locale)) {
      res = I18n.use(I18nLocale.ZH_CN);
    } else {
      res = I18n.use(I18nLocale.EN_US);
    }

    // 验证username
    if (StrKit.isBlank(username)) {
      String string = res.get("The_username_or_password_cannot_be_empty");
      return TableResult.fail(string);
    }

    if (StrKit.isBlank(password)) {
      return TableResult.fail(res.get("The_username_or_password_cannot_be_empty"));
    }

    if (StrKit.isBlank(locale)) {
      return TableResult.fail(res.get("The_locale_cannot_be_empty"));
    }

    if (userChannel == null) {
      return TableResult.fail(res.get("The_userChannel_cannot_be_empty"));
    }
    if (appId == null) {
      return TableResult.fail(res.get("The_appId_cannot_be_empty"));
    }

    if (userFrom == null) {
      return TableResult.fail(res.get("The_userFrom_cannot_be_empty"));
    }

    if (v == null) {
      return TableResult.fail(res.get("The_v_cannot_be_empty"));
    }

    if (username.length() < 6) {
      return TableResult.fail("The_username_at_least_6_characters");
    }

    if (password.length() < 8) {
      return TableResult.fail("The_password_at_least_8_characters");
    }

    // 至少包含大小,小写和数字
    if (!PasswordValidator.validatePassword(password)) {
      return TableResult.fail(res.get("The_password_should_contain_uppercase_letters,_lowercase_letters_and_numbers"));
    }

    if (locale.length() < 5) {
      return TableResult.fail(res.get("The_locale_at_least_5_characters"));
    }

    // 判断用户名是否存在
    if (SysUserInfoDao.existsUsername(username)) {
      return TableResult.fail(res.get("The_username_exists"));
    }
    return null;
  }
}

package com.example.auth.app.validator;
import org.junit.Test;

import com.jfinal.kit.StrKit;
import nexus.io.tio.boot.i18n.I18n;
import nexus.io.tio.boot.i18n.I18nLocale;
import nexus.io.tio.boot.i18n.Res;

public class UserRegisteValidatorTest {

  @Test
  public void test() {
    String locale = I18nLocale.ZH_CN;
    String username = null;
    Res res = null;
    if (I18nLocale.ZH_CN.equals(locale)) {
      res = I18n.use(I18nLocale.ZH_CN);
    } else {
      res = I18n.use(I18nLocale.EN_US);
    }

    // 验证username
    String string = null;
    if (StrKit.isBlank(username)) {
      string = res.get("The_username_or_password_cannot_be_empty");
    }
    System.out.println(string);
  }
}

package com.example.auth.app.controller;

import com.example.auth.app.model.LoginVo;
import com.example.auth.app.services.AuthService;
import com.example.auth.app.validator.AuthValidator;
import com.jfinal.kit.Kv;
import nexus.io.annotation.Inject;
import nexus.io.annotation.RequestPath;
import nexus.io.db.TableResult;
import nexus.io.model.body.RespBodyVo;
import nexus.io.tio.boot.http.TioRequestContext;
import nexus.io.tio.http.common.HttpResponse;
import nexus.io.tio.http.server.util.Resps;

import cn.dev33.satoken.stp.StpUtil;
import lombok.extern.slf4j.Slf4j;

@RequestPath("/auth")
@Slf4j
public class AuthController {
  @Inject
  private AuthValidator authValidator;
  @Inject
  private AuthService authService;

  public HttpResponse login(String username, String password, String locale) {
    log.info("username:{}", username);
    RespBodyVo respVo = authValidator.doLogin(username, password, locale);
    if (respVo != null) {
      return Resps.json(TioRequestContext.getResponse(), TableResult.fail(-1, respVo.getMsg()));
    }
    TableResult<Kv> result = authService.doLogin(username, password);
    return Resps.json(TioRequestContext.getResponse(), result);
  }

  public HttpResponse logout() {
    StpUtil.logout();
    HttpResponse response = TioRequestContext.getResponse();
    return Resps.json(response, RespBodyVo.ok());
  }

  public RespBodyVo validateToken() {
    try {
      StpUtil.checkActiveTimeout();
      return RespBodyVo.ok();
    } catch (Exception e) {
      return RespBodyVo.fail();
    }
  }
}

package com.example.auth.app.validator;

import com.jfinal.kit.StrKit;
import nexus.io.model.body.RespBodyVo;
import nexus.io.tio.boot.i18n.I18n;
import nexus.io.tio.boot.i18n.I18nLocale;
import nexus.io.tio.boot.i18n.Res;

public class AuthValidator {

  public RespBodyVo doLogin(String username, String password,String locale) {
    Res res = null;
    if (I18nLocale.ZH_CN.equals(locale)) {
      res = I18n.use(I18nLocale.ZH_CN);
    } else {
      res = I18n.use(I18nLocale.EN_US);
    }

    if (StrKit.isBlank(locale)) {
      return RespBodyVo.fail(res.get("The_locale_cannot_be_empty"));
    }

    if(StrKit.isBlank(username) || StrKit.isBlank(password)) {
      String string = res.get("The_username_or_password_cannot_be_empty");
      return RespBodyVo.fail(string);
    }
    return null;
  }

}