feat: create save new agent in db

This commit is contained in:
2026-04-19 16:16:03 +03:00
parent 52766cf7e8
commit e289365ce8
14 changed files with 314 additions and 72 deletions
+75
View File
@@ -36,3 +36,78 @@ func (q *Queries) CreateAgent(ctx context.Context, arg CreateAgentParams) error
)
return err
}
const getAgentByAgentID = `-- name: GetAgentByAgentID :one
SELECT id, agent_id, agent_name, architecture, system, hostname, version, capabilities, registered_at from agents
WHERE agent_id=$1
`
func (q *Queries) GetAgentByAgentID(ctx context.Context, agentID string) (Agent, error) {
row := q.db.QueryRow(ctx, getAgentByAgentID, agentID)
var i Agent
err := row.Scan(
&i.ID,
&i.AgentID,
&i.AgentName,
&i.Architecture,
&i.System,
&i.Hostname,
&i.Version,
&i.Capabilities,
&i.RegisteredAt,
)
return i, err
}
const getAgentByID = `-- name: GetAgentByID :one
SELECT id, agent_id, agent_name, architecture, system, hostname, version, capabilities, registered_at from agents
WHERE id=$1
`
func (q *Queries) GetAgentByID(ctx context.Context, id int32) (Agent, error) {
row := q.db.QueryRow(ctx, getAgentByID, id)
var i Agent
err := row.Scan(
&i.ID,
&i.AgentID,
&i.AgentName,
&i.Architecture,
&i.System,
&i.Hostname,
&i.Version,
&i.Capabilities,
&i.RegisteredAt,
)
return i, err
}
const updateAgentByID = `-- name: UpdateAgentByID :exec
UPDATE agents
SET agent_id=$1, agent_name=$2, architecture=$3, system=$4, hostname=$5, version=$6, capabilities=$7
WHERE id=$8
`
type UpdateAgentByIDParams struct {
AgentID string
AgentName *string
Architecture string
System string
Hostname string
Version string
Capabilities []byte
ID int32
}
func (q *Queries) UpdateAgentByID(ctx context.Context, arg UpdateAgentByIDParams) error {
_, err := q.db.Exec(ctx, updateAgentByID,
arg.AgentID,
arg.AgentName,
arg.Architecture,
arg.System,
arg.Hostname,
arg.Version,
arg.Capabilities,
arg.ID,
)
return err
}
+1 -1
View File
@@ -9,7 +9,7 @@ import (
)
type Agent struct {
ID int64
ID int32
AgentID string
AgentName *string
Architecture string