From ae86ef519c5dc557914c29bd1786b1c87d8dcb95 Mon Sep 17 00:00:00 2001
From: adamdottv <2363879+adamdottv@users.noreply.github.com>
Date: Tue, 13 May 2025 10:27:09 -0500
Subject: chore: refactoring

---
 internal/db/logs.sql.go  | 20 +++++++++++++++-----
 internal/db/querier.go   |  2 +-
 internal/db/sql/logs.sql |  6 +++---
 3 files changed, 19 insertions(+), 9 deletions(-)

(limited to 'internal/db')

diff --git a/internal/db/logs.sql.go b/internal/db/logs.sql.go
index d227b472a..aa89d7d22 100644
--- a/internal/db/logs.sql.go
+++ b/internal/db/logs.sql.go
@@ -10,7 +10,7 @@ import (
 	"database/sql"
 )
 
-const createLog = `-- name: CreateLog :exec
+const createLog = `-- name: CreateLog :one
 INSERT INTO logs (
     id,
     session_id,
@@ -27,7 +27,7 @@ INSERT INTO logs (
     ?,
     ?,
     ?
-)
+) RETURNING id, session_id, timestamp, level, message, attributes, created_at
 `
 
 type CreateLogParams struct {
@@ -40,8 +40,8 @@ type CreateLogParams struct {
 	CreatedAt  int64          `json:"created_at"`
 }
 
-func (q *Queries) CreateLog(ctx context.Context, arg CreateLogParams) error {
-	_, err := q.exec(ctx, q.createLogStmt, createLog,
+func (q *Queries) CreateLog(ctx context.Context, arg CreateLogParams) (Log, error) {
+	row := q.queryRow(ctx, q.createLogStmt, createLog,
 		arg.ID,
 		arg.SessionID,
 		arg.Timestamp,
@@ -50,7 +50,17 @@ func (q *Queries) CreateLog(ctx context.Context, arg CreateLogParams) error {
 		arg.Attributes,
 		arg.CreatedAt,
 	)
-	return err
+	var i Log
+	err := row.Scan(
+		&i.ID,
+		&i.SessionID,
+		&i.Timestamp,
+		&i.Level,
+		&i.Message,
+		&i.Attributes,
+		&i.CreatedAt,
+	)
+	return i, err
 }
 
 const listAllLogs = `-- name: ListAllLogs :many
diff --git a/internal/db/querier.go b/internal/db/querier.go
index 08ad41ec7..7b089f021 100644
--- a/internal/db/querier.go
+++ b/internal/db/querier.go
@@ -11,7 +11,7 @@ import (
 
 type Querier interface {
 	CreateFile(ctx context.Context, arg CreateFileParams) (File, error)
-	CreateLog(ctx context.Context, arg CreateLogParams) error
+	CreateLog(ctx context.Context, arg CreateLogParams) (Log, error)
 	CreateMessage(ctx context.Context, arg CreateMessageParams) (Message, error)
 	CreateSession(ctx context.Context, arg CreateSessionParams) (Session, error)
 	DeleteFile(ctx context.Context, id string) error
diff --git a/internal/db/sql/logs.sql b/internal/db/sql/logs.sql
index 1a0655f7e..58414f88a 100644
--- a/internal/db/sql/logs.sql
+++ b/internal/db/sql/logs.sql
@@ -1,4 +1,4 @@
--- name: CreateLog :exec
+-- name: CreateLog :one
 INSERT INTO logs (
     id,
     session_id,
@@ -15,7 +15,7 @@ INSERT INTO logs (
     ?,
     ?,
     ?
-);
+) RETURNING *;
 
 -- name: ListLogsBySession :many
 SELECT * FROM logs
@@ -25,4 +25,4 @@ ORDER BY timestamp ASC;
 -- name: ListAllLogs :many
 SELECT * FROM logs
 ORDER BY timestamp DESC
-LIMIT ?;
\ No newline at end of file
+LIMIT ?;
-- 
cgit v1.2.3