Complete the datasource code

chat2db-server/.apifox-helper.properties

@@ -13,3 +13,8 @@ constant.field.ignore=groovy:it.name()=="serialVersionUID"
 
 # sprnSupport for Jackson annotations
 json.rule.field.ignore=@com.fasterxml.jackson.annotation.JsonIgnore#value
+
+field.demo=#demo
+field.default.value=#default
+#apifox mock
+field.mock=#mock

chat2db-server/chat2db-server-domain/chat2db-server-domain-api/src/main/java/ai/chat2db/server/domain/api/enums/RoleCodeEnum.java

@@ -13,25 +13,29 @@ public enum RoleCodeEnum implements BaseEnum<String> {
     /**
      * DESKTOP
      */
-    DESKTOP("DESKTOP", 1L),
+    DESKTOP("DESKTOP", 1L, "_desktop_default_user_name", "_desktop_default_user_name"),
 
     /**
      * ADMIN
      */
-    ADMIN("ADMIN", 2L),
+    ADMIN("ADMIN", 2L, "chat2db", "chat2db"),
 
     /**
      * USER
      */
-    USER("USER", null),
+    USER("USER", null, null, null),
 
     ;
     final String description;
     final Long defaultUserId;
+    final String userName;
+    final String password;
 
-    RoleCodeEnum(String description, Long defaultUserId) {
+    RoleCodeEnum(String description, Long defaultUserId, String userName, String password) {
         this.description = description;
         this.defaultUserId = defaultUserId;
+        this.userName = userName;
+        this.password = password;
     }
 
     @Override

chat2db-server/chat2db-server-domain/chat2db-server-domain-core/src/main/java/ai/chat2db/server/domain/core/cache/MemoryCacheManage.java

@@ -33,7 +33,7 @@ public class MemoryCacheManage {
             .createCache(CACHE,
                 CacheConfigurationBuilder.newCacheConfigurationBuilder(String.class, byte[].class,
                         ResourcePoolsBuilder.newResourcePoolsBuilder()
-                            .offheap(10, MemoryUnit.MB))
+                            .offheap(5, MemoryUnit.MB))
                     .withExpiry(ExpiryPolicyBuilder.timeToLiveExpiration(Duration.ofMinutes(10))));
     }
 

chat2db-server/chat2db-server-domain/chat2db-server-domain-repository/src/main/java/ai/chat2db/server/domain/repository/entity/DataSourceDO.java

@@ -14,7 +14,7 @@ import lombok.Setter;
  * </p>
  *
  * @author chat2db
- * @since 2023-07-30
+ * @since 2023-08-05
  */
 @Getter
 @Setter

chat2db-server/chat2db-server-domain/chat2db-server-domain-repository/src/main/java/ai/chat2db/server/domain/repository/mapper/DataSourceCustomMapper.java

@@ -1,6 +1,7 @@
 package ai.chat2db.server.domain.repository.mapper;
 
 import ai.chat2db.server.domain.repository.entity.DataSourceDO;
+import com.baomidou.mybatisplus.core.mapper.Mapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import org.apache.ibatis.annotations.Param;
 
@@ -9,8 +10,9 @@ import org.apache.ibatis.annotations.Param;
  *
  * @author Jiaju Zhuang
  */
-public interface DataSourceCustomMapper {
-
-    IPage<DataSourceDO> selectPageWithPermission(IPage<DataSourceDO> page, @Param("admin") Boolean admin, @Param("userId") Long userId,
+public interface DataSourceCustomMapper extends Mapper<DataSourceDO> {
+    IPage<DataSourceDO> selectPageWithPermission(IPage<DataSourceDO> page, @Param("admin") Boolean admin,
+        @Param("userId") Long userId,
         @Param("searchKey") String searchKey);
+
 }

chat2db-server/chat2db-server-domain/chat2db-server-domain-repository/src/main/resources/ai/chat2db/server/domain/repository/DataSourceCustomMapper.xml

@@ -1,32 +0,0 @@
-<?xml version="1.0" encoding="UTF-8"?>
-<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
-<mapper namespace="ai.chat2db.server.domain.repository.mapper.DataSourceCustomMapper">
-
-    <select id="selectPageWithPermission" resultType="ai.chat2db.server.domain.repository.entity.DataSourceDO">
-        select ds.*
-        from DATA_SOURCE ds
-        <where>
-            <if test="admin != true">
-                <choose>
-                    <when test="userId == 1">
-                        ds.USER_ID = #{userId}
-                    </when>
-                    <otherwise>
-                        (ds.USER_ID = #{userId}
-                        or exists(select 1 from DATA_SOURCE_ACCESS dsa where dsa.ACCESS_OBJECT_TYPE = 'USER' and
-                        dsa.ACCESS_OBJECT_ID = #{userId})
-                        or exists(select 1
-                        from DATA_SOURCE_ACCESS dsa
-                        LEFT JOIN TEAM_USER tu on tu.ID = dsa.ACCESS_OBJECT_ID and dsa.ACCESS_OBJECT_TYPE = 'TEAM'
-                        where tu.USER_ID = #{userId})
-                        )
-                    </otherwise>
-                </choose>
-            </if>
-            <if test="searchKey != '' and searchKey != null ">
-                and (ds.alias like concat('%',#{searchKey},'%') or ds.url like concat('%',#{searchKey},'%'))
-            </if>
-        </where>
-
-    </select>
-</mapper>

chat2db-server/chat2db-server-domain/chat2db-server-domain-repository/src/main/resources/mapper/DataSourceCustomMapper.xml

@@ -0,0 +1,36 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
+<mapper namespace="ai.chat2db.server.domain.repository.mapper.DataSourceCustomMapper">
+
+    <select id="selectPageWithPermission" resultType="ai.chat2db.server.domain.repository.entity.DataSourceDO">
+        select ds.*
+        from DATA_SOURCE ds
+        <where>
+            <choose>
+                <when test="admin == true">
+                    and ds.USER_ID != 1
+                </when>
+                <otherwise>
+                    <choose>
+                        <when test="userId == 1">
+                            and ds.USER_ID = #{userId}
+                        </when>
+                        <otherwise>
+                            and (ds.USER_ID = #{userId}
+                            or exists(select 1 from DATA_SOURCE_ACCESS dsa where dsa.ACCESS_OBJECT_TYPE = 'USER' and
+                            dsa.ACCESS_OBJECT_ID = #{userId})
+                            or exists(select 1
+                            from DATA_SOURCE_ACCESS dsa
+                            LEFT JOIN TEAM_USER tu on tu.ID = dsa.ACCESS_OBJECT_ID and dsa.ACCESS_OBJECT_TYPE = 'TEAM'
+                            where tu.USER_ID = #{userId})
+                            )
+                        </otherwise>
+                    </choose>
+                </otherwise>
+            </choose>
+            <if test="searchKey != '' and searchKey != null ">
+                and (ds.alias like concat('%',#{searchKey},'%') or ds.url like concat('%',#{searchKey},'%'))
+            </if>
+        </where>
+    </select>
+</mapper>

chat2db-server/chat2db-server-start/pom.xml

@@ -28,7 +28,10 @@
             <groupId>ai.chat2db</groupId>
             <artifactId>chat2db-server-web-api</artifactId>
         </dependency>
-
+        <dependency>
+            <groupId>ai.chat2db</groupId>
+            <artifactId>chat2db-server-admin-api</artifactId>
+        </dependency>
         <!-- 需要加载的服务 -->
         <dependency>
             <groupId>ai.chat2db</groupId>

chat2db-server/chat2db-server-start/src/main/java/ai/chat2db/server/start/config/config/Chat2dbWebMvcConfigurer.java

@@ -77,8 +77,9 @@ public class Chat2dbWebMvcConfigurer implements WebMvcConfigurer {
                         } else {
                             return true;
                         }
-                    }
+                    } else {
                         userId = Long.parseLong(userIdString);
+                    }
                     Long finalUserId = userId;
                     LoginUser loginUser = MemoryCacheManage.computeIfAbsent(CacheKey.getLoginUserKey(userId), () -> {
                         User user = userService.query(finalUserId).getData();

chat2db-server/chat2db-server-start/src/main/java/ai/chat2db/server/start/controller/oauth/OauthController.java

@@ -1,5 +1,6 @@
 package ai.chat2db.server.start.controller.oauth;
 
+import ai.chat2db.server.domain.api.enums.RoleCodeEnum;
 import jakarta.annotation.Resource;
 
 import ai.chat2db.server.domain.api.model.User;
@@ -47,14 +48,26 @@ public class OauthController {
         //   查询用户
         User user = userService.query(request.getUserName()).getData();
         if (user == null) {
-            throw new BusinessException("当前用户不存在。");
+            throw new BusinessException("oauth.userNameNotExits");
         }
+        if (RoleCodeEnum.DESKTOP.getDefaultUserId().equals(user.getId())) {
+            throw new BusinessException("oauth.IllegalUserName");
+        }
+        // Successfully logged in without modifying the administrator password
+        if (RoleCodeEnum.ADMIN.getDefaultUserId().equals(user.getId()) && RoleCodeEnum.ADMIN.getPassword().equals(
+            user.getPassword())) {
+            return DataResult.of(doLogin(user));
+        }
+
         if (!DigestUtil.bcryptCheck(request.getPassword(), user.getPassword())) {
-            throw new BusinessException("您输入的密码有误。");
+            throw new BusinessException("oauth.passwordIncorrect");
         }
+        return DataResult.of(doLogin(user));
+    }
+
+    private Object doLogin(User user) {
         StpUtil.login(user.getId());
-        Object token = SaHolder.getStorage().get(SaTokenConsts.JUST_CREATED_NOT_PREFIX);
-        return DataResult.of(token);
+        return SaHolder.getStorage().get(SaTokenConsts.JUST_CREATED_NOT_PREFIX);
     }
 
     /**

chat2db-server/chat2db-server-start/src/main/resources/application.yml

@@ -41,6 +41,12 @@ logging:
   level:
     org:
       flywaydb: debug
+    ai:
+      chat2db:
+        server:
+          domain:
+            repository:
+              mapper: debug
 # 登录功能
 sa-token:
   # token名称 (同时也是cookie名称)

chat2db-server/chat2db-server-start/src/main/resources/db/migration/V2_1_0__支持环境、用户权限.sql

@@ -38,8 +38,11 @@ ALTER TABLE `dbhub_user`
 
 
 update dbhub_user
-set role_code= 'DESKTOP'
+set role_code= 'DESKTOP',user_name='_desktop_default_user_name',password='_desktop_default_user_name',nick_name='桌面端用户'
 where id = 1;
+INSERT INTO DBHUB_USER (USER_NAME, PASSWORD, NICK_NAME, EMAIL, ROLE_CODE) VALUES ('chat2db', 'chat2db', '管理员', null, 'ADMIN');
+
+create UNIQUE INDEX uk_user_user_name on dbhub_user (user_name);
 
 
 CREATE TABLE IF NOT EXISTS `team`

chat2db-server/chat2db-server-start/src/main/resources/i18n/messages.properties

@@ -8,6 +8,11 @@ common.paramDetailError=The parameter: {0} is incorrect
 common.paramCheckError=The following parameters are not valid:
 common.maxUploadSize=The file exceeds the maximum limit
 common.permissionDenied=Permission denied
+
+oauth.userNameNotExits=The current account does not exist
+oauth.IllegalUserName=The current account cannot be logged in. Please change your account
+oauth.passwordIncorrect=The password you entered is incorrect
+
 # dataSource
 dataSource.sqlAnalysisError=Invalid statements
 # connection

chat2db-server/chat2db-server-start/src/main/resources/i18n/messages_en_US.properties

@@ -8,6 +8,10 @@ common.paramCheckError=The following parameters are not valid
 common.maxUploadSize=The file exceeds the maximum limit
 common.permissionDenied=Permission denied
 
+oauth.userNameNotExits=The current account does not exist
+oauth.IllegalUserName=The current account cannot be logged in. Please change your account
+oauth.passwordError=The password you entered is incorrect
+
 dataSource.sqlAnalysisError=Invalid statements
 connection.error=Connection failed, please check the connection information
 connection.ssh.error=SSH connection failed, please check the connection information

chat2db-server/chat2db-server-start/src/main/resources/i18n/messages_zh_CN.properties

@@ -8,6 +8,10 @@ common.paramCheckError=请检查以下参数:
 common.maxUploadSize=您输入的文件超过最大限制
 common.permissionDenied=您没有权限访问该页面
 
+oauth.userNameNotExits=当前账号不存在
+oauth.IllegalUserName=当前账号无法登录，请换一个账号
+oauth.passwordIncorrect=您输入的密码有误
+
 dataSource.sqlAnalysisError=不合法的执行语句
 connection.error=数据库链接异常，请检查数据库配置
 connection.ssh.error=SSH 链接异常，请检查SSH配置

chat2db-server/chat2db-server-start/src/test/java/ai/chat2db/server/start/test/mybatis/MybatisGeneratorTest.java

@@ -34,7 +34,7 @@ public class MybatisGeneratorTest extends BaseTest {
         //doGenerator(Lists.newArrayList("operation_saved"));
         //doGenerator(Lists.newArrayList("environment","data_source","team","team_dbhub_user","data_source_access",
         // "dbhub_user"));
-        doGenerator(Lists.newArrayList("team_user"));
+        doGenerator(Lists.newArrayList("data_source"));
     }
 
     private void doGenerator(List<String> tableList) {
@@ -45,7 +45,7 @@ public class MybatisGeneratorTest extends BaseTest {
             + "/java";
         String xmlDir = System.getProperty("user.dir")
             + "/../chat2db-server-domain/chat2db-server-domain-repository/src/main"
-            + "/resources/ai/chat2db/server/domain/repository";
+            + "/resources/mapper";
 
         // 不要生成service controller
         Map<OutputFile, String> pathInfo = new HashMap<>();
@@ -83,14 +83,12 @@ public class MybatisGeneratorTest extends BaseTest {
                     //开启实体类配置
                     .entityBuilder()
                     .formatFileName("%sDO")
-                    // 覆盖文件
-                    .enableFileOverride()
                     //.addTableFills(new Column("gmt_create", FieldFill.INSERT)) // 表字段填充
                     //.addTableFills(new Column("update_time", FieldFill.INSERT_UPDATE)) // 表字段填充
                     //开启lombok
                     .enableLombok()
                     .mapperBuilder()
-                    // 覆盖文件
+                    //// 覆盖文件
                     .enableFileOverride()
                 ;
 

chat2db-server/chat2db-server-tools/chat2db-server-tools-base/src/main/java/ai/chat2db/server/tools/base/wrapper/request/PageQueryRequest.java

@@ -3,7 +3,6 @@ package ai.chat2db.server.tools.base.wrapper.request;
 import java.io.Serial;
 import java.io.Serializable;
 
-
 import ai.chat2db.server.tools.base.constant.EasyToolsConstant;
 
 import jakarta.validation.constraints.NotNull;
@@ -26,11 +25,15 @@ public class PageQueryRequest implements Serializable {
     private static final long serialVersionUID = EasyToolsConstant.SERIAL_VERSION_UID;
     /**
      * 页码
+     *
+     * @mock 1
      */
     @NotNull(message = "分页页码不能为空")
     private Integer pageNo;
     /**
      * 分页条数
+     *
+     * @demo 10
      */
     @NotNull(message = "分页大小不能为空")
     @Range(min = 1, max = EasyToolsConstant.MAX_PAGE_SIZE,

chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/datasource/DataSourceAdminController.java

@@ -46,7 +46,7 @@ public class DataSourceAdminController {
      */
     @GetMapping("/page")
     public WebPageResult<DataSourcePageQueryVO> page(@Valid CommonPageQueryRequest request) {
-        return dataSourceService.queryPage(dataSourceAdminConverter.request2param(request), null)
+        return dataSourceService.queryPageWithPermission(dataSourceAdminConverter.request2param(request), null)
             .mapToWeb(dataSourceAdminConverter::dto2vo);
     }
 
@@ -81,6 +81,7 @@ public class DataSourceAdminController {
      *
      * @param request
      * @return
+     * @version 2.1.0
      */
     @PostMapping("/clone")
     public DataResult<Long> clone(@RequestBody DataSourceCloneRequest request) {
@@ -92,6 +93,7 @@ public class DataSourceAdminController {
      *
      * @param id
      * @return
+     * @version 2.1.0
      */
     @DeleteMapping("/{id}")
     public ActionResult delete(@PathVariable Long id) {

chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/user/UserAdminController.java

@@ -2,7 +2,7 @@
 package ai.chat2db.server.admin.api.controller.user;
 
 import ai.chat2db.server.admin.api.controller.common.request.CommonPageQueryRequest;
-import ai.chat2db.server.admin.api.controller.user.converter.DataSourceAdminConverter;
+import ai.chat2db.server.admin.api.controller.user.converter.UserAdminConverter;
 import ai.chat2db.server.admin.api.controller.user.request.UserCreateRequest;
 import ai.chat2db.server.admin.api.controller.user.request.UserUpdateRequest;
 import ai.chat2db.server.admin.api.controller.user.vo.UserPageQueryVO;
@@ -32,7 +32,7 @@ public class UserAdminController {
     @Resource
     private DataSourceService dataSourceService;
     @Resource
-    private DataSourceAdminConverter dataSourceAdminConverter;
+    private UserAdminConverter dataSourceAdminConverter;
 
     /**
      * Pagination query

chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/user/converter/DataSourceAdminConverter.java

@@ -1,66 +0,0 @@
-package ai.chat2db.server.admin.api.controller.user.converter;
-
-import ai.chat2db.server.admin.api.controller.common.request.CommonPageQueryRequest;
-import ai.chat2db.server.admin.api.controller.datasource.request.DataSourceUpdateRequest;
-import ai.chat2db.server.admin.api.controller.user.request.UserCreateRequest;
-import ai.chat2db.server.domain.api.param.DataSourceCreateParam;
-import ai.chat2db.server.domain.api.param.DataSourcePageQueryParam;
-import ai.chat2db.server.domain.api.param.DataSourceUpdateParam;
-import org.mapstruct.Mapper;
-import org.mapstruct.Mapping;
-import org.mapstruct.Mappings;
-
-/**
- * converter
- *
- * @author Jiaju Zhuang
- */
-@Mapper(componentModel = "spring")
-public abstract class DataSourceAdminConverter {
-
-    /**
-     * conversion
-     *
-     * @param request
-     * @return
-     */
-    public abstract DataSourcePageQueryParam request2param(CommonPageQueryRequest request);
-
-    /**
-     * conversion
-     *
-     * @param request
-     * @return
-     */
-    public abstract DataSourcePageQueryParam request2paramAccess(CommonPageQueryRequest request);
-
-    ///**
-    // * conversion
-    // *
-    // * @param dto
-    // * @return
-    // */
-    //public abstract DataSourcePageQueryVO dto2vo(DataSource dto);
-
-    /**
-     * 参数转换
-     *
-     * @param request
-     * @return
-     */
-    @Mappings({
-        @Mapping(source = "user", target = "userName")
-    })
-    public abstract DataSourceCreateParam createReq2param(UserCreateRequest request);
-
-    /**
-     * 参数转换
-     *
-     * @param request
-     * @return
-     */
-    @Mappings({
-        @Mapping(source = "user", target = "userName")
-    })
-    public abstract DataSourceUpdateParam updateReq2param(DataSourceUpdateRequest request);
-}

chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/user/converter/UserAdminConverter.java

@@ -0,0 +1,58 @@
+package ai.chat2db.server.admin.api.controller.user.converter;
+
+import org.mapstruct.Mapper;
+
+/**
+ * converter
+ *
+ * @author Jiaju Zhuang
+ */
+@Mapper(componentModel = "spring")
+public abstract class UserAdminConverter {
+
+    ///**
+    // * conversion
+    // *
+    // * @param request
+    // * @return
+    // */
+    //public abstract DataSourcePageQueryParam request2param(CommonPageQueryRequest request);
+    //
+    ///**
+    // * conversion
+    // *
+    // * @param request
+    // * @return
+    // */
+    //public abstract DataSourcePageQueryParam request2paramAccess(CommonPageQueryRequest request);
+
+    ///**
+    // * conversion
+    // *
+    // * @param dto
+    // * @return
+    // */
+    //public abstract DataSourcePageQueryVO dto2vo(DataSource dto);
+    //
+    ///**
+    // * 参数转换
+    // *
+    // * @param request
+    // * @return
+    // */
+    //@Mappings({
+    //    @Mapping(source = "user", target = "userName")
+    //})
+    //public abstract DataSourceCreateParam createReq2param(UserCreateRequest request);
+    //
+    ///**
+    // * 参数转换
+    // *
+    // * @param request
+    // * @return
+    // */
+    //@Mappings({
+    //    @Mapping(source = "user", target = "userName")
+    //})
+    //public abstract DataSourceUpdateParam updateReq2param(DataSourceUpdateRequest request);
+}