t
18c9354095
- 注入数据库、运行时配置、统一响应、文件存储和平台 AI 能力，补充业务读写工具与客户快捷操作契约。 - 移除模块内重复的组织、客户、工单、标签、技能、旧工作流、MCP 和迁移实现，将身份权限与业务主体交由宿主管理。 - 使用 libSQL 重构向量存储，并完善图片消息、访客身份、排队调度、企业微信和支持聊天页面。 - 统一 HTTP、DTO 与 WebSocket 的 snake_case 协议，补齐模块初始化、业务动作和公共载荷等回归测试。
49 lines
1.3 KiB
Go
49 lines
1.3 KiB
Go
package services_test
|
|
|
|
import (
|
|
"context"
|
|
"slices"
|
|
"sync"
|
|
|
|
"code.tczkiot.com/wlw/ai-agent/identity"
|
|
"code.tczkiot.com/wlw/ai-agent/internal/pkg/enums"
|
|
"code.tczkiot.com/wlw/ai-agent/internal/services"
|
|
)
|
|
|
|
var testExternalSubjects sync.Map
|
|
|
|
func registerTestExternalSubject(id int64, username, name string, status enums.Status) {
|
|
registerTestSubject(identity.Subject{
|
|
Type: identity.SubjectAdmin,
|
|
Category: identity.CategorySystem,
|
|
ID: id,
|
|
Username: username,
|
|
Name: name,
|
|
Identifier: username,
|
|
Enabled: status == enums.StatusOk,
|
|
})
|
|
}
|
|
|
|
func registerTestSubject(subject identity.Subject) {
|
|
testExternalSubjects.Store(string(subject.Type)+":"+subject.Identifier, subject)
|
|
services.SetQuerySubjects(func(_ context.Context, query identity.Query) ([]identity.Subject, error) {
|
|
results := make([]identity.Subject, 0)
|
|
testExternalSubjects.Range(func(_, value any) bool {
|
|
subject := value.(identity.Subject)
|
|
if len(query.Types) > 0 && !slices.Contains(query.Types, subject.Type) {
|
|
return true
|
|
}
|
|
if len(query.IDs) > 0 && !slices.Contains(query.IDs, subject.ID) {
|
|
return true
|
|
}
|
|
if query.EnabledOnly && !subject.Enabled {
|
|
return true
|
|
}
|
|
results = append(results, subject)
|
|
return true
|
|
})
|
|
return results, nil
|
|
})
|
|
services.SetAuthorize(func(_ context.Context, _ string) error { return nil })
|
|
}
|