Compare commits
2 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| c23adb0150 | |||
| 4f69a3ad45 |
35
cmd/main.go
35
cmd/main.go
@@ -1,17 +1,14 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
|
||||||
"database/sql"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/google/uuid"
|
"github.com/google/uuid"
|
||||||
httpSwagger "github.com/swaggo/http-swagger"
|
httpSwagger "github.com/swaggo/http-swagger"
|
||||||
_ "go-nkode/docs"
|
_ "go-nkode/docs"
|
||||||
"go-nkode/internal/api"
|
"go-nkode/internal/api"
|
||||||
|
"go-nkode/internal/db"
|
||||||
"go-nkode/internal/email"
|
"go-nkode/internal/email"
|
||||||
"go-nkode/internal/models"
|
"go-nkode/internal/models"
|
||||||
"go-nkode/internal/repository"
|
|
||||||
sqliteQueue "go-nkode/internal/sqlc"
|
|
||||||
"log"
|
"log"
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
@@ -40,42 +37,24 @@ const (
|
|||||||
// @securityDefinitions.apiKey ApiKeyAuth
|
// @securityDefinitions.apiKey ApiKeyAuth
|
||||||
// @in header
|
// @in header
|
||||||
// @name Authorization
|
// @name Authorization
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
dbPath := os.Getenv("SQLITE_DB")
|
dbPath := os.Getenv("SQLITE_DB")
|
||||||
if dbPath == "" {
|
if dbPath == "" {
|
||||||
log.Fatal("SQLITE_DB=/path/to/nkode.db not set")
|
log.Fatalf("SQLITE_DB=/path/to/nkode.db not set")
|
||||||
}
|
}
|
||||||
|
sqlitedb, err := db.NewSqliteDB(dbPath)
|
||||||
sqliteDb, err := sql.Open("sqlite3", dbPath)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatalf("failed to open database: %v", err)
|
log.Fatalf("%v", err)
|
||||||
}
|
}
|
||||||
|
defer sqlitedb.Close()
|
||||||
if err := sqliteDb.Ping(); err != nil {
|
|
||||||
log.Fatalf("failed to connect to database: %v", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
ctx := context.Background()
|
|
||||||
queue, err := sqliteQueue.NewQueue(sqliteDb, ctx)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
queue.Start()
|
|
||||||
|
|
||||||
defer func(queue *sqliteQueue.Queue) {
|
|
||||||
if err := queue.Stop(); err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
}(queue)
|
|
||||||
|
|
||||||
sesClient := email.NewSESClient()
|
sesClient := email.NewSESClient()
|
||||||
emailQueue := email.NewEmailQueue(emailQueueBufferSize, maxEmailsPerSecond, &sesClient)
|
emailQueue := email.NewEmailQueue(emailQueueBufferSize, maxEmailsPerSecond, &sesClient)
|
||||||
emailQueue.Start()
|
emailQueue.Start()
|
||||||
defer emailQueue.Stop()
|
defer emailQueue.Stop()
|
||||||
|
|
||||||
sqlitedb := repository.NewSqliteRepository(queue, ctx)
|
nkodeApi := api.NewNKodeAPI(sqlitedb, emailQueue)
|
||||||
nkodeApi := api.NewNKodeAPI(&sqlitedb, emailQueue)
|
|
||||||
|
|
||||||
AddDefaultCustomer(nkodeApi)
|
AddDefaultCustomer(nkodeApi)
|
||||||
handler := api.NKodeHandler{Api: nkodeApi}
|
handler := api.NKodeHandler{Api: nkodeApi}
|
||||||
|
|
||||||
|
|||||||
7
go.mod
7
go.mod
@@ -1,6 +1,8 @@
|
|||||||
module go-nkode
|
module go-nkode
|
||||||
|
|
||||||
go 1.23.0
|
go 1.22.0
|
||||||
|
|
||||||
|
toolchain go1.23.0
|
||||||
|
|
||||||
require (
|
require (
|
||||||
github.com/aws/aws-sdk-go-v2 v1.31.0
|
github.com/aws/aws-sdk-go-v2 v1.31.0
|
||||||
@@ -10,7 +12,7 @@ require (
|
|||||||
github.com/google/uuid v1.6.0
|
github.com/google/uuid v1.6.0
|
||||||
github.com/mattn/go-sqlite3 v1.14.22
|
github.com/mattn/go-sqlite3 v1.14.22
|
||||||
github.com/patrickmn/go-cache v2.1.0+incompatible
|
github.com/patrickmn/go-cache v2.1.0+incompatible
|
||||||
github.com/stretchr/testify v1.10.0
|
github.com/stretchr/testify v1.9.0
|
||||||
github.com/swaggo/http-swagger v1.3.4
|
github.com/swaggo/http-swagger v1.3.4
|
||||||
github.com/swaggo/swag v1.16.4
|
github.com/swaggo/swag v1.16.4
|
||||||
github.com/swaggo/swag/example/celler v0.0.0-20241025062444-99698582709d
|
github.com/swaggo/swag/example/celler v0.0.0-20241025062444-99698582709d
|
||||||
@@ -18,7 +20,6 @@ require (
|
|||||||
)
|
)
|
||||||
|
|
||||||
require (
|
require (
|
||||||
github.com/DonovanKelly/sugar-n-spice v1.0.1 // indirect
|
|
||||||
github.com/KyleBanks/depth v1.2.1 // indirect
|
github.com/KyleBanks/depth v1.2.1 // indirect
|
||||||
github.com/aws/aws-sdk-go-v2/credentials v1.17.35 // indirect
|
github.com/aws/aws-sdk-go-v2/credentials v1.17.35 // indirect
|
||||||
github.com/aws/aws-sdk-go-v2/feature/ec2/imds v1.16.14 // indirect
|
github.com/aws/aws-sdk-go-v2/feature/ec2/imds v1.16.14 // indirect
|
||||||
|
|||||||
4
go.sum
4
go.sum
@@ -1,5 +1,3 @@
|
|||||||
github.com/DonovanKelly/sugar-n-spice v1.0.1 h1:VsybiCHSziAqyPtbYF6GtkiJYYECWMHKN+EyEa6UVpA=
|
|
||||||
github.com/DonovanKelly/sugar-n-spice v1.0.1/go.mod h1:/HQWoablLFCwsa4gwfzVBu80cI5A3dyO1uCiB11sup0=
|
|
||||||
github.com/KyleBanks/depth v1.2.1 h1:5h8fQADFrWtarTdtDudMmGsC7GPbOAu6RVB3ffsVFHc=
|
github.com/KyleBanks/depth v1.2.1 h1:5h8fQADFrWtarTdtDudMmGsC7GPbOAu6RVB3ffsVFHc=
|
||||||
github.com/KyleBanks/depth v1.2.1/go.mod h1:jzSb9d0L43HxTQfT+oSA1EEp2q+ne2uh6XgeJcm8brE=
|
github.com/KyleBanks/depth v1.2.1/go.mod h1:jzSb9d0L43HxTQfT+oSA1EEp2q+ne2uh6XgeJcm8brE=
|
||||||
github.com/aws/aws-sdk-go-v2 v1.31.0 h1:3V05LbxTSItI5kUqNwhJrrrY1BAXxXt0sN0l72QmG5U=
|
github.com/aws/aws-sdk-go-v2 v1.31.0 h1:3V05LbxTSItI5kUqNwhJrrrY1BAXxXt0sN0l72QmG5U=
|
||||||
@@ -118,8 +116,6 @@ github.com/stretchr/testify v1.8.2/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o
|
|||||||
github.com/stretchr/testify v1.8.3/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
|
github.com/stretchr/testify v1.8.3/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
|
||||||
github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
|
github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
|
||||||
github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
|
github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
|
||||||
github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA=
|
|
||||||
github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
|
|
||||||
github.com/swaggo/files v1.0.1 h1:J1bVJ4XHZNq0I46UU90611i9/YzdrF7x92oX1ig5IdE=
|
github.com/swaggo/files v1.0.1 h1:J1bVJ4XHZNq0I46UU90611i9/YzdrF7x92oX1ig5IdE=
|
||||||
github.com/swaggo/files v1.0.1/go.mod h1:0qXmMNH6sXNf+73t65aKeB+ApmgxdnkQzVTAj2uaMUg=
|
github.com/swaggo/files v1.0.1/go.mod h1:0qXmMNH6sXNf+73t65aKeB+ApmgxdnkQzVTAj2uaMUg=
|
||||||
github.com/swaggo/http-swagger v1.3.4 h1:q7t/XLx0n15H1Q9/tk3Y9L4n210XzJF5WtnDX64a5ww=
|
github.com/swaggo/http-swagger v1.3.4 h1:q7t/XLx0n15H1Q9/tk3Y9L4n210XzJF5WtnDX64a5ww=
|
||||||
|
|||||||
29
instructions/task_120924.md
Normal file
29
instructions/task_120924.md
Normal file
@@ -0,0 +1,29 @@
|
|||||||
|
# Task: Get nKode running on you system and document the process for setting up in the Readme.md
|
||||||
|
|
||||||
|
Prereq:
|
||||||
|
1. sign into gitea
|
||||||
|
2. clone go-nkode
|
||||||
|
3. install docker
|
||||||
|
4. install sqlite3
|
||||||
|
|
||||||
|
Do this stuff later
|
||||||
|
- install flutter
|
||||||
|
- clone flutter-nkode
|
||||||
|
|
||||||
|
## Install Swaggo
|
||||||
|
|
||||||
|
Install [swaggo](https://github.com/swaggo/swag)
|
||||||
|
|
||||||
|
you'll know you're successful when you see this in the cli:
|
||||||
|
```
|
||||||
|
➜ go-nkode git:(main) ✗ swag --version
|
||||||
|
swag version v1.16.4
|
||||||
|
```
|
||||||
|
|
||||||
|
## Setup Sqlite DB
|
||||||
|
Follow this [video](https://1drv.ms/v/s!Agm1aJyS_E11lPVCt_uoJFEn9HPFrA?e=mvntuY)
|
||||||
|
Download flaticon zip [here](https://1drv.ms/u/s!Agm1aJyS_E11lPVAI9Af279oZGnC2w?e=WbY1Xd)
|
||||||
|
|
||||||
|
Add this to your environment variables in your Goland config for sqlite_init.go
|
||||||
|
`DB_PATH=/path/to/databases/nkode.db;SVG_DIR=/path/to/flaticon_colored_svgs;TEST_DB_PATH=/path/to/databases/test.db`
|
||||||
|
|
||||||
@@ -5,10 +5,10 @@ import (
|
|||||||
"github.com/google/uuid"
|
"github.com/google/uuid"
|
||||||
"github.com/patrickmn/go-cache"
|
"github.com/patrickmn/go-cache"
|
||||||
"go-nkode/config"
|
"go-nkode/config"
|
||||||
|
"go-nkode/internal/db"
|
||||||
"go-nkode/internal/email"
|
"go-nkode/internal/email"
|
||||||
"go-nkode/internal/entities"
|
"go-nkode/internal/entities"
|
||||||
"go-nkode/internal/models"
|
"go-nkode/internal/models"
|
||||||
"go-nkode/internal/repository"
|
|
||||||
"go-nkode/internal/security"
|
"go-nkode/internal/security"
|
||||||
"log"
|
"log"
|
||||||
"os"
|
"os"
|
||||||
@@ -21,12 +21,12 @@ const (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type NKodeAPI struct {
|
type NKodeAPI struct {
|
||||||
Db repository.CustomerUserRepository
|
Db db.CustomerUserRepository
|
||||||
SignupSessionCache *cache.Cache
|
SignupSessionCache *cache.Cache
|
||||||
EmailQueue *email.Queue
|
EmailQueue *email.Queue
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewNKodeAPI(db repository.CustomerUserRepository, queue *email.Queue) NKodeAPI {
|
func NewNKodeAPI(db db.CustomerUserRepository, queue *email.Queue) NKodeAPI {
|
||||||
return NKodeAPI{
|
return NKodeAPI{
|
||||||
Db: db,
|
Db: db,
|
||||||
EmailQueue: queue,
|
EmailQueue: queue,
|
||||||
|
|||||||
@@ -1,15 +1,12 @@
|
|||||||
package api
|
package api
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
|
"go-nkode/internal/db"
|
||||||
"go-nkode/internal/email"
|
"go-nkode/internal/email"
|
||||||
"go-nkode/internal/entities"
|
"go-nkode/internal/entities"
|
||||||
"go-nkode/internal/models"
|
"go-nkode/internal/models"
|
||||||
"go-nkode/internal/repository"
|
|
||||||
"go-nkode/internal/security"
|
"go-nkode/internal/security"
|
||||||
sqlite_queue "go-nkode/internal/sqlc"
|
|
||||||
"log"
|
|
||||||
"os"
|
"os"
|
||||||
"testing"
|
"testing"
|
||||||
)
|
)
|
||||||
@@ -18,31 +15,22 @@ func TestNKodeAPI(t *testing.T) {
|
|||||||
//db1 := NewInMemoryDb()
|
//db1 := NewInMemoryDb()
|
||||||
//testNKodeAPI(t, &db1)
|
//testNKodeAPI(t, &db1)
|
||||||
|
|
||||||
dbPath := os.Getenv("TEST_DB")
|
dbFile := os.Getenv("TEST_DB")
|
||||||
ctx := context.Background()
|
|
||||||
sqliteDb, err := sqlite_queue.OpenSqliteDb(dbPath)
|
|
||||||
assert.NoError(t, err)
|
|
||||||
|
|
||||||
queue, err := sqlite_queue.NewQueue(sqliteDb, ctx)
|
db2, err := db.NewSqliteDB(dbFile)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
queue.Start()
|
defer db2.Close()
|
||||||
defer func(queue *sqlite_queue.Queue) {
|
testNKodeAPI(t, db2)
|
||||||
if err := queue.Stop(); err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
}(queue)
|
|
||||||
sqlitedb := repository.NewSqliteRepository(queue, ctx)
|
|
||||||
testNKodeAPI(t, &sqlitedb)
|
|
||||||
|
|
||||||
//if _, err := os.Stat(dbPath); err == nil {
|
//if _, err := os.Stat(dbFile); err == nil {
|
||||||
// err = os.Remove(dbPath)
|
// err = os.Remove(dbFile)
|
||||||
// assert.NoError(t, err)
|
// assert.NoError(t, err)
|
||||||
//} else {
|
//} else {
|
||||||
// assert.NoError(t, err)
|
// assert.NoError(t, err)
|
||||||
//}
|
//}
|
||||||
}
|
}
|
||||||
|
|
||||||
func testNKodeAPI(t *testing.T, db repository.CustomerUserRepository) {
|
func testNKodeAPI(t *testing.T, db db.CustomerUserRepository) {
|
||||||
bufferSize := 100
|
bufferSize := 100
|
||||||
emailsPerSec := 14
|
emailsPerSec := 14
|
||||||
testClient := email.TestEmailClient{}
|
testClient := email.TestEmailClient{}
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
package repository
|
package db
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"go-nkode/internal/entities"
|
"go-nkode/internal/entities"
|
||||||
@@ -1,4 +1,4 @@
|
|||||||
package repository
|
package db
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
@@ -1,4 +1,4 @@
|
|||||||
package repository
|
package db
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
@@ -14,21 +14,82 @@ import (
|
|||||||
"go-nkode/internal/sqlc"
|
"go-nkode/internal/sqlc"
|
||||||
"go-nkode/internal/utils"
|
"go-nkode/internal/utils"
|
||||||
"log"
|
"log"
|
||||||
|
"sync"
|
||||||
)
|
)
|
||||||
|
|
||||||
type SqliteRepository struct {
|
const writeBufferSize = 100
|
||||||
Queue *sqlc.Queue
|
|
||||||
ctx context.Context
|
type sqlcGeneric func(*sqlc.Queries, context.Context, any) error
|
||||||
|
|
||||||
|
// WriteTx represents a write transaction
|
||||||
|
type WriteTx struct {
|
||||||
|
ErrChan chan error
|
||||||
|
Query sqlcGeneric
|
||||||
|
Args interface{}
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewSqliteRepository(queue *sqlc.Queue, ctx context.Context) SqliteRepository {
|
// SqliteDB represents the SQLite database connection and write queue
|
||||||
return SqliteRepository{
|
type SqliteDB struct {
|
||||||
Queue: queue,
|
queries *sqlc.Queries
|
||||||
|
db *sql.DB
|
||||||
|
writeQueue chan WriteTx
|
||||||
|
wg sync.WaitGroup
|
||||||
|
ctx context.Context
|
||||||
|
cancel context.CancelFunc
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewSqliteDB initializes a new SqliteDB instance
|
||||||
|
func NewSqliteDB(path string) (*SqliteDB, error) {
|
||||||
|
if path == "" {
|
||||||
|
return nil, errors.New("database path is required")
|
||||||
|
}
|
||||||
|
|
||||||
|
db, err := sql.Open("sqlite3", path)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to open database: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := db.Ping(); err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to connect to database: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx, cancel := context.WithCancel(context.Background())
|
||||||
|
sqldb := &SqliteDB{
|
||||||
|
queries: sqlc.New(db),
|
||||||
|
db: db,
|
||||||
|
writeQueue: make(chan WriteTx, writeBufferSize),
|
||||||
ctx: ctx,
|
ctx: ctx,
|
||||||
|
cancel: cancel,
|
||||||
|
}
|
||||||
|
|
||||||
|
sqldb.wg.Add(1)
|
||||||
|
go sqldb.processWriteQueue()
|
||||||
|
|
||||||
|
return sqldb, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// processWriteQueue handles write transactions from the queue
|
||||||
|
func (d *SqliteDB) processWriteQueue() {
|
||||||
|
defer d.wg.Done()
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-d.ctx.Done():
|
||||||
|
return
|
||||||
|
case writeTx := <-d.writeQueue:
|
||||||
|
err := writeTx.Query(d.queries, d.ctx, writeTx.Args)
|
||||||
|
writeTx.ErrChan <- err
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *SqliteRepository) CreateCustomer(c entities.Customer) error {
|
func (d *SqliteDB) Close() error {
|
||||||
|
d.cancel()
|
||||||
|
d.wg.Wait()
|
||||||
|
close(d.writeQueue)
|
||||||
|
return d.db.Close()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *SqliteDB) CreateCustomer(c entities.Customer) error {
|
||||||
queryFunc := func(q *sqlc.Queries, ctx context.Context, args any) error {
|
queryFunc := func(q *sqlc.Queries, ctx context.Context, args any) error {
|
||||||
params, ok := args.(sqlc.CreateCustomerParams)
|
params, ok := args.(sqlc.CreateCustomerParams)
|
||||||
if !ok {
|
if !ok {
|
||||||
@@ -37,10 +98,10 @@ func (d *SqliteRepository) CreateCustomer(c entities.Customer) error {
|
|||||||
return q.CreateCustomer(ctx, params)
|
return q.CreateCustomer(ctx, params)
|
||||||
}
|
}
|
||||||
|
|
||||||
return d.Queue.EnqueueWriteTx(queryFunc, c.ToSqlcCreateCustomerParams())
|
return d.enqueueWriteTx(queryFunc, c.ToSqlcCreateCustomerParams())
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *SqliteRepository) WriteNewUser(u entities.User) error {
|
func (d *SqliteDB) WriteNewUser(u entities.User) error {
|
||||||
queryFunc := func(q *sqlc.Queries, ctx context.Context, args any) error {
|
queryFunc := func(q *sqlc.Queries, ctx context.Context, args any) error {
|
||||||
params, ok := args.(sqlc.CreateUserParams)
|
params, ok := args.(sqlc.CreateUserParams)
|
||||||
if !ok {
|
if !ok {
|
||||||
@@ -48,7 +109,7 @@ func (d *SqliteRepository) WriteNewUser(u entities.User) error {
|
|||||||
}
|
}
|
||||||
return q.CreateUser(ctx, params)
|
return q.CreateUser(ctx, params)
|
||||||
}
|
}
|
||||||
// Use the wrapped function in EnqueueWriteTx
|
// Use the wrapped function in enqueueWriteTx
|
||||||
|
|
||||||
renew := 0
|
renew := 0
|
||||||
if u.Renew {
|
if u.Renew {
|
||||||
@@ -75,10 +136,10 @@ func (d *SqliteRepository) WriteNewUser(u entities.User) error {
|
|||||||
SvgIDInterface: security.IntArrToByteArr(u.Interface.SvgId),
|
SvgIDInterface: security.IntArrToByteArr(u.Interface.SvgId),
|
||||||
CreatedAt: sql.NullString{String: utils.TimeStamp(), Valid: true},
|
CreatedAt: sql.NullString{String: utils.TimeStamp(), Valid: true},
|
||||||
}
|
}
|
||||||
return d.Queue.EnqueueWriteTx(queryFunc, params)
|
return d.enqueueWriteTx(queryFunc, params)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *SqliteRepository) UpdateUserNKode(u entities.User) error {
|
func (d *SqliteDB) UpdateUserNKode(u entities.User) error {
|
||||||
queryFunc := func(q *sqlc.Queries, ctx context.Context, args any) error {
|
queryFunc := func(q *sqlc.Queries, ctx context.Context, args any) error {
|
||||||
params, ok := args.(sqlc.UpdateUserParams)
|
params, ok := args.(sqlc.UpdateUserParams)
|
||||||
if !ok {
|
if !ok {
|
||||||
@@ -86,7 +147,7 @@ func (d *SqliteRepository) UpdateUserNKode(u entities.User) error {
|
|||||||
}
|
}
|
||||||
return q.UpdateUser(ctx, params)
|
return q.UpdateUser(ctx, params)
|
||||||
}
|
}
|
||||||
// Use the wrapped function in EnqueueWriteTx
|
// Use the wrapped function in enqueueWriteTx
|
||||||
renew := 0
|
renew := 0
|
||||||
if u.Renew {
|
if u.Renew {
|
||||||
renew = 1
|
renew = 1
|
||||||
@@ -109,10 +170,10 @@ func (d *SqliteRepository) UpdateUserNKode(u entities.User) error {
|
|||||||
IdxInterface: security.IntArrToByteArr(u.Interface.IdxInterface),
|
IdxInterface: security.IntArrToByteArr(u.Interface.IdxInterface),
|
||||||
SvgIDInterface: security.IntArrToByteArr(u.Interface.SvgId),
|
SvgIDInterface: security.IntArrToByteArr(u.Interface.SvgId),
|
||||||
}
|
}
|
||||||
return d.Queue.EnqueueWriteTx(queryFunc, params)
|
return d.enqueueWriteTx(queryFunc, params)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *SqliteRepository) UpdateUserInterface(id models.UserId, ui entities.UserInterface) error {
|
func (d *SqliteDB) UpdateUserInterface(id models.UserId, ui entities.UserInterface) error {
|
||||||
queryFunc := func(q *sqlc.Queries, ctx context.Context, args any) error {
|
queryFunc := func(q *sqlc.Queries, ctx context.Context, args any) error {
|
||||||
params, ok := args.(sqlc.UpdateUserInterfaceParams)
|
params, ok := args.(sqlc.UpdateUserInterfaceParams)
|
||||||
if !ok {
|
if !ok {
|
||||||
@@ -126,10 +187,10 @@ func (d *SqliteRepository) UpdateUserInterface(id models.UserId, ui entities.Use
|
|||||||
ID: uuid.UUID(id).String(),
|
ID: uuid.UUID(id).String(),
|
||||||
}
|
}
|
||||||
|
|
||||||
return d.Queue.EnqueueWriteTx(queryFunc, params)
|
return d.enqueueWriteTx(queryFunc, params)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *SqliteRepository) UpdateUserRefreshToken(id models.UserId, refreshToken string) error {
|
func (d *SqliteDB) UpdateUserRefreshToken(id models.UserId, refreshToken string) error {
|
||||||
queryFunc := func(q *sqlc.Queries, ctx context.Context, args any) error {
|
queryFunc := func(q *sqlc.Queries, ctx context.Context, args any) error {
|
||||||
params, ok := args.(sqlc.UpdateUserRefreshTokenParams)
|
params, ok := args.(sqlc.UpdateUserRefreshTokenParams)
|
||||||
if !ok {
|
if !ok {
|
||||||
@@ -144,10 +205,10 @@ func (d *SqliteRepository) UpdateUserRefreshToken(id models.UserId, refreshToken
|
|||||||
},
|
},
|
||||||
ID: uuid.UUID(id).String(),
|
ID: uuid.UUID(id).String(),
|
||||||
}
|
}
|
||||||
return d.Queue.EnqueueWriteTx(queryFunc, params)
|
return d.enqueueWriteTx(queryFunc, params)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *SqliteRepository) RenewCustomer(renewParams sqlc.RenewCustomerParams) error {
|
func (d *SqliteDB) RenewCustomer(renewParams sqlc.RenewCustomerParams) error {
|
||||||
queryFunc := func(q *sqlc.Queries, ctx context.Context, args any) error {
|
queryFunc := func(q *sqlc.Queries, ctx context.Context, args any) error {
|
||||||
params, ok := args.(sqlc.RenewCustomerParams)
|
params, ok := args.(sqlc.RenewCustomerParams)
|
||||||
if !ok {
|
if !ok {
|
||||||
@@ -155,16 +216,16 @@ func (d *SqliteRepository) RenewCustomer(renewParams sqlc.RenewCustomerParams) e
|
|||||||
}
|
}
|
||||||
return q.RenewCustomer(ctx, params)
|
return q.RenewCustomer(ctx, params)
|
||||||
}
|
}
|
||||||
return d.Queue.EnqueueWriteTx(queryFunc, renewParams)
|
return d.enqueueWriteTx(queryFunc, renewParams)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *SqliteRepository) Renew(id models.CustomerId) error {
|
func (d *SqliteDB) Renew(id models.CustomerId) error {
|
||||||
setXor, attrXor, err := d.renewCustomer(id)
|
setXor, attrXor, err := d.renewCustomer(id)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
customerId := models.CustomerIdToString(id)
|
customerId := models.CustomerIdToString(id)
|
||||||
userRenewRows, err := d.Queue.Queries.GetUserRenew(d.ctx, customerId)
|
userRenewRows, err := d.queries.GetUserRenew(d.ctx, customerId)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
@@ -204,14 +265,14 @@ func (d *SqliteRepository) Renew(id models.CustomerId) error {
|
|||||||
Renew: 1,
|
Renew: 1,
|
||||||
ID: uuid.UUID(user.Id).String(),
|
ID: uuid.UUID(user.Id).String(),
|
||||||
}
|
}
|
||||||
if err = d.Queue.EnqueueWriteTx(queryFunc, params); err != nil {
|
if err = d.enqueueWriteTx(queryFunc, params); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *SqliteRepository) renewCustomer(id models.CustomerId) ([]uint64, []uint64, error) {
|
func (d *SqliteDB) renewCustomer(id models.CustomerId) ([]uint64, []uint64, error) {
|
||||||
customer, err := d.GetCustomer(id)
|
customer, err := d.GetCustomer(id)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, nil, err
|
return nil, nil, err
|
||||||
@@ -234,13 +295,13 @@ func (d *SqliteRepository) renewCustomer(id models.CustomerId) ([]uint64, []uint
|
|||||||
ID: uuid.UUID(customer.Id).String(),
|
ID: uuid.UUID(customer.Id).String(),
|
||||||
}
|
}
|
||||||
|
|
||||||
if err = d.Queue.EnqueueWriteTx(queryFunc, params); err != nil {
|
if err = d.enqueueWriteTx(queryFunc, params); err != nil {
|
||||||
return nil, nil, err
|
return nil, nil, err
|
||||||
}
|
}
|
||||||
return setXor, attrXor, nil
|
return setXor, attrXor, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *SqliteRepository) RefreshUserPasscode(user entities.User, passcodeIdx []int, customerAttr entities.CustomerAttributes) error {
|
func (d *SqliteDB) RefreshUserPasscode(user entities.User, passcodeIdx []int, customerAttr entities.CustomerAttributes) error {
|
||||||
if err := user.RefreshPasscode(passcodeIdx, customerAttr); err != nil {
|
if err := user.RefreshPasscode(passcodeIdx, customerAttr); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
@@ -262,11 +323,11 @@ func (d *SqliteRepository) RefreshUserPasscode(user entities.User, passcodeIdx [
|
|||||||
Salt: user.CipherKeys.Salt,
|
Salt: user.CipherKeys.Salt,
|
||||||
ID: uuid.UUID(user.Id).String(),
|
ID: uuid.UUID(user.Id).String(),
|
||||||
}
|
}
|
||||||
return d.Queue.EnqueueWriteTx(queryFunc, params)
|
return d.enqueueWriteTx(queryFunc, params)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *SqliteRepository) GetCustomer(id models.CustomerId) (*entities.Customer, error) {
|
func (d *SqliteDB) GetCustomer(id models.CustomerId) (*entities.Customer, error) {
|
||||||
customer, err := d.Queue.Queries.GetCustomer(d.ctx, uuid.UUID(id).String())
|
customer, err := d.queries.GetCustomer(d.ctx, uuid.UUID(id).String())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@@ -285,8 +346,8 @@ func (d *SqliteRepository) GetCustomer(id models.CustomerId) (*entities.Customer
|
|||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *SqliteRepository) GetUser(email models.UserEmail, customerId models.CustomerId) (*entities.User, error) {
|
func (d *SqliteDB) GetUser(email models.UserEmail, customerId models.CustomerId) (*entities.User, error) {
|
||||||
userRow, err := d.Queue.Queries.GetUser(d.ctx, sqlc.GetUserParams{
|
userRow, err := d.queries.GetUser(d.ctx, sqlc.GetUserParams{
|
||||||
Email: string(email),
|
Email: string(email),
|
||||||
CustomerID: uuid.UUID(customerId).String(),
|
CustomerID: uuid.UUID(customerId).String(),
|
||||||
})
|
})
|
||||||
@@ -335,7 +396,7 @@ func (d *SqliteRepository) GetUser(email models.UserEmail, customerId models.Cus
|
|||||||
return &user, nil
|
return &user, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *SqliteRepository) RandomSvgInterface(kp entities.KeypadDimension) ([]string, error) {
|
func (d *SqliteDB) RandomSvgInterface(kp entities.KeypadDimension) ([]string, error) {
|
||||||
ids, err := d.getRandomIds(kp.TotalAttrs())
|
ids, err := d.getRandomIds(kp.TotalAttrs())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
@@ -343,18 +404,18 @@ func (d *SqliteRepository) RandomSvgInterface(kp entities.KeypadDimension) ([]st
|
|||||||
return d.getSvgsById(ids)
|
return d.getSvgsById(ids)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *SqliteRepository) RandomSvgIdxInterface(kp entities.KeypadDimension) (models.SvgIdInterface, error) {
|
func (d *SqliteDB) RandomSvgIdxInterface(kp entities.KeypadDimension) (models.SvgIdInterface, error) {
|
||||||
return d.getRandomIds(kp.TotalAttrs())
|
return d.getRandomIds(kp.TotalAttrs())
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *SqliteRepository) GetSvgStringInterface(idxs models.SvgIdInterface) ([]string, error) {
|
func (d *SqliteDB) GetSvgStringInterface(idxs models.SvgIdInterface) ([]string, error) {
|
||||||
return d.getSvgsById(idxs)
|
return d.getSvgsById(idxs)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *SqliteRepository) getSvgsById(ids []int) ([]string, error) {
|
func (d *SqliteDB) getSvgsById(ids []int) ([]string, error) {
|
||||||
svgs := make([]string, len(ids))
|
svgs := make([]string, len(ids))
|
||||||
for idx, id := range ids {
|
for idx, id := range ids {
|
||||||
svg, err := d.Queue.Queries.GetSvgId(d.ctx, int64(id))
|
svg, err := d.queries.GetSvgId(d.ctx, int64(id))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@@ -363,8 +424,25 @@ func (d *SqliteRepository) getSvgsById(ids []int) ([]string, error) {
|
|||||||
return svgs, nil
|
return svgs, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *SqliteRepository) getRandomIds(count int) ([]int, error) {
|
func (d *SqliteDB) enqueueWriteTx(queryFunc sqlcGeneric, args any) error {
|
||||||
tx, err := d.Queue.Db.Begin()
|
select {
|
||||||
|
case <-d.ctx.Done():
|
||||||
|
return errors.New("database is shutting down")
|
||||||
|
default:
|
||||||
|
}
|
||||||
|
|
||||||
|
errChan := make(chan error, 1)
|
||||||
|
writeTx := WriteTx{
|
||||||
|
Query: queryFunc,
|
||||||
|
Args: args,
|
||||||
|
ErrChan: errChan,
|
||||||
|
}
|
||||||
|
d.writeQueue <- writeTx
|
||||||
|
return <-errChan
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *SqliteDB) getRandomIds(count int) ([]int, error) {
|
||||||
|
tx, err := d.db.Begin()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Print(err)
|
log.Print(err)
|
||||||
return nil, config.ErrSqliteTx
|
return nil, config.ErrSqliteTx
|
||||||
@@ -1,31 +1,28 @@
|
|||||||
package repository
|
package db
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
"go-nkode/internal/entities"
|
"go-nkode/internal/entities"
|
||||||
"go-nkode/internal/models"
|
"go-nkode/internal/models"
|
||||||
sqlite_queue "go-nkode/internal/sqlc"
|
|
||||||
"os"
|
"os"
|
||||||
"testing"
|
"testing"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestNewSqliteDB(t *testing.T) {
|
func TestNewSqliteDB(t *testing.T) {
|
||||||
dbPath := os.Getenv("TEST_DB")
|
dbFile := os.Getenv("TEST_DB")
|
||||||
// sql_driver.MakeTables(dbFile)
|
// sql_driver.MakeTables(dbFile)
|
||||||
ctx := context.Background()
|
db, err := NewSqliteDB(dbFile)
|
||||||
sqliteDb, err := sqlite_queue.OpenSqliteDb(dbPath)
|
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
defer db.Close()
|
||||||
|
|
||||||
queue, err := sqlite_queue.NewQueue(sqliteDb, ctx)
|
testSignupLoginRenew(t, db)
|
||||||
assert.NoError(t, err)
|
testSqliteDBRandomSvgInterface(t, db)
|
||||||
|
// if _, err := os.Stat(dbFile); err == nil {
|
||||||
queue.Start()
|
// err = os.Remove(dbFile)
|
||||||
defer queue.Stop()
|
// assert.NoError(t, err)
|
||||||
db := NewSqliteRepository(queue, ctx)
|
// } else {
|
||||||
assert.NoError(t, err)
|
// assert.NoError(t, err)
|
||||||
testSignupLoginRenew(t, &db)
|
// }
|
||||||
testSqliteDBRandomSvgInterface(t, &db)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func testSignupLoginRenew(t *testing.T, db CustomerUserRepository) {
|
func testSignupLoginRenew(t *testing.T, db CustomerUserRepository) {
|
||||||
@@ -163,6 +163,6 @@ func (q *Queue) Stop() {
|
|||||||
q.stop = true
|
q.stop = true
|
||||||
// Wait for all emails to be processed
|
// Wait for all emails to be processed
|
||||||
q.wg.Wait()
|
q.wg.Wait()
|
||||||
// Stop the email queue
|
// Close the email queue
|
||||||
close(q.emailQueue)
|
close(q.emailQueue)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -22,7 +22,7 @@ func TestEmailQueue(t *testing.T) {
|
|||||||
}
|
}
|
||||||
queue.AddEmail(email)
|
queue.AddEmail(email)
|
||||||
}
|
}
|
||||||
// Stop the queue after all emails are processed
|
// Close the queue after all emails are processed
|
||||||
queue.Stop()
|
queue.Stop()
|
||||||
|
|
||||||
assert.Equal(t, queue.FailedSendCount, 0)
|
assert.Equal(t, queue.FailedSendCount, 0)
|
||||||
|
|||||||
@@ -1,7 +1,6 @@
|
|||||||
package entities
|
package entities
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/DonovanKelly/sugar-n-spice/set"
|
|
||||||
"github.com/google/uuid"
|
"github.com/google/uuid"
|
||||||
"go-nkode/config"
|
"go-nkode/config"
|
||||||
"go-nkode/internal/models"
|
"go-nkode/internal/models"
|
||||||
@@ -39,8 +38,8 @@ func (c *Customer) IsValidNKode(kp KeypadDimension, passcodeAttrIdx []int) error
|
|||||||
if validIdx := kp.ValidateAttributeIndices(passcodeAttrIdx); !validIdx {
|
if validIdx := kp.ValidateAttributeIndices(passcodeAttrIdx); !validIdx {
|
||||||
return config.ErrInvalidNKodeIdx
|
return config.ErrInvalidNKodeIdx
|
||||||
}
|
}
|
||||||
passcodeSetVals := make(set.Set[uint64])
|
passcodeSetVals := make(utils.Set[uint64])
|
||||||
passcodeAttrVals := make(set.Set[uint64])
|
passcodeAttrVals := make(utils.Set[uint64])
|
||||||
attrVals, err := c.Attributes.AttrValsForKp(kp)
|
attrVals, err := c.Attributes.AttrValsForKp(kp)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
|
|||||||
@@ -1,8 +1,8 @@
|
|||||||
package entities
|
package entities
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/DonovanKelly/sugar-n-spice/all"
|
|
||||||
"go-nkode/config"
|
"go-nkode/config"
|
||||||
|
py "go-nkode/internal/utils"
|
||||||
)
|
)
|
||||||
|
|
||||||
type KeypadDimension struct {
|
type KeypadDimension struct {
|
||||||
@@ -26,13 +26,13 @@ func (kp *KeypadDimension) IsValidKeypadDimension() error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (kp *KeypadDimension) ValidKeySelections(selectedKeys []int) bool {
|
func (kp *KeypadDimension) ValidKeySelections(selectedKeys []int) bool {
|
||||||
return all.All[int](selectedKeys, func(idx int) bool {
|
return py.All[int](selectedKeys, func(idx int) bool {
|
||||||
return 0 <= idx && idx < kp.NumbOfKeys
|
return 0 <= idx && idx < kp.NumbOfKeys
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
func (kp *KeypadDimension) ValidateAttributeIndices(attrIndicies []int) bool {
|
func (kp *KeypadDimension) ValidateAttributeIndices(attrIndicies []int) bool {
|
||||||
return all.All[int](attrIndicies, func(i int) bool {
|
return py.All[int](attrIndicies, func(i int) bool {
|
||||||
return i >= 0 && i < kp.TotalAttrs()
|
return i >= 0 && i < kp.TotalAttrs()
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,10 +1,10 @@
|
|||||||
package entities
|
package entities
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/DonovanKelly/sugar-n-spice/set"
|
|
||||||
"go-nkode/config"
|
"go-nkode/config"
|
||||||
"go-nkode/internal/models"
|
"go-nkode/internal/models"
|
||||||
"go-nkode/internal/security"
|
"go-nkode/internal/security"
|
||||||
|
"go-nkode/internal/utils"
|
||||||
"log"
|
"log"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -122,15 +122,15 @@ func (u *UserInterface) randomAttributeRotation() error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (u *UserInterface) AttributeAdjacencyGraph() (map[int]set.Set[int], error) {
|
func (u *UserInterface) AttributeAdjacencyGraph() (map[int]utils.Set[int], error) {
|
||||||
interfaceKeypad, err := u.InterfaceMatrix()
|
interfaceKeypad, err := u.InterfaceMatrix()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
graph := make(map[int]set.Set[int])
|
graph := make(map[int]utils.Set[int])
|
||||||
|
|
||||||
for _, key := range interfaceKeypad {
|
for _, key := range interfaceKeypad {
|
||||||
keySet := set.NewSetFromSlice(key)
|
keySet := utils.NewSetFromSlice(key)
|
||||||
for _, attr := range key {
|
for _, attr := range key {
|
||||||
attrAdjacency := keySet.Copy()
|
attrAdjacency := keySet.Copy()
|
||||||
attrAdjacency.Remove(attr)
|
attrAdjacency.Remove(attr)
|
||||||
@@ -154,7 +154,7 @@ func (u *UserInterface) LoginShuffle() error {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
selectedSets := set.NewSetFromSlice[int](setIdxs[:numbOfSelectedSets])
|
selectedSets := utils.NewSetFromSlice[int](setIdxs[:numbOfSelectedSets])
|
||||||
|
|
||||||
for keyIdx, key := range keypadSet1 {
|
for keyIdx, key := range keypadSet1 {
|
||||||
for idx := range key {
|
for idx := range key {
|
||||||
|
|||||||
@@ -1,12 +1,11 @@
|
|||||||
package entities
|
package entities
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/DonovanKelly/sugar-n-spice/all"
|
|
||||||
"github.com/DonovanKelly/sugar-n-spice/set"
|
|
||||||
"github.com/google/uuid"
|
"github.com/google/uuid"
|
||||||
"go-nkode/config"
|
"go-nkode/config"
|
||||||
"go-nkode/internal/models"
|
"go-nkode/internal/models"
|
||||||
"go-nkode/internal/security"
|
"go-nkode/internal/security"
|
||||||
|
py "go-nkode/internal/utils"
|
||||||
"log"
|
"log"
|
||||||
"sort"
|
"sort"
|
||||||
)
|
)
|
||||||
@@ -51,7 +50,7 @@ func NewSignupResetSession(userEmail models.UserEmail, kp KeypadDimension, custo
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (s *UserSignSession) DeducePasscode(confirmKeyEntry models.KeySelection) ([]int, error) {
|
func (s *UserSignSession) DeducePasscode(confirmKeyEntry models.KeySelection) ([]int, error) {
|
||||||
validEntry := all.All[int](confirmKeyEntry, func(i int) bool {
|
validEntry := py.All[int](confirmKeyEntry, func(i int) bool {
|
||||||
return 0 <= i && i < s.Kp.NumbOfKeys
|
return 0 <= i && i < s.Kp.NumbOfKeys
|
||||||
})
|
})
|
||||||
|
|
||||||
@@ -94,8 +93,8 @@ func (s *UserSignSession) DeducePasscode(confirmKeyEntry models.KeySelection) ([
|
|||||||
passcode := make([]int, passcodeLen)
|
passcode := make([]int, passcodeLen)
|
||||||
|
|
||||||
for idx := 0; idx < passcodeLen; idx++ {
|
for idx := 0; idx < passcodeLen; idx++ {
|
||||||
setKey := set.NewSetFromSlice[int](setKeyVals[idx])
|
setKey := py.NewSetFromSlice[int](setKeyVals[idx])
|
||||||
confirmKey := set.NewSetFromSlice[int](confirmKeyVals[idx])
|
confirmKey := py.NewSetFromSlice[int](confirmKeyVals[idx])
|
||||||
intersection := setKey.Intersect(confirmKey)
|
intersection := setKey.Intersect(confirmKey)
|
||||||
if intersection.Size() < 1 {
|
if intersection.Size() < 1 {
|
||||||
log.Printf("set and confirm do not intersect at index %d", idx)
|
log.Printf("set and confirm do not intersect at index %d", idx)
|
||||||
@@ -112,7 +111,7 @@ func (s *UserSignSession) DeducePasscode(confirmKeyEntry models.KeySelection) ([
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (s *UserSignSession) SetUserNKode(keySelection models.KeySelection) (models.IdxInterface, error) {
|
func (s *UserSignSession) SetUserNKode(keySelection models.KeySelection) (models.IdxInterface, error) {
|
||||||
validKeySelection := all.All[int](keySelection, func(i int) bool {
|
validKeySelection := py.All[int](keySelection, func(i int) bool {
|
||||||
return 0 <= i && i < s.Kp.NumbOfKeys
|
return 0 <= i && i < s.Kp.NumbOfKeys
|
||||||
})
|
})
|
||||||
if !validKeySelection {
|
if !validKeySelection {
|
||||||
|
|||||||
@@ -1,9 +1,9 @@
|
|||||||
package entities
|
package entities
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/DonovanKelly/sugar-n-spice/all"
|
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
"go-nkode/internal/models"
|
"go-nkode/internal/models"
|
||||||
|
py "go-nkode/internal/utils"
|
||||||
"testing"
|
"testing"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -120,12 +120,12 @@ func TestUserInterface_PartialInterfaceShuffle(t *testing.T) {
|
|||||||
shuffleCompare[idx] = val == postShuffle[idx]
|
shuffleCompare[idx] = val == postShuffle[idx]
|
||||||
}
|
}
|
||||||
|
|
||||||
allTrue := all.All[bool](shuffleCompare, func(n bool) bool {
|
allTrue := py.All[bool](shuffleCompare, func(n bool) bool {
|
||||||
return n == true
|
return n == true
|
||||||
})
|
})
|
||||||
assert.False(t, allTrue)
|
assert.False(t, allTrue)
|
||||||
|
|
||||||
allFalse := all.All[bool](shuffleCompare, func(n bool) bool {
|
allFalse := py.All[bool](shuffleCompare, func(n bool) bool {
|
||||||
return n == false
|
return n == false
|
||||||
})
|
})
|
||||||
|
|
||||||
|
|||||||
@@ -6,7 +6,7 @@ import (
|
|||||||
"encoding/binary"
|
"encoding/binary"
|
||||||
"encoding/hex"
|
"encoding/hex"
|
||||||
"errors"
|
"errors"
|
||||||
"github.com/DonovanKelly/sugar-n-spice/set"
|
"go-nkode/internal/utils"
|
||||||
"log"
|
"log"
|
||||||
"math/big"
|
"math/big"
|
||||||
r "math/rand"
|
r "math/rand"
|
||||||
@@ -84,7 +84,7 @@ func GenerateRandomNonRepeatingUint64(listLen int) ([]uint64, error) {
|
|||||||
if listLen > int(1)<<32 {
|
if listLen > int(1)<<32 {
|
||||||
return nil, ErrRandNonRepeatingUint64
|
return nil, ErrRandNonRepeatingUint64
|
||||||
}
|
}
|
||||||
listSet := make(set.Set[uint64])
|
listSet := make(utils.Set[uint64])
|
||||||
for {
|
for {
|
||||||
if listSet.Size() == listLen {
|
if listSet.Size() == listLen {
|
||||||
break
|
break
|
||||||
@@ -104,7 +104,7 @@ func GenerateRandomNonRepeatingInt(listLen int) ([]int, error) {
|
|||||||
if listLen > int(1)<<31 {
|
if listLen > int(1)<<31 {
|
||||||
return nil, ErrRandNonRepeatingInt
|
return nil, ErrRandNonRepeatingInt
|
||||||
}
|
}
|
||||||
listSet := make(set.Set[int])
|
listSet := make(utils.Set[int])
|
||||||
for {
|
for {
|
||||||
if listSet.Size() == listLen {
|
if listSet.Size() == listLen {
|
||||||
break
|
break
|
||||||
|
|||||||
@@ -1,93 +0,0 @@
|
|||||||
package sqlc
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"database/sql"
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
"sync"
|
|
||||||
)
|
|
||||||
|
|
||||||
const writeBufferSize = 100
|
|
||||||
|
|
||||||
type SqlcGeneric func(*Queries, context.Context, any) error
|
|
||||||
|
|
||||||
type WriteTx struct {
|
|
||||||
ErrChan chan error
|
|
||||||
Query SqlcGeneric
|
|
||||||
Args interface{}
|
|
||||||
}
|
|
||||||
|
|
||||||
type Queue struct {
|
|
||||||
Queries *Queries
|
|
||||||
Db *sql.DB
|
|
||||||
WriteQueue chan WriteTx
|
|
||||||
wg sync.WaitGroup
|
|
||||||
ctx context.Context
|
|
||||||
cancel context.CancelFunc
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewQueue(sqlDb *sql.DB, ctx context.Context) (*Queue, error) {
|
|
||||||
ctx, cancel := context.WithCancel(context.Background())
|
|
||||||
sqldb := &Queue{
|
|
||||||
Queries: New(sqlDb),
|
|
||||||
Db: sqlDb,
|
|
||||||
WriteQueue: make(chan WriteTx, writeBufferSize),
|
|
||||||
ctx: ctx,
|
|
||||||
cancel: cancel,
|
|
||||||
}
|
|
||||||
|
|
||||||
return sqldb, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *Queue) Start() {
|
|
||||||
d.wg.Add(1)
|
|
||||||
defer d.wg.Done()
|
|
||||||
go func() {
|
|
||||||
for {
|
|
||||||
select {
|
|
||||||
case <-d.ctx.Done():
|
|
||||||
return
|
|
||||||
case writeTx := <-d.WriteQueue:
|
|
||||||
err := writeTx.Query(d.Queries, d.ctx, writeTx.Args)
|
|
||||||
writeTx.ErrChan <- err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *Queue) Stop() error {
|
|
||||||
d.cancel()
|
|
||||||
d.wg.Wait()
|
|
||||||
close(d.WriteQueue)
|
|
||||||
return d.Db.Close()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *Queue) EnqueueWriteTx(queryFunc SqlcGeneric, args any) error {
|
|
||||||
select {
|
|
||||||
case <-d.ctx.Done():
|
|
||||||
return errors.New("database is shutting down")
|
|
||||||
default:
|
|
||||||
}
|
|
||||||
|
|
||||||
errChan := make(chan error, 1)
|
|
||||||
writeTx := WriteTx{
|
|
||||||
Query: queryFunc,
|
|
||||||
Args: args,
|
|
||||||
ErrChan: errChan,
|
|
||||||
}
|
|
||||||
d.WriteQueue <- writeTx
|
|
||||||
return <-errChan
|
|
||||||
}
|
|
||||||
|
|
||||||
func OpenSqliteDb(dbPath string) (*sql.DB, error) {
|
|
||||||
sqliteDb, err := sql.Open("sqlite3", dbPath)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("failed to open database: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := sqliteDb.Ping(); err != nil {
|
|
||||||
return nil, fmt.Errorf("failed to connect to database: %w", err)
|
|
||||||
}
|
|
||||||
return sqliteDb, nil
|
|
||||||
}
|
|
||||||
63
internal/utils/hashset.go
Normal file
63
internal/utils/hashset.go
Normal file
@@ -0,0 +1,63 @@
|
|||||||
|
package utils
|
||||||
|
|
||||||
|
type Set[T comparable] map[T]struct{}
|
||||||
|
|
||||||
|
func (s *Set[T]) Add(element T) {
|
||||||
|
(*s)[element] = struct{}{}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Set[T]) Remove(element T) {
|
||||||
|
delete(*s, element)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Set[T]) Contains(element T) bool {
|
||||||
|
_, exists := (*s)[element]
|
||||||
|
return exists
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Set[T]) Size() int {
|
||||||
|
return len(*s)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Set[T]) ToSlice() []T {
|
||||||
|
list := make([]T, 0, len(*s))
|
||||||
|
for key := range *s {
|
||||||
|
list = append(list, key)
|
||||||
|
}
|
||||||
|
return list
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewSetFromSlice[T comparable](slice []T) Set[T] {
|
||||||
|
set := make(Set[T])
|
||||||
|
for _, val := range slice {
|
||||||
|
set.Add(val)
|
||||||
|
}
|
||||||
|
return set
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Set[T]) Copy() Set[T] {
|
||||||
|
newSet := make(Set[T])
|
||||||
|
for key, val := range *s {
|
||||||
|
newSet[key] = val
|
||||||
|
}
|
||||||
|
return newSet
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Set[T]) IsDisjoint(otherSet Set[T]) bool {
|
||||||
|
for attr := range *s {
|
||||||
|
if otherSet.Contains(attr) {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Set[T]) Intersect(otherSet Set[T]) Set[T] {
|
||||||
|
intersect := make(Set[T])
|
||||||
|
for val := range *s {
|
||||||
|
if otherSet.Contains(val) {
|
||||||
|
intersect.Add(val)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return intersect
|
||||||
|
}
|
||||||
35
internal/utils/hashset_test.go
Normal file
35
internal/utils/hashset_test.go
Normal file
@@ -0,0 +1,35 @@
|
|||||||
|
package utils
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestSet(t *testing.T) {
|
||||||
|
intSet := make(Set[int])
|
||||||
|
intSet.Add(1)
|
||||||
|
intSet.Add(2)
|
||||||
|
assert.EqualValues(t, intSet.Size(), 2)
|
||||||
|
intSet.Add(3)
|
||||||
|
intSet.Add(3)
|
||||||
|
assert.EqualValues(t, intSet.Size(), 3)
|
||||||
|
intSet.Remove(2)
|
||||||
|
assert.EqualValues(t, intSet.Size(), 2)
|
||||||
|
assert.False(t, intSet.Contains(2))
|
||||||
|
assert.True(t, intSet.Contains(1))
|
||||||
|
|
||||||
|
list := intSet.ToSlice()
|
||||||
|
assert.Contains(t, list, 1)
|
||||||
|
assert.Contains(t, list, 3)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSet_Copy(t *testing.T) {
|
||||||
|
intSet := NewSetFromSlice[int]([]int{1, 2, 3})
|
||||||
|
|
||||||
|
copySet := intSet.Copy()
|
||||||
|
|
||||||
|
intSet.Remove(1)
|
||||||
|
assert.Equal(t, intSet.Size(), 2)
|
||||||
|
assert.Equal(t, copySet.Size(), 3)
|
||||||
|
|
||||||
|
}
|
||||||
11
internal/utils/py-builtin.go
Normal file
11
internal/utils/py-builtin.go
Normal file
@@ -0,0 +1,11 @@
|
|||||||
|
package utils
|
||||||
|
|
||||||
|
func All[T comparable](slice []T, condition func(T) bool) bool {
|
||||||
|
|
||||||
|
for _, v := range slice {
|
||||||
|
if !condition(v) {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user