reorganize repository

Author: UltimateForm

cmd/api/auth.go

@@ -3,8 +3,8 @@ package api
 import (
 	"net/http"
 
-	"github.com/UltimateForm/gopen-api/internal/authrep"
 	"github.com/UltimateForm/gopen-api/internal/core"
+	"github.com/UltimateForm/gopen-api/internal/repository/authrep"
 )
 
 type LoginRequest struct {

internal/authrep/main.go

@@ -7,12 +7,14 @@ import (
 	"github.com/neo4j/neo4j-go-driver/v5/neo4j"
 )
 
-func init() {
-	log.Println("DB PACKAGE INIT")
-}
+var Driver neo4j.DriverWithContext
 
-func GetDriver() (neo4j.DriverWithContext, error) {
+func init() {
 	dbCfg := config.LoadDbConfig()
 	driver, driverErr := neo4j.NewDriverWithContext(dbCfg.Uri, neo4j.BasicAuth(dbCfg.Username, dbCfg.Password, ""))
-	return driver, driverErr
+	if driverErr != nil {
+		panic(driverErr)
+	}
+	Driver = driver
+	log.Println("DB PACKAGE INIT")
 }

internal/db/main.go

@@ -0,0 +1,38 @@
+package authrep
+
+import (
+	"context"
+	"errors"
+
+	"github.com/UltimateForm/gopen-api/internal/repository"
+	"github.com/neo4j/neo4j-go-driver/v5/neo4j"
+)
+
+func txUsrExistsAtomically(ctx context.Context, user User) neo4j.ManagedTransactionWorkT[bool] {
+	query := queryComputeUserExistsAtomically(user)
+
+	return func(tx neo4j.ManagedTransaction) (bool, error) {
+		res, err := query.Execute(tx, ctx)
+		if err != nil {
+			return false, err
+		}
+		record, singleErr := res.Single(ctx)
+		if singleErr != nil {
+			return false, singleErr
+		}
+		exists, found := record.Get("exists")
+		if !found {
+			return false, nil
+		}
+		boolValue, isBoolValue := exists.(bool)
+		if !isBoolValue {
+			return false, errors.New("expected 'exists' bool value from DB row")
+		}
+		return boolValue, nil
+	}
+
+}
+
+func UserExistsAtomically(ctx context.Context, user User) (bool, error) {
+	return repository.ExecuteReadWithDriver(ctx, txUsrExistsAtomically(ctx, user))
+}

internal/repository/authrep/main.go

@@ -0,0 +1,18 @@
+package authrep
+
+import (
+	"context"
+
+	"github.com/UltimateForm/gopen-api/internal/repository"
+
+	"github.com/neo4j/neo4j-go-driver/v5/neo4j"
+)
+
+func queryComputeUserExistsAtomically(user User) repository.QueryRunner {
+	return func(tx neo4j.ManagedTransaction, ctx context.Context) (neo4j.ResultWithContext, error) {
+		return tx.Run(
+			ctx,
+			"MATCH (u:User {email:$email, password:$password}) RETURN count(u) > 0 as exists",
+			map[string]any{"email": user.Email, "password": user.Password})
+	}
+}

internal/repository/authrep/queries.go

@@ -0,0 +1,17 @@
+package repository
+
+import (
+	"context"
+	"time"
+
+	"github.com/UltimateForm/gopen-api/internal/db"
+	"github.com/neo4j/neo4j-go-driver/v5/neo4j"
+)
+
+func ExecuteReadWithDriver[T any](ctx context.Context, tx neo4j.ManagedTransactionWorkT[T]) (T, error) {
+	readCtx, cancel := context.WithDeadline(ctx, time.Now().Add(time.Second*10))
+	defer cancel()
+	session := db.Driver.NewSession(readCtx, neo4j.SessionConfig{})
+	defer session.Close(readCtx)
+	return neo4j.ExecuteRead(readCtx, session, tx)
+}

internal/authrep/types.go internal/repository/authrep/types.gointernal/authrep/types.go renamed to internal/repository/authrep/types.go

@@ -0,0 +1,13 @@
+package repository
+
+import (
+	"context"
+
+	"github.com/neo4j/neo4j-go-driver/v5/neo4j"
+)
+
+type QueryRunner func(tx neo4j.ManagedTransaction, ctx context.Context) (neo4j.ResultWithContext, error)
+
+func (src QueryRunner) Execute(tx neo4j.ManagedTransaction, ctx context.Context) (neo4j.ResultWithContext, error) {
+	return src(tx, ctx)
+}