feature: add sqlite kv db
This commit is contained in:
parent
3671bf1f80
commit
759aae3b54
9 changed files with 152 additions and 2 deletions
3
.gitignore
vendored
3
.gitignore
vendored
|
@ -1,3 +1,2 @@
|
||||||
.idea
|
.idea
|
||||||
badger
|
/db
|
||||||
db
|
|
||||||
|
|
6
embed.go
Normal file
6
embed.go
Normal file
|
@ -0,0 +1,6 @@
|
||||||
|
package discordBots
|
||||||
|
|
||||||
|
import _ "embed"
|
||||||
|
|
||||||
|
//go:embed schema.sql
|
||||||
|
var Schema string
|
1
go.mod
1
go.mod
|
@ -8,6 +8,7 @@ require (
|
||||||
github.com/disgoorg/log v1.2.1
|
github.com/disgoorg/log v1.2.1
|
||||||
github.com/disgoorg/snowflake/v2 v2.0.1
|
github.com/disgoorg/snowflake/v2 v2.0.1
|
||||||
golang.org/x/net v0.27.0
|
golang.org/x/net v0.27.0
|
||||||
|
modernc.org/sqlite v1.30.1
|
||||||
)
|
)
|
||||||
|
|
||||||
require (
|
require (
|
||||||
|
|
31
pkg/db/db.go
Normal file
31
pkg/db/db.go
Normal file
|
@ -0,0 +1,31 @@
|
||||||
|
// Code generated by sqlc. DO NOT EDIT.
|
||||||
|
// versions:
|
||||||
|
// sqlc v1.25.0
|
||||||
|
|
||||||
|
package db
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"database/sql"
|
||||||
|
)
|
||||||
|
|
||||||
|
type DBTX interface {
|
||||||
|
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 {
|
||||||
|
return &Queries{db: db}
|
||||||
|
}
|
||||||
|
|
||||||
|
type Queries struct {
|
||||||
|
db DBTX
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) WithTx(tx *sql.Tx) *Queries {
|
||||||
|
return &Queries{
|
||||||
|
db: tx,
|
||||||
|
}
|
||||||
|
}
|
16
pkg/db/models.go
Normal file
16
pkg/db/models.go
Normal file
|
@ -0,0 +1,16 @@
|
||||||
|
// Code generated by sqlc. DO NOT EDIT.
|
||||||
|
// versions:
|
||||||
|
// sqlc v1.25.0
|
||||||
|
|
||||||
|
package db
|
||||||
|
|
||||||
|
import (
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Store struct {
|
||||||
|
ID string
|
||||||
|
Data []byte
|
||||||
|
CreatedAt time.Time
|
||||||
|
UpdatedAt time.Time
|
||||||
|
}
|
58
pkg/db/query.sql.go
Normal file
58
pkg/db/query.sql.go
Normal file
|
@ -0,0 +1,58 @@
|
||||||
|
// Code generated by sqlc. DO NOT EDIT.
|
||||||
|
// versions:
|
||||||
|
// sqlc v1.25.0
|
||||||
|
// source: query.sql
|
||||||
|
|
||||||
|
package db
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
)
|
||||||
|
|
||||||
|
const countItems = `-- name: CountItems :one
|
||||||
|
SELECT count(*)
|
||||||
|
FROM store
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) CountItems(ctx context.Context) (int64, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, countItems)
|
||||||
|
var count int64
|
||||||
|
err := row.Scan(&count)
|
||||||
|
return count, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const createItem = `-- name: CreateItem :exec
|
||||||
|
INSERT INTO store (id, data)
|
||||||
|
VALUES (?1, ?2)
|
||||||
|
ON CONFLICT(id)
|
||||||
|
DO UPDATE SET data = ?2
|
||||||
|
`
|
||||||
|
|
||||||
|
type CreateItemParams struct {
|
||||||
|
ID string
|
||||||
|
Data []byte
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) CreateItem(ctx context.Context, arg CreateItemParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, createItem, arg.ID, arg.Data)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getItem = `-- name: GetItem :one
|
||||||
|
SELECT id, data, created_at, updated_at
|
||||||
|
FROM store
|
||||||
|
WHERE id = ?
|
||||||
|
LIMIT 1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) GetItem(ctx context.Context, id string) (Store, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getItem, id)
|
||||||
|
var i Store
|
||||||
|
err := row.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.Data,
|
||||||
|
&i.CreatedAt,
|
||||||
|
&i.UpdatedAt,
|
||||||
|
)
|
||||||
|
return i, err
|
||||||
|
}
|
15
query.sql
Normal file
15
query.sql
Normal file
|
@ -0,0 +1,15 @@
|
||||||
|
-- name: GetItem :one
|
||||||
|
SELECT *
|
||||||
|
FROM store
|
||||||
|
WHERE id = ?
|
||||||
|
LIMIT 1;
|
||||||
|
|
||||||
|
-- name: CountItems :one
|
||||||
|
SELECT count(*)
|
||||||
|
FROM store;
|
||||||
|
|
||||||
|
-- name: CreateItem :exec
|
||||||
|
INSERT INTO store (id, data)
|
||||||
|
VALUES (?1, ?2)
|
||||||
|
ON CONFLICT(id)
|
||||||
|
DO UPDATE SET data = ?2;
|
15
schema.sql
Normal file
15
schema.sql
Normal file
|
@ -0,0 +1,15 @@
|
||||||
|
CREATE TABLE IF NOT EXISTS store
|
||||||
|
(
|
||||||
|
id TEXT NOT NULL PRIMARY KEY,
|
||||||
|
data BLOB NOT NULL,
|
||||||
|
created_at DATE NOT NULL DEFAULT current_timestamp,
|
||||||
|
updated_at DATE NOT NULL DEFAULT current_timestamp
|
||||||
|
);
|
||||||
|
|
||||||
|
-- automatic updated at
|
||||||
|
CREATE TRIGGER IF NOT EXISTS update_store_updated_at
|
||||||
|
AFTER UPDATE ON store
|
||||||
|
WHEN OLD.updated_at <> current_timestamp
|
||||||
|
BEGIN
|
||||||
|
UPDATE store SET updated_at = CURRENT_TIMESTAMP WHERE id = OLD.id;
|
||||||
|
END;
|
9
sqlc.yaml
Normal file
9
sqlc.yaml
Normal file
|
@ -0,0 +1,9 @@
|
||||||
|
version: "2"
|
||||||
|
sql:
|
||||||
|
- engine: "sqlite"
|
||||||
|
queries: "query.sql"
|
||||||
|
schema: "schema.sql"
|
||||||
|
gen:
|
||||||
|
go:
|
||||||
|
package: "db"
|
||||||
|
out: "pkg/db"
|
Loading…
Reference in a new issue