updating methods #1

Merged
jake merged 1 commits from fix/connect-methodology into main 2025-07-15 23:01:10 +00:00
Showing only changes of commit efc047c965 - Show all commits

View File

@ -3,7 +3,6 @@ package lite
import ( import (
"context" "context"
"database/sql" "database/sql"
"errors"
"fmt" "fmt"
"os" "os"
@ -12,10 +11,6 @@ import (
_ "github.com/ncruces/go-sqlite3/embed" _ "github.com/ncruces/go-sqlite3/embed"
) )
var (
ErrNotConnected = errors.New("the connect method must be called before any queries can be executed")
)
type database struct { type database struct {
file string file string
path string path string
@ -23,11 +18,10 @@ type database struct {
} }
type Database interface { type Database interface {
Connect() error
Close() error Close() error
Exec(ctx context.Context, query string, args ...any) (sql.Result, error) Exec(ctx context.Context, query string, args ...any) (sql.Result, error)
Query(ctx context.Context, query string, args ...any) (*sqlx.Rows, error) Query(ctx context.Context, query string, args ...any) (*sqlx.Rows, error)
QueryRow(ctx context.Context, query string, args ...any) (*sqlx.Row, error) QueryRow(ctx context.Context, query string, args ...any) *sqlx.Row
QueryIn(ctx context.Context, query string, args ...any) (*sqlx.Rows, error) QueryIn(ctx context.Context, query string, args ...any) (*sqlx.Rows, error)
} }
@ -48,19 +42,13 @@ func New(opts ...Option) (Database, error) {
opt(db) opt(db)
} }
return db, nil slite, err := sqlx.Open("sqlite3", fmt.Sprintf("file:%s/%s", db.path, db.file))
}
// connects to database file
func (d *database) Connect() error {
db, err := sqlx.Open("sqlite3", fmt.Sprintf("file:%s/%s", d.path, d.file))
if err != nil { if err != nil {
return err return nil, err
} }
db.sqlite = slite
d.sqlite = db return db, nil
return nil
} }
// closes database file connection // closes database file connection
@ -70,33 +58,21 @@ func (d *database) Close() error {
// runs any updates, inserts, or any other queries that do not return rows // runs any updates, inserts, or any other queries that do not return rows
func (d *database) Exec(ctx context.Context, query string, args ...any) (sql.Result, error) { func (d *database) Exec(ctx context.Context, query string, args ...any) (sql.Result, error) {
if d.sqlite == nil {
return nil, ErrNotConnected
}
return d.sqlite.ExecContext(ctx, query, args...) return d.sqlite.ExecContext(ctx, query, args...)
} }
// runs basic data queries // runs basic data queries
func (d *database) Query(ctx context.Context, query string, args ...any) (*sqlx.Rows, error) { func (d *database) Query(ctx context.Context, query string, args ...any) (*sqlx.Rows, error) {
if d.sqlite == nil {
return nil, ErrNotConnected
}
return d.sqlite.QueryxContext(ctx, query, args...) return d.sqlite.QueryxContext(ctx, query, args...)
} }
// retrieves a single row from database // retrieves a single row from database
func (d *database) QueryRow(ctx context.Context, query string, args ...any) (*sqlx.Row, error) { func (d *database) QueryRow(ctx context.Context, query string, args ...any) *sqlx.Row {
if d.sqlite == nil { return d.sqlite.QueryRowxContext(ctx, query, args...)
return nil, ErrNotConnected
}
return d.sqlite.QueryRowxContext(ctx, query, args...), nil
} }
// runs queries that require parameters in an IN clause // runs queries that require parameters in an IN clause
func (d *database) QueryIn(ctx context.Context, query string, args ...any) (*sqlx.Rows, error) { func (d *database) QueryIn(ctx context.Context, query string, args ...any) (*sqlx.Rows, error) {
if d.sqlite == nil {
return nil, ErrNotConnected
}
q, ar, err := sqlx.In(query, args...) q, ar, err := sqlx.In(query, args...)
if err != nil { if err != nil {
return nil, err return nil, err