Compare commits
7 Commits
Author | SHA1 | Date | |
---|---|---|---|
6b4bd5a6f8
|
|||
6df4f5f2a1
|
|||
780905ba35
|
|||
c679797765
|
|||
401aad9fa4
|
|||
645113d553
|
|||
4a33986b6a
|
@@ -38,7 +38,7 @@ var rexIDChecksumConst = rext.W(regexp.MustCompile(`const ChecksumIDGenerator =
|
|||||||
//go:embed id-generate.template
|
//go:embed id-generate.template
|
||||||
var templateIDGenerateText string
|
var templateIDGenerateText string
|
||||||
|
|
||||||
func GenerateIDSpecs(sourceDir string, destFile string, opt *IDGenOptions) error {
|
func GenerateIDSpecs(sourceDir string, destFile string, opt IDGenOptions) error {
|
||||||
|
|
||||||
debugOutput := langext.Coalesce(opt.DebugOutput, false)
|
debugOutput := langext.Coalesce(opt.DebugOutput, false)
|
||||||
|
|
||||||
|
@@ -34,10 +34,10 @@ func TestGenerateIDSpecs(t *testing.T) {
|
|||||||
_, err = cmdext.Runner("tar").Arg("-xvzf").Arg(tmpFile).Arg("-C").Arg(tmpDir).FailOnExitCode().FailOnTimeout().Timeout(time.Minute).Run()
|
_, err = cmdext.Runner("tar").Arg("-xvzf").Arg(tmpFile).Arg("-C").Arg(tmpDir).FailOnExitCode().FailOnTimeout().Timeout(time.Minute).Run()
|
||||||
tst.AssertNoErr(t, err)
|
tst.AssertNoErr(t, err)
|
||||||
|
|
||||||
err = GenerateIDSpecs(tmpDir, tmpDir+"/id_gen.go", &IDGenOptions{DebugOutput: langext.PTrue})
|
err = GenerateIDSpecs(tmpDir, tmpDir+"/id_gen.go", IDGenOptions{DebugOutput: langext.PTrue})
|
||||||
tst.AssertNoErr(t, err)
|
tst.AssertNoErr(t, err)
|
||||||
|
|
||||||
err = GenerateIDSpecs(tmpDir, tmpDir+"/id_gen.go", &IDGenOptions{DebugOutput: langext.PTrue})
|
err = GenerateIDSpecs(tmpDir, tmpDir+"/id_gen.go", IDGenOptions{DebugOutput: langext.PTrue})
|
||||||
tst.AssertNoErr(t, err)
|
tst.AssertNoErr(t, err)
|
||||||
|
|
||||||
fmt.Println()
|
fmt.Println()
|
||||||
|
@@ -39,6 +39,7 @@ type ginRouteSpec struct {
|
|||||||
type Options struct {
|
type Options struct {
|
||||||
AllowCors *bool // Add cors handler to allow all CORS requests on the default http methods
|
AllowCors *bool // Add cors handler to allow all CORS requests on the default http methods
|
||||||
GinDebug *bool // Set gin.debug to true (adds more logs)
|
GinDebug *bool // Set gin.debug to true (adds more logs)
|
||||||
|
SuppressGinLogs *bool // Suppress our custom gin logs (even if GinDebug == true)
|
||||||
BufferBody *bool // Buffers the input body stream, this way the ginext error handler can later include the whole request body
|
BufferBody *bool // Buffers the input body stream, this way the ginext error handler can later include the whole request body
|
||||||
Timeout *time.Duration // The default handler timeout
|
Timeout *time.Duration // The default handler timeout
|
||||||
ListenerBeforeRequest []func(g *gin.Context) // Register listener that are called before the handler method
|
ListenerBeforeRequest []func(g *gin.Context) // Register listener that are called before the handler method
|
||||||
@@ -51,7 +52,7 @@ func NewEngine(opt Options) *GinWrapper {
|
|||||||
|
|
||||||
wrapper := &GinWrapper{
|
wrapper := &GinWrapper{
|
||||||
engine: engine,
|
engine: engine,
|
||||||
suppressGinLogs: false,
|
suppressGinLogs: langext.Coalesce(opt.SuppressGinLogs, false),
|
||||||
allowCors: langext.Coalesce(opt.AllowCors, false),
|
allowCors: langext.Coalesce(opt.AllowCors, false),
|
||||||
ginDebug: langext.Coalesce(opt.GinDebug, true),
|
ginDebug: langext.Coalesce(opt.GinDebug, true),
|
||||||
bufferBody: langext.Coalesce(opt.BufferBody, false),
|
bufferBody: langext.Coalesce(opt.BufferBody, false),
|
||||||
@@ -73,12 +74,12 @@ func NewEngine(opt Options) *GinWrapper {
|
|||||||
if !wrapper.ginDebug {
|
if !wrapper.ginDebug {
|
||||||
gin.SetMode(gin.ReleaseMode)
|
gin.SetMode(gin.ReleaseMode)
|
||||||
|
|
||||||
|
if !wrapper.suppressGinLogs {
|
||||||
ginlogger := gin.Logger()
|
ginlogger := gin.Logger()
|
||||||
engine.Use(func(context *gin.Context) {
|
engine.Use(func(context *gin.Context) {
|
||||||
if !wrapper.suppressGinLogs {
|
|
||||||
ginlogger(context)
|
ginlogger(context)
|
||||||
}
|
|
||||||
})
|
})
|
||||||
|
}
|
||||||
} else {
|
} else {
|
||||||
gin.SetMode(gin.DebugMode)
|
gin.SetMode(gin.DebugMode)
|
||||||
}
|
}
|
||||||
|
@@ -1,5 +1,5 @@
|
|||||||
package goext
|
package goext
|
||||||
|
|
||||||
const GoextVersion = "0.0.406"
|
const GoextVersion = "0.0.413"
|
||||||
|
|
||||||
const GoextVersionTimestamp = "2024-03-10T16:44:21+0100"
|
const GoextVersionTimestamp = "2024-03-11T21:00:30+0100"
|
||||||
|
@@ -5,12 +5,76 @@ import (
|
|||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
func Coalesce[T any](v *T, def T) T {
|
func Coalesce[T any](v1 *T, def T) T {
|
||||||
if v == nil {
|
if v1 != nil {
|
||||||
return def
|
return *v1
|
||||||
} else {
|
|
||||||
return *v
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
return def
|
||||||
|
}
|
||||||
|
|
||||||
|
func CoalesceOpt[T any](v1 *T, v2 *T) *T {
|
||||||
|
if v1 != nil {
|
||||||
|
return v1
|
||||||
|
}
|
||||||
|
|
||||||
|
return v2
|
||||||
|
}
|
||||||
|
|
||||||
|
func Coalesce3[T any](v1 *T, v2 *T, def T) T {
|
||||||
|
if v1 != nil {
|
||||||
|
return *v1
|
||||||
|
}
|
||||||
|
|
||||||
|
if v2 != nil {
|
||||||
|
return *v2
|
||||||
|
}
|
||||||
|
|
||||||
|
return def
|
||||||
|
}
|
||||||
|
|
||||||
|
func Coalesce3Opt[T any](v1 *T, v2 *T, v3 *T) *T {
|
||||||
|
if v1 != nil {
|
||||||
|
return v1
|
||||||
|
}
|
||||||
|
|
||||||
|
if v2 != nil {
|
||||||
|
return v2
|
||||||
|
}
|
||||||
|
|
||||||
|
return v3
|
||||||
|
}
|
||||||
|
|
||||||
|
func Coalesce4[T any](v1 *T, v2 *T, v3 *T, def T) T {
|
||||||
|
if v1 != nil {
|
||||||
|
return *v1
|
||||||
|
}
|
||||||
|
|
||||||
|
if v2 != nil {
|
||||||
|
return *v2
|
||||||
|
}
|
||||||
|
|
||||||
|
if v3 != nil {
|
||||||
|
return *v3
|
||||||
|
}
|
||||||
|
|
||||||
|
return def
|
||||||
|
}
|
||||||
|
|
||||||
|
func Coalesce4Opt[T any](v1 *T, v2 *T, v3 *T, v4 *T) *T {
|
||||||
|
if v1 != nil {
|
||||||
|
return v1
|
||||||
|
}
|
||||||
|
|
||||||
|
if v2 != nil {
|
||||||
|
return v2
|
||||||
|
}
|
||||||
|
|
||||||
|
if v3 != nil {
|
||||||
|
return v3
|
||||||
|
}
|
||||||
|
|
||||||
|
return v4
|
||||||
}
|
}
|
||||||
|
|
||||||
func CoalesceString(s *string, def string) string {
|
func CoalesceString(s *string, def string) string {
|
||||||
|
@@ -47,3 +47,10 @@ func NewSimplePaginateFilter(filterClause string, filterParams PP, sort []Filter
|
|||||||
},
|
},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func NewEmptyPaginateFilter() PaginateFilter {
|
||||||
|
return genericPaginateFilter{
|
||||||
|
sql: func(params PP) (string, string, []string) { return "1=1", "", nil },
|
||||||
|
sort: func() []FilterSort { return make([]FilterSort, 0) },
|
||||||
|
}
|
||||||
|
}
|
||||||
|
48
sq/list.go
Normal file
48
sq/list.go
Normal file
@@ -0,0 +1,48 @@
|
|||||||
|
package sq
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"gogs.mikescher.com/BlackForestBytes/goext/exerr"
|
||||||
|
)
|
||||||
|
|
||||||
|
func Iterate[TData any](ctx context.Context, q Queryable, table string, filter PaginateFilter, scanMode StructScanMode, scanSec StructScanSafety, page int, limit *int, consumer func(ctx context.Context, v TData) error) (int, error) {
|
||||||
|
if filter == nil {
|
||||||
|
filter = NewEmptyPaginateFilter()
|
||||||
|
}
|
||||||
|
|
||||||
|
prepParams := PP{}
|
||||||
|
|
||||||
|
sortOrder := filter.Sort()
|
||||||
|
sortCond := ""
|
||||||
|
if len(sortOrder) > 0 {
|
||||||
|
sortCond = "ORDER BY "
|
||||||
|
for i, v := range sortOrder {
|
||||||
|
if i > 0 {
|
||||||
|
sortCond += ", "
|
||||||
|
}
|
||||||
|
sortCond += v.Field + " " + string(v.Direction)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pageCond := ""
|
||||||
|
if limit != nil {
|
||||||
|
pageCond += fmt.Sprintf("LIMIT :%s OFFSET :%s", prepParams.Add(*limit+1), prepParams.Add(*limit*(page-1)))
|
||||||
|
}
|
||||||
|
|
||||||
|
filterCond, joinCond, joinTables := filter.SQL(prepParams)
|
||||||
|
|
||||||
|
selectCond := table + ".*"
|
||||||
|
for _, v := range joinTables {
|
||||||
|
selectCond += ", " + v + ".*"
|
||||||
|
}
|
||||||
|
|
||||||
|
sqlQueryData := "SELECT " + selectCond + " FROM " + table + " " + joinCond + " WHERE ( " + filterCond + " ) " + sortCond + " " + pageCond
|
||||||
|
|
||||||
|
rows, err := q.Query(ctx, sqlQueryData, prepParams)
|
||||||
|
if err != nil {
|
||||||
|
return 0, exerr.Wrap(err, "failed to list paginated entries from DB").Str("table", table).Any("filter", filter).Int("page", page).Any("limit", limit).Build()
|
||||||
|
}
|
||||||
|
|
||||||
|
return IterateAll[TData](ctx, q, rows, scanMode, scanSec, true, consumer)
|
||||||
|
}
|
@@ -9,6 +9,10 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
func Paginate[TData any](ctx context.Context, q Queryable, table string, filter PaginateFilter, scanMode StructScanMode, scanSec StructScanSafety, page int, limit *int) ([]TData, pag.Pagination, error) {
|
func Paginate[TData any](ctx context.Context, q Queryable, table string, filter PaginateFilter, scanMode StructScanMode, scanSec StructScanSafety, page int, limit *int) ([]TData, pag.Pagination, error) {
|
||||||
|
if filter == nil {
|
||||||
|
filter = NewEmptyPaginateFilter()
|
||||||
|
}
|
||||||
|
|
||||||
prepParams := PP{}
|
prepParams := PP{}
|
||||||
|
|
||||||
sortOrder := filter.Sort()
|
sortOrder := filter.Sort()
|
||||||
@@ -90,6 +94,10 @@ func Paginate[TData any](ctx context.Context, q Queryable, table string, filter
|
|||||||
}
|
}
|
||||||
|
|
||||||
func Count(ctx context.Context, q Queryable, table string, filter PaginateFilter) (int, error) {
|
func Count(ctx context.Context, q Queryable, table string, filter PaginateFilter) (int, error) {
|
||||||
|
if filter == nil {
|
||||||
|
filter = NewEmptyPaginateFilter()
|
||||||
|
}
|
||||||
|
|
||||||
prepParams := PP{}
|
prepParams := PP{}
|
||||||
|
|
||||||
filterCond, joinCond, _ := filter.SQL(prepParams)
|
filterCond, joinCond, _ := filter.SQL(prepParams)
|
||||||
|
@@ -333,3 +333,79 @@ func ScanAll[TData any](ctx context.Context, q Queryable, rows *sqlx.Rows, mode
|
|||||||
}
|
}
|
||||||
return res, nil
|
return res, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func IterateAll[TData any](ctx context.Context, q Queryable, rows *sqlx.Rows, mode StructScanMode, sec StructScanSafety, close bool, consumer func(ctx context.Context, v TData) error) (int, error) {
|
||||||
|
var strscan *StructScanner
|
||||||
|
|
||||||
|
if sec == Safe {
|
||||||
|
strscan = NewStructScanner(rows, false)
|
||||||
|
var data TData
|
||||||
|
err := strscan.Start(&data)
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
} else if sec == Unsafe {
|
||||||
|
strscan = NewStructScanner(rows, true)
|
||||||
|
var data TData
|
||||||
|
err := strscan.Start(&data)
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
return 0, errors.New("unknown value for <sec>")
|
||||||
|
}
|
||||||
|
|
||||||
|
rcount := 0
|
||||||
|
|
||||||
|
for rows.Next() {
|
||||||
|
|
||||||
|
if err := ctx.Err(); err != nil {
|
||||||
|
return rcount, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if mode == SModeFast {
|
||||||
|
var data TData
|
||||||
|
err := strscan.StructScanBase(&data)
|
||||||
|
if err != nil {
|
||||||
|
return rcount, err
|
||||||
|
}
|
||||||
|
|
||||||
|
err = consumer(ctx, data)
|
||||||
|
if err != nil {
|
||||||
|
return rcount, exerr.Wrap(err, "").Build()
|
||||||
|
}
|
||||||
|
|
||||||
|
rcount++
|
||||||
|
|
||||||
|
} else if mode == SModeExtended {
|
||||||
|
var data TData
|
||||||
|
err := strscan.StructScanExt(q, &data)
|
||||||
|
if err != nil {
|
||||||
|
return rcount, err
|
||||||
|
}
|
||||||
|
|
||||||
|
err = consumer(ctx, data)
|
||||||
|
if err != nil {
|
||||||
|
return rcount, exerr.Wrap(err, "").Build()
|
||||||
|
}
|
||||||
|
|
||||||
|
rcount++
|
||||||
|
|
||||||
|
} else {
|
||||||
|
return rcount, errors.New("unknown value for <mode>")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if close {
|
||||||
|
err := strscan.rows.Close()
|
||||||
|
if err != nil {
|
||||||
|
return rcount, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := rows.Err(); err != nil {
|
||||||
|
return rcount, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return rcount, nil
|
||||||
|
}
|
||||||
|
Reference in New Issue
Block a user