xieyinlu/ruoyi-ai
1
0
Fork 0
mirror of https://gitcode.com/ageerle/ruoyi-ai.git synced 2026-03-13 20:53:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #143 from violateer/feature/knowledge-role

Browse Source 
添加知识库权限控制功能
This commit is contained in:
ageerle
2025-08-02 15:56:42 +08:00
committed by GitHub
parent 85930f4b12 999282210e
commit cc4ca69640
32 changed files with 2008 additions and 337 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
ruoyi-admin/src/main/resources/application.yml
View File

@@ -1,4 +1,3 @@
# 项目相关配置
ruoyi:
# 名称
@@ -162,6 +161,9 @@ tenant:
- sys_user_post
- sys_user_role
knowledge-role:
enable: false
# MyBatisPlus配置
# https://baomidou.com/config/
mybatis-plus:

10
ruoyi-common/ruoyi-common-core/src/main/java/org/ruoyi/common/core/domain/model/LoginUser.java
View File

@@ -117,6 +117,16 @@ public class LoginUser implements Serializable {
*/
private Long roleId;
/**
* 关联角色类型
*/
private String kroleGroupType;
/**
* 关联角色id
*/
private String kroleGroupIds;
/**
* 获取登录id
*/

62
ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/KnowledgeRole.java Normal file
View File

@@ -0,0 +1,62 @@
package org.ruoyi.domain;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableLogic;
import com.baomidou.mybatisplus.annotation.TableName;
import lombok.Data;
import lombok.EqualsAndHashCode;
import org.ruoyi.core.domain.BaseEntity;
import java.io.Serial;
import java.util.List;
/**
* 知识库角色对象 knowledge_role
*
* @author ageerle
* @date 2025-07-19
*/
@Data
@EqualsAndHashCode(callSuper = true)
@TableName("knowledge_role")
public class KnowledgeRole extends BaseEntity {
@Serial
private static final long serialVersionUID = 1L;
/**
* 知识库角色id
*/
@TableId(value = "id")
private Long id;
/**
* 知识库角色组id
*/
private Long groupId;
/**
* 知识库角色name
*/
private String name;
/**
* 删除标志（0代表存在 2代表删除）
*/
@TableLogic
private String delFlag;
/**
* 备注
*/
private String remark;
/**
* 知识库id列表
*/
@TableField(exist = false)
private List<Long> knowledgeIds;
}

49
ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/KnowledgeRoleGroup.java Normal file
View File

@@ -0,0 +1,49 @@
package org.ruoyi.domain;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableLogic;
import com.baomidou.mybatisplus.annotation.TableName;
import lombok.Data;
import lombok.EqualsAndHashCode;
import org.ruoyi.core.domain.BaseEntity;
import java.io.Serial;
/**
* 知识库角色组对象 knowledge_role_group
*
* @author ageerle
* @date 2025-07-19
*/
@Data
@EqualsAndHashCode(callSuper = true)
@TableName("knowledge_role_group")
public class KnowledgeRoleGroup extends BaseEntity {
@Serial
private static final long serialVersionUID = 1L;
/**
* 知识库角色组id
*/
@TableId(value = "id")
private Long id;
/**
* 知识库角色组name
*/
private String name;
/**
* 删除标志（0代表存在 2代表删除）
*/
@TableLogic
private String delFlag;
/**
* 备注
*/
private String remark;
}

54
ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/KnowledgeRoleRelation.java Normal file
View File

@@ -0,0 +1,54 @@
package org.ruoyi.domain;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableLogic;
import com.baomidou.mybatisplus.annotation.TableName;
import lombok.Data;
import lombok.EqualsAndHashCode;
import org.ruoyi.core.domain.BaseEntity;
import java.io.Serial;
/**
* 知识库角色与知识库关联对象 knowledge_role_relation
*
* @author ageerle
* @date 2025-07-19
*/
@Data
@EqualsAndHashCode(callSuper = true)
@TableName("knowledge_role_relation")
public class KnowledgeRoleRelation extends BaseEntity {
@Serial
private static final long serialVersionUID = 1L;
/**
* id
*/
@TableId(value = "id")
private Long id;
/**
* 删除标志（0代表存在 2代表删除）
*/
@TableLogic
private String delFlag;
/**
* 备注
*/
private String remark;
/**
* 知识库角色id
*/
private Long knowledgeRoleId;
/**
* 知识库id
*/
private Long knowledgeId;
}

54
ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/bo/KnowledgeRoleBo.java Normal file
View File

@@ -0,0 +1,54 @@
package org.ruoyi.domain.bo;
import io.github.linpeilie.annotations.AutoMapper;
import jakarta.validation.constraints.NotBlank;
import jakarta.validation.constraints.NotNull;
import lombok.Data;
import lombok.EqualsAndHashCode;
import org.ruoyi.common.core.validate.AddGroup;
import org.ruoyi.common.core.validate.EditGroup;
import org.ruoyi.core.domain.BaseEntity;
import org.ruoyi.domain.KnowledgeRole;
import java.util.List;
/**
* 知识库角色业务对象 knowledge_role
*
* @author ageerle
* @date 2025-07-19
*/
@Data
@EqualsAndHashCode(callSuper = true)
@AutoMapper(target = KnowledgeRole.class, reverseConvertGenerate = false)
public class KnowledgeRoleBo extends BaseEntity {
/**
* 知识库角色id
*/
@NotNull(message = "知识库角色id不能为空", groups = {EditGroup.class})
private Long id;
/**
* 知识库角色组id
*/
@NotNull(message = "知识库角色组id不能为空", groups = {AddGroup.class, EditGroup.class})
private Long groupId;
/**
* 知识库角色name
*/
@NotBlank(message = "知识库角色name不能为空", groups = {AddGroup.class, EditGroup.class})
private String name;
/**
* 备注
*/
private String remark;
/**
* 知识库id列表
*/
private List<Long> knowledgeIds;
}

42
ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/bo/KnowledgeRoleGroupBo.java Normal file
View File

@@ -0,0 +1,42 @@
package org.ruoyi.domain.bo;
import io.github.linpeilie.annotations.AutoMapper;
import jakarta.validation.constraints.NotBlank;
import jakarta.validation.constraints.NotNull;
import lombok.Data;
import lombok.EqualsAndHashCode;
import org.ruoyi.common.core.validate.AddGroup;
import org.ruoyi.common.core.validate.EditGroup;
import org.ruoyi.core.domain.BaseEntity;
import org.ruoyi.domain.KnowledgeRoleGroup;
/**
* 知识库角色组业务对象 knowledge_role_group
*
* @author ageerle
* @date 2025-07-19
*/
@Data
@EqualsAndHashCode(callSuper = true)
@AutoMapper(target = KnowledgeRoleGroup.class, reverseConvertGenerate = false)
public class KnowledgeRoleGroupBo extends BaseEntity {
/**
* 知识库角色组id
*/
@NotNull(message = "知识库角色组id不能为空", groups = {EditGroup.class})
private Long id;
/**
* 知识库角色组name
*/
@NotBlank(message = "知识库角色组name不能为空", groups = {AddGroup.class, EditGroup.class})
private String name;
/**
* 备注
*/
private String remark;
}

49
ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/bo/KnowledgeRoleRelationBo.java Normal file
View File

@@ -0,0 +1,49 @@
package org.ruoyi.domain.bo;
import io.github.linpeilie.annotations.AutoMapper;
import jakarta.validation.constraints.NotBlank;
import jakarta.validation.constraints.NotNull;
import lombok.Data;
import lombok.EqualsAndHashCode;
import org.ruoyi.common.core.validate.AddGroup;
import org.ruoyi.common.core.validate.EditGroup;
import org.ruoyi.core.domain.BaseEntity;
import org.ruoyi.domain.KnowledgeRoleRelation;
/**
* 知识库角色与知识库关联业务对象 knowledge_role_relation
*
* @author ageerle
* @date 2025-07-19
*/
@Data
@EqualsAndHashCode(callSuper = true)
@AutoMapper(target = KnowledgeRoleRelation.class, reverseConvertGenerate = false)
public class KnowledgeRoleRelationBo extends BaseEntity {
/**
* id
*/
@NotNull(message = "id不能为空", groups = {EditGroup.class})
private Long id;
/**
* 备注
*/
@NotBlank(message = "备注不能为空", groups = {AddGroup.class, EditGroup.class})
private String remark;
/**
* 知识库角色id
*/
@NotNull(message = "知识库角色id不能为空", groups = {AddGroup.class, EditGroup.class})
private Long knowledgeRoleId;
/**
* 知识库id
*/
@NotNull(message = "知识库id不能为空", groups = {AddGroup.class, EditGroup.class})
private Long knowledgeId;
}

71
ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/vo/KnowledgeRoleGroupVo.java Normal file
View File

@@ -0,0 +1,71 @@
package org.ruoyi.domain.vo;
import com.alibaba.excel.annotation.ExcelIgnoreUnannotated;
import com.alibaba.excel.annotation.ExcelProperty;
import io.github.linpeilie.annotations.AutoMapper;
import lombok.Data;
import org.ruoyi.domain.KnowledgeRoleGroup;
import java.io.Serial;
import java.io.Serializable;
import java.util.Date;
/**
* 知识库角色组视图对象 knowledge_role_group
*
* @author ageerle
* @date 2025-07-19
*/
@Data
@ExcelIgnoreUnannotated
@AutoMapper(target = KnowledgeRoleGroup.class)
public class KnowledgeRoleGroupVo implements Serializable {
@Serial
private static final long serialVersionUID = 1L;
/**
* 知识库角色组id
*/
@ExcelProperty(value = "知识库角色组id")
private Long id;
/**
* 知识库角色组name
*/
@ExcelProperty(value = "知识库角色组名称")
private String name;
/**
* 创建者
*/
@ExcelProperty(value = "创建者")
private Long createBy;
/**
* 创建时间
*/
@ExcelProperty(value = "创建时间")
private Date createTime;
/**
* 更新者
*/
@ExcelProperty(value = "更新时间")
private Long updateBy;
/**
* 更新时间
*/
@ExcelProperty(value = "更新时间")
private Date updateTime;
/**
* 备注
*/
@ExcelProperty(value = "备注")
private String remark;
}

52
ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/vo/KnowledgeRoleRelationVo.java Normal file
View File

@@ -0,0 +1,52 @@
package org.ruoyi.domain.vo;
import com.alibaba.excel.annotation.ExcelIgnoreUnannotated;
import com.alibaba.excel.annotation.ExcelProperty;
import io.github.linpeilie.annotations.AutoMapper;
import lombok.Data;
import org.ruoyi.domain.KnowledgeRoleRelation;
import java.io.Serial;
import java.io.Serializable;
/**
* 知识库角色与知识库关联视图对象 knowledge_role_relation
*
* @author ageerle
* @date 2025-07-19
*/
@Data
@ExcelIgnoreUnannotated
@AutoMapper(target = KnowledgeRoleRelation.class)
public class KnowledgeRoleRelationVo implements Serializable {
@Serial
private static final long serialVersionUID = 1L;
/**
* id
*/
@ExcelProperty(value = "id")
private Long id;
/**
* 备注
*/
@ExcelProperty(value = "备注")
private String remark;
/**
* 知识库角色id
*/
@ExcelProperty(value = "知识库角色id")
private Long knowledgeRoleId;
/**
* 知识库id
*/
@ExcelProperty(value = "知识库id")
private Long knowledgeId;
}

89
ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/vo/KnowledgeRoleVo.java Normal file
View File

@@ -0,0 +1,89 @@
package org.ruoyi.domain.vo;
import com.alibaba.excel.annotation.ExcelIgnoreUnannotated;
import com.alibaba.excel.annotation.ExcelProperty;
import io.github.linpeilie.annotations.AutoMapper;
import lombok.Data;
import org.ruoyi.domain.KnowledgeRole;
import java.io.Serial;
import java.io.Serializable;
import java.util.Date;
import java.util.List;
/**
* 知识库角色视图对象 knowledge_role
*
* @author ageerle
* @date 2025-07-19
*/
@Data
@ExcelIgnoreUnannotated
@AutoMapper(target = KnowledgeRole.class)
public class KnowledgeRoleVo implements Serializable {
@Serial
private static final long serialVersionUID = 1L;
/**
* 知识库角色id
*/
@ExcelProperty(value = "知识库角色id")
private Long id;
/**
* 知识库角色组id
*/
@ExcelProperty(value = "知识库角色组id")
private Long groupId;
/**
* 知识库角色name
*/
@ExcelProperty(value = "知识库角色name")
private String name;
/**
* 创建者
*/
@ExcelProperty(value = "创建者")
private Long createBy;
/**
* 创建时间
*/
@ExcelProperty(value = "创建时间")
private Date createTime;
/**
* 更新者
*/
@ExcelProperty(value = "更新时间")
private Long updateBy;
/**
* 更新时间
*/
@ExcelProperty(value = "更新时间")
private Date updateTime;
/**
* 备注
*/
@ExcelProperty(value = "备注")
private String remark;
/**
* 知识库id列表
*/
private List<Long> knowledgeIds;
/**
* 角色组名称
*/
private String groupName;
}

15
ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/mapper/KnowledgeRoleGroupMapper.java Normal file
View File

@@ -0,0 +1,15 @@
package org.ruoyi.mapper;
import org.ruoyi.core.mapper.BaseMapperPlus;
import org.ruoyi.domain.KnowledgeRoleGroup;
import org.ruoyi.domain.vo.KnowledgeRoleGroupVo;
/**
* 知识库角色组Mapper接口
*
* @author ageerle
* @date 2025-07-19
*/
public interface KnowledgeRoleGroupMapper extends BaseMapperPlus<KnowledgeRoleGroup, KnowledgeRoleGroupVo> {
}

15
ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/mapper/KnowledgeRoleMapper.java Normal file
View File

@@ -0,0 +1,15 @@
package org.ruoyi.mapper;
import org.ruoyi.core.mapper.BaseMapperPlus;
import org.ruoyi.domain.KnowledgeRole;
import org.ruoyi.domain.vo.KnowledgeRoleVo;
/**
* 知识库角色Mapper接口
*
* @author ageerle
* @date 2025-07-19
*/
public interface KnowledgeRoleMapper extends BaseMapperPlus<KnowledgeRole, KnowledgeRoleVo> {
}

38
ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/mapper/KnowledgeRoleRelationMapper.java Normal file
View File

@@ -0,0 +1,38 @@
package org.ruoyi.mapper;
import org.apache.ibatis.annotations.Delete;
import org.apache.ibatis.annotations.Param;
import org.apache.ibatis.annotations.Select;
import org.ruoyi.core.mapper.BaseMapperPlus;
import org.ruoyi.domain.KnowledgeRoleRelation;
import org.ruoyi.domain.vo.KnowledgeRoleRelationVo;
import java.util.List;
/**
* 知识库角色与知识库关联Mapper接口
*
* @author ageerle
* @date 2025-07-19
*/
public interface KnowledgeRoleRelationMapper extends BaseMapperPlus<KnowledgeRoleRelation, KnowledgeRoleRelationVo> {
@Select("SELECT knowledge_id FROM knowledge_role_relation WHERE knowledge_role_id = #{knowledgeRoleId}")
List<Long> selectKnowledgeIdsByRoleId(@Param("knowledgeRoleId") Long knowledgeRoleId);
/**
* 根据 roleId 删除所有关联
*/
@Delete("DELETE FROM knowledge_role_relation WHERE knowledge_role_id = #{knowledgeRoleId}")
void deleteByRoleId(@Param("knowledgeRoleId") Long knowledgeRoleId);
@Delete({
"<script>",
"DELETE FROM knowledge_role_relation",
"WHERE knowledge_role_id IN",
"<foreach collection='knowledgeRoleIds' item='id' open='(' separator=',' close=')'>",
"#{id}",
"</foreach>",
"</script>"
})
void deleteByRoleIds(@Param("knowledgeRoleIds") List<Long> knowledgeRoleIds);
}

5
ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/IKnowledgeInfoService.java
View File

@@ -28,6 +28,11 @@ public interface IKnowledgeInfoService {
*/
TableDataInfo<KnowledgeInfoVo> queryPageList(KnowledgeInfoBo bo, PageQuery pageQuery);
/**
* 查询知识库列表
*/
TableDataInfo<KnowledgeInfoVo> queryPageListByRole(PageQuery pageQuery);
/**
* 查询知识库列表
*/

48
ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/IKnowledgeRoleGroupService.java Normal file
View File

@@ -0,0 +1,48 @@
package org.ruoyi.service;
import org.ruoyi.core.page.PageQuery;
import org.ruoyi.core.page.TableDataInfo;
import org.ruoyi.domain.bo.KnowledgeRoleGroupBo;
import org.ruoyi.domain.vo.KnowledgeRoleGroupVo;
import java.util.Collection;
import java.util.List;
/**
* 知识库角色组Service接口
*
* @author ageerle
* @date 2025-07-19
*/
public interface IKnowledgeRoleGroupService {
/**
* 查询知识库角色组
*/
KnowledgeRoleGroupVo queryById(Long id);
/**
* 查询知识库角色组列表
*/
TableDataInfo<KnowledgeRoleGroupVo> queryPageList(KnowledgeRoleGroupBo bo, PageQuery pageQuery);
/**
* 查询知识库角色组列表
*/
List<KnowledgeRoleGroupVo> queryList(KnowledgeRoleGroupBo bo);
/**
* 新增知识库角色组
*/
Boolean insertByBo(KnowledgeRoleGroupBo bo);
/**
* 修改知识库角色组
*/
Boolean updateByBo(KnowledgeRoleGroupBo bo);
/**
* 校验并批量删除知识库角色组信息
*/
Boolean deleteWithValidByIds(Collection<Long> ids, Boolean isValid);
}

48
ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/IKnowledgeRoleRelationService.java Normal file
View File

@@ -0,0 +1,48 @@
package org.ruoyi.service;
import org.ruoyi.core.page.PageQuery;
import org.ruoyi.core.page.TableDataInfo;
import org.ruoyi.domain.bo.KnowledgeRoleRelationBo;
import org.ruoyi.domain.vo.KnowledgeRoleRelationVo;
import java.util.Collection;
import java.util.List;
/**
* 知识库角色与知识库关联Service接口
*
* @author ageerle
* @date 2025-07-19
*/
public interface IKnowledgeRoleRelationService {
/**
* 查询知识库角色与知识库关联
*/
KnowledgeRoleRelationVo queryById(Long id);
/**
* 查询知识库角色与知识库关联列表
*/
TableDataInfo<KnowledgeRoleRelationVo> queryPageList(KnowledgeRoleRelationBo bo, PageQuery pageQuery);
/**
* 查询知识库角色与知识库关联列表
*/
List<KnowledgeRoleRelationVo> queryList(KnowledgeRoleRelationBo bo);
/**
* 新增知识库角色与知识库关联
*/
Boolean insertByBo(KnowledgeRoleRelationBo bo);
/**
* 修改知识库角色与知识库关联
*/
Boolean updateByBo(KnowledgeRoleRelationBo bo);
/**
* 校验并批量删除知识库角色与知识库关联信息
*/
Boolean deleteWithValidByIds(Collection<Long> ids, Boolean isValid);
}

48
ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/IKnowledgeRoleService.java Normal file
View File

@@ -0,0 +1,48 @@
package org.ruoyi.service;
import org.ruoyi.core.page.PageQuery;
import org.ruoyi.core.page.TableDataInfo;
import org.ruoyi.domain.bo.KnowledgeRoleBo;
import org.ruoyi.domain.vo.KnowledgeRoleVo;
import java.util.Collection;
import java.util.List;
/**
* 知识库角色Service接口
*
* @author ageerle
* @date 2025-07-19
*/
public interface IKnowledgeRoleService {
/**
* 查询知识库角色
*/
KnowledgeRoleVo queryById(Long id);
/**
* 查询知识库角色列表
*/
TableDataInfo<KnowledgeRoleVo> queryPageList(KnowledgeRoleBo bo, PageQuery pageQuery);
/**
* 查询知识库角色列表
*/
List<KnowledgeRoleVo> queryList(KnowledgeRoleBo bo);
/**
* 新增知识库角色
*/
Boolean insertByBo(KnowledgeRoleBo bo);
/**
* 修改知识库角色
*/
Boolean updateByBo(KnowledgeRoleBo bo);
/**
* 校验并批量删除知识库角色信息
*/
Boolean deleteWithValidByIds(Collection<Long> ids, Boolean isValid);
}

109
ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/impl/KnowledgeRoleGroupServiceImpl.java Normal file
View File

@@ -0,0 +1,109 @@
package org.ruoyi.service.impl;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.RequiredArgsConstructor;
import org.ruoyi.common.core.utils.MapstructUtils;
import org.ruoyi.common.core.utils.StringUtils;
import org.ruoyi.core.page.PageQuery;
import org.ruoyi.core.page.TableDataInfo;
import org.ruoyi.domain.KnowledgeRoleGroup;
import org.ruoyi.domain.bo.KnowledgeRoleGroupBo;
import org.ruoyi.domain.vo.KnowledgeRoleGroupVo;
import org.ruoyi.mapper.KnowledgeRoleGroupMapper;
import org.ruoyi.service.IKnowledgeRoleGroupService;
import org.springframework.stereotype.Service;
import java.util.Collection;
import java.util.List;
import java.util.Map;
/**
* 知识库角色组Service业务层处理
*
* @author ageerle
* @date 2025-07-19
*/
@RequiredArgsConstructor
@Service
public class KnowledgeRoleGroupServiceImpl implements IKnowledgeRoleGroupService {
private final KnowledgeRoleGroupMapper baseMapper;
/**
* 查询知识库角色组
*/
@Override
public KnowledgeRoleGroupVo queryById(Long id) {
return baseMapper.selectVoById(id);
}
/**
* 查询知识库角色组列表
*/
@Override
public TableDataInfo<KnowledgeRoleGroupVo> queryPageList(KnowledgeRoleGroupBo bo, PageQuery pageQuery) {
LambdaQueryWrapper<KnowledgeRoleGroup> lqw = buildQueryWrapper(bo);
Page<KnowledgeRoleGroupVo> result = baseMapper.selectVoPage(pageQuery.build(), lqw);
return TableDataInfo.build(result);
}
/**
* 查询知识库角色组列表
*/
@Override
public List<KnowledgeRoleGroupVo> queryList(KnowledgeRoleGroupBo bo) {
LambdaQueryWrapper<KnowledgeRoleGroup> lqw = buildQueryWrapper(bo);
return baseMapper.selectVoList(lqw);
}
private LambdaQueryWrapper<KnowledgeRoleGroup> buildQueryWrapper(KnowledgeRoleGroupBo bo) {
Map<String, Object> params = bo.getParams();
LambdaQueryWrapper<KnowledgeRoleGroup> lqw = Wrappers.lambdaQuery();
lqw.like(StringUtils.isNotBlank(bo.getName()), KnowledgeRoleGroup::getName, bo.getName());
return lqw;
}
/**
* 新增知识库角色组
*/
@Override
public Boolean insertByBo(KnowledgeRoleGroupBo bo) {
KnowledgeRoleGroup add = MapstructUtils.convert(bo, KnowledgeRoleGroup.class);
validEntityBeforeSave(add);
boolean flag = baseMapper.insert(add) > 0;
if (flag) {
bo.setId(add.getId());
}
return flag;
}
/**
* 修改知识库角色组
*/
@Override
public Boolean updateByBo(KnowledgeRoleGroupBo bo) {
KnowledgeRoleGroup update = MapstructUtils.convert(bo, KnowledgeRoleGroup.class);
validEntityBeforeSave(update);
return baseMapper.updateById(update) > 0;
}
/**
* 保存前的数据校验
*/
private void validEntityBeforeSave(KnowledgeRoleGroup entity) {
//TODO 做一些数据校验,如唯一约束
}
/**
* 批量删除知识库角色组
*/
@Override
public Boolean deleteWithValidByIds(Collection<Long> ids, Boolean isValid) {
if (isValid) {
//TODO 做一些业务上的校验,判断是否需要校验
}
return baseMapper.deleteBatchIds(ids) > 0;
}
}

133
ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/impl/KnowledgeRoleRelationServiceImpl.java Normal file
View File

@@ -0,0 +1,133 @@
package org.ruoyi.service.impl;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.RequiredArgsConstructor;
import org.ruoyi.common.core.utils.MapstructUtils;
import org.ruoyi.core.page.PageQuery;
import org.ruoyi.core.page.TableDataInfo;
import org.ruoyi.domain.KnowledgeRoleRelation;
import org.ruoyi.domain.bo.KnowledgeRoleRelationBo;
import org.ruoyi.domain.vo.KnowledgeRoleRelationVo;
import org.ruoyi.mapper.KnowledgeRoleRelationMapper;
import org.ruoyi.service.IKnowledgeRoleRelationService;
import org.springframework.stereotype.Service;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Map;
/**
* 知识库角色与知识库关联Service业务层处理
*
* @author ageerle
* @date 2025-07-19
*/
@RequiredArgsConstructor
@Service
public class KnowledgeRoleRelationServiceImpl implements IKnowledgeRoleRelationService {
private final KnowledgeRoleRelationMapper baseMapper;
private final KnowledgeRoleRelationMapper knowledgeRoleRelationMapper;
/**
* 查询知识库角色与知识库关联
*/
@Override
public KnowledgeRoleRelationVo queryById(Long id) {
return baseMapper.selectVoById(id);
}
/**
* 查询知识库角色与知识库关联列表
*/
@Override
public TableDataInfo<KnowledgeRoleRelationVo> queryPageList(KnowledgeRoleRelationBo bo, PageQuery pageQuery) {
LambdaQueryWrapper<KnowledgeRoleRelation> lqw = buildQueryWrapper(bo);
Page<KnowledgeRoleRelationVo> result = baseMapper.selectVoPage(pageQuery.build(), lqw);
return TableDataInfo.build(result);
}
/**
* 查询知识库角色与知识库关联列表
*/
@Override
public List<KnowledgeRoleRelationVo> queryList(KnowledgeRoleRelationBo bo) {
LambdaQueryWrapper<KnowledgeRoleRelation> lqw = buildQueryWrapper(bo);
return baseMapper.selectVoList(lqw);
}
private LambdaQueryWrapper<KnowledgeRoleRelation> buildQueryWrapper(KnowledgeRoleRelationBo bo) {
Map<String, Object> params = bo.getParams();
LambdaQueryWrapper<KnowledgeRoleRelation> lqw = Wrappers.lambdaQuery();
lqw.eq(bo.getKnowledgeRoleId() != null, KnowledgeRoleRelation::getKnowledgeRoleId, bo.getKnowledgeRoleId());
lqw.eq(bo.getKnowledgeId() != null, KnowledgeRoleRelation::getKnowledgeId, bo.getKnowledgeId());
return lqw;
}
/**
* 新增知识库角色与知识库关联
*/
@Override
public Boolean insertByBo(KnowledgeRoleRelationBo bo) {
KnowledgeRoleRelation add = MapstructUtils.convert(bo, KnowledgeRoleRelation.class);
validEntityBeforeSave(add);
boolean flag = baseMapper.insert(add) > 0;
if (flag) {
bo.setId(add.getId());
}
return flag;
}
/**
* 修改知识库角色与知识库关联
*/
@Override
public Boolean updateByBo(KnowledgeRoleRelationBo bo) {
KnowledgeRoleRelation update = MapstructUtils.convert(bo, KnowledgeRoleRelation.class);
validEntityBeforeSave(update);
return baseMapper.updateById(update) > 0;
}
/**
* 保存前的数据校验
*/
private void validEntityBeforeSave(KnowledgeRoleRelation entity) {
//TODO 做一些数据校验,如唯一约束
}
/**
* 批量删除知识库角色与知识库关联
*/
@Override
public Boolean deleteWithValidByIds(Collection<Long> ids, Boolean isValid) {
if (isValid) {
//TODO 做一些业务上的校验,判断是否需要校验
}
return baseMapper.deleteBatchIds(ids) > 0;
}
public void saveRoleKnowledgeRelations(Long knowledgeRoleId, List<Long> knowledgeIds) {
if (knowledgeRoleId == null) {
throw new IllegalArgumentException("knowledgeRoleId不能为空");
}
// 1. 删除旧的关联记录
knowledgeRoleRelationMapper.deleteByRoleId(knowledgeRoleId);
// 2. 插入新的关联记录
if (CollectionUtils.isNotEmpty(knowledgeIds)) {
List<KnowledgeRoleRelation> insertList = new ArrayList<>();
knowledgeIds.forEach(knowledgeId -> {
KnowledgeRoleRelation knowledgeRoleRelation = new KnowledgeRoleRelation();
knowledgeRoleRelation.setKnowledgeId(knowledgeId);
knowledgeRoleRelation.setKnowledgeRoleId(knowledgeRoleId);
insertList.add(knowledgeRoleRelation);
});
knowledgeRoleRelationMapper.insertBatch(insertList);
}
}
}

213
ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/impl/KnowledgeRoleServiceImpl.java Normal file
View File

@@ -0,0 +1,213 @@
package org.ruoyi.service.impl;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.RequiredArgsConstructor;
import org.ruoyi.common.core.utils.MapstructUtils;
import org.ruoyi.core.page.PageQuery;
import org.ruoyi.core.page.TableDataInfo;
import org.ruoyi.domain.KnowledgeRole;
import org.ruoyi.domain.KnowledgeRoleGroup;
import org.ruoyi.domain.KnowledgeRoleRelation;
import org.ruoyi.domain.bo.KnowledgeRoleBo;
import org.ruoyi.domain.vo.KnowledgeRoleVo;
import org.ruoyi.mapper.KnowledgeRoleGroupMapper;
import org.ruoyi.mapper.KnowledgeRoleMapper;
import org.ruoyi.mapper.KnowledgeRoleRelationMapper;
import org.ruoyi.service.IKnowledgeRoleService;
import org.springframework.stereotype.Service;
import java.util.*;
import java.util.stream.Collectors;
/**
* 知识库角色Service业务层处理
*
* @author ageerle
* @date 2025-07-19
*/
@RequiredArgsConstructor
@Service
public class KnowledgeRoleServiceImpl implements IKnowledgeRoleService {
private final KnowledgeRoleMapper baseMapper;
private final KnowledgeRoleGroupMapper knowledgeRoleGroupMapper;
private final KnowledgeRoleRelationMapper knowledgeRoleRelationMapper;
private final KnowledgeRoleRelationServiceImpl knowledgeRoleRelationServiceImpl;
/**
* 查询知识库角色
*/
@Override
public KnowledgeRoleVo queryById(Long id) {
KnowledgeRoleVo vo = baseMapper.selectVoById(id);
fillKnowledgeIds(vo);
fillKnowledgeRoleGroupName(vo);
return vo;
}
/**
* 查询知识库角色列表
*/
@Override
public TableDataInfo<KnowledgeRoleVo> queryPageList(KnowledgeRoleBo bo, PageQuery pageQuery) {
LambdaQueryWrapper<KnowledgeRole> lqw = buildQueryWrapper(bo);
Page<KnowledgeRoleVo> result = baseMapper.selectVoPage(pageQuery.build(), lqw);
fillKnowledgeIds(result.getRecords());
fillKnowledgeRoleGroupName(result.getRecords());
return TableDataInfo.build(result);
}
/**
* 查询知识库角色列表
*/
@Override
public List<KnowledgeRoleVo> queryList(KnowledgeRoleBo bo) {
LambdaQueryWrapper<KnowledgeRole> lqw = buildQueryWrapper(bo);
List<KnowledgeRoleVo> knowledgeRoleVos = baseMapper.selectVoList(lqw);
fillKnowledgeIds(knowledgeRoleVos);
fillKnowledgeRoleGroupName(knowledgeRoleVos);
return knowledgeRoleVos;
}
private LambdaQueryWrapper<KnowledgeRole> buildQueryWrapper(KnowledgeRoleBo bo) {
Map<String, Object> params = bo.getParams();
LambdaQueryWrapper<KnowledgeRole> lqw = Wrappers.lambdaQuery();
lqw.eq(bo.getGroupId() != null, KnowledgeRole::getGroupId, bo.getGroupId());
return lqw;
}
/**
* 新增知识库角色
*/
@Override
public Boolean insertByBo(KnowledgeRoleBo bo) {
KnowledgeRole add = MapstructUtils.convert(bo, KnowledgeRole.class);
validEntityBeforeSave(add);
boolean flag = baseMapper.insert(add) > 0;
if (flag) {
bo.setId(add.getId());
// 生成relation数据
knowledgeRoleRelationServiceImpl.saveRoleKnowledgeRelations(bo.getId(), bo.getKnowledgeIds());
}
return flag;
}
/**
* 修改知识库角色
*/
@Override
public Boolean updateByBo(KnowledgeRoleBo bo) {
KnowledgeRole update = MapstructUtils.convert(bo, KnowledgeRole.class);
validEntityBeforeSave(update);
int count = baseMapper.updateById(update);
if (count > 0) {
// 生成relation数据
knowledgeRoleRelationServiceImpl.saveRoleKnowledgeRelations(bo.getId(), bo.getKnowledgeIds());
}
return count > 0;
}
/**
* 保存前的数据校验
*/
private void validEntityBeforeSave(KnowledgeRole entity) {
//TODO 做一些数据校验,如唯一约束
}
/**
* 批量删除知识库角色
*/
@Override
public Boolean deleteWithValidByIds(Collection<Long> ids, Boolean isValid) {
if (isValid) {
//TODO 做一些业务上的校验,判断是否需要校验
}
int count = baseMapper.deleteBatchIds(ids);
if (count > 0) {
knowledgeRoleRelationMapper.deleteByRoleIds(ids.stream().toList());
}
return count > 0;
}
/**
* 填充 VO 的 knowledgeIds 字段（从中间表查询）
*/
private void fillKnowledgeIds(KnowledgeRoleVo vo) {
if (vo == null) return;
List<Long> knowledgeIds = knowledgeRoleRelationMapper.selectKnowledgeIdsByRoleId(vo.getId());
vo.setKnowledgeIds(knowledgeIds);
}
private void fillKnowledgeIds(List<KnowledgeRoleVo> list) {
if (CollectionUtils.isEmpty(list)) return;
// 1. 提取所有 roleId
Set<Long> roleIds = list.stream()
.map(KnowledgeRoleVo::getId)
.filter(Objects::nonNull)
.collect(Collectors.toSet());
if (roleIds.isEmpty()) return;
// 2. 一次性查询所有 roleId 对应的 knowledgeId 列表
List<KnowledgeRoleRelation> relations = knowledgeRoleRelationMapper.selectList(
new LambdaQueryWrapper<KnowledgeRoleRelation>()
.in(KnowledgeRoleRelation::getKnowledgeRoleId, roleIds)
);
// 3. 转为 Map<roleId, List<knowledgeId>>
Map<Long, List<Long>> roleIdToKnowledgeIds = relations.stream()
.collect(Collectors.groupingBy(
KnowledgeRoleRelation::getKnowledgeRoleId,
Collectors.mapping(KnowledgeRoleRelation::getKnowledgeId, Collectors.toList())
));
// 4. 回填到 VO 中
for (KnowledgeRoleVo vo : list) {
vo.setKnowledgeIds(roleIdToKnowledgeIds.getOrDefault(vo.getId(), Collections.emptyList()));
}
}
/**
* 填充 VO 的 knowledgeRoleGroupName 字段
*/
private void fillKnowledgeRoleGroupName(KnowledgeRoleVo vo) {
if (vo == null || vo.getGroupId() == null) return;
KnowledgeRoleGroup group = knowledgeRoleGroupMapper.selectById(vo.getGroupId());
if (group != null) {
vo.setGroupName(group.getName());
}
}
private void fillKnowledgeRoleGroupName(List<KnowledgeRoleVo> list) {
if (CollectionUtils.isEmpty(list)) return;
// 1. 提取所有 groupId（去重）
Set<Long> groupIds = list.stream()
.map(KnowledgeRoleVo::getGroupId)
.filter(Objects::nonNull)
.collect(Collectors.toSet());
if (groupIds.isEmpty()) return;
// 2. 一次性查出所有角色组信息
List<KnowledgeRoleGroup> groupList = knowledgeRoleGroupMapper.selectBatchIds(groupIds);
// 3. 转为 Map<id, name>
Map<Long, String> groupNameMap = groupList.stream()
.collect(Collectors.toMap(KnowledgeRoleGroup::getId, KnowledgeRoleGroup::getName));
// 4. 回填到每个 VO
for (KnowledgeRoleVo vo : list) {
vo.setGroupName(groupNameMap.get(vo.getGroupId()));
}
}
}

28
ruoyi-modules-api/ruoyi-system-api/src/main/java/org/ruoyi/system/domain/SysUser.java
View File

@@ -76,9 +76,9 @@ public class SysUser extends TenantEntity {
* 密码
*/
@TableField(
insertStrategy = FieldStrategy.NOT_EMPTY,
updateStrategy = FieldStrategy.NOT_EMPTY,
whereStrategy = FieldStrategy.NOT_EMPTY
insertStrategy = FieldStrategy.NOT_EMPTY,
updateStrategy = FieldStrategy.NOT_EMPTY,
whereStrategy = FieldStrategy.NOT_EMPTY
)
private String password;
@@ -113,13 +113,19 @@ public class SysUser extends TenantEntity {
*/
private String remark;
/** 普通用户的标识,对当前开发者帐号唯一。一个openid对应一个公众号或小程序 */
/**
* 普通用户的标识,对当前开发者帐号唯一。一个openid对应一个公众号或小程序
*/
private String openId;
/** 用户余额 */
/**
* 用户余额
*/
private Double userBalance;
/** 用户等级 */
/**
* 用户等级
*/
private String userGrade;
public SysUser(Long userId) {
@@ -130,4 +136,14 @@ public class SysUser extends TenantEntity {
return UserConstants.SUPER_ADMIN_ID.equals(this.userId);
}
/**
* 知识库角色组类型（role/roleGroup）
*/
private String kroleGroupType;
/**
* 知识库角色组id（role/roleGroup）
*/
private String kroleGroupIds;
}

22
ruoyi-modules-api/ruoyi-system-api/src/main/java/org/ruoyi/system/domain/bo/SysUserBo.java
View File

@@ -120,13 +120,19 @@ public class SysUserBo extends BaseEntity {
*/
private Long roleId;
/** 普通用户的标识,对当前开发者帐号唯一。一个openid对应一个公众号或小程序 */
/**
* 普通用户的标识,对当前开发者帐号唯一。一个openid对应一个公众号或小程序
*/
private String openId;
/** 用户等级 */
/**
* 用户等级
*/
private String userGrade;
/** 用户余额 */
/**
* 用户余额
*/
private Double userBalance;
public SysUserBo(Long userId) {
@@ -137,4 +143,14 @@ public class SysUserBo extends BaseEntity {
return UserConstants.SUPER_ADMIN_ID.equals(this.userId);
}
/**
* 知识库角色组类型（role/roleGroup）
*/
private String kroleGroupType;
/**
* 知识库角色组id（role/roleGroup）
*/
private String kroleGroupIds;
}

18
ruoyi-modules-api/ruoyi-system-api/src/main/java/org/ruoyi/system/domain/vo/SysUserVo.java
View File

@@ -147,9 +147,23 @@ public class SysUserVo implements Serializable {
*/
private Long roleId;
/** 用户等级 */
/**
* 用户等级
*/
private String userGrade;
/** 用户余额 */
/**
* 用户余额
*/
private Double userBalance;
/**
* 知识库角色组类型（role/roleGroup）
*/
private String kroleGroupType;
/**
* 知识库角色组id（role/roleGroup）
*/
private String kroleGroupIds;
}

95
ruoyi-modules-api/ruoyi-system-api/src/main/resources/mapper/SysUserMapper.xml
View File

@@ -44,6 +44,8 @@
u.create_by,
u.create_time,
u.remark,
u.krole_group_type,
u.krole_group_ids,
d.dept_id,
d.parent_id,
d.ancestors,
@@ -58,50 +60,99 @@
r.data_scope,
r.status as role_status
from sys_user u
left join sys_dept d on u.dept_id = d.dept_id
left join sys_user_role sur on u.user_id = sur.user_id
left join sys_role r on r.role_id = sur.role_id
left join sys_dept d on u.dept_id = d.dept_id
left join sys_user_role sur on u.user_id = sur.user_id
left join sys_role r on r.role_id = sur.role_id
</sql>
<update id="updateXcxUser">
update sys_user
set nick_name = #{nickName},
wx_avatar = #{wxAvatar}
WHERE user_id = #{userId}
WHERE user_id = #{userId}
</update>
<select id="selectPageUserList" resultMap="SysUserResult">
select u.user_id, u.dept_id, u.nick_name, u.user_name, u.email, u.avatar, u.phonenumber, u.sex,u.user_balance,u.user_grade,u.domain_name,
u.status, u.del_flag, u.login_ip, u.login_date, u.create_by, u.create_time, u.remark, d.dept_name, d.leader
select u.user_id,
u.dept_id,
u.nick_name,
u.user_name,
u.email,
u.avatar,
u.phonenumber,
u.sex,
u.user_balance,
u.user_grade,
u.domain_name,
u.status,
u.del_flag,
u.login_ip,
u.login_date,
u.create_by,
u.create_time,
u.remark,
d.dept_name,
d.leader
from sys_user u
left join sys_dept d on u.dept_id = d.dept_id
${ew.getCustomSqlSegment}
left join sys_dept d on u.dept_id = d.dept_id
${ew.getCustomSqlSegment}
</select>
<select id="selectUserList" resultMap="SysUserResult">
select u.user_id, u.dept_id, u.nick_name, u.user_name, u.email, u.avatar, u.phonenumber, u.sex,u.user_grade,u.user_balance,
u.status, u.del_flag, u.login_ip, u.login_date, u.create_by, u.create_time, u.remark, d.dept_name, d.leader
select u.user_id,
u.dept_id,
u.nick_name,
u.user_name,
u.email,
u.avatar,
u.phonenumber,
u.sex,
u.user_grade,
u.user_balance,
u.status,
u.del_flag,
u.login_ip,
u.login_date,
u.create_by,
u.create_time,
u.remark,
d.dept_name,
d.leader
from sys_user u
left join sys_dept d on u.dept_id = d.dept_id
${ew.getCustomSqlSegment}
left join sys_dept d on u.dept_id = d.dept_id
${ew.getCustomSqlSegment}
</select>
<select id="selectAllocatedList" resultMap="SysUserResult">
select distinct u.user_id, u.dept_id, u.user_name, u.nick_name, u.email, u.phonenumber, u.status, u.create_time
select distinct u.user_id,
u.dept_id,
u.user_name,
u.nick_name,
u.email,
u.phonenumber,
u.status,
u.create_time
from sys_user u
left join sys_dept d on u.dept_id = d.dept_id
left join sys_user_role sur on u.user_id = sur.user_id
left join sys_role r on r.role_id = sur.role_id
${ew.getCustomSqlSegment}
left join sys_dept d on u.dept_id = d.dept_id
left join sys_user_role sur on u.user_id = sur.user_id
left join sys_role r on r.role_id = sur.role_id
${ew.getCustomSqlSegment}
</select>
<select id="selectUnallocatedList" resultMap="SysUserResult">
select distinct u.user_id, u.dept_id, u.user_name, u.nick_name, u.email, u.phonenumber, u.status, u.create_time
select distinct u.user_id,
u.dept_id,
u.user_name,
u.nick_name,
u.email,
u.phonenumber,
u.status,
u.create_time
from sys_user u
left join sys_dept d on u.dept_id = d.dept_id
left join sys_user_role sur on u.user_id = sur.user_id
left join sys_role r on r.role_id = sur.role_id
${ew.getCustomSqlSegment}
left join sys_dept d on u.dept_id = d.dept_id
left join sys_user_role sur on u.user_id = sur.user_id
left join sys_role r on r.role_id = sur.role_id
${ew.getCustomSqlSegment}
</select>
<select id="selectUserByUserName" parameterType="String" resultMap="SysUserResult">

17
ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/config/KnowledgeRoleConfig.java Normal file
View File

@@ -0,0 +1,17 @@
package org.ruoyi.chat.config;
import lombok.Data;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
/**
* @Description:
* @Author: violateer
* @Date: 2025/7/20
*/
@Data
@Component
public class KnowledgeRoleConfig {
@Value("${knowledge-role.enable}")
private Boolean enable;
}

223
ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/controller/knowledge/KnowledgeController.java
View File

@@ -5,7 +5,9 @@ import jakarta.servlet.http.HttpServletResponse;
import jakarta.validation.constraints.NotEmpty;
import jakarta.validation.constraints.NotNull;
import lombok.RequiredArgsConstructor;
import org.ruoyi.chat.config.KnowledgeRoleConfig;
import org.ruoyi.common.core.domain.R;
import org.ruoyi.common.core.domain.model.LoginUser;
import org.ruoyi.common.core.validate.AddGroup;
import org.ruoyi.common.excel.utils.ExcelUtil;
import org.ruoyi.common.log.annotation.Log;
@@ -42,120 +44,141 @@ import java.util.List;
@RequestMapping("/knowledge")
public class KnowledgeController extends BaseController {
private final IKnowledgeInfoService knowledgeInfoService;
private final IKnowledgeInfoService knowledgeInfoService;
private final IKnowledgeAttachService attachService;
private final IKnowledgeAttachService attachService;
private final IKnowledgeFragmentService fragmentService;
private final IKnowledgeFragmentService fragmentService;
/**
* 根据用户信息查询本地知识库
*/
@GetMapping("/list")
public TableDataInfo<KnowledgeInfoVo> list(KnowledgeInfoBo bo, PageQuery pageQuery) {
if (!StpUtil.isLogin()) {
throw new SecurityException("请先去登录!");
private final KnowledgeRoleConfig knowledgeRoleConfig;
/**
* 根据用户信息查询本地知识库
*/
@GetMapping("/list")
public TableDataInfo<KnowledgeInfoVo> list(KnowledgeInfoBo bo, PageQuery pageQuery) {
if (!StpUtil.isLogin()) {
throw new SecurityException("请先去登录!");
}
bo.setUid(LoginHelper.getUserId());
return knowledgeInfoService.queryPageList(bo, pageQuery);
}
bo.setUid(LoginHelper.getUserId());
return knowledgeInfoService.queryPageList(bo, pageQuery);
}
/**
* 新增知识库
*/
@Log(title = "知识库", businessType = BusinessType.INSERT)
@PostMapping("/save")
public R<Void> save(@Validated(AddGroup.class) @RequestBody KnowledgeInfoBo bo) {
knowledgeInfoService.saveOne(bo);
return R.ok();
}
/**
* 根据用户信息及知识库角色查询本地知识库
*/
@GetMapping("/listByRole")
public TableDataInfo<KnowledgeInfoVo> listByRole(KnowledgeInfoBo bo, PageQuery pageQuery) {
if (!StpUtil.isLogin()) {
throw new SecurityException("请先去登录!");
}
LoginUser loginUser = LoginHelper.getLoginUser();
/**
* 删除知识库
*/
@PostMapping("/remove/{id}")
public R<String> remove(@PathVariable String id) {
knowledgeInfoService.removeKnowledge(id);
return R.ok("删除知识库成功!");
}
// 管理员跳过权限
if (loginUser.getUserId().equals(1L) || !knowledgeRoleConfig.getEnable()) {
bo.setUid(LoginHelper.getUserId());
return knowledgeInfoService.queryPageList(bo, pageQuery);
} else {
return knowledgeInfoService.queryPageListByRole(pageQuery);
}
}
/**
* 修改知识库
*/
@Log(title = "知识库", businessType = BusinessType.UPDATE)
@PostMapping("/edit")
public R<Void> edit(@RequestBody KnowledgeInfoBo bo) {
return toAjax(knowledgeInfoService.updateByBo(bo));
}
/**
* 新增知识库
*/
@Log(title = "知识库", businessType = BusinessType.INSERT)
@PostMapping("/save")
public R<Void> save(@Validated(AddGroup.class) @RequestBody KnowledgeInfoBo bo) {
knowledgeInfoService.saveOne(bo);
return R.ok();
}
/**
* 导出知识库列表
*/
@Log(title = "知识库", businessType = BusinessType.EXPORT)
@PostMapping("/export")
public void export(KnowledgeInfoBo bo, HttpServletResponse response) {
List<KnowledgeInfoVo> list = knowledgeInfoService.queryList(bo);
ExcelUtil.exportExcel(list, "知识库", KnowledgeInfoVo.class, response);
}
/**
* 删除知识库
*/
@PostMapping("/remove/{id}")
public R<String> remove(@PathVariable String id) {
knowledgeInfoService.removeKnowledge(id);
return R.ok("删除知识库成功!");
}
/**
* 查询知识附件信息
*/
@GetMapping("/detail/{kid}")
public TableDataInfo<KnowledgeAttachVo> attach(KnowledgeAttachBo bo, PageQuery pageQuery,
@PathVariable String kid) {
bo.setKid(kid);
return attachService.queryPageList(bo, pageQuery);
}
/**
* 修改知识库
*/
@Log(title = "知识库", businessType = BusinessType.UPDATE)
@PostMapping("/edit")
public R<Void> edit(@RequestBody KnowledgeInfoBo bo) {
return toAjax(knowledgeInfoService.updateByBo(bo));
}
/**
* 上传知识库附件
*/
@PostMapping(value = "/attach/upload")
public R<String> upload(KnowledgeInfoUploadBo bo) throws Exception {
knowledgeInfoService.upload(bo);
return R.ok("上传知识库附件成功!");
}
/**
* 导出知识库列表
*/
@Log(title = "知识库", businessType = BusinessType.EXPORT)
@PostMapping("/export")
public void export(KnowledgeInfoBo bo, HttpServletResponse response) {
List<KnowledgeInfoVo> list = knowledgeInfoService.queryList(bo);
ExcelUtil.exportExcel(list, "知识库", KnowledgeInfoVo.class, response);
}
/**
* 获取知识库附件详细信息
*
* @param id 主键
*/
@GetMapping("attach/info/{id}")
public R<KnowledgeAttachVo> getAttachInfo(@NotNull(message = "主键不能为空")
@PathVariable Long id) {
return R.ok(attachService.queryById(id));
}
/**
* 查询知识附件信息
*/
@GetMapping("/detail/{kid}")
public TableDataInfo<KnowledgeAttachVo> attach(KnowledgeAttachBo bo, PageQuery pageQuery,
@PathVariable String kid) {
bo.setKid(kid);
return attachService.queryPageList(bo, pageQuery);
}
/**
* 删除知识库附件
*/
@PostMapping("attach/remove/{kid}")
public R<Void> removeAttach(@NotEmpty(message = "主键不能为空")
@PathVariable String kid) {
attachService.removeKnowledgeAttach(kid);
return R.ok();
}
/**
* 上传知识库附件
*/
@PostMapping(value = "/attach/upload")
public R<String> upload(KnowledgeInfoUploadBo bo) throws Exception {
knowledgeInfoService.upload(bo);
return R.ok("上传知识库附件成功!");
}
/**
* 获取知识库附件详细信息
*
* @param id 主键
*/
@GetMapping("attach/info/{id}")
public R<KnowledgeAttachVo> getAttachInfo(@NotNull(message = "主键不能为空")
@PathVariable Long id) {
return R.ok(attachService.queryById(id));
}
/**
* 删除知识库附件
*/
@PostMapping("attach/remove/{kid}")
public R<Void> removeAttach(@NotEmpty(message = "主键不能为空")
@PathVariable String kid) {
attachService.removeKnowledgeAttach(kid);
return R.ok();
}
/**
* 查询知识片段
*/
@GetMapping("/fragment/list/{docId}")
public TableDataInfo<KnowledgeFragmentVo> fragmentList(KnowledgeFragmentBo bo,
PageQuery pageQuery, @PathVariable String docId) {
bo.setDocId(docId);
return fragmentService.queryPageList(bo, pageQuery);
}
/**
* 查询知识片段
*/
@GetMapping("/fragment/list/{docId}")
public TableDataInfo<KnowledgeFragmentVo> fragmentList(KnowledgeFragmentBo bo,
PageQuery pageQuery, @PathVariable String docId) {
bo.setDocId(docId);
return fragmentService.queryPageList(bo, pageQuery);
}
/**
* 上传文件翻译
*/
@PostMapping("/translationByFile")
@ResponseBody
public String translationByFile(@RequestParam("file") MultipartFile file, String targetLanguage) {
return attachService.translationByFile(file, targetLanguage);
}
/**
* 上传文件翻译
*/
@PostMapping("/translationByFile")
@ResponseBody
public String translationByFile(@RequestParam("file") MultipartFile file, String targetLanguage) {
return attachService.translationByFile(file, targetLanguage);
}
}

99
ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/controller/knowledge/KnowledgeRoleController.java Normal file
View File

@@ -0,0 +1,99 @@
package org.ruoyi.chat.controller.knowledge;
import jakarta.servlet.http.HttpServletResponse;
import jakarta.validation.constraints.NotEmpty;
import jakarta.validation.constraints.NotNull;
import lombok.RequiredArgsConstructor;
import org.ruoyi.common.core.domain.R;
import org.ruoyi.common.core.validate.AddGroup;
import org.ruoyi.common.core.validate.EditGroup;
import org.ruoyi.common.excel.utils.ExcelUtil;
import org.ruoyi.common.idempotent.annotation.RepeatSubmit;
import org.ruoyi.common.log.annotation.Log;
import org.ruoyi.common.log.enums.BusinessType;
import org.ruoyi.common.web.core.BaseController;
import org.ruoyi.core.page.PageQuery;
import org.ruoyi.core.page.TableDataInfo;
import org.ruoyi.domain.bo.KnowledgeRoleBo;
import org.ruoyi.domain.vo.KnowledgeRoleVo;
import org.ruoyi.service.IKnowledgeRoleService;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
import java.util.List;
/**
* 知识库角色
*
* @author ageerle
* @date 2025-07-19
*/
@Validated
@RequiredArgsConstructor
@RestController
@RequestMapping("/knowledgeRole")
public class KnowledgeRoleController extends BaseController {
private final IKnowledgeRoleService knowledgeRoleService;
/**
* 查询知识库角色列表
*/
@GetMapping("/list")
public TableDataInfo<KnowledgeRoleVo> list(KnowledgeRoleBo bo, PageQuery pageQuery) {
return knowledgeRoleService.queryPageList(bo, pageQuery);
}
/**
* 导出知识库角色列表
*/
@Log(title = "知识库角色", businessType = BusinessType.EXPORT)
@PostMapping("/export")
public void export(KnowledgeRoleBo bo, HttpServletResponse response) {
List<KnowledgeRoleVo> list = knowledgeRoleService.queryList(bo);
ExcelUtil.exportExcel(list, "知识库角色", KnowledgeRoleVo.class, response);
}
/**
* 获取知识库角色详细信息
*
* @param id 主键
*/
@GetMapping("/{id}")
public R<KnowledgeRoleVo> getInfo(@NotNull(message = "主键不能为空")
@PathVariable Long id) {
return R.ok(knowledgeRoleService.queryById(id));
}
/**
* 新增知识库角色
*/
@Log(title = "知识库角色", businessType = BusinessType.INSERT)
@RepeatSubmit()
@PostMapping()
public R<Void> add(@Validated(AddGroup.class) @RequestBody KnowledgeRoleBo bo) {
return toAjax(knowledgeRoleService.insertByBo(bo));
}
/**
* 修改知识库角色
*/
@Log(title = "知识库角色", businessType = BusinessType.UPDATE)
@RepeatSubmit()
@PutMapping()
public R<Void> edit(@Validated(EditGroup.class) @RequestBody KnowledgeRoleBo bo) {
return toAjax(knowledgeRoleService.updateByBo(bo));
}
/**
* 删除知识库角色
*
* @param ids 主键串
*/
@Log(title = "知识库角色", businessType = BusinessType.DELETE)
@DeleteMapping("/{ids}")
public R<Void> remove(@NotEmpty(message = "主键不能为空")
@PathVariable Long[] ids) {
return toAjax(knowledgeRoleService.deleteWithValidByIds(List.of(ids), true));
}
}

99
ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/controller/knowledge/KnowledgeRoleGroupController.java Normal file
View File

@@ -0,0 +1,99 @@
package org.ruoyi.chat.controller.knowledge;
import jakarta.servlet.http.HttpServletResponse;
import jakarta.validation.constraints.NotEmpty;
import jakarta.validation.constraints.NotNull;
import lombok.RequiredArgsConstructor;
import org.ruoyi.common.core.domain.R;
import org.ruoyi.common.core.validate.AddGroup;
import org.ruoyi.common.core.validate.EditGroup;
import org.ruoyi.common.excel.utils.ExcelUtil;
import org.ruoyi.common.idempotent.annotation.RepeatSubmit;
import org.ruoyi.common.log.annotation.Log;
import org.ruoyi.common.log.enums.BusinessType;
import org.ruoyi.common.web.core.BaseController;
import org.ruoyi.core.page.PageQuery;
import org.ruoyi.core.page.TableDataInfo;
import org.ruoyi.domain.bo.KnowledgeRoleGroupBo;
import org.ruoyi.domain.vo.KnowledgeRoleGroupVo;
import org.ruoyi.service.IKnowledgeRoleGroupService;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
import java.util.List;
/**
* 知识库角色组
*
* @author ageerle
* @date 2025-07-19
*/
@Validated
@RequiredArgsConstructor
@RestController
@RequestMapping("/knowledgeRoleGroup")
public class KnowledgeRoleGroupController extends BaseController {
private final IKnowledgeRoleGroupService knowledgeRoleGroupService;
/**
* 查询知识库角色组列表
*/
@GetMapping("/list")
public TableDataInfo<KnowledgeRoleGroupVo> list(KnowledgeRoleGroupBo bo, PageQuery pageQuery) {
return knowledgeRoleGroupService.queryPageList(bo, pageQuery);
}
/**
* 导出知识库角色组列表
*/
@Log(title = "知识库角色组", businessType = BusinessType.EXPORT)
@PostMapping("/export")
public void export(KnowledgeRoleGroupBo bo, HttpServletResponse response) {
List<KnowledgeRoleGroupVo> list = knowledgeRoleGroupService.queryList(bo);
ExcelUtil.exportExcel(list, "知识库角色组", KnowledgeRoleGroupVo.class, response);
}
/**
* 获取知识库角色组详细信息
*
* @param id 主键
*/
@GetMapping("/{id}")
public R<KnowledgeRoleGroupVo> getInfo(@NotNull(message = "主键不能为空")
@PathVariable Long id) {
return R.ok(knowledgeRoleGroupService.queryById(id));
}
/**
* 新增知识库角色组
*/
@Log(title = "知识库角色组", businessType = BusinessType.INSERT)
@RepeatSubmit()
@PostMapping()
public R<Void> add(@Validated(AddGroup.class) @RequestBody KnowledgeRoleGroupBo bo) {
return toAjax(knowledgeRoleGroupService.insertByBo(bo));
}
/**
* 修改知识库角色组
*/
@Log(title = "知识库角色组", businessType = BusinessType.UPDATE)
@RepeatSubmit()
@PutMapping()
public R<Void> edit(@Validated(EditGroup.class) @RequestBody KnowledgeRoleGroupBo bo) {
return toAjax(knowledgeRoleGroupService.updateByBo(bo));
}
/**
* 删除知识库角色组
*
* @param ids 主键串
*/
@Log(title = "知识库角色组", businessType = BusinessType.DELETE)
@DeleteMapping("/{ids}")
public R<Void> remove(@NotEmpty(message = "主键不能为空")
@PathVariable Long[] ids) {
return toAjax(knowledgeRoleGroupService.deleteWithValidByIds(List.of(ids), true));
}
}

449
ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/knowledge/KnowledgeInfoServiceImpl.java
View File

@@ -3,6 +3,7 @@ package org.ruoyi.chat.service.knowledge;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.RandomUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.RequiredArgsConstructor;
@@ -14,17 +15,13 @@ import org.ruoyi.common.core.utils.StringUtils;
import org.ruoyi.common.satoken.utils.LoginHelper;
import org.ruoyi.core.page.PageQuery;
import org.ruoyi.core.page.TableDataInfo;
import org.ruoyi.domain.KnowledgeAttach;
import org.ruoyi.domain.KnowledgeFragment;
import org.ruoyi.domain.KnowledgeInfo;
import org.ruoyi.domain.*;
import org.ruoyi.domain.bo.KnowledgeInfoBo;
import org.ruoyi.domain.bo.KnowledgeInfoUploadBo;
import org.ruoyi.domain.bo.StoreEmbeddingBo;
import org.ruoyi.domain.vo.ChatModelVo;
import org.ruoyi.domain.vo.KnowledgeInfoVo;
import org.ruoyi.mapper.KnowledgeAttachMapper;
import org.ruoyi.mapper.KnowledgeFragmentMapper;
import org.ruoyi.mapper.KnowledgeInfoMapper;
import org.ruoyi.mapper.*;
import org.ruoyi.service.IChatModelService;
import org.ruoyi.service.IKnowledgeInfoService;
import org.ruoyi.service.VectorStoreService;
@@ -37,6 +34,7 @@ import org.springframework.web.multipart.MultipartFile;
import java.io.IOException;
import java.util.*;
import java.util.stream.Collectors;
/**
@@ -49,219 +47,266 @@ import java.util.*;
@Service
public class KnowledgeInfoServiceImpl implements IKnowledgeInfoService {
private static final Logger log = LoggerFactory.getLogger(KnowledgeInfoServiceImpl.class);
private static final Logger log = LoggerFactory.getLogger(KnowledgeInfoServiceImpl.class);
private final KnowledgeInfoMapper baseMapper;
private final KnowledgeInfoMapper baseMapper;
private final VectorStoreService vectorStoreService;
private final VectorStoreService vectorStoreService;
private final ResourceLoaderFactory resourceLoaderFactory;
private final ResourceLoaderFactory resourceLoaderFactory;
private final KnowledgeFragmentMapper fragmentMapper;
private final KnowledgeFragmentMapper fragmentMapper;
private final KnowledgeAttachMapper attachMapper;
private final KnowledgeAttachMapper attachMapper;
private final IChatModelService chatModelService;
private final IChatModelService chatModelService;
private final ISysOssService ossService;
private final ISysOssService ossService;
/**
* 查询知识库
*/
@Override
public KnowledgeInfoVo queryById(Long id) {
return baseMapper.selectVoById(id);
}
private final KnowledgeRoleMapper knowledgeRoleMapper;
/**
* 查询知识库列表
*/
@Override
public TableDataInfo<KnowledgeInfoVo> queryPageList(KnowledgeInfoBo bo, PageQuery pageQuery) {
LambdaQueryWrapper<KnowledgeInfo> lqw = buildQueryWrapper(bo);
Page<KnowledgeInfoVo> result = baseMapper.selectVoPage(pageQuery.build(), lqw);
return TableDataInfo.build(result);
}
private final KnowledgeRoleRelationMapper knowledgeRoleRelationMapper;
/**
* 查询知识库列表
*/
@Override
public List<KnowledgeInfoVo> queryList(KnowledgeInfoBo bo) {
LambdaQueryWrapper<KnowledgeInfo> lqw = buildQueryWrapper(bo);
return baseMapper.selectVoList(lqw);
}
private LambdaQueryWrapper<KnowledgeInfo> buildQueryWrapper(KnowledgeInfoBo bo) {
Map<String, Object> params = bo.getParams();
LambdaQueryWrapper<KnowledgeInfo> lqw = Wrappers.lambdaQuery();
lqw.eq(StringUtils.isNotBlank(bo.getKid()), KnowledgeInfo::getKid, bo.getKid());
lqw.eq(bo.getUid() != null, KnowledgeInfo::getUid, bo.getUid());
lqw.like(StringUtils.isNotBlank(bo.getKname()), KnowledgeInfo::getKname, bo.getKname());
lqw.eq(bo.getShare() != null, KnowledgeInfo::getShare, bo.getShare());
lqw.eq(StringUtils.isNotBlank(bo.getDescription()), KnowledgeInfo::getDescription,
bo.getDescription());
lqw.eq(StringUtils.isNotBlank(bo.getKnowledgeSeparator()), KnowledgeInfo::getKnowledgeSeparator,
bo.getKnowledgeSeparator());
lqw.eq(StringUtils.isNotBlank(bo.getQuestionSeparator()), KnowledgeInfo::getQuestionSeparator,
bo.getQuestionSeparator());
lqw.eq(bo.getOverlapChar() != null, KnowledgeInfo::getOverlapChar, bo.getOverlapChar());
lqw.eq(bo.getRetrieveLimit() != null, KnowledgeInfo::getRetrieveLimit, bo.getRetrieveLimit());
lqw.eq(bo.getTextBlockSize() != null, KnowledgeInfo::getTextBlockSize, bo.getTextBlockSize());
return lqw;
}
/**
* 新增知识库
*/
@Override
public Boolean insertByBo(KnowledgeInfoBo bo) {
KnowledgeInfo add = MapstructUtils.convert(bo, KnowledgeInfo.class);
validEntityBeforeSave(add);
boolean flag = baseMapper.insert(add) > 0;
if (flag) {
bo.setId(add.getId());
/**
* 查询知识库
*/
@Override
public KnowledgeInfoVo queryById(Long id) {
return baseMapper.selectVoById(id);
}
return flag;
}
/**
* 修改知识库
*/
@Override
public Boolean updateByBo(KnowledgeInfoBo bo) {
KnowledgeInfo update = MapstructUtils.convert(bo, KnowledgeInfo.class);
validEntityBeforeSave(update);
return baseMapper.updateById(update) > 0;
}
/**
* 保存前的数据校验
*/
private void validEntityBeforeSave(KnowledgeInfo entity) {
//TODO 做一些数据校验,如唯一约束
}
/**
* 批量删除知识库
*/
@Override
public Boolean deleteWithValidByIds(Collection<Long> ids, Boolean isValid) {
if (isValid) {
//TODO 做一些业务上的校验,判断是否需要校验
/**
* 查询知识库列表
*/
@Override
public TableDataInfo<KnowledgeInfoVo> queryPageList(KnowledgeInfoBo bo, PageQuery pageQuery) {
LambdaQueryWrapper<KnowledgeInfo> lqw = buildQueryWrapper(bo);
Page<KnowledgeInfoVo> result = baseMapper.selectVoPage(pageQuery.build(), lqw);
return TableDataInfo.build(result);
}
return baseMapper.deleteBatchIds(ids) > 0;
}
@Override
@Transactional(rollbackFor = Exception.class)
public void saveOne(KnowledgeInfoBo bo) {
KnowledgeInfo knowledgeInfo = MapstructUtils.convert(bo, KnowledgeInfo.class);
if (StringUtils.isBlank(bo.getKid())) {
String kid = RandomUtil.randomString(10);
if (knowledgeInfo != null) {
knowledgeInfo.setKid(kid);
knowledgeInfo.setUid(LoginHelper.getLoginUser().getUserId());
}
baseMapper.insert(knowledgeInfo);
if (knowledgeInfo != null) {
vectorStoreService.createSchema(String.valueOf(knowledgeInfo.getId()),
bo.getVectorModelName());
}
} else {
baseMapper.updateById(knowledgeInfo);
}
}
@Override
@Transactional(rollbackFor = Exception.class)
public void removeKnowledge(String id) {
Map<String,Object> map = new HashMap<>();
KnowledgeInfo knowledgeInfo = baseMapper.selectById(id);
check(knowledgeInfo);
map.put("kid",knowledgeInfo.getId());
// 删除向量数据
vectorStoreService.removeById(String.valueOf(knowledgeInfo.getId()),knowledgeInfo.getVectorModelName());
// 删除附件和知识片段
fragmentMapper.deleteByMap(map);
attachMapper.deleteByMap(map);
// 删除知识库
map.put("kid",knowledgeInfo.getKid());
baseMapper.deleteByMap(map);
}
@Override
public void upload(KnowledgeInfoUploadBo bo) {
storeContent(bo.getFile(), bo.getKid());
}
public void storeContent(MultipartFile file, String kid) {
String fileName = file.getOriginalFilename();
List<String> chunkList = new ArrayList<>();
KnowledgeAttach knowledgeAttach = new KnowledgeAttach();
knowledgeAttach.setKid(kid);
String docId = RandomUtil.randomString(10);
knowledgeAttach.setDocId(docId);
knowledgeAttach.setDocName(fileName);
knowledgeAttach.setDocType(fileName.substring(fileName.lastIndexOf(".")+1));
String content = "";
ResourceLoader resourceLoader = resourceLoaderFactory.getLoaderByFileType(knowledgeAttach.getDocType());
List<String> fids = new ArrayList<>();
try {
content = resourceLoader.getContent(file.getInputStream());
chunkList = resourceLoader.getChunkList(content, kid);
List<KnowledgeFragment> knowledgeFragmentList = new ArrayList<>();
if (CollUtil.isNotEmpty(chunkList)) {
for (int i = 0; i < chunkList.size(); i++) {
String fid = RandomUtil.randomString(10);
fids.add(fid);
KnowledgeFragment knowledgeFragment = new KnowledgeFragment();
knowledgeFragment.setKid(kid);
knowledgeFragment.setDocId(docId);
knowledgeFragment.setFid(fid);
knowledgeFragment.setIdx(i);
knowledgeFragment.setContent(chunkList.get(i));
knowledgeFragment.setCreateTime(new Date());
knowledgeFragmentList.add(knowledgeFragment);
/**
* 根据知识库角色查询知识库列表
*/
@Override
public TableDataInfo<KnowledgeInfoVo> queryPageListByRole(PageQuery pageQuery) {
// 查询用户关联角色
LoginUser loginUser = LoginHelper.getLoginUser();
if (StringUtils.isEmpty(loginUser.getKroleGroupIds()) || StringUtils.isEmpty(loginUser.getKroleGroupType())) {
return new TableDataInfo<>();
}
}
fragmentMapper.insertBatch(knowledgeFragmentList);
} catch (IOException e) {
log.error("保存知识库信息失败！{}", e.getMessage());
// 角色/角色组id列表
List<String> groupIdList = Arrays.stream(loginUser.getKroleGroupIds().split(","))
.filter(StringUtils::isNotEmpty)
.toList();
List<KnowledgeRole> knowledgeRoles;
LambdaQueryWrapper<KnowledgeRole> roleLqw = Wrappers.lambdaQuery();
if ("role".equals(loginUser.getKroleGroupType())) {
roleLqw.in(KnowledgeRole::getId, groupIdList);
} else {
roleLqw.in(KnowledgeRole::getGroupId, groupIdList);
}
knowledgeRoles = knowledgeRoleMapper.selectList(roleLqw);
if (CollectionUtils.isEmpty(knowledgeRoles)) {
return new TableDataInfo<>();
}
// 查询知识库id列表
LambdaQueryWrapper<KnowledgeRoleRelation> relationLqw = Wrappers.lambdaQuery();
relationLqw.in(KnowledgeRoleRelation::getKnowledgeRoleId, knowledgeRoles.stream().map(KnowledgeRole::getId).filter(Objects::nonNull).collect(Collectors.toList()));
List<KnowledgeRoleRelation> knowledgeRoleRelations = knowledgeRoleRelationMapper.selectList(relationLqw);
if (CollectionUtils.isEmpty(knowledgeRoleRelations)) {
return new TableDataInfo<>();
}
LambdaQueryWrapper<KnowledgeInfo> lqw = Wrappers.lambdaQuery();
lqw.in(KnowledgeInfo::getId, knowledgeRoleRelations.stream().map(KnowledgeRoleRelation::getKnowledgeId).filter(Objects::nonNull).collect(Collectors.toList()));
Page<KnowledgeInfoVo> result = baseMapper.selectVoPage(pageQuery.build(), lqw);
return TableDataInfo.build(result);
}
knowledgeAttach.setContent(content);
knowledgeAttach.setCreateTime(new Date());
attachMapper.insert(knowledgeAttach);
// 通过kid查询知识库信息
KnowledgeInfoVo knowledgeInfoVo = baseMapper.selectVoOne(Wrappers.<KnowledgeInfo>lambdaQuery()
.eq(KnowledgeInfo::getId, kid));
// 通过向量模型查询模型信息
ChatModelVo chatModelVo = chatModelService.selectModelByName(knowledgeInfoVo.getEmbeddingModelName());
StoreEmbeddingBo storeEmbeddingBo = new StoreEmbeddingBo();
storeEmbeddingBo.setKid(kid);
storeEmbeddingBo.setDocId(docId);
storeEmbeddingBo.setFids(fids);
storeEmbeddingBo.setChunkList(chunkList);
storeEmbeddingBo.setVectorModelName(knowledgeInfoVo.getVectorModelName());
storeEmbeddingBo.setEmbeddingModelName(knowledgeInfoVo.getEmbeddingModelName());
storeEmbeddingBo.setApiKey(chatModelVo.getApiKey());
storeEmbeddingBo.setBaseUrl(chatModelVo.getApiHost());
vectorStoreService.storeEmbeddings(storeEmbeddingBo);
}
/**
* 检查用户是否有删除知识库权限
*
* @param knowledgeInfo 知识库
*/
public void check( KnowledgeInfo knowledgeInfo) {
LoginUser loginUser = LoginHelper.getLoginUser();
if (!knowledgeInfo.getUid().equals(loginUser.getUserId())) {
throw new SecurityException("权限不足");
/**
* 查询知识库列表
*/
@Override
public List<KnowledgeInfoVo> queryList(KnowledgeInfoBo bo) {
LambdaQueryWrapper<KnowledgeInfo> lqw = buildQueryWrapper(bo);
return baseMapper.selectVoList(lqw);
}
private LambdaQueryWrapper<KnowledgeInfo> buildQueryWrapper(KnowledgeInfoBo bo) {
Map<String, Object> params = bo.getParams();
LambdaQueryWrapper<KnowledgeInfo> lqw = Wrappers.lambdaQuery();
lqw.eq(StringUtils.isNotBlank(bo.getKid()), KnowledgeInfo::getKid, bo.getKid());
lqw.eq(bo.getUid() != null, KnowledgeInfo::getUid, bo.getUid());
lqw.like(StringUtils.isNotBlank(bo.getKname()), KnowledgeInfo::getKname, bo.getKname());
lqw.eq(bo.getShare() != null, KnowledgeInfo::getShare, bo.getShare());
lqw.eq(StringUtils.isNotBlank(bo.getDescription()), KnowledgeInfo::getDescription,
bo.getDescription());
lqw.eq(StringUtils.isNotBlank(bo.getKnowledgeSeparator()), KnowledgeInfo::getKnowledgeSeparator,
bo.getKnowledgeSeparator());
lqw.eq(StringUtils.isNotBlank(bo.getQuestionSeparator()), KnowledgeInfo::getQuestionSeparator,
bo.getQuestionSeparator());
lqw.eq(bo.getOverlapChar() != null, KnowledgeInfo::getOverlapChar, bo.getOverlapChar());
lqw.eq(bo.getRetrieveLimit() != null, KnowledgeInfo::getRetrieveLimit, bo.getRetrieveLimit());
lqw.eq(bo.getTextBlockSize() != null, KnowledgeInfo::getTextBlockSize, bo.getTextBlockSize());
return lqw;
}
/**
* 新增知识库
*/
@Override
public Boolean insertByBo(KnowledgeInfoBo bo) {
KnowledgeInfo add = MapstructUtils.convert(bo, KnowledgeInfo.class);
validEntityBeforeSave(add);
boolean flag = baseMapper.insert(add) > 0;
if (flag) {
bo.setId(add.getId());
}
return flag;
}
/**
* 修改知识库
*/
@Override
public Boolean updateByBo(KnowledgeInfoBo bo) {
KnowledgeInfo update = MapstructUtils.convert(bo, KnowledgeInfo.class);
validEntityBeforeSave(update);
return baseMapper.updateById(update) > 0;
}
/**
* 保存前的数据校验
*/
private void validEntityBeforeSave(KnowledgeInfo entity) {
//TODO 做一些数据校验,如唯一约束
}
/**
* 批量删除知识库
*/
@Override
public Boolean deleteWithValidByIds(Collection<Long> ids, Boolean isValid) {
if (isValid) {
//TODO 做一些业务上的校验,判断是否需要校验
}
return baseMapper.deleteBatchIds(ids) > 0;
}
@Override
@Transactional(rollbackFor = Exception.class)
public void saveOne(KnowledgeInfoBo bo) {
KnowledgeInfo knowledgeInfo = MapstructUtils.convert(bo, KnowledgeInfo.class);
if (StringUtils.isBlank(bo.getKid())) {
String kid = RandomUtil.randomString(10);
if (knowledgeInfo != null) {
knowledgeInfo.setKid(kid);
knowledgeInfo.setUid(LoginHelper.getLoginUser().getUserId());
}
baseMapper.insert(knowledgeInfo);
if (knowledgeInfo != null) {
vectorStoreService.createSchema(String.valueOf(knowledgeInfo.getId()),
bo.getVectorModelName());
}
} else {
baseMapper.updateById(knowledgeInfo);
}
}
@Override
@Transactional(rollbackFor = Exception.class)
public void removeKnowledge(String id) {
Map<String, Object> map = new HashMap<>();
KnowledgeInfo knowledgeInfo = baseMapper.selectById(id);
check(knowledgeInfo);
map.put("kid", knowledgeInfo.getId());
// 删除向量数据
vectorStoreService.removeById(String.valueOf(knowledgeInfo.getId()), knowledgeInfo.getVectorModelName());
// 删除附件和知识片段
fragmentMapper.deleteByMap(map);
attachMapper.deleteByMap(map);
// 删除知识库
map.put("kid", knowledgeInfo.getKid());
baseMapper.deleteByMap(map);
}
@Override
public void upload(KnowledgeInfoUploadBo bo) {
storeContent(bo.getFile(), bo.getKid());
}
public void storeContent(MultipartFile file, String kid) {
String fileName = file.getOriginalFilename();
List<String> chunkList = new ArrayList<>();
KnowledgeAttach knowledgeAttach = new KnowledgeAttach();
knowledgeAttach.setKid(kid);
String docId = RandomUtil.randomString(10);
knowledgeAttach.setDocId(docId);
knowledgeAttach.setDocName(fileName);
knowledgeAttach.setDocType(fileName.substring(fileName.lastIndexOf(".") + 1));
String content = "";
ResourceLoader resourceLoader = resourceLoaderFactory.getLoaderByFileType(knowledgeAttach.getDocType());
List<String> fids = new ArrayList<>();
try {
content = resourceLoader.getContent(file.getInputStream());
chunkList = resourceLoader.getChunkList(content, kid);
List<KnowledgeFragment> knowledgeFragmentList = new ArrayList<>();
if (CollUtil.isNotEmpty(chunkList)) {
for (int i = 0; i < chunkList.size(); i++) {
String fid = RandomUtil.randomString(10);
fids.add(fid);
KnowledgeFragment knowledgeFragment = new KnowledgeFragment();
knowledgeFragment.setKid(kid);
knowledgeFragment.setDocId(docId);
knowledgeFragment.setFid(fid);
knowledgeFragment.setIdx(i);
knowledgeFragment.setContent(chunkList.get(i));
knowledgeFragment.setCreateTime(new Date());
knowledgeFragmentList.add(knowledgeFragment);
}
}
fragmentMapper.insertBatch(knowledgeFragmentList);
} catch (IOException e) {
log.error("保存知识库信息失败！{}", e.getMessage());
}
knowledgeAttach.setContent(content);
knowledgeAttach.setCreateTime(new Date());
attachMapper.insert(knowledgeAttach);
// 通过kid查询知识库信息
KnowledgeInfoVo knowledgeInfoVo = baseMapper.selectVoOne(Wrappers.<KnowledgeInfo>lambdaQuery()
.eq(KnowledgeInfo::getId, kid));
// 通过向量模型查询模型信息
ChatModelVo chatModelVo = chatModelService.selectModelByName(knowledgeInfoVo.getEmbeddingModelName());
StoreEmbeddingBo storeEmbeddingBo = new StoreEmbeddingBo();
storeEmbeddingBo.setKid(kid);
storeEmbeddingBo.setDocId(docId);
storeEmbeddingBo.setFids(fids);
storeEmbeddingBo.setChunkList(chunkList);
storeEmbeddingBo.setVectorModelName(knowledgeInfoVo.getVectorModelName());
storeEmbeddingBo.setEmbeddingModelName(knowledgeInfoVo.getEmbeddingModelName());
storeEmbeddingBo.setApiKey(chatModelVo.getApiKey());
storeEmbeddingBo.setBaseUrl(chatModelVo.getApiHost());
vectorStoreService.storeEmbeddings(storeEmbeddingBo);
}
/**
* 检查用户是否有删除知识库权限
*
* @param knowledgeInfo 知识库
*/
public void check(KnowledgeInfo knowledgeInfo) {
LoginUser loginUser = LoginHelper.getLoginUser();
if (!knowledgeInfo.getUid().equals(loginUser.getUserId())) {
throw new SecurityException("权限不足");
}
}
}
}

4
ruoyi-modules/ruoyi-system/src/main/java/org/ruoyi/system/service/SysLoginService.java
View File

@@ -124,7 +124,7 @@ public class SysLoginService {
TenantHelper.clearDynamic();
}
StpUtil.logout();
if (loginUser !=null) {
if (loginUser != null) {
recordLogininfor(loginUser.getTenantId(), loginUser.getUsername(), Constants.LOGOUT, MessageUtils.message("user.logout.success"));
}
} catch (NotLoginException ignored) {
@@ -251,6 +251,8 @@ public class SysLoginService {
loginUser.setUsername(user.getUserName());
loginUser.setAvatar(user.getAvatar());
loginUser.setUserType(user.getUserType());
loginUser.setKroleGroupIds(user.getKroleGroupIds());
loginUser.setKroleGroupType(user.getKroleGroupType());
loginUser.setMenuPermission(permissionService.getMenuPermission(user.getUserId()));
loginUser.setRolePermission(permissionService.getRolePermission(user.getUserId()));
loginUser.setDeptName(ObjectUtil.isNull(user.getDept()) ? "" : user.getDept().getDeptName());

83
script/sql/update/knowledge-role.sql Normal file
View File

@@ -0,0 +1,83 @@
/*
Navicat Premium Dump SQL
Source Server : mysql-local-study
Source Server Type : MySQL
Source Server Version : 80405 (8.4.5)
Source Host : 100.168.0.1:3500
Source Schema : ruoyi-ai
Target Server Type : MySQL
Target Server Version : 80405 (8.4.5)
File Encoding : 65001
Date: 20/07/2025 10:01:42
*/
SET NAMES utf8mb4;
SET FOREIGN_KEY_CHECKS = 0;
-- ----------------------------
-- Table structure for knowledge_role
-- ----------------------------
DROP TABLE IF EXISTS `knowledge_role`;
CREATE TABLE `knowledge_role` (
`id` bigint NOT NULL COMMENT '知识库角色id',
`name` varchar(100) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT NULL COMMENT '知识库角色name',
`del_flag` char(1) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT '0' COMMENT '删除标志（0代表存在 2代表删除）',
`create_dept` bigint NULL DEFAULT NULL COMMENT '创建部门',
`create_by` bigint NULL DEFAULT NULL COMMENT '创建者',
`create_time` datetime NULL DEFAULT NULL COMMENT '创建时间',
`update_by` bigint NULL DEFAULT NULL COMMENT '更新者',
`update_time` datetime NULL DEFAULT NULL COMMENT '更新时间',
`remark` varchar(500) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT NULL COMMENT '备注',
`group_id` bigint NULL DEFAULT NULL COMMENT '知识库角色组id',
PRIMARY KEY (`id`) USING BTREE
) ENGINE = InnoDB CHARACTER SET = utf8mb4 COLLATE = utf8mb4_general_ci COMMENT = '知识库角色表' ROW_FORMAT = DYNAMIC;
-- ----------------------------
-- Table structure for knowledge_role_group
-- ----------------------------
DROP TABLE IF EXISTS `knowledge_role_group`;
CREATE TABLE `knowledge_role_group` (
`id` bigint NOT NULL COMMENT '知识库角色组id',
`name` varchar(100) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT NULL COMMENT '知识库角色组name',
`del_flag` char(1) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT '0' COMMENT '删除标志（0代表存在 2代表删除）',
`create_dept` bigint NULL DEFAULT NULL COMMENT '创建部门',
`create_by` bigint NULL DEFAULT NULL COMMENT '创建者',
`create_time` datetime NULL DEFAULT NULL COMMENT '创建时间',
`update_by` bigint NULL DEFAULT NULL COMMENT '更新者',
`update_time` datetime NULL DEFAULT NULL COMMENT '更新时间',
`remark` varchar(500) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT NULL COMMENT '备注',
PRIMARY KEY (`id`) USING BTREE
) ENGINE = InnoDB CHARACTER SET = utf8mb4 COLLATE = utf8mb4_general_ci COMMENT = '知识库角色组表' ROW_FORMAT = DYNAMIC;
-- ----------------------------
-- Table structure for knowledge_role_relation
-- ----------------------------
DROP TABLE IF EXISTS `knowledge_role_relation`;
CREATE TABLE `knowledge_role_relation` (
`id` bigint NOT NULL COMMENT 'id',
`del_flag` char(1) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT '0' COMMENT '删除标志（0代表存在 2代表删除）',
`create_dept` bigint NULL DEFAULT NULL COMMENT '创建部门',
`create_by` bigint NULL DEFAULT NULL COMMENT '创建者',
`create_time` datetime NULL DEFAULT NULL COMMENT '创建时间',
`update_by` bigint NULL DEFAULT NULL COMMENT '更新者',
`update_time` datetime NULL DEFAULT NULL COMMENT '更新时间',
`remark` varchar(500) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT NULL COMMENT '备注',
`knowledge_role_id` bigint NULL DEFAULT NULL COMMENT '知识库角色id',
`knowledge_id` bigint NULL DEFAULT NULL COMMENT '知识库id',
PRIMARY KEY (`id`) USING BTREE
) ENGINE = InnoDB CHARACTER SET = utf8mb4 COLLATE = utf8mb4_general_ci COMMENT = '知识库角色与知识库关联表' ROW_FORMAT = DYNAMIC;
SET FOREIGN_KEY_CHECKS = 1;
-- 菜单
INSERT INTO `sys_menu` (`menu_id`, `menu_name`, `parent_id`, `order_num`, `path`, `component`, `query_param`, `is_frame`, `is_cache`, `menu_type`, `visible`, `status`, `perms`, `icon`, `create_dept`, `create_by`, `create_time`, `update_by`, `update_time`, `remark`) VALUES (1946483381643743233, '知识库角色管理', 1775500307898949634, '12', 'knowledgeRole', 'system/knowledgeRole/index', NULL, 1, 0, 'C', '0', '0', NULL, 'ri:user-3-fill', 103, 1, '2025-07-19 16:41:17', NULL, NULL, '知识库角色管理');
-- 用户表添加字段
ALTER TABLE sys_user
ADD COLUMN `krole_group_type` VARCHAR(50) COMMENT '关联知识库角色/角色组',
ADD COLUMN `krole_group_id` TEXT COMMENT '关联知识库角色/角色组id';