Compare commits
3 Commits
a199e34730
...
88ab880a8c
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
88ab880a8c | ||
|
|
e6d9e356ca | ||
|
|
b85df7bd31 |
@@ -28,6 +28,7 @@ func One(ctx context.Context) error {
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
_ = feeds
|
||||
|
||||
return io.EOF
|
||||
}
|
||||
|
||||
48
src/db/db.go
48
src/db/db.go
@@ -5,6 +5,7 @@ import (
|
||||
"database/sql"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"strings"
|
||||
)
|
||||
|
||||
func QueryOne[T any](ctx context.Context, q string, args ...any) (T, error) {
|
||||
@@ -33,12 +34,33 @@ func Query[T any](ctx context.Context, q string, args ...any) ([]T, error) {
|
||||
}
|
||||
scanners := func(columns []string) ([]any, error) {
|
||||
s := make([]any, len(columns))
|
||||
for i, k := range columns {
|
||||
v, ok := m[k]
|
||||
if !ok {
|
||||
return nil, fmt.Errorf("cannot scan column %s to %T (%+v)", k, a, m)
|
||||
i := 0
|
||||
for i < len(columns) {
|
||||
k := columns[i]
|
||||
if strings.Contains(k, ".") {
|
||||
columns := strings.SplitN(k, ".", 2)
|
||||
m2, ok := m[columns[0]]
|
||||
if !ok {
|
||||
return nil, fmt.Errorf("no column %s in %T (%+v)", columns[0], a, m)
|
||||
}
|
||||
m3, ok := m2.(map[string]any)
|
||||
if !ok {
|
||||
return nil, fmt.Errorf("cannot scan subfield %s of %s of %T (%+v)", columns[1], columns[0], a, m)
|
||||
}
|
||||
v, ok := m3[columns[1]]
|
||||
if !ok {
|
||||
return nil, fmt.Errorf("no subfield %s of %s of %T (%+v)", columns[1], columns[0], a, m)
|
||||
}
|
||||
s[i] = &v
|
||||
i += 1
|
||||
} else {
|
||||
v, ok := m[k]
|
||||
if !ok {
|
||||
return nil, fmt.Errorf("no column %s in %T (%+v)", k, a, m)
|
||||
}
|
||||
s[i] = &v
|
||||
i += 1
|
||||
}
|
||||
s[i] = &v
|
||||
}
|
||||
return s, nil
|
||||
}
|
||||
@@ -67,7 +89,21 @@ func Query[T any](ctx context.Context, q string, args ...any) ([]T, error) {
|
||||
|
||||
m := map[string]any{}
|
||||
for i, column := range columns {
|
||||
m[column] = scanners[i]
|
||||
if !strings.Contains(column, ".") {
|
||||
m[column] = scanners[i]
|
||||
} else {
|
||||
columns := strings.SplitN(column, ".", 2)
|
||||
m2, ok := m[columns[0]]
|
||||
if !ok {
|
||||
m2 = map[string]any{}
|
||||
}
|
||||
m3, ok := m2.(map[string]any)
|
||||
if !ok {
|
||||
return fmt.Errorf("%s is not a submap", columns[0])
|
||||
}
|
||||
m3[columns[1]] = scanners[i]
|
||||
m[columns[0]] = m3
|
||||
}
|
||||
}
|
||||
|
||||
var a T
|
||||
|
||||
@@ -46,4 +46,15 @@ func TestDB(t *testing.T) {
|
||||
} else if gots[1].K != "b" {
|
||||
t.Errorf("expected [1]='b' but got %q", gots[1].K)
|
||||
}
|
||||
|
||||
type NestedResult struct {
|
||||
Nest struct {
|
||||
K string `json:"k"`
|
||||
}
|
||||
}
|
||||
if got, err := db.QueryOne[NestedResult](ctx, `SELECT k AS "Nest.k" FROM test WHERE k='a'`); err != nil {
|
||||
t.Errorf("failed nested query one: %v", err)
|
||||
} else if got.Nest.K != "a" {
|
||||
t.Errorf("bad nested query one: %+v", got)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -128,20 +128,20 @@ func (f *Feeds) Get(ctx context.Context, id string) (Feed, error) {
|
||||
versions.created_at AS "Version.Created",
|
||||
versions.url AS "Version.URL",
|
||||
versions.cron AS "Version.Cron",
|
||||
(
|
||||
SELECT executed_at
|
||||
FROM "feed.executions"
|
||||
WHERE entries_id = entry.ID
|
||||
(
|
||||
SELECT executed_at
|
||||
FROM "feed.executions"
|
||||
WHERE entries_id = entry.ID
|
||||
ORDER BY executed_at DESC
|
||||
LIMIT 1
|
||||
) AS "Execution.Executed",
|
||||
(
|
||||
SELECT versions_created_at
|
||||
FROM "feed.executions"
|
||||
WHERE entries_id = entry.ID
|
||||
) AS "Execution.Executed",
|
||||
(
|
||||
SELECT versions_created_at
|
||||
FROM "feed.executions"
|
||||
WHERE entries_id = entry.ID
|
||||
ORDER BY executed_at DESC
|
||||
LIMIT 1
|
||||
) AS "Execution.Version"
|
||||
) AS "Execution.Version"
|
||||
FROM entry
|
||||
JOIN "feed.versions" version_entries_id ON
|
||||
version_entries_id.entries_id=entry.ID
|
||||
@@ -196,22 +196,21 @@ func (f *Feeds) oldGet(ctx context.Context, id string) (Feed, error) {
|
||||
func (f *Feeds) Insert(ctx context.Context, url, cron string) (string, error) {
|
||||
now := time.Now()
|
||||
id := uuid.New().String()
|
||||
q := `
|
||||
BEGIN;
|
||||
INSERT INTO "feed.entries" (
|
||||
id,
|
||||
created_at,
|
||||
updated_at
|
||||
) VALUES (?, ?, ?);
|
||||
INSERT INTO "feed.versions" (
|
||||
entries_id,
|
||||
created_at,
|
||||
url,
|
||||
cron
|
||||
) VALUES (?, ?, ?, ?);
|
||||
COMMIT;
|
||||
`
|
||||
return id, db.Exec(ctx, q,
|
||||
return id, db.Exec(ctx, `
|
||||
BEGIN;
|
||||
INSERT INTO "feed.entries" (
|
||||
id,
|
||||
created_at,
|
||||
updated_at
|
||||
) VALUES ($1, $2, $3);
|
||||
INSERT INTO "feed.versions" (
|
||||
entries_id,
|
||||
created_at,
|
||||
url,
|
||||
cron
|
||||
) VALUES ($4, $5, $6, $7);
|
||||
COMMIT;
|
||||
`,
|
||||
id, now, now,
|
||||
id, now, url, cron,
|
||||
)
|
||||
|
||||
@@ -51,6 +51,36 @@ func TestFeeds(t *testing.T) {
|
||||
if err != nil {
|
||||
t.Fatal("cannot get:", err)
|
||||
}
|
||||
t.Errorf("%+v", got)
|
||||
t.Logf("%+v", got)
|
||||
|
||||
if got.Entry.ID == "" {
|
||||
t.Error("no entry.id")
|
||||
}
|
||||
if got.Entry.Created.IsZero() {
|
||||
t.Error("no entry.created")
|
||||
}
|
||||
if got.Entry.Updated.IsZero() {
|
||||
t.Error("no entry.updated")
|
||||
}
|
||||
if !got.Entry.Deleted.IsZero() {
|
||||
t.Error("entry.deleted")
|
||||
}
|
||||
|
||||
if got.Version.Created.IsZero() {
|
||||
t.Error("no version.created")
|
||||
}
|
||||
if got.Version.URL != "url" {
|
||||
t.Error("no version.url")
|
||||
}
|
||||
if got.Version.Cron != "cron" {
|
||||
t.Error("no version.cron")
|
||||
}
|
||||
|
||||
if !got.Execution.Executed.IsZero() {
|
||||
t.Error("execution.executed")
|
||||
}
|
||||
if !got.Execution.Version.IsZero() {
|
||||
t.Error("execution.version")
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user