4ceb74da9e
- Standardized field formatting in TriggerResult and Execution structs for improved code clarity. - Added test cases in robot_test.go to ensure Robot can run with nil config and quota, verifying default behavior. - Enhanced comments in the Goals struct to clarify task objectives and improve documentation consistency.
106 lines
2.3 KiB
Go
106 lines
2.3 KiB
Go
package cache
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/yaoapp/yao/agent/robot/types"
|
|
)
|
|
|
|
// Cache implements types.Cache interface
|
|
// This is a stub implementation for Phase 2
|
|
type Cache struct {
|
|
robots map[string]*types.Robot // memberID -> Robot
|
|
byTeam map[string][]string // teamID -> memberIDs
|
|
mu sync.RWMutex
|
|
}
|
|
|
|
// New creates a new cache instance
|
|
func New() *Cache {
|
|
return &Cache{
|
|
robots: make(map[string]*types.Robot),
|
|
byTeam: make(map[string][]string),
|
|
}
|
|
}
|
|
|
|
// Load loads all active robots from database
|
|
// Stub: returns nil (will be implemented in Phase 3)
|
|
func (c *Cache) Load(ctx *types.Context) error {
|
|
return nil
|
|
}
|
|
|
|
// Get returns a robot by member ID
|
|
// Stub: returns nil (will be implemented in Phase 3)
|
|
func (c *Cache) Get(memberID string) *types.Robot {
|
|
c.mu.RLock()
|
|
defer c.mu.RUnlock()
|
|
return c.robots[memberID]
|
|
}
|
|
|
|
// List returns all robots for a team
|
|
// Stub: returns empty slice (will be implemented in Phase 3)
|
|
func (c *Cache) List(teamID string) []*types.Robot {
|
|
c.mu.RLock()
|
|
defer c.mu.RUnlock()
|
|
|
|
memberIDs := c.byTeam[teamID]
|
|
robots := make([]*types.Robot, 0, len(memberIDs))
|
|
for _, memberID := range memberIDs {
|
|
if robot := c.robots[memberID]; robot != nil {
|
|
robots = append(robots, robot)
|
|
}
|
|
}
|
|
return robots
|
|
}
|
|
|
|
// Refresh refreshes a single robot's config from database
|
|
// Stub: returns nil (will be implemented in Phase 3)
|
|
func (c *Cache) Refresh(ctx *types.Context, memberID string) error {
|
|
return nil
|
|
}
|
|
|
|
// Add adds or updates a robot in cache
|
|
func (c *Cache) Add(robot *types.Robot) {
|
|
c.mu.Lock()
|
|
defer c.mu.Unlock()
|
|
|
|
c.robots[robot.MemberID] = robot
|
|
|
|
// Update team index
|
|
if _, exists := c.byTeam[robot.TeamID]; !exists {
|
|
c.byTeam[robot.TeamID] = []string{}
|
|
}
|
|
|
|
// Check if member ID already in team list
|
|
found := false
|
|
for _, id := range c.byTeam[robot.TeamID] {
|
|
if id == robot.MemberID {
|
|
found = true
|
|
break
|
|
}
|
|
}
|
|
if !found {
|
|
c.byTeam[robot.TeamID] = append(c.byTeam[robot.TeamID], robot.MemberID)
|
|
}
|
|
}
|
|
|
|
// Remove removes a robot from cache
|
|
func (c *Cache) Remove(memberID string) {
|
|
c.mu.Lock()
|
|
defer c.mu.Unlock()
|
|
|
|
robot := c.robots[memberID]
|
|
if robot == nil {
|
|
return
|
|
}
|
|
|
|
delete(c.robots, memberID)
|
|
|
|
// Remove from team index
|
|
teamMembers := c.byTeam[robot.TeamID]
|
|
for i, id := range teamMembers {
|
|
if id == memberID {
|
|
c.byTeam[robot.TeamID] = append(teamMembers[:i], teamMembers[i+1:]...)
|
|
break
|
|
}
|
|
}
|
|
}
|