whm
cb76824e94
Create and restart paths enable IsSystemDefault channels; SyncPage auto-starts after save. Co-authored-by: Cursor <cursoragent@cursor.com>
92 lines
2.9 KiB
Go
92 lines
2.9 KiB
Go
package applogic
|
||
|
||
import (
|
||
"fmt"
|
||
"strings"
|
||
|
||
"aijianzhan/platform/internal/agentstore"
|
||
"aijianzhan/platform/internal/dbsync"
|
||
"aijianzhan/platform/internal/svc"
|
||
"aijianzhan/platform/internal/types"
|
||
"aijianzhan/platform/internal/userstore"
|
||
)
|
||
|
||
// fillAgentSyncOnToken Z12a：换票带回同步落点。
|
||
func fillAgentSyncOnToken(resp *types.TokenResp, acc *agentstore.Account) {
|
||
if resp == nil || acc == nil {
|
||
return
|
||
}
|
||
resp.ChannelID = strings.TrimSpace(acc.ChannelID)
|
||
resp.OnlineDBID = strings.TrimSpace(acc.OnlineDBID)
|
||
resp.DatabaseName = strings.TrimSpace(acc.DatabaseName)
|
||
resp.SyncBound = resp.ChannelID != "" && resp.OnlineDBID != ""
|
||
}
|
||
|
||
// fillUserSyncOnToken：人类登录优先从本人 Binding 带回落点。
|
||
func fillUserSyncOnToken(svcCtx *svc.ServiceContext, resp *types.TokenResp, u *userstore.User) {
|
||
if resp == nil || u == nil || svcCtx == nil || svcCtx.DBSync == nil || u.TenantID <= 0 {
|
||
return
|
||
}
|
||
list, err := svcCtx.DBSync.Store().ListBindingsFiltered(u.TenantID, u.UserID, "")
|
||
if err != nil || len(list) == 0 {
|
||
return
|
||
}
|
||
b := list[0]
|
||
for i := range list {
|
||
if list[i].ChannelID != "" && list[i].OnlineDBID != "" {
|
||
b = list[i]
|
||
break
|
||
}
|
||
}
|
||
resp.ChannelID = strings.TrimSpace(b.ChannelID)
|
||
resp.OnlineDBID = strings.TrimSpace(b.OnlineDBID)
|
||
resp.DatabaseName = strings.TrimSpace(b.DatabaseName)
|
||
if resp.DatabaseName == "" {
|
||
resp.DatabaseName = strings.TrimSpace(b.DisplayName)
|
||
}
|
||
resp.SyncBound = resp.ChannelID != "" && resp.OnlineDBID != ""
|
||
}
|
||
|
||
// ensureAgentSyncBind Z12c：启用/创建时若无通道则自动创建公司默认同步通道并写回智能体。
|
||
func (l *AgentAdminLogic) ensureAgentSyncBind(acc *agentstore.Account) (*agentstore.Account, error) {
|
||
if acc == nil {
|
||
return nil, fmt.Errorf("agent nil")
|
||
}
|
||
if strings.TrimSpace(acc.ChannelID) != "" && strings.TrimSpace(acc.OnlineDBID) != "" {
|
||
return acc, nil
|
||
}
|
||
if l.svcCtx.DBSync == nil {
|
||
return acc, nil
|
||
}
|
||
cfg := l.svcCtx.Config.DBSync
|
||
driver := dbsync.Driver(strings.TrimSpace(cfg.DefaultRemoteDriver))
|
||
if driver == "" {
|
||
driver = dbsync.DriverPostgres
|
||
}
|
||
ch, err := l.svcCtx.DBSync.EnsureAndStartSystemDefaultChannel(dbsync.DefaultChannelOpts{
|
||
TenantID: acc.TenantID,
|
||
AgentID: acc.AgentID,
|
||
Name: fmt.Sprintf("默认同步 · %s", acc.Name),
|
||
RemoteDriver: driver,
|
||
RemoteDSN: strings.TrimSpace(cfg.DefaultRemoteDSN),
|
||
DatabaseName: acc.DatabaseName,
|
||
})
|
||
if err != nil {
|
||
return nil, fmt.Errorf("ensure default channel: %w", err)
|
||
}
|
||
online := dbsync.ResolveOnlineDBID(acc.OnlineDBID, ch.ID)
|
||
dbName := strings.TrimSpace(acc.DatabaseName)
|
||
if dbName == "" {
|
||
dbName = fmt.Sprintf("agent_%d", acc.AgentID)
|
||
}
|
||
st, err := l.store()
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
updated, err := st.AttachSyncBind(l.ctx, acc.AgentID, acc.TenantID, ch.ID, online, dbName, acc.Status == agentstore.StatusActive || acc.Status == "")
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return updated, nil
|
||
}
|