lorsan/HomeOps-Hub
1
0
Fork 0
mirror of https://github.com/lorsanstand/HomeOps-Hub.git synced 2026-09-18 15:08:21 +03:00
Code Issues Packages Projects Releases Wiki Activity

refactor: migrate to sqlite

Browse Source
This commit is contained in:
lorsan
2026-05-05 20:32:29 +03:00
parent 91c49fb563
commit 4988aacc56
13 changed files with 207 additions and 126 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hub/internal/store/mapper.go
+12 -3
View File
@@ -9,6 +9,8 @@ import (
)
func toDBAgent(agent domainHub.CreateAgentModel) gen2.CreateAgentParams {
jsonCaps := toJSONCapabilities(agent.Capabilities)
capsStr := string(jsonCaps)
return gen2.CreateAgentParams{
AgentID: agent.AgentID,
AgentName: &agent.AgentName,
@@ -16,11 +18,13 @@ func toDBAgent(agent domainHub.CreateAgentModel) gen2.CreateAgentParams {
System: agent.System,
Hostname: agent.Hostname,
Version: agent.Version,
Capabilities: toJSONCapabilities(agent.Capabilities),
Capabilities: &capsStr,
}
}
func toUpdateDBAgent(agent domainHub.CreateAgentModel) gen2.UpdateAgentByIDParams {
jsonCaps := toJSONCapabilities(agent.Capabilities)
capsStr := string(jsonCaps)
return gen2.UpdateAgentByIDParams{
AgentID: agent.AgentID,
AgentName: &agent.AgentName,
@@ -28,7 +32,7 @@ func toUpdateDBAgent(agent domainHub.CreateAgentModel) gen2.UpdateAgentByIDParam
System: agent.System,
Hostname: agent.Hostname,
Version: agent.Version,
Capabilities: toJSONCapabilities(agent.Capabilities),
Capabilities: &capsStr,
}
}
@@ -47,6 +51,11 @@ func toAgentModel(dbAgent gen2.Agent) domainHub.AgentModel {
dbAgentName = *dbAgent.AgentName
}
var capsBytes []byte
if dbAgent.Capabilities != nil {
capsBytes = []byte(*dbAgent.Capabilities)
}
return domainHub.AgentModel{
ID: int(dbAgent.ID),
AgentID: dbAgent.AgentID,
@@ -54,7 +63,7 @@ func toAgentModel(dbAgent gen2.Agent) domainHub.AgentModel {
Architecture: dbAgent.Architecture,
System: dbAgent.System,
Hostname: dbAgent.Hostname,
Capabilities: toDomainCapabilities(dbAgent.Capabilities),
Capabilities: toDomainCapabilities(capsBytes),
}
}
hub/internal/store/sqlc/gen/agent.sql.go
+20 -14
View File
@@ -1,7 +1,7 @@
// Code generated by sqlc. DO NOT EDIT.
// versions:
// sqlc v1.30.0
// source: cmd.sql
// source: agent.sql
package gen
@@ -11,7 +11,7 @@ import (
const createAgent = `-- name: CreateAgent :exec
INSERT INTO agents (agent_id, agent_name, architecture, system, hostname, version, capabilities)
VALUES ($1, $2, $3, $4, $5, $6, $7)
VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7)
`
type CreateAgentParams struct {
@@ -21,11 +21,11 @@ type CreateAgentParams struct {
System string
Hostname string
Version string
Capabilities []byte
Capabilities *string
}
func (q *Queries) CreateAgent(ctx context.Context, arg CreateAgentParams) error {
_, err := q.db.Exec(ctx, createAgent,
_, err := q.db.ExecContext(ctx, createAgent,
arg.AgentID,
arg.AgentName,
arg.Architecture,
@@ -39,11 +39,11 @@ func (q *Queries) CreateAgent(ctx context.Context, arg CreateAgentParams) error
const getAgentByAgentID = `-- name: GetAgentByAgentID :one
SELECT id, agent_id, agent_name, architecture, system, hostname, version, capabilities, registered_at from agents
WHERE agent_id=$1
WHERE agent_id = ?1
`
func (q *Queries) GetAgentByAgentID(ctx context.Context, agentID string) (Agent, error) {
row := q.db.QueryRow(ctx, getAgentByAgentID, agentID)
row := q.db.QueryRowContext(ctx, getAgentByAgentID, agentID)
var i Agent
err := row.Scan(
&i.ID,
@@ -61,11 +61,11 @@ func (q *Queries) GetAgentByAgentID(ctx context.Context, agentID string) (Agent,
const getAgentByID = `-- name: GetAgentByID :one
SELECT id, agent_id, agent_name, architecture, system, hostname, version, capabilities, registered_at from agents
WHERE id=$1
WHERE id = ?1
`
func (q *Queries) GetAgentByID(ctx context.Context, id int32) (Agent, error) {
row := q.db.QueryRow(ctx, getAgentByID, id)
func (q *Queries) GetAgentByID(ctx context.Context, id int64) (Agent, error) {
row := q.db.QueryRowContext(ctx, getAgentByID, id)
var i Agent
err := row.Scan(
&i.ID,
@@ -83,8 +83,14 @@ func (q *Queries) GetAgentByID(ctx context.Context, id int32) (Agent, error) {
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
SET agent_id = ?1,
agent_name = ?2,
architecture = ?3,
system = ?4,
hostname = ?5,
version = ?6,
capabilities = ?7
WHERE id = ?8
`
type UpdateAgentByIDParams struct {
@@ -94,12 +100,12 @@ type UpdateAgentByIDParams struct {
System string
Hostname string
Version string
Capabilities []byte
ID int32
Capabilities *string
ID int64
}
func (q *Queries) UpdateAgentByID(ctx context.Context, arg UpdateAgentByIDParams) error {
_, err := q.db.Exec(ctx, updateAgentByID,
_, err := q.db.ExecContext(ctx, updateAgentByID,
arg.AgentID,
arg.AgentName,
arg.Architecture,
hub/internal/store/sqlc/gen/db.go
+6 -7
View File
@@ -6,15 +6,14 @@ package gen
import (
"context"
"github.com/jackc/pgx/v5"
"github.com/jackc/pgx/v5/pgconn"
"database/sql"
)
type DBTX interface {
Exec(context.Context, string, ...interface{}) (pgconn.CommandTag, error)
Query(context.Context, string, ...interface{}) (pgx.Rows, error)
QueryRow(context.Context, string, ...interface{}) pgx.Row
ExecContext(context.Context, string, ...interface{}) (sql.Result, error)
PrepareContext(context.Context, string) (*sql.Stmt, error)
QueryContext(context.Context, string, ...interface{}) (*sql.Rows, error)
QueryRowContext(context.Context, string, ...interface{}) *sql.Row
}
func New(db DBTX) *Queries {
@@ -25,7 +24,7 @@ type Queries struct {
db DBTX
}
func (q *Queries) WithTx(tx pgx.Tx) *Queries {
func (q *Queries) WithTx(tx *sql.Tx) *Queries {
return &Queries{
db: tx,
}
hub/internal/store/sqlc/gen/models.go
+4 -4
View File
@@ -5,17 +5,17 @@
package gen
import (
"github.com/jackc/pgx/v5/pgtype"
"time"
)
type Agent struct {
ID int32
ID int64
AgentID string
AgentName *string
Architecture string
System string
Hostname string
Version string
Capabilities []byte
RegisteredAt pgtype.Timestamp
Capabilities *string
RegisteredAt time.Time
}
hub/internal/store/sqlc/queries/agent.sql
+11 -5
View File
@@ -1,16 +1,22 @@
-- name: CreateAgent :exec
INSERT INTO agents (agent_id, agent_name, architecture, system, hostname, version, capabilities)
VALUES ($1, $2, $3, $4, $5, $6, $7);
VALUES (:agent_id, :agent_name, :architecture, :system, :hostname, :version, :capabilities);
-- name: GetAgentByID :one
SELECT * from agents
WHERE id=$1;
WHERE id = :id;
-- name: GetAgentByAgentID :one
SELECT * from agents
WHERE agent_id=$1;
WHERE agent_id = :agent_id;
-- 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;
SET agent_id = :agent_id,
agent_name = :agent_name,
architecture = :architecture,
system = :system,
hostname = :hostname,
version = :version,
capabilities = :capabilities
WHERE id = :id;
hub/internal/store/store.go
+3 -3
View File
@@ -2,8 +2,8 @@ package store
import (
"context"
"database/sql"
"github.com/jackc/pgx/v5/pgxpool"
domainHub "github.com/lorsanstand/HomeOps-Hub/hub/internal/domain"
"github.com/lorsanstand/HomeOps-Hub/hub/internal/store/sqlc/gen"
)
@@ -12,7 +12,7 @@ type HubStore struct {
queries *gen.Queries
}
func NewHubStore(db *pgxpool.Pool) *HubStore {
func NewHubStore(db *sql.DB) *HubStore {
queries := gen.New(db)
return &HubStore{queries}
}
@@ -31,6 +31,6 @@ func (h *HubStore) GetAgentByAgentID(ctx context.Context, AgentID string) (domai
func (h *HubStore) UpdateAgentByID(ctx context.Context, ID int, updateAgent domainHub.CreateAgentModel) error {
data := toUpdateDBAgent(updateAgent)
data.ID = int32(ID)
data.ID = int64(ID)
return h.queries.UpdateAgentByID(ctx, data)
}