0529

admin

2025-05-29 6ca3d7c61ee8d763f00847d90f4365e348f43078

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
package tech.aiflowy.ai.controller;
 
import tech.aiflowy.ai.entity.AiBotPlugins;
import tech.aiflowy.ai.entity.AiPlugins;
import tech.aiflowy.ai.service.AiBotPluginsService;
import tech.aiflowy.ai.service.AiPluginsService;
import tech.aiflowy.common.domain.Result;
import tech.aiflowy.common.entity.LoginAccount;
import tech.aiflowy.common.web.controller.BaseCurdController;
import tech.aiflowy.common.satoken.util.SaTokenUtil;
import com.mybatisflex.core.query.QueryWrapper;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
 
import javax.annotation.Resource;
import java.io.Serializable;
import java.util.Collection;
import java.util.Date;
 
/**
 * 插件 控制层。
 *
 * @author ArkLight
 * @since 2025-04-01
 */
@RestController
@RequestMapping("/api/v1/aiPlugins")
public class AiPluginsController extends BaseCurdController<AiPluginsService, AiPlugins> {
 
    public AiPluginsController(AiPluginsService service) {
        super(service);
    }
 
    @Resource
    private AiBotPluginsService aiBotPluginsService;
 
    @Override
    protected Result onSaveOrUpdateBefore(AiPlugins entity, boolean isSave) {
        LoginAccount loginUser = SaTokenUtil.getLoginAccount();
        if (isSave) {
            commonFiled(entity,loginUser.getId(),loginUser.getTenantId(), loginUser.getDeptId());
        } else {
            entity.setModified(new Date());
            entity.setModifiedBy(loginUser.getId());
        }
        return super.onSaveOrUpdateBefore(entity, isSave);
    }
 
    @Override
    protected void onRemoveAfter(Collection<Serializable> ids) {
        QueryWrapper wrapper = QueryWrapper.create();
        wrapper.in(AiBotPlugins::getPluginToolId, ids);
        aiBotPluginsService.remove(wrapper);
    }
}