diff --git a/internal/repo/agent_repo.go b/internal/repo/agent_repo.go index d227b9d..82151a0 100644 --- a/internal/repo/agent_repo.go +++ b/internal/repo/agent_repo.go @@ -48,7 +48,7 @@ func (r *AgentRepo) FindOnlineAgents(ctx context.Context) ([]models.Agent, error func (r *AgentRepo) FindByIP(ctx context.Context, ip string) (*models.Agent, error) { var agent models.Agent err := r.db.WithContext(ctx). - Where("ip = ? or ipv4 = ? OR ipv6 = ?", ip, ip). + Where("ip = ? or ipv4 = ? OR ipv6 = ?", ip, ip, ip). First(&agent).Error if err != nil { return nil, err