Compare commits
6 Commits
Author | SHA1 | Date | |
---|---|---|---|
295a098eb4
|
|||
b69a082bb1
|
|||
a4a8c83d17
|
|||
e952176bb0
|
|||
d99adb203b
|
|||
f1f91f4cfa
|
263
cryptext/pronouncablePassword.go
Normal file
263
cryptext/pronouncablePassword.go
Normal file
@@ -0,0 +1,263 @@
|
|||||||
|
package cryptext
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto/rand"
|
||||||
|
"io"
|
||||||
|
"math/big"
|
||||||
|
mathrand "math/rand"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
ppStartChar = "BCDFGHJKLMNPQRSTVWXZ"
|
||||||
|
ppEndChar = "ABDEFIKMNORSTUXYZ"
|
||||||
|
ppVowel = "AEIOUY"
|
||||||
|
ppConsonant = "BCDFGHJKLMNPQRSTVWXZ"
|
||||||
|
ppSegmentLenMin = 3
|
||||||
|
ppSegmentLenMax = 7
|
||||||
|
ppMaxRepeatedVowel = 2
|
||||||
|
ppMaxRepeatedConsonant = 2
|
||||||
|
)
|
||||||
|
|
||||||
|
var ppContinuation = map[uint8]string{
|
||||||
|
'A': "BCDFGHJKLMNPRSTVWXYZ",
|
||||||
|
'B': "ADFIKLMNORSTUY",
|
||||||
|
'C': "AEIKOUY",
|
||||||
|
'D': "AEILORSUYZ",
|
||||||
|
'E': "BCDFGHJKLMNPRSTVWXYZ",
|
||||||
|
'F': "ADEGIKLOPRTUY",
|
||||||
|
'G': "ABDEFHILMNORSTUY",
|
||||||
|
'H': "AEIOUY",
|
||||||
|
'I': "BCDFGHJKLMNPRSTVWXZ",
|
||||||
|
'J': "AEIOUY",
|
||||||
|
'K': "ADEFHILMNORSTUY",
|
||||||
|
'L': "ADEFGIJKMNOPSTUVWYZ",
|
||||||
|
'M': "ABEFIKOPSTUY",
|
||||||
|
'N': "ABEFIKOPSTUY",
|
||||||
|
'O': "BCDFGHJKLMNPRSTVWXYZ",
|
||||||
|
'P': "AEFIJLORSTUY",
|
||||||
|
'Q': "AEIOUY",
|
||||||
|
'R': "ADEFGHIJKLMNOPSTUVYZ",
|
||||||
|
'S': "ACDEIKLOPTUYZ",
|
||||||
|
'T': "AEHIJOPRSUWY",
|
||||||
|
'U': "BCDFGHJKLMNPRSTVWXZ",
|
||||||
|
'V': "AEIOUY",
|
||||||
|
'W': "AEIOUY",
|
||||||
|
'X': "AEIOUY",
|
||||||
|
'Y': "ABCDFGHKLMNPRSTVXZ",
|
||||||
|
'Z': "AEILOTUY",
|
||||||
|
}
|
||||||
|
|
||||||
|
var ppLog2Map = map[int]float64{
|
||||||
|
1: 0.00000000,
|
||||||
|
2: 1.00000000,
|
||||||
|
3: 1.58496250,
|
||||||
|
4: 2.00000000,
|
||||||
|
5: 2.32192809,
|
||||||
|
6: 2.58496250,
|
||||||
|
7: 2.80735492,
|
||||||
|
8: 3.00000000,
|
||||||
|
9: 3.16992500,
|
||||||
|
10: 3.32192809,
|
||||||
|
11: 3.45943162,
|
||||||
|
12: 3.58496250,
|
||||||
|
13: 3.70043972,
|
||||||
|
14: 3.80735492,
|
||||||
|
15: 3.90689060,
|
||||||
|
16: 4.00000000,
|
||||||
|
17: 4.08746284,
|
||||||
|
18: 4.16992500,
|
||||||
|
19: 4.24792751,
|
||||||
|
20: 4.32192809,
|
||||||
|
21: 4.39231742,
|
||||||
|
22: 4.45943162,
|
||||||
|
23: 4.52356196,
|
||||||
|
24: 4.58496250,
|
||||||
|
25: 4.64385619,
|
||||||
|
26: 4.70043972,
|
||||||
|
27: 4.75488750,
|
||||||
|
28: 4.80735492,
|
||||||
|
29: 4.85798100,
|
||||||
|
30: 4.90689060,
|
||||||
|
31: 4.95419631,
|
||||||
|
32: 5.00000000,
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
ppVowelMap = ppMakeSet(ppVowel)
|
||||||
|
ppConsonantMap = ppMakeSet(ppConsonant)
|
||||||
|
ppEndCharMap = ppMakeSet(ppEndChar)
|
||||||
|
)
|
||||||
|
|
||||||
|
func ppMakeSet(v string) map[uint8]bool {
|
||||||
|
mp := make(map[uint8]bool, len(v))
|
||||||
|
for _, chr := range v {
|
||||||
|
mp[uint8(chr)] = true
|
||||||
|
}
|
||||||
|
return mp
|
||||||
|
}
|
||||||
|
|
||||||
|
func ppRandInt(rng io.Reader, max int) int {
|
||||||
|
v, err := rand.Int(rng, big.NewInt(int64(max)))
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
return int(v.Int64())
|
||||||
|
}
|
||||||
|
|
||||||
|
func ppRand(rng io.Reader, chars string, entropy *float64) uint8 {
|
||||||
|
chr := chars[ppRandInt(rng, len(chars))]
|
||||||
|
|
||||||
|
*entropy = *entropy + ppLog2Map[len(chars)]
|
||||||
|
|
||||||
|
return chr
|
||||||
|
}
|
||||||
|
|
||||||
|
func ppCharType(chr uint8) (bool, bool) {
|
||||||
|
_, ok1 := ppVowelMap[chr]
|
||||||
|
_, ok2 := ppConsonantMap[chr]
|
||||||
|
|
||||||
|
return ok1, ok2
|
||||||
|
}
|
||||||
|
|
||||||
|
func ppCharsetRemove(cs string, set map[uint8]bool, allowEmpty bool) string {
|
||||||
|
result := ""
|
||||||
|
for _, chr := range cs {
|
||||||
|
if _, ok := set[uint8(chr)]; !ok {
|
||||||
|
result += string(chr)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if result == "" && !allowEmpty {
|
||||||
|
return cs
|
||||||
|
}
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
|
func ppCharsetFilter(cs string, set map[uint8]bool, allowEmpty bool) string {
|
||||||
|
result := ""
|
||||||
|
for _, chr := range cs {
|
||||||
|
if _, ok := set[uint8(chr)]; ok {
|
||||||
|
result += string(chr)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if result == "" && !allowEmpty {
|
||||||
|
return cs
|
||||||
|
}
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
|
func PronouncablePasswordExt(rng io.Reader, pwlen int) (string, float64) {
|
||||||
|
|
||||||
|
// kinda pseudo markov-chain - with a few extra rules and no weights...
|
||||||
|
|
||||||
|
if pwlen <= 0 {
|
||||||
|
return "", 0
|
||||||
|
}
|
||||||
|
|
||||||
|
vowelCount := 0
|
||||||
|
consoCount := 0
|
||||||
|
entropy := float64(0)
|
||||||
|
|
||||||
|
startChar := ppRand(rng, ppStartChar, &entropy)
|
||||||
|
|
||||||
|
result := string(startChar)
|
||||||
|
currentChar := startChar
|
||||||
|
|
||||||
|
isVowel, isConsonant := ppCharType(currentChar)
|
||||||
|
if isVowel {
|
||||||
|
vowelCount = 1
|
||||||
|
}
|
||||||
|
if isConsonant {
|
||||||
|
consoCount = ppMaxRepeatedConsonant
|
||||||
|
}
|
||||||
|
|
||||||
|
segmentLen := 1
|
||||||
|
|
||||||
|
segmentLenTarget := ppSegmentLenMin + ppRandInt(rng, ppSegmentLenMax-ppSegmentLenMin)
|
||||||
|
|
||||||
|
for len(result) < pwlen {
|
||||||
|
|
||||||
|
charset := ppContinuation[currentChar]
|
||||||
|
if vowelCount >= ppMaxRepeatedVowel {
|
||||||
|
charset = ppCharsetRemove(charset, ppVowelMap, false)
|
||||||
|
}
|
||||||
|
if consoCount >= ppMaxRepeatedConsonant {
|
||||||
|
charset = ppCharsetRemove(charset, ppConsonantMap, false)
|
||||||
|
}
|
||||||
|
|
||||||
|
lastOfSegment := false
|
||||||
|
newSegment := false
|
||||||
|
|
||||||
|
if len(result)+1 == pwlen {
|
||||||
|
// last of result
|
||||||
|
charset = ppCharsetFilter(charset, ppEndCharMap, false)
|
||||||
|
} else if segmentLen+1 == segmentLenTarget {
|
||||||
|
// last of segment
|
||||||
|
charsetNew := ppCharsetFilter(charset, ppEndCharMap, true)
|
||||||
|
if charsetNew != "" {
|
||||||
|
charset = charsetNew
|
||||||
|
lastOfSegment = true
|
||||||
|
}
|
||||||
|
} else if segmentLen >= segmentLenTarget {
|
||||||
|
// (perhaps) start of new segment
|
||||||
|
if _, ok := ppEndCharMap[currentChar]; ok {
|
||||||
|
charset = ppStartChar
|
||||||
|
newSegment = true
|
||||||
|
} else {
|
||||||
|
// continue segment for one more char to (hopefully) find an end-char
|
||||||
|
charsetNew := ppCharsetFilter(charset, ppEndCharMap, true)
|
||||||
|
if charsetNew != "" {
|
||||||
|
charset = charsetNew
|
||||||
|
lastOfSegment = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// normal continuation
|
||||||
|
}
|
||||||
|
|
||||||
|
newChar := ppRand(rng, charset, &entropy)
|
||||||
|
if lastOfSegment {
|
||||||
|
currentChar = newChar
|
||||||
|
segmentLen++
|
||||||
|
result += strings.ToLower(string(newChar))
|
||||||
|
} else if newSegment {
|
||||||
|
currentChar = newChar
|
||||||
|
segmentLen = 1
|
||||||
|
result += strings.ToUpper(string(newChar))
|
||||||
|
segmentLenTarget = ppSegmentLenMin + ppRandInt(rng, ppSegmentLenMax-ppSegmentLenMin)
|
||||||
|
vowelCount = 0
|
||||||
|
consoCount = 0
|
||||||
|
} else {
|
||||||
|
currentChar = newChar
|
||||||
|
segmentLen++
|
||||||
|
result += strings.ToLower(string(newChar))
|
||||||
|
}
|
||||||
|
|
||||||
|
isVowel, isConsonant := ppCharType(currentChar)
|
||||||
|
if isVowel {
|
||||||
|
vowelCount++
|
||||||
|
consoCount = 0
|
||||||
|
}
|
||||||
|
if isConsonant {
|
||||||
|
vowelCount = 0
|
||||||
|
if newSegment {
|
||||||
|
consoCount = ppMaxRepeatedConsonant
|
||||||
|
} else {
|
||||||
|
consoCount++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return result, entropy
|
||||||
|
}
|
||||||
|
|
||||||
|
func PronouncablePassword(len int) string {
|
||||||
|
v, _ := PronouncablePasswordExt(rand.Reader, len)
|
||||||
|
return v
|
||||||
|
}
|
||||||
|
|
||||||
|
func PronouncablePasswordSeeded(seed int64, len int) string {
|
||||||
|
|
||||||
|
v, _ := PronouncablePasswordExt(mathrand.New(mathrand.NewSource(seed)), len)
|
||||||
|
return v
|
||||||
|
}
|
35
cryptext/pronouncablePassword_test.go
Normal file
35
cryptext/pronouncablePassword_test.go
Normal file
@@ -0,0 +1,35 @@
|
|||||||
|
package cryptext
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"math/rand"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestPronouncablePasswordExt(t *testing.T) {
|
||||||
|
for i := 0; i < 20; i++ {
|
||||||
|
pw, entropy := PronouncablePasswordExt(rand.New(rand.NewSource(int64(i))), 16)
|
||||||
|
fmt.Printf("[%.2f] => %s\n", entropy, pw)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestPronouncablePasswordSeeded(t *testing.T) {
|
||||||
|
for i := 0; i < 20; i++ {
|
||||||
|
pw := PronouncablePasswordSeeded(int64(i), 8)
|
||||||
|
fmt.Printf("%s\n", pw)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestPronouncablePassword(t *testing.T) {
|
||||||
|
for i := 0; i < 20; i++ {
|
||||||
|
pw := PronouncablePassword(i + 1)
|
||||||
|
fmt.Printf("%s\n", pw)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestPronouncablePasswordWrongLen(t *testing.T) {
|
||||||
|
PronouncablePassword(0)
|
||||||
|
PronouncablePassword(-1)
|
||||||
|
PronouncablePassword(-2)
|
||||||
|
PronouncablePassword(-3)
|
||||||
|
}
|
@@ -50,6 +50,7 @@ var (
|
|||||||
|
|
||||||
TypeSQLQuery = NewType("SQL_QUERY", langext.Ptr(500))
|
TypeSQLQuery = NewType("SQL_QUERY", langext.Ptr(500))
|
||||||
TypeSQLBuild = NewType("SQL_BUILD", langext.Ptr(500))
|
TypeSQLBuild = NewType("SQL_BUILD", langext.Ptr(500))
|
||||||
|
TypeSQLDecode = NewType("SQL_DECODE", langext.Ptr(500))
|
||||||
|
|
||||||
TypeWrap = NewType("Wrap", nil)
|
TypeWrap = NewType("Wrap", nil)
|
||||||
|
|
||||||
|
2
go.mod
2
go.mod
@@ -23,7 +23,7 @@ require (
|
|||||||
github.com/gin-contrib/sse v0.1.0 // indirect
|
github.com/gin-contrib/sse v0.1.0 // indirect
|
||||||
github.com/go-playground/locales v0.14.1 // indirect
|
github.com/go-playground/locales v0.14.1 // indirect
|
||||||
github.com/go-playground/universal-translator v0.18.1 // indirect
|
github.com/go-playground/universal-translator v0.18.1 // indirect
|
||||||
github.com/go-playground/validator/v10 v10.16.0 // indirect
|
github.com/go-playground/validator/v10 v10.17.0 // indirect
|
||||||
github.com/goccy/go-json v0.10.2 // indirect
|
github.com/goccy/go-json v0.10.2 // indirect
|
||||||
github.com/golang/snappy v0.0.4 // indirect
|
github.com/golang/snappy v0.0.4 // indirect
|
||||||
github.com/google/uuid v1.5.0 // indirect
|
github.com/google/uuid v1.5.0 // indirect
|
||||||
|
2
go.sum
2
go.sum
@@ -31,6 +31,8 @@ github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJn
|
|||||||
github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY=
|
github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY=
|
||||||
github.com/go-playground/validator/v10 v10.16.0 h1:x+plE831WK4vaKHO/jpgUGsvLKIqRRkz6M78GuJAfGE=
|
github.com/go-playground/validator/v10 v10.16.0 h1:x+plE831WK4vaKHO/jpgUGsvLKIqRRkz6M78GuJAfGE=
|
||||||
github.com/go-playground/validator/v10 v10.16.0/go.mod h1:9iXMNT7sEkjXb0I+enO7QXmzG6QCsPWY4zveKFVRSyU=
|
github.com/go-playground/validator/v10 v10.16.0/go.mod h1:9iXMNT7sEkjXb0I+enO7QXmzG6QCsPWY4zveKFVRSyU=
|
||||||
|
github.com/go-playground/validator/v10 v10.17.0 h1:SmVVlfAOtlZncTxRuinDPomC2DkXJ4E5T9gDA0AIH74=
|
||||||
|
github.com/go-playground/validator/v10 v10.17.0/go.mod h1:9iXMNT7sEkjXb0I+enO7QXmzG6QCsPWY4zveKFVRSyU=
|
||||||
github.com/go-sql-driver/mysql v1.6.0 h1:BCTh4TKNUYmOmMUcQ3IipzF5prigylS7XXjEkfCHuOE=
|
github.com/go-sql-driver/mysql v1.6.0 h1:BCTh4TKNUYmOmMUcQ3IipzF5prigylS7XXjEkfCHuOE=
|
||||||
github.com/go-sql-driver/mysql v1.6.0/go.mod h1:DCzpHaOWr8IXmIStZouvnhqoel9Qv2LBy8hT2VhHyBg=
|
github.com/go-sql-driver/mysql v1.6.0/go.mod h1:DCzpHaOWr8IXmIStZouvnhqoel9Qv2LBy8hT2VhHyBg=
|
||||||
github.com/goccy/go-json v0.10.2 h1:CrxCmQqYDkv1z7lO7Wbh2HN93uovUHgrECaO5ZrCXAU=
|
github.com/goccy/go-json v0.10.2 h1:CrxCmQqYDkv1z7lO7Wbh2HN93uovUHgrECaO5ZrCXAU=
|
||||||
|
@@ -1,5 +1,5 @@
|
|||||||
package goext
|
package goext
|
||||||
|
|
||||||
const GoextVersion = "0.0.371"
|
const GoextVersion = "0.0.377"
|
||||||
|
|
||||||
const GoextVersionTimestamp = "2024-01-13T14:19:19+0100"
|
const GoextVersionTimestamp = "2024-01-14T17:06:42+0100"
|
||||||
|
@@ -5,7 +5,7 @@ import (
|
|||||||
"go.mongodb.org/mongo-driver/mongo"
|
"go.mongodb.org/mongo-driver/mongo"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Filter interface {
|
type MongoFilter interface {
|
||||||
FilterQuery() mongo.Pipeline
|
FilterQuery() mongo.Pipeline
|
||||||
Sort() bson.D
|
Sort() bson.D
|
||||||
}
|
}
|
||||||
@@ -23,6 +23,6 @@ func (d dynamicFilter) Sort() bson.D {
|
|||||||
return d.sort
|
return d.sort
|
||||||
}
|
}
|
||||||
|
|
||||||
func CreateFilter(pipeline mongo.Pipeline, sort bson.D) Filter {
|
func CreateFilter(pipeline mongo.Pipeline, sort bson.D) MongoFilter {
|
||||||
return dynamicFilter{pipeline: pipeline, sort: sort}
|
return dynamicFilter{pipeline: pipeline, sort: sort}
|
||||||
}
|
}
|
||||||
|
@@ -69,3 +69,52 @@ func BuildUpdateStatement(q Queryable, tableName string, obj any, idColumn strin
|
|||||||
//goland:noinspection SqlNoDataSourceInspection
|
//goland:noinspection SqlNoDataSourceInspection
|
||||||
return fmt.Sprintf("UPDATE %s SET %s WHERE %s", tableName, strings.Join(setClauses, ", "), matchClause), params, nil
|
return fmt.Sprintf("UPDATE %s SET %s WHERE %s", tableName, strings.Join(setClauses, ", "), matchClause), params, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func BuildInsertStatement(q Queryable, tableName string, obj any) (string, PP, error) {
|
||||||
|
rval := reflect.ValueOf(obj)
|
||||||
|
rtyp := rval.Type()
|
||||||
|
|
||||||
|
params := PP{}
|
||||||
|
|
||||||
|
fields := make([]string, 0)
|
||||||
|
values := make([]string, 0)
|
||||||
|
|
||||||
|
for i := 0; i < rtyp.NumField(); i++ {
|
||||||
|
|
||||||
|
rsfield := rtyp.Field(i)
|
||||||
|
rvfield := rval.Field(i)
|
||||||
|
|
||||||
|
if !rsfield.IsExported() {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
columnName := rsfield.Tag.Get("db")
|
||||||
|
if columnName == "" || columnName == "-" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if rsfield.Type.Kind() == reflect.Ptr && rvfield.IsNil() {
|
||||||
|
|
||||||
|
fields = append(fields, columnName)
|
||||||
|
values = append(values, "NULL")
|
||||||
|
|
||||||
|
} else {
|
||||||
|
|
||||||
|
val, err := convertValueToDB(q, rvfield.Interface())
|
||||||
|
if err != nil {
|
||||||
|
return "", nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
fields = append(fields, columnName)
|
||||||
|
values = append(values, ":"+params.Add(val))
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(fields) == 0 {
|
||||||
|
return "", nil, exerr.New(exerr.TypeSQLBuild, "no fields found in object").Build()
|
||||||
|
}
|
||||||
|
|
||||||
|
//goland:noinspection SqlNoDataSourceInspection
|
||||||
|
return fmt.Sprintf("INSERT INTO %s (%s) VALUES (%s)", tableName, strings.Join(fields, ", "), strings.Join(values, ", ")), params, nil
|
||||||
|
}
|
||||||
|
@@ -18,9 +18,9 @@ type DBTypeConverter interface {
|
|||||||
DBToModel(v any) (any, error)
|
DBToModel(v any) (any, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
var ConverterBoolToBit = NewDBTypeConverter[bool, int](func(v bool) (int, error) {
|
var ConverterBoolToBit = NewDBTypeConverter[bool, int64](func(v bool) (int64, error) {
|
||||||
return langext.Conditional(v, 1, 0), nil
|
return langext.Conditional(v, int64(1), int64(0)), nil
|
||||||
}, func(v int) (bool, error) {
|
}, func(v int64) (bool, error) {
|
||||||
if v == 0 {
|
if v == 0 {
|
||||||
return false, nil
|
return false, nil
|
||||||
}
|
}
|
||||||
|
126
sq/paginate.go
Normal file
126
sq/paginate.go
Normal file
@@ -0,0 +1,126 @@
|
|||||||
|
package sq
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
ct "gogs.mikescher.com/BlackForestBytes/goext/cursortoken"
|
||||||
|
"gogs.mikescher.com/BlackForestBytes/goext/exerr"
|
||||||
|
"gogs.mikescher.com/BlackForestBytes/goext/langext"
|
||||||
|
pag "gogs.mikescher.com/BlackForestBytes/goext/pagination"
|
||||||
|
)
|
||||||
|
|
||||||
|
type PaginateFilter interface {
|
||||||
|
SQL(params PP) (filterClause string, joinClause string, joinTables []string)
|
||||||
|
Sort() []FilterSort
|
||||||
|
}
|
||||||
|
|
||||||
|
type FilterSort struct {
|
||||||
|
Field string
|
||||||
|
Direction ct.SortDirection
|
||||||
|
}
|
||||||
|
|
||||||
|
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) {
|
||||||
|
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
|
||||||
|
sqlQueryCount := "SELECT " + "COUNT(*)" + " FROM " + table + " " + joinCond + " WHERE ( " + filterCond + " ) "
|
||||||
|
|
||||||
|
rows, err := q.Query(ctx, sqlQueryData, prepParams)
|
||||||
|
if err != nil {
|
||||||
|
return nil, pag.Pagination{}, exerr.Wrap(err, "failed to list paginated entries from DB").Str("table", table).Any("filter", filter).Int("page", page).Any("limit", limit).Build()
|
||||||
|
}
|
||||||
|
|
||||||
|
entities, err := ScanAll[TData](ctx, q, rows, scanMode, scanSec, true)
|
||||||
|
if err != nil {
|
||||||
|
return nil, pag.Pagination{}, exerr.Wrap(err, "failed to decode paginated entries from DB").Str("table", table).Int("page", page).Any("limit", limit).Str("scanMode", string(scanMode)).Str("scanSec", string(scanSec)).Build()
|
||||||
|
}
|
||||||
|
|
||||||
|
if page == 1 && (limit == nil || len(entities) <= *limit) {
|
||||||
|
return entities, pag.Pagination{
|
||||||
|
Page: 1,
|
||||||
|
Limit: langext.Coalesce(limit, len(entities)),
|
||||||
|
TotalPages: 1,
|
||||||
|
TotalItems: len(entities),
|
||||||
|
CurrentPageCount: 1,
|
||||||
|
}, nil
|
||||||
|
} else {
|
||||||
|
|
||||||
|
countRows, err := q.Query(ctx, sqlQueryCount, prepParams)
|
||||||
|
if err != nil {
|
||||||
|
return nil, pag.Pagination{}, exerr.Wrap(err, "failed to query total-count of paginated entries from DB").Str("table", table).Build()
|
||||||
|
}
|
||||||
|
|
||||||
|
if !countRows.Next() {
|
||||||
|
return nil, pag.Pagination{}, exerr.New(exerr.TypeSQLDecode, "SQL COUNT(*) query returned no rows").Str("table", table).Any("filter", filter).Build()
|
||||||
|
}
|
||||||
|
|
||||||
|
var countRes int
|
||||||
|
err = countRows.Scan(&countRes)
|
||||||
|
if err != nil {
|
||||||
|
return nil, pag.Pagination{}, exerr.Wrap(err, "failed to decode total-count of paginated entries from DB").Str("table", table).Build()
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(entities) > *limit {
|
||||||
|
entities = entities[:*limit]
|
||||||
|
}
|
||||||
|
|
||||||
|
paginationObj := pag.Pagination{
|
||||||
|
Page: page,
|
||||||
|
Limit: langext.Coalesce(limit, countRes),
|
||||||
|
TotalPages: pag.CalcPaginationTotalPages(countRes, langext.Coalesce(limit, countRes)),
|
||||||
|
TotalItems: countRes,
|
||||||
|
CurrentPageCount: len(entities),
|
||||||
|
}
|
||||||
|
|
||||||
|
return entities, paginationObj, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func Count(ctx context.Context, q Queryable, table string, filter PaginateFilter) (int, error) {
|
||||||
|
prepParams := PP{}
|
||||||
|
|
||||||
|
filterCond, joinCond, _ := filter.SQL(prepParams)
|
||||||
|
|
||||||
|
sqlQueryCount := "SELECT " + "COUNT(*)" + " FROM " + table + " " + joinCond + " WHERE ( " + filterCond + " )"
|
||||||
|
|
||||||
|
countRows, err := q.Query(ctx, sqlQueryCount, prepParams)
|
||||||
|
if err != nil {
|
||||||
|
return 0, exerr.Wrap(err, "failed to query count of entries from DB").Str("table", table).Build()
|
||||||
|
}
|
||||||
|
|
||||||
|
if !countRows.Next() {
|
||||||
|
return 0, exerr.New(exerr.TypeSQLDecode, "SQL COUNT(*) query returned no rows").Str("table", table).Any("filter", filter).Build()
|
||||||
|
}
|
||||||
|
|
||||||
|
var countRes int
|
||||||
|
err = countRows.Scan(&countRes)
|
||||||
|
if err != nil {
|
||||||
|
return 0, exerr.Wrap(err, "failed to decode count of entries from DB").Str("table", table).Build()
|
||||||
|
}
|
||||||
|
|
||||||
|
return countRes, nil
|
||||||
|
}
|
@@ -9,7 +9,7 @@ import (
|
|||||||
pag "gogs.mikescher.com/BlackForestBytes/goext/pagination"
|
pag "gogs.mikescher.com/BlackForestBytes/goext/pagination"
|
||||||
)
|
)
|
||||||
|
|
||||||
func (c *Coll[TData]) Paginate(ctx context.Context, filter pag.Filter, page int, limit *int) ([]TData, pag.Pagination, error) {
|
func (c *Coll[TData]) Paginate(ctx context.Context, filter pag.MongoFilter, page int, limit *int) ([]TData, pag.Pagination, error) {
|
||||||
type totalCountResult struct {
|
type totalCountResult struct {
|
||||||
Count int `bson:"count"`
|
Count int `bson:"count"`
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user