Przeglądaj źródła

1、知识库基础管理模块,CRUD
2、通过DIFY API 同步知识库列表

lihao16 5 miesięcy temu
rodzic
commit
2d2da1ad9f

+ 3 - 3
smsb-admin/src/main/resources/application-dev.yml

@@ -15,7 +15,7 @@ spring.boot.admin.client:
 
 --- # snail-job 配置
 snail-job:
-  enabled: false
+  enabled: true
   # 需要在 SnailJob 后台组管理创建对应名称的组,然后创建任务的时候选择对应的组,才能正确分派任务
   # group: "ruoyi_group"
   group: "smsb_group"
@@ -54,10 +54,10 @@ spring:
           driverClassName: com.mysql.cj.jdbc.Driver
           # jdbc 所有参数配置参考 https://lionli.blog.csdn.net/article/details/122018562
           # rewriteBatchedStatements=true 批处理优化 大幅提升批量插入更新删除性能(对数据库有性能损耗 使用批量操作应考虑性能问题)
-#          url: jdbc:mysql://localhost:3306/smsb-plus?useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&useSSL=true&serverTimezone=GMT%2B8&autoReconnect=true&rewriteBatchedStatements=true&allowPublicKeyRetrieval=true&nullCatalogMeansCurrent=true
+          # url: jdbc:mysql://localhost:3306/smsb-plus?useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&useSSL=true&serverTimezone=GMT%2B8&autoReconnect=true&rewriteBatchedStatements=true&allowPublicKeyRetrieval=true&nullCatalogMeansCurrent=true
           url: jdbc:mysql://117.73.3.135:3306/smsb-plus?useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&useSSL=true&serverTimezone=GMT%2B8&autoReconnect=true&rewriteBatchedStatements=true&allowPublicKeyRetrieval=true&nullCatalogMeansCurrent=true
           username: root
-#          password: password
+          # password: password
           password: Hycpb@123
         # 从库数据源
         slave:

+ 12 - 3
smsb-admin/src/main/resources/application.yml

@@ -22,10 +22,10 @@ captcha:
 # 开发环境配置
 server:
   # 服务器的HTTP端口,默认为8080
-  port: 8080
+  port: 8084
   servlet:
     # 应用的访问路径
-    context-path: /
+    context-path: /prod-api
   # undertow 配置
   undertow:
     # HTTP post内容的最大大小。当值为-1时,默认值为大小是无限的
@@ -99,6 +99,14 @@ spring:
       # 允许对象忽略json中不存在的属性
       fail_on_unknown_properties: false
 
+# Dify 对接参数
+dify:
+  url: http://192.168.32.114
+  # 知识库配置
+  datasets:
+    apiKey: dataset-pBh1o1eHctOU0QUs6zhkN2UV
+
+
 # Sa-Token配置
 sa-token:
   # token名称 (同时也是cookie名称)
@@ -147,6 +155,7 @@ tenant:
     - smsb_device_login
     - smsb_device_auth
     - smsb_device_heart_record
+    - smsb_dify_datasets
 
 # MyBatisPlus配置
 # https://baomidou.com/config/
@@ -204,7 +213,7 @@ springdoc:
     # 描述
     description: '描述:用于管理集团旗下公司的人员信息,具体包括XXX,XXX模块...'
     # 版本
-    version: '版本号: ${ruoyi.version}'
+    version: '版本号: 5.2.3'
     # 作者信息
     contact:
       name: Lion Li

+ 110 - 0
smsb-modules/smsb-device/src/main/java/com/inspur/device/controller/SmsbDifyDatasetsController.java

@@ -0,0 +1,110 @@
+package com.inspur.device.controller;
+
+import java.util.List;
+
+import lombok.RequiredArgsConstructor;
+import jakarta.servlet.http.HttpServletResponse;
+import jakarta.validation.constraints.*;
+import cn.dev33.satoken.annotation.SaCheckPermission;
+import org.springframework.web.bind.annotation.*;
+import org.springframework.validation.annotation.Validated;
+import org.dromara.common.idempotent.annotation.RepeatSubmit;
+import org.dromara.common.log.annotation.Log;
+import org.dromara.common.web.core.BaseController;
+import org.dromara.common.mybatis.core.page.PageQuery;
+import org.dromara.common.core.domain.R;
+import org.dromara.common.core.validate.AddGroup;
+import org.dromara.common.core.validate.EditGroup;
+import org.dromara.common.log.enums.BusinessType;
+import org.dromara.common.excel.utils.ExcelUtil;
+import com.inspur.device.domain.vo.SmsbDifyDatasetsVo;
+import com.inspur.device.domain.bo.SmsbDifyDatasetsBo;
+import com.inspur.device.service.ISmsbDifyDatasetsService;
+import org.dromara.common.mybatis.core.page.TableDataInfo;
+
+/**
+ * 知识库管理
+ *
+ * @author Hao Li
+ * @date 2025-05-14
+ */
+@Validated
+@RequiredArgsConstructor
+@RestController
+@RequestMapping("/smsb/device/datasets")
+public class SmsbDifyDatasetsController extends BaseController {
+
+    private final ISmsbDifyDatasetsService smsbDifyDatasetsService;
+
+    /**
+     * 查询知识库管理列表
+     */
+    @SaCheckPermission("device:datasets:list")
+    @GetMapping("/list")
+    public TableDataInfo<SmsbDifyDatasetsVo> list(SmsbDifyDatasetsBo bo, PageQuery pageQuery) {
+        return smsbDifyDatasetsService.queryPageList(bo, pageQuery);
+    }
+
+    @GetMapping("/sync")
+    public R<Void> sync() {
+        return toAjax(smsbDifyDatasetsService.syncFromDify());
+    }
+
+    /**
+     * 导出知识库管理列表
+     */
+    @SaCheckPermission("device:datasets:export")
+    @Log(title = "知识库管理", businessType = BusinessType.EXPORT)
+    @PostMapping("/export")
+    public void export(SmsbDifyDatasetsBo bo, HttpServletResponse response) {
+        List<SmsbDifyDatasetsVo> list = smsbDifyDatasetsService.queryList(bo);
+        ExcelUtil.exportExcel(list, "知识库管理", SmsbDifyDatasetsVo.class, response);
+    }
+
+    /**
+     * 获取知识库管理详细信息
+     *
+     * @param id 主键
+     */
+    @SaCheckPermission("device:datasets:query")
+    @GetMapping("/{id}")
+    public R<SmsbDifyDatasetsVo> getInfo(@NotNull(message = "主键不能为空")
+                                         @PathVariable Long id) {
+        return R.ok(smsbDifyDatasetsService.queryById(id));
+    }
+
+    /**
+     * 新增知识库管理
+     */
+    @SaCheckPermission("device:datasets:add")
+    @Log(title = "知识库管理", businessType = BusinessType.INSERT)
+    @RepeatSubmit()
+    @PostMapping()
+    public R<Void> add(@Validated(AddGroup.class) @RequestBody SmsbDifyDatasetsBo bo) {
+        return toAjax(smsbDifyDatasetsService.insertByBo(bo));
+    }
+
+    /**
+     * 修改知识库管理
+     */
+    @SaCheckPermission("device:datasets:edit")
+    @Log(title = "知识库管理", businessType = BusinessType.UPDATE)
+    @RepeatSubmit()
+    @PutMapping()
+    public R<Void> edit(@Validated(EditGroup.class) @RequestBody SmsbDifyDatasetsBo bo) {
+        return toAjax(smsbDifyDatasetsService.updateByBo(bo));
+    }
+
+    /**
+     * 删除知识库管理
+     *
+     * @param ids 主键串
+     */
+    @SaCheckPermission("device:datasets:remove")
+    @Log(title = "知识库管理", businessType = BusinessType.DELETE)
+    @DeleteMapping("/{ids}")
+    public R<Void> remove(@NotEmpty(message = "主键不能为空")
+                          @PathVariable Long[] ids) {
+        return toAjax(smsbDifyDatasetsService.deleteWithValidByIds(List.of(ids), true));
+    }
+}

+ 92 - 0
smsb-modules/smsb-device/src/main/java/com/inspur/device/domain/SmsbDifyDatasets.java

@@ -0,0 +1,92 @@
+package com.inspur.device.domain;
+
+import com.baomidou.mybatisplus.annotation.TableId;
+import com.baomidou.mybatisplus.annotation.TableName;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+import org.dromara.common.mybatis.core.domain.BaseEntity;
+
+import java.io.Serial;
+
+/**
+ * 知识库管理对象 smsb_dify_datasets
+ *
+ * @author Hao Li
+ * @date 2025-05-14
+ */
+@Data
+@EqualsAndHashCode(callSuper = true)
+@TableName("smsb_dify_datasets")
+public class SmsbDifyDatasets extends BaseEntity {
+
+    @Serial
+    private static final long serialVersionUID = 1L;
+
+    /**
+     * 主键ID
+     */
+    @TableId(value = "id")
+    private Long id;
+
+    /**
+     * 知识库名称
+     */
+    private String name;
+
+    /**
+     * 描述
+     */
+    private String description;
+
+    /**
+     * 索引模式
+     */
+    private String indexingTechnique;
+
+    /**
+     * 权限
+     */
+    private String permission;
+
+    /**
+     * 提供者
+     */
+    private String provider;
+
+    /**
+     * 数据类型
+     */
+    private String dataSourceType;
+
+    /**
+     * 文字数量
+     */
+    private Integer wordCount;
+
+    /**
+     * 文档数量
+     */
+    private Integer documentCount;
+
+    /**
+     * APP数量
+     */
+    private Integer appCount;
+
+    /**
+     * dify ID
+     */
+    private String difyId;
+
+    /**
+     * 创建人姓名
+     */
+    private String createUser;
+
+    /**
+     * 更新人姓名
+     */
+    private String updateUser;
+
+
+}

+ 76 - 0
smsb-modules/smsb-device/src/main/java/com/inspur/device/domain/bo/SmsbDifyDatasetsBo.java

@@ -0,0 +1,76 @@
+package com.inspur.device.domain.bo;
+
+import com.inspur.device.domain.SmsbDifyDatasets;
+import io.github.linpeilie.annotations.AutoMapper;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+import org.dromara.common.mybatis.core.domain.BaseEntity;
+
+/**
+ * 知识库管理业务对象 smsb_dify_datasets
+ *
+ * @author Hao Li
+ * @date 2025-05-14
+ */
+@Data
+@EqualsAndHashCode(callSuper = true)
+@AutoMapper(target = SmsbDifyDatasets.class, reverseConvertGenerate = false)
+public class SmsbDifyDatasetsBo extends BaseEntity {
+
+    /**
+     * 主键ID
+     */
+    private Long id;
+
+    /**
+     * 知识库名称
+     */
+    private String name;
+
+    /**
+     * 描述
+     */
+    private String description;
+
+    /**
+     * 索引模式
+     */
+    private String indexingTechnique;
+
+    /**
+     * 权限
+     */
+    private String permission;
+
+    /**
+     * 提供者
+     */
+    private String provider;
+
+    /**
+     * 数据类型
+     */
+    private String dataSourceType;
+
+    /**
+     * 文字数量
+     */
+    private Integer wordCount;
+
+    /**
+     * 文档数量
+     */
+    private Integer documentCount;
+
+    /**
+     * APP数量
+     */
+    private Integer appCount;
+
+    /**
+     * dify ID
+     */
+    private String difyId;
+
+
+}

+ 32 - 0
smsb-modules/smsb-device/src/main/java/com/inspur/device/domain/vo/DifyDatasetsRspData.java

@@ -0,0 +1,32 @@
+package com.inspur.device.domain.vo;
+
+import lombok.Data;
+
+/**
+ * Dify 获取知识库列表 返回数据
+ * @author lihao16
+ */
+@Data
+public class DifyDatasetsRspData {
+
+    private String id;
+
+    private String name;
+
+    private String description;
+
+    private String provider;
+
+    private String permission;
+
+    private String data_source_type;
+
+    private String indexing_technique;
+
+    private Integer app_count;
+
+    private Integer document_count;
+
+    private Integer word_count;
+
+}

+ 119 - 0
smsb-modules/smsb-device/src/main/java/com/inspur/device/domain/vo/SmsbDifyDatasetsVo.java

@@ -0,0 +1,119 @@
+package com.inspur.device.domain.vo;
+
+import com.alibaba.excel.annotation.ExcelIgnoreUnannotated;
+import com.alibaba.excel.annotation.ExcelProperty;
+import com.inspur.device.domain.SmsbDifyDatasets;
+import io.github.linpeilie.annotations.AutoMapper;
+import lombok.Data;
+
+import java.io.Serial;
+import java.io.Serializable;
+import java.util.Date;
+
+
+/**
+ * 知识库管理视图对象 smsb_dify_datasets
+ *
+ * @author Hao Li
+ * @date 2025-05-14
+ */
+@Data
+@ExcelIgnoreUnannotated
+@AutoMapper(target = SmsbDifyDatasets.class)
+public class SmsbDifyDatasetsVo implements Serializable {
+
+    @Serial
+    private static final long serialVersionUID = 1L;
+
+    /**
+     * 主键ID
+     */
+    @ExcelProperty(value = "主键ID")
+    private Long id;
+
+    /**
+     * 知识库名称
+     */
+    @ExcelProperty(value = "知识库名称")
+    private String name;
+
+    /**
+     * 描述
+     */
+    @ExcelProperty(value = "描述")
+    private String description;
+
+    /**
+     * 索引模式
+     */
+    @ExcelProperty(value = "索引模式")
+    private String indexingTechnique;
+
+    /**
+     * 权限
+     */
+    @ExcelProperty(value = "权限")
+    private String permission;
+
+    /**
+     * 提供者
+     */
+    @ExcelProperty(value = "提供者")
+    private String provider;
+
+    /**
+     * 数据类型
+     */
+    @ExcelProperty(value = "数据类型")
+    private String dataSourceType;
+
+    /**
+     * 文字数量
+     */
+    @ExcelProperty(value = "文字数量")
+    private Integer wordCount;
+
+    /**
+     * 文档数量
+     */
+    @ExcelProperty(value = "文档数量")
+    private Integer documentCount;
+
+    /**
+     * APP数量
+     */
+    @ExcelProperty(value = "APP数量")
+    private Integer appCount;
+
+    /**
+     * dify ID
+     */
+    @ExcelProperty(value = "dify ID")
+    private String difyId;
+
+    /**
+     * 创建时间
+     */
+    @ExcelProperty(value = "创建时间")
+    private Date createTime;
+
+    /**
+     * 创建人姓名
+     */
+    @ExcelProperty(value = "创建人姓名")
+    private String createUser;
+
+    /**
+     * 更新时间
+     */
+    @ExcelProperty(value = "更新时间")
+    private Date updateTime;
+
+    /**
+     * 更新人姓名
+     */
+    @ExcelProperty(value = "更新人姓名")
+    private String updateUser;
+
+
+}

+ 15 - 0
smsb-modules/smsb-device/src/main/java/com/inspur/device/mapper/SmsbDifyDatasetsMapper.java

@@ -0,0 +1,15 @@
+package com.inspur.device.mapper;
+
+import com.inspur.device.domain.SmsbDifyDatasets;
+import com.inspur.device.domain.vo.SmsbDifyDatasetsVo;
+import org.dromara.common.mybatis.core.mapper.BaseMapperPlus;
+
+/**
+ * 知识库管理Mapper接口
+ *
+ * @author Hao Li
+ * @date 2025-05-14
+ */
+public interface SmsbDifyDatasetsMapper extends BaseMapperPlus<SmsbDifyDatasets, SmsbDifyDatasetsVo> {
+
+}

+ 74 - 0
smsb-modules/smsb-device/src/main/java/com/inspur/device/service/ISmsbDifyDatasetsService.java

@@ -0,0 +1,74 @@
+package com.inspur.device.service;
+
+import com.inspur.device.domain.bo.SmsbDifyDatasetsBo;
+import com.inspur.device.domain.vo.SmsbDifyDatasetsVo;
+import org.dromara.common.mybatis.core.page.PageQuery;
+import org.dromara.common.mybatis.core.page.TableDataInfo;
+
+import java.util.Collection;
+import java.util.List;
+
+/**
+ * 知识库管理Service接口
+ *
+ * @author Hao Li
+ * @date 2025-05-14
+ */
+public interface ISmsbDifyDatasetsService {
+
+    /**
+     * 查询知识库管理
+     *
+     * @param id 主键
+     * @return 知识库管理
+     */
+    SmsbDifyDatasetsVo queryById(Long id);
+
+    /**
+     * 分页查询知识库管理列表
+     *
+     * @param bo        查询条件
+     * @param pageQuery 分页参数
+     * @return 知识库管理分页列表
+     */
+    TableDataInfo<SmsbDifyDatasetsVo> queryPageList(SmsbDifyDatasetsBo bo, PageQuery pageQuery);
+
+    /**
+     * 查询符合条件的知识库管理列表
+     *
+     * @param bo 查询条件
+     * @return 知识库管理列表
+     */
+    List<SmsbDifyDatasetsVo> queryList(SmsbDifyDatasetsBo bo);
+
+    /**
+     * 新增知识库管理
+     *
+     * @param bo 知识库管理
+     * @return 是否新增成功
+     */
+    Boolean insertByBo(SmsbDifyDatasetsBo bo);
+
+    /**
+     * 修改知识库管理
+     *
+     * @param bo 知识库管理
+     * @return 是否修改成功
+     */
+    Boolean updateByBo(SmsbDifyDatasetsBo bo);
+
+    /**
+     * 校验并批量删除知识库管理信息
+     *
+     * @param ids     待删除的主键集合
+     * @param isValid 是否进行有效性校验
+     * @return 是否删除成功
+     */
+    Boolean deleteWithValidByIds(Collection<Long> ids, Boolean isValid);
+
+    /**
+     * 从Dify同步数据
+     * @return
+     */
+    boolean syncFromDify();
+}

+ 205 - 0
smsb-modules/smsb-device/src/main/java/com/inspur/device/service/impl/SmsbDifyDatasetsServiceImpl.java

@@ -0,0 +1,205 @@
+package com.inspur.device.service.impl;
+
+import cn.hutool.http.HttpRequest;
+import cn.hutool.http.HttpResponse;
+import com.alibaba.fastjson2.JSON;
+import com.alibaba.fastjson2.JSONObject;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.inspur.device.domain.SmsbDifyDatasets;
+import com.inspur.device.domain.bo.SmsbDifyDatasetsBo;
+import com.inspur.device.domain.vo.DifyDatasetsRspData;
+import com.inspur.device.domain.vo.SmsbDifyDatasetsVo;
+import com.inspur.device.mapper.SmsbDifyDatasetsMapper;
+import com.inspur.device.service.ISmsbDifyDatasetsService;
+import lombok.RequiredArgsConstructor;
+import org.dromara.common.core.utils.MapstructUtils;
+import org.dromara.common.core.utils.StringUtils;
+import org.dromara.common.mybatis.core.page.PageQuery;
+import org.dromara.common.mybatis.core.page.TableDataInfo;
+import org.springframework.beans.BeanUtils;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.stereotype.Service;
+import org.springframework.util.CollectionUtils;
+
+import java.nio.charset.StandardCharsets;
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * 知识库管理Service业务层处理
+ *
+ * @author Hao Li
+ * @date 2025-05-14
+ */
+@RequiredArgsConstructor
+@Service
+public class SmsbDifyDatasetsServiceImpl implements ISmsbDifyDatasetsService {
+
+    @Value("${dify.url}")
+    private String difyUrl;
+
+    @Value("${dify.datasets.apiKey}")
+    private String datasetsApiKey;
+
+    private final static String API_DATASETS_LIST = "/v1/datasets";
+
+
+    private final SmsbDifyDatasetsMapper baseMapper;
+
+
+    /**
+     * 从Dify同步数据
+     *
+     * @return
+     */
+    @Override
+    public boolean syncFromDify() {
+        String requestUrl = difyUrl + API_DATASETS_LIST + "?page=1&limit=100";
+        HttpRequest request = HttpRequest.get(requestUrl)
+            .header("Authorization", "Bearer " + datasetsApiKey);
+        HttpResponse response = request.execute();
+        if (!response.isOk()) {
+            return false;
+        }
+        String responseBody = response.body();
+        JSONObject jsonObject = JSON.parseObject(responseBody);
+        // 1 如果结果为0 直接返回
+        Integer total = jsonObject.getIntValue("total");
+        if (total == 0) {
+            return true;
+        }
+        // data中取出知识库列表
+        List<DifyDatasetsRspData> datasets = jsonObject.getJSONArray("data").toJavaList(DifyDatasetsRspData.class);
+        // 获取数据库中的所有知识库
+        List<SmsbDifyDatasetsVo> smsbDifyDatasetsVos = baseMapper.selectVoList(null);
+        List<SmsbDifyDatasets> insertList = new ArrayList<>();
+        // List<SmsbDifyDatasets> updateList = new ArrayList<>();
+        // 数据库为空 则全部新增
+        if (CollectionUtils.isEmpty(smsbDifyDatasetsVos)) {
+            for (DifyDatasetsRspData dataset : datasets) {
+                SmsbDifyDatasets smsbDifyDatasets = buildDatasetByDifyData(dataset);
+                insertList.add(smsbDifyDatasets);
+            }
+            baseMapper.insertBatch(insertList);
+            return true;
+        }
+        List<String> dbDifyIds = smsbDifyDatasetsVos.stream().map(SmsbDifyDatasetsVo::getDifyId).toList();
+        // 数据库不为空 则遍历数据库中的知识库
+        for (DifyDatasetsRspData rspData : datasets) {
+            // 数据库中不包含这个id 则进行新增
+            if (!dbDifyIds.contains(rspData.getId())) {
+                SmsbDifyDatasets smsbDifyDatasets = buildDatasetByDifyData(rspData);
+                insertList.add(smsbDifyDatasets);
+            }else {
+                // 数据库中包含这个id 则进行更新
+                SmsbDifyDatasets smsbDifyDatasets = buildDatasetByDifyData(rspData);
+                smsbDifyDatasets.setId(smsbDifyDatasetsVos.stream().filter(vo -> vo.getDifyId().equals(rspData.getId())).findFirst().get().getId());
+                baseMapper.updateById(smsbDifyDatasets);
+            }
+        }
+        baseMapper.insertBatch(insertList);
+        return true;
+    }
+
+    private SmsbDifyDatasets buildDatasetByDifyData(DifyDatasetsRspData dataset) {
+        SmsbDifyDatasets smsbDifyDatasets = new SmsbDifyDatasets();
+        BeanUtils.copyProperties(dataset, smsbDifyDatasets);
+        smsbDifyDatasets.setName(new String(dataset.getName().getBytes(StandardCharsets.UTF_8)));
+        smsbDifyDatasets.setDataSourceType(dataset.getData_source_type());
+        smsbDifyDatasets.setIndexingTechnique(dataset.getIndexing_technique());
+        smsbDifyDatasets.setAppCount(dataset.getApp_count());
+        smsbDifyDatasets.setDocumentCount(dataset.getDocument_count());
+        smsbDifyDatasets.setWordCount(dataset.getWord_count());
+        smsbDifyDatasets.setDifyId(dataset.getId());
+        return smsbDifyDatasets;
+    }
+
+    /**
+     * 查询知识库管理
+     *
+     * @param id 主键
+     * @return 知识库管理
+     */
+    @Override
+    public SmsbDifyDatasetsVo queryById(Long id) {
+        return baseMapper.selectVoById(id);
+    }
+
+    /**
+     * 分页查询知识库管理列表
+     *
+     * @param bo        查询条件
+     * @param pageQuery 分页参数
+     * @return 知识库管理分页列表
+     */
+    @Override
+    public TableDataInfo<SmsbDifyDatasetsVo> queryPageList(SmsbDifyDatasetsBo bo, PageQuery pageQuery) {
+        LambdaQueryWrapper<SmsbDifyDatasets> lqw = buildQueryWrapper(bo);
+        Page<SmsbDifyDatasetsVo> result = baseMapper.selectVoPage(pageQuery.build(), lqw);
+        return TableDataInfo.build(result);
+    }
+
+    /**
+     * 查询符合条件的知识库管理列表
+     *
+     * @param bo 查询条件
+     * @return 知识库管理列表
+     */
+    @Override
+    public List<SmsbDifyDatasetsVo> queryList(SmsbDifyDatasetsBo bo) {
+        LambdaQueryWrapper<SmsbDifyDatasets> lqw = buildQueryWrapper(bo);
+        return baseMapper.selectVoList(lqw);
+    }
+
+    private LambdaQueryWrapper<SmsbDifyDatasets> buildQueryWrapper(SmsbDifyDatasetsBo bo) {
+        Map<String, Object> params = bo.getParams();
+        LambdaQueryWrapper<SmsbDifyDatasets> lqw = Wrappers.lambdaQuery();
+        lqw.like(StringUtils.isNotBlank(bo.getName()), SmsbDifyDatasets::getName, bo.getName());
+        return lqw;
+    }
+
+    /**
+     * 新增知识库管理
+     *
+     * @param bo 知识库管理
+     * @return 是否新增成功
+     */
+    @Override
+    public Boolean insertByBo(SmsbDifyDatasetsBo bo) {
+        SmsbDifyDatasets add = MapstructUtils.convert(bo, SmsbDifyDatasets.class);
+        boolean flag = baseMapper.insert(add) > 0;
+        if (flag) {
+            bo.setId(add.getId());
+        }
+        return flag;
+    }
+
+    /**
+     * 修改知识库管理
+     *
+     * @param bo 知识库管理
+     * @return 是否修改成功
+     */
+    @Override
+    public Boolean updateByBo(SmsbDifyDatasetsBo bo) {
+        SmsbDifyDatasets update = MapstructUtils.convert(bo, SmsbDifyDatasets.class);
+        return baseMapper.updateById(update) > 0;
+    }
+
+
+    /**
+     * 校验并批量删除知识库管理信息
+     *
+     * @param ids     待删除的主键集合
+     * @param isValid 是否进行有效性校验
+     * @return 是否删除成功
+     */
+    @Override
+    public Boolean deleteWithValidByIds(Collection<Long> ids, Boolean isValid) {
+        return baseMapper.deleteByIds(ids) > 0;
+    }
+}

+ 7 - 0
smsb-modules/smsb-device/src/main/resources/mapper/device/SmsbDifyDatasetsMapper.xml

@@ -0,0 +1,7 @@
+<?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="com.inspur.device.mapper.SmsbDifyDatasetsMapper">
+
+</mapper>

+ 70 - 0
smsb-plus-ui/src/api/smsb/device/datasets.ts

@@ -0,0 +1,70 @@
+import request from '@/utils/request';
+import { AxiosPromise } from 'axios';
+import { DifyDatasetsVO, DifyDatasetsForm, DifyDatasetsQuery } from '@/api/smsb/device/datasets_type';
+
+/**
+ * 查询知识库管理列表
+ * @param query
+ * @returns {*}
+ */
+
+export const listDifyDatasets = (query?: DifyDatasetsQuery): AxiosPromise<DifyDatasetsVO[]> => {
+  return request({
+    url: '/smsb/device/datasets/list',
+    method: 'get',
+    params: query
+  });
+};
+
+export const syncDifyDatasets = () => {
+  return request({
+    url: '/smsb/device/datasets/sync',
+    method: 'get'
+  });
+}
+
+/**
+ * 查询知识库管理详细
+ * @param id
+ */
+export const getDifyDatasets = (id: string | number): AxiosPromise<DifyDatasetsVO> => {
+  return request({
+    url: '/smsb/device/datasets/' + id,
+    method: 'get'
+  });
+};
+
+/**
+ * 新增知识库管理
+ * @param data
+ */
+export const addDifyDatasets = (data: DifyDatasetsForm) => {
+  return request({
+    url: '/smsb/device/datasets',
+    method: 'post',
+    data: data
+  });
+};
+
+/**
+ * 修改知识库管理
+ * @param data
+ */
+export const updateDifyDatasets = (data: DifyDatasetsForm) => {
+  return request({
+    url: '/smsb/device/datasets',
+    method: 'put',
+    data: data
+  });
+};
+
+/**
+ * 删除知识库管理
+ * @param id
+ */
+export const delDifyDatasets = (id: string | number | Array<string | number>) => {
+  return request({
+    url: '/smsb/device/datasets/' + id,
+    method: 'delete'
+  });
+};

+ 151 - 0
smsb-plus-ui/src/api/smsb/device/datasets_type.ts

@@ -0,0 +1,151 @@
+export interface DifyDatasetsVO {
+  /**
+   * 主键ID
+   */
+  id: string | number;
+
+  /**
+   * 知识库名称
+   */
+  name: string;
+
+  /**
+   * 描述
+   */
+  description: string;
+
+  /**
+   * 索引模式
+   */
+  indexingTechnique: string;
+
+  /**
+   * 权限
+   */
+  permission: string;
+
+  /**
+   * 提供者
+   */
+  provider: string | number;
+
+  /**
+   * 数据类型
+   */
+  dataSourceType: string;
+
+  /**
+   * 文字数量
+   */
+  wordCount: number;
+
+  /**
+   * 文档数量
+   */
+  documentCount: number;
+
+  /**
+   * APP数量
+   */
+  appCount: number;
+
+  /**
+   * dify ID
+   */
+  difyId: string | number;
+
+  /**
+   * 创建时间
+   */
+  createTime: string;
+
+  /**
+   * 创建人姓名
+   */
+  createUser: string;
+
+  /**
+   * 更新时间
+   */
+  updateTime: string;
+
+  /**
+   * 更新人姓名
+   */
+  updateUser: string;
+
+}
+
+export interface DifyDatasetsForm extends BaseEntity {
+  /**
+   * 主键ID
+   */
+  id?: string | number;
+
+  /**
+   * 知识库名称
+   */
+  name?: string;
+
+  /**
+   * 描述
+   */
+  description?: string;
+
+  /**
+   * 索引模式
+   */
+  indexingTechnique?: string;
+
+  /**
+   * 权限
+   */
+  permission?: string;
+
+  /**
+   * 提供者
+   */
+  provider?: string | number;
+
+  /**
+   * 数据类型
+   */
+  dataSourceType?: string;
+
+  /**
+   * 文字数量
+   */
+  wordCount?: number;
+
+  /**
+   * 文档数量
+   */
+  documentCount?: number;
+
+  /**
+   * APP数量
+   */
+  appCount?: number;
+
+  /**
+   * dify ID
+   */
+  difyId?: string | number;
+
+}
+
+export interface DifyDatasetsQuery extends PageQuery {
+
+  /**
+   * 知识库名称
+   */
+  name?: string;
+
+  /**
+   * 日期范围参数
+   */
+  params?: any;
+}
+
+
+

+ 250 - 0
smsb-plus-ui/src/views/smsb/datasets/index.vue

@@ -0,0 +1,250 @@
+<template>
+  <div class="p-2">
+    <transition :enter-active-class="proxy?.animate.searchAnimate.enter"
+                :leave-active-class="proxy?.animate.searchAnimate.leave">
+      <div v-show="showSearch" class="mb-[10px]">
+        <el-card shadow="hover" :style="{ marginTop: '10px', height: '60px' }">
+          <el-form ref="queryFormRef" :model="queryParams" :inline="true">
+            <el-form-item label="知识库" prop="name">
+              <el-input v-model="queryParams.name" placeholder="请输入知识库名称" clearable @keyup.enter="handleQuery"/>
+            </el-form-item>
+            <el-form-item>
+              <el-button type="primary" icon="Search" @click="handleQuery">搜索</el-button>
+              <el-button icon="Refresh" @click="resetQuery">重置</el-button>
+              <el-button type="primary" plain icon="Plus" @click="handleAdd" v-hasPermi="['device:datasets:add']">
+                新增
+              </el-button>
+              <el-button type="danger" plain icon="Delete" :disabled="multiple" @click="handleDelete()"
+                         v-hasPermi="['device:datasets:remove']">删除
+              </el-button>
+              <el-button type="warning" plain icon="Refresh" @click="handleSync()"
+                         v-hasPermi="['device:datasets:add']">同步
+              </el-button>
+            </el-form-item>
+          </el-form>
+        </el-card>
+      </div>
+    </transition>
+
+    <el-card shadow="never">
+      <div class="table-content">
+        <el-table v-loading="loading" :data="difyDatasetsList" @selection-change="handleSelectionChange">
+          <el-table-column type="selection" width="55" align="center"/>
+<!--          <el-table-column label="ID" align="left" prop="id" v-if="true" width="175"/>-->
+          <el-table-column label="知识库ID" align="left" prop="difyId" width="180" :show-overflow-tooltip = "true"/>
+          <el-table-column label="知识库" align="left" prop="name" width="200" :show-overflow-tooltip="true"/>
+          <el-table-column label="描述" align="left" prop="description" :show-overflow-tooltip="true"/>
+          <el-table-column label="索引模式" align="center" prop="indexingTechnique" width="80">
+            <template #default="scope">
+              <span>{{ scope.row.indexingTechnique === 'high_quality' ? '高质量' : '经济' }}</span>
+            </template>
+          </el-table-column>
+          <el-table-column label="权限" align="center" prop="permission" width="80"/>
+          <el-table-column label="提供类型" align="center" prop="provider" width="80">
+            <template #default="scope">
+              <span>{{ scope.row.provider === 'vendor' ? '上传文件' : ' 外部知识库' }}</span>
+            </template>
+          </el-table-column>
+          <el-table-column label="数据源" align="center" prop="dataSourceType" width="120"/>
+          <el-table-column label="文字数量" align="center" prop="wordCount" width="80"/>
+          <el-table-column label="文档数量" align="center" prop="documentCount" width="80"/>
+          <el-table-column label="APP数量" align="center" prop="appCount" width="80"/>
+          <el-table-column label="创建时间" align="left" prop="createTime" width="160"/>
+          <el-table-column label="更新时间" align="left" prop="updateTime" width="160"/>
+          <el-table-column label="操作" align="center" class-name="small-padding fixed-width" width="100">
+            <template #default="scope">
+              <!--              <el-tooltip content="修改" placement="top">
+                              <el-button link type="primary" icon="Edit" @click="handleUpdate(scope.row)"
+                                         v-hasPermi="['device:datasets:edit']"></el-button>
+                            </el-tooltip>-->
+              <el-tooltip content="删除" placement="top">
+                <el-button link type="primary" icon="Delete" @click="handleDelete(scope.row)"
+                           v-hasPermi="['device:datasets:remove']"></el-button>
+              </el-tooltip>
+            </template>
+          </el-table-column>
+        </el-table>
+      </div>
+
+      <pagination v-show="total > 0" :total="total" v-model:page="queryParams.pageNum"
+                  v-model:limit="queryParams.pageSize" @pagination="getList"/>
+    </el-card>
+    <!-- 添加或修改知识库管理对话框 -->
+    <el-dialog :title="dialog.title" v-model="dialog.visible" width="700px" append-to-body>
+      <el-form ref="difyDatasetsFormRef" :model="form" :rules="rules" label-width="40px">
+        <el-form-item label="名称" prop="name">
+          <el-input v-model="form.name" placeholder="请输入知识库名称"/>
+        </el-form-item>
+        <el-form-item label="描述" prop="description">
+          <el-input type="textarea" :rows="5" v-model="form.description" placeholder="请输入描述"/>
+        </el-form-item>
+      </el-form>
+      <template #footer>
+        <div class="dialog-footer">
+          <el-button :loading="buttonLoading" type="primary" @click="submitForm">确 定</el-button>
+          <el-button @click="cancel">取 消</el-button>
+        </div>
+      </template>
+    </el-dialog>
+  </div>
+</template>
+
+<script setup name="DifyDatasets" lang="ts">
+import {
+  addDifyDatasets,
+  delDifyDatasets,
+  getDifyDatasets,
+  listDifyDatasets, syncDifyDatasets,
+  updateDifyDatasets
+} from '@/api/smsb/device/datasets';
+import {DifyDatasetsForm, DifyDatasetsQuery, DifyDatasetsVO} from '@/api/smsb/device/datasets_type';
+
+const {proxy} = getCurrentInstance() as ComponentInternalInstance;
+
+const difyDatasetsList = ref<DifyDatasetsVO[]>([]);
+const buttonLoading = ref(false);
+const loading = ref(true);
+const showSearch = ref(true);
+const ids = ref<Array<string | number>>([]);
+const single = ref(true);
+const multiple = ref(true);
+const total = ref(0);
+
+const queryFormRef = ref<ElFormInstance>();
+const difyDatasetsFormRef = ref<ElFormInstance>();
+
+const dialog = reactive<DialogOption>({
+  visible: false,
+  title: ''
+});
+
+const initFormData: DifyDatasetsForm = {
+  id: undefined,
+  name: undefined,
+  description: undefined,
+  indexingTechnique: undefined,
+  permission: undefined,
+  provider: undefined,
+  dataSourceType: undefined,
+  wordCount: undefined,
+  documentCount: undefined,
+  appCount: undefined,
+  difyId: undefined,
+}
+const data = reactive<PageData<DifyDatasetsForm, DifyDatasetsQuery>>({
+  form: {...initFormData},
+  queryParams: {
+    pageNum: 1,
+    pageSize: 10,
+    name: undefined,
+    params: {}
+  },
+  rules: {}
+});
+
+const {queryParams, form, rules} = toRefs(data);
+
+/** 查询知识库管理列表 */
+const getList = async () => {
+  loading.value = true;
+  const res = await listDifyDatasets(queryParams.value);
+  difyDatasetsList.value = res.rows;
+  total.value = res.total;
+  loading.value = false;
+}
+
+/** 取消按钮 */
+const cancel = () => {
+  reset();
+  dialog.visible = false;
+}
+
+/** 表单重置 */
+const reset = () => {
+  form.value = {...initFormData};
+  difyDatasetsFormRef.value?.resetFields();
+}
+
+/** 搜索按钮操作 */
+const handleQuery = () => {
+  queryParams.value.pageNum = 1;
+  getList();
+}
+
+/** 重置按钮操作 */
+const resetQuery = () => {
+  queryFormRef.value?.resetFields();
+  handleQuery();
+}
+
+/** 多选框选中数据 */
+const handleSelectionChange = (selection: DifyDatasetsVO[]) => {
+  ids.value = selection.map(item => item.id);
+  single.value = selection.length != 1;
+  multiple.value = !selection.length;
+}
+
+/** 新增按钮操作 */
+const handleAdd = () => {
+  reset();
+  dialog.visible = true;
+  dialog.title = "添加知识库";
+}
+
+const handleSync = async () => {
+  await proxy?.$modal.confirm('是否确认同步知识库数据?').finally(() => loading.value = false);
+  loading.value = true;
+  await syncDifyDatasets();
+  proxy?.$modal.msgSuccess("同步成功");
+  await getList();
+  loading.value = false;
+}
+
+
+/** 修改按钮操作 */
+const handleUpdate = async (row?: DifyDatasetsVO) => {
+  reset();
+  const _id = row?.id || ids.value[0]
+  const res = await getDifyDatasets(_id);
+  Object.assign(form.value, res.data);
+  dialog.visible = true;
+  dialog.title = "修改知识库";
+}
+
+/** 提交按钮 */
+const submitForm = () => {
+  difyDatasetsFormRef.value?.validate(async (valid: boolean) => {
+    if (valid) {
+      buttonLoading.value = true;
+      if (form.value.id) {
+        await updateDifyDatasets(form.value).finally(() => buttonLoading.value = false);
+      } else {
+        await addDifyDatasets(form.value).finally(() => buttonLoading.value = false);
+      }
+      proxy?.$modal.msgSuccess("操作成功");
+      dialog.visible = false;
+      await getList();
+    }
+  });
+}
+
+/** 删除按钮操作 */
+const handleDelete = async (row?: DifyDatasetsVO) => {
+  const _ids = row?.id || ids.value;
+  await proxy?.$modal.confirm('是否确认删除知识库编号为"' + _ids + '"的数据项?').finally(() => loading.value = false);
+  await delDifyDatasets(_ids);
+  proxy?.$modal.msgSuccess("删除成功");
+  await getList();
+}
+
+/** 导出按钮操作 */
+const handleExport = () => {
+  proxy?.download('device/difyDatasets/export', {
+    ...queryParams.value
+  }, `difyDatasets_${new Date().getTime()}.xlsx`)
+}
+
+onMounted(() => {
+  getList();
+});
+</script>

+ 1 - 1
smsb-plus-ui/vite.config.ts

@@ -25,7 +25,7 @@ export default defineConfig(({ mode, command }: ConfigEnv): UserConfig => {
       open: true,
       proxy: {
         [env.VITE_APP_BASE_API]: {
-          target: 'http://localhost:8080',
+          target: 'http://localhost:8084/prod-api',
           changeOrigin: true,
           ws: true,
           rewrite: (path) => path.replace(new RegExp('^' + env.VITE_APP_BASE_API), '')