fix: agent tasks cmd

This commit is contained in:
engigu
2025-12-30 21:09:36 +08:00
parent 7f83401b2e
commit 94c6e1b7fb
3 changed files with 24 additions and 0 deletions
+9
View File
@@ -230,6 +230,15 @@ func (s *AgentService) GetByToken(token string) *models.Agent {
return &agent
}
// GetByMachineID 根据 MachineID 获取 Agent
func (s *AgentService) GetByMachineID(machineID string) *models.Agent {
var agent models.Agent
if err := database.DB.Where("machine_id = ?", machineID).First(&agent).Error; err != nil {
return nil
}
return &agent
}
// List 获取 Agent 列表
func (s *AgentService) List() []models.Agent {
var agents []models.Agent