From 5d16d9a0e7240ff4e6346bfee4890159da5a764e Mon Sep 17 00:00:00 2001
From: zhou zhou <3272660260@qq.com>
Date: 星期四, 19 三月 2026 11:40:51 +0800
Subject: [PATCH] #AI.记忆治理
---
rsf-server/src/main/java/com/vincent/rsf/server/ai/controller/AiPromptController.java | 34 +++++++++++++++++++++++++++-------
1 files changed, 27 insertions(+), 7 deletions(-)
diff --git a/rsf-server/src/main/java/com/vincent/rsf/server/ai/controller/AiPromptController.java b/rsf-server/src/main/java/com/vincent/rsf/server/ai/controller/AiPromptController.java
index 8c40c39..d35f879 100644
--- a/rsf-server/src/main/java/com/vincent/rsf/server/ai/controller/AiPromptController.java
+++ b/rsf-server/src/main/java/com/vincent/rsf/server/ai/controller/AiPromptController.java
@@ -1,6 +1,7 @@
package com.vincent.rsf.server.ai.controller;
import com.vincent.rsf.framework.common.R;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.vincent.rsf.server.ai.entity.AiPrompt;
import com.vincent.rsf.server.ai.service.AiPromptService;
import com.vincent.rsf.server.common.annotation.OperationLog;
@@ -28,32 +29,46 @@
public R page(@RequestBody Map<String, Object> map) {
BaseParam baseParam = buildParam(map, BaseParam.class);
PageParam<AiPrompt, BaseParam> pageParam = new PageParam<>(baseParam, AiPrompt.class);
- return R.ok().add(aiPromptService.page(pageParam, pageParam.buildWrapper(true)));
+ return R.ok().add(aiPromptService.page(pageParam, pageParam.buildWrapper(true)
+ .eq("tenant_id", getTenantId())
+ .eq("deleted", 0)));
}
@PreAuthorize("hasAuthority('system:aiPrompt:list')")
@PostMapping("/aiPrompt/list")
public R list(@RequestBody Map<String, Object> map) {
- return R.ok().add(aiPromptService.list());
+ return R.ok().add(aiPromptService.list(new LambdaQueryWrapper<AiPrompt>()
+ .eq(AiPrompt::getTenantId, getTenantId())
+ .eq(AiPrompt::getDeleted, 0)
+ .orderByDesc(AiPrompt::getCreateTime)
+ .orderByDesc(AiPrompt::getId)));
}
@PreAuthorize("hasAuthority('system:aiPrompt:list')")
@PostMapping({"/aiPrompt/many/{ids}", "/aiPrompts/many/{ids}"})
public R many(@PathVariable Long[] ids) {
- return R.ok().add(aiPromptService.listByIds(Arrays.asList(ids)));
+ return R.ok().add(aiPromptService.list(new LambdaQueryWrapper<AiPrompt>()
+ .eq(AiPrompt::getTenantId, getTenantId())
+ .eq(AiPrompt::getDeleted, 0)
+ .in(AiPrompt::getId, Arrays.asList(ids))));
}
@PreAuthorize("hasAuthority('system:aiPrompt:list')")
@GetMapping("/aiPrompt/{id}")
public R get(@PathVariable("id") Long id) {
- return R.ok().add(aiPromptService.getById(id));
+ return R.ok().add(aiPromptService.getOne(new LambdaQueryWrapper<AiPrompt>()
+ .eq(AiPrompt::getId, id)
+ .eq(AiPrompt::getTenantId, getTenantId())
+ .eq(AiPrompt::getDeleted, 0)
+ .last("limit 1")));
}
@PreAuthorize("hasAuthority('system:aiPrompt:save')")
@OperationLog("Create AiPrompt")
@PostMapping("/aiPrompt/save")
public R save(@RequestBody AiPrompt aiPrompt) {
- aiPromptService.validateBeforeSave(aiPrompt);
+ aiPrompt.setTenantId(getTenantId());
+ aiPromptService.validateBeforeSave(aiPrompt, getTenantId());
aiPrompt.setCreateBy(getLoginUserId());
aiPrompt.setCreateTime(new Date());
aiPrompt.setUpdateBy(getLoginUserId());
@@ -68,7 +83,8 @@
@OperationLog("Update AiPrompt")
@PostMapping("/aiPrompt/update")
public R update(@RequestBody AiPrompt aiPrompt) {
- aiPromptService.validateBeforeUpdate(aiPrompt);
+ aiPrompt.setTenantId(getTenantId());
+ aiPromptService.validateBeforeUpdate(aiPrompt, getTenantId());
aiPrompt.setUpdateBy(getLoginUserId());
aiPrompt.setUpdateTime(new Date());
if (!aiPromptService.updateById(aiPrompt)) {
@@ -90,6 +106,10 @@
@PreAuthorize("hasAuthority('system:aiPrompt:list')")
@PostMapping("/aiPrompt/export")
public void export(@RequestBody Map<String, Object> map, HttpServletResponse response) throws Exception {
- ExcelUtil.build(ExcelUtil.create(aiPromptService.list(), AiPrompt.class), response);
+ ExcelUtil.build(ExcelUtil.create(aiPromptService.list(new LambdaQueryWrapper<AiPrompt>()
+ .eq(AiPrompt::getTenantId, getTenantId())
+ .eq(AiPrompt::getDeleted, 0)
+ .orderByDesc(AiPrompt::getCreateTime)
+ .orderByDesc(AiPrompt::getId)), AiPrompt.class), response);
}
}
--
Gitblit v1.9.1