Compare commits
	
		
			11 Commits
		
	
	
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| fff609db4a | |||
| 5e99e07f40 | |||
| bdb181cb3a | |||
| 3552acd38b | |||
| c42324c58f | |||
| 3a9c3f4e9e | |||
| becd8f1ebc | |||
| e733f30c38 | |||
| 1a9e5c70fc | |||
| f3700a772d | |||
| 2c69b33547 | 
							
								
								
									
										136
									
								
								confext/confParser.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										136
									
								
								confext/confParser.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,136 @@ | ||||
| package confext | ||||
|  | ||||
| import ( | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"gogs.mikescher.com/BlackForestBytes/goext/timeext" | ||||
| 	"math/bits" | ||||
| 	"os" | ||||
| 	"reflect" | ||||
| 	"strconv" | ||||
| 	"time" | ||||
| ) | ||||
|  | ||||
| // ApplyEnvOverrides overrides field values from environment variables | ||||
| // | ||||
| // fields must be tagged with `env:"env_key"` | ||||
| func ApplyEnvOverrides[T any](c *T) error { | ||||
|  | ||||
| 	rval := reflect.ValueOf(c).Elem() | ||||
| 	rtyp := rval.Type() | ||||
|  | ||||
| 	for i := 0; i < rtyp.NumField(); i++ { | ||||
|  | ||||
| 		rsfield := rtyp.Field(i) | ||||
| 		rvfield := rval.Field(i) | ||||
|  | ||||
| 		envkey := rsfield.Tag.Get("env") | ||||
| 		if envkey == "" { | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		envval, efound := os.LookupEnv(envkey) | ||||
| 		if !efound { | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		if rvfield.Type() == reflect.TypeOf("") { | ||||
|  | ||||
| 			rvfield.Set(reflect.ValueOf(envval)) | ||||
|  | ||||
| 			fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", envkey, envval) | ||||
|  | ||||
| 		} else if rvfield.Type() == reflect.TypeOf(int(0)) { | ||||
|  | ||||
| 			envint, err := strconv.ParseInt(envval, 10, bits.UintSize) | ||||
| 			if err != nil { | ||||
| 				return errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to int (value := '%s')", envkey, envval)) | ||||
| 			} | ||||
|  | ||||
| 			rvfield.Set(reflect.ValueOf(int(envint))) | ||||
|  | ||||
| 			fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", envkey, envval) | ||||
|  | ||||
| 		} else if rvfield.Type() == reflect.TypeOf(int64(0)) { | ||||
|  | ||||
| 			envint, err := strconv.ParseInt(envval, 10, 64) | ||||
| 			if err != nil { | ||||
| 				return errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to int64 (value := '%s')", envkey, envval)) | ||||
| 			} | ||||
|  | ||||
| 			rvfield.Set(reflect.ValueOf(int64(envint))) | ||||
|  | ||||
| 			fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", envkey, envval) | ||||
|  | ||||
| 		} else if rvfield.Type() == reflect.TypeOf(int32(0)) { | ||||
|  | ||||
| 			envint, err := strconv.ParseInt(envval, 10, 32) | ||||
| 			if err != nil { | ||||
| 				return errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to int32 (value := '%s')", envkey, envval)) | ||||
| 			} | ||||
|  | ||||
| 			rvfield.Set(reflect.ValueOf(int32(envint))) | ||||
|  | ||||
| 			fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", envkey, envval) | ||||
|  | ||||
| 		} else if rvfield.Type() == reflect.TypeOf(int8(0)) { | ||||
|  | ||||
| 			envint, err := strconv.ParseInt(envval, 10, 8) | ||||
| 			if err != nil { | ||||
| 				return errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to int32 (value := '%s')", envkey, envval)) | ||||
| 			} | ||||
|  | ||||
| 			rvfield.Set(reflect.ValueOf(int8(envint))) | ||||
|  | ||||
| 			fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", envkey, envval) | ||||
|  | ||||
| 		} else if rvfield.Type() == reflect.TypeOf(time.Duration(0)) { | ||||
|  | ||||
| 			dur, err := timeext.ParseDurationShortString(envval) | ||||
| 			if err != nil { | ||||
| 				return errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to duration (value := '%s')", envkey, envval)) | ||||
| 			} | ||||
|  | ||||
| 			rvfield.Set(reflect.ValueOf(dur)) | ||||
|  | ||||
| 			fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", envkey, dur.String()) | ||||
|  | ||||
| 		} else if rvfield.Type() == reflect.TypeOf(time.UnixMilli(0)) { | ||||
|  | ||||
| 			tim, err := time.Parse(time.RFC3339Nano, envval) | ||||
| 			if err != nil { | ||||
| 				return errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to time.time (value := '%s')", envkey, envval)) | ||||
| 			} | ||||
|  | ||||
| 			rvfield.Set(reflect.ValueOf(tim)) | ||||
|  | ||||
| 			fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", envkey, tim.String()) | ||||
|  | ||||
| 		} else if rvfield.Type().ConvertibleTo(reflect.TypeOf(int(0))) { | ||||
|  | ||||
| 			envint, err := strconv.ParseInt(envval, 10, 8) | ||||
| 			if err != nil { | ||||
| 				return errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to <%s, ,int> (value := '%s')", rvfield.Type().Name(), envkey, envval)) | ||||
| 			} | ||||
|  | ||||
| 			envcvl := reflect.ValueOf(envint).Convert(rvfield.Type()) | ||||
|  | ||||
| 			rvfield.Set(envcvl) | ||||
|  | ||||
| 			fmt.Printf("[CONF] Overwrite config '%s' with '%v'\n", envkey, envcvl.Interface()) | ||||
|  | ||||
| 		} else if rvfield.Type().ConvertibleTo(reflect.TypeOf("")) { | ||||
|  | ||||
| 			envcvl := reflect.ValueOf(envval).Convert(rvfield.Type()) | ||||
|  | ||||
| 			rvfield.Set(envcvl) | ||||
|  | ||||
| 			fmt.Printf("[CONF] Overwrite config '%s' with '%v'\n", envkey, envcvl.Interface()) | ||||
|  | ||||
| 		} else { | ||||
| 			return errors.New(fmt.Sprintf("Unknown kind/type in config: [ %s | %s ]", rvfield.Kind().String(), rvfield.Type().String())) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
							
								
								
									
										116
									
								
								confext/confParser_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										116
									
								
								confext/confParser_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,116 @@ | ||||
| package confext | ||||
|  | ||||
| import ( | ||||
| 	"testing" | ||||
| 	"time" | ||||
| ) | ||||
|  | ||||
| func TestApplyEnvOverridesNoop(t *testing.T) { | ||||
|  | ||||
| 	type aliasint int | ||||
| 	type aliasstring string | ||||
|  | ||||
| 	type testdata struct { | ||||
| 		V1 int           `env:"TEST_V1"` | ||||
| 		VX string        `` | ||||
| 		V2 string        `env:"TEST_V2"` | ||||
| 		V3 int8          `env:"TEST_V3"` | ||||
| 		V4 int32         `env:"TEST_V4"` | ||||
| 		V5 int64         `env:"TEST_V5"` | ||||
| 		V6 aliasint      `env:"TEST_V6"` | ||||
| 		VY aliasint      `` | ||||
| 		V7 aliasstring   `env:"TEST_V7"` | ||||
| 		V8 time.Duration `env:"TEST_V8"` | ||||
| 		V9 time.Time     `env:"TEST_V9"` | ||||
| 	} | ||||
|  | ||||
| 	input := testdata{ | ||||
| 		V1: 1, | ||||
| 		VX: "X", | ||||
| 		V2: "2", | ||||
| 		V3: 3, | ||||
| 		V4: 4, | ||||
| 		V5: 5, | ||||
| 		V6: 6, | ||||
| 		VY: 99, | ||||
| 		V7: "7", | ||||
| 		V8: 9, | ||||
| 		V9: time.Unix(1671102873, 0), | ||||
| 	} | ||||
|  | ||||
| 	output := input | ||||
|  | ||||
| 	err := ApplyEnvOverrides(&output) | ||||
| 	if err != nil { | ||||
| 		t.Errorf("%v", err) | ||||
| 		t.FailNow() | ||||
| 	} | ||||
|  | ||||
| 	assertEqual(t, input, output) | ||||
| } | ||||
|  | ||||
| func TestApplyEnvOverridesSimple(t *testing.T) { | ||||
|  | ||||
| 	type aliasint int | ||||
| 	type aliasstring string | ||||
|  | ||||
| 	type testdata struct { | ||||
| 		V1 int           `env:"TEST_V1"` | ||||
| 		VX string        `` | ||||
| 		V2 string        `env:"TEST_V2"` | ||||
| 		V3 int8          `env:"TEST_V3"` | ||||
| 		V4 int32         `env:"TEST_V4"` | ||||
| 		V5 int64         `env:"TEST_V5"` | ||||
| 		V6 aliasint      `env:"TEST_V6"` | ||||
| 		VY aliasint      `` | ||||
| 		V7 aliasstring   `env:"TEST_V7"` | ||||
| 		V8 time.Duration `env:"TEST_V8"` | ||||
| 		V9 time.Time     `env:"TEST_V9"` | ||||
| 	} | ||||
|  | ||||
| 	data := testdata{ | ||||
| 		V1: 1, | ||||
| 		VX: "X", | ||||
| 		V2: "2", | ||||
| 		V3: 3, | ||||
| 		V4: 4, | ||||
| 		V5: 5, | ||||
| 		V6: 6, | ||||
| 		VY: 99, | ||||
| 		V7: "7", | ||||
| 		V8: 9, | ||||
| 		V9: time.Unix(1671102873, 0), | ||||
| 	} | ||||
|  | ||||
| 	t.Setenv("TEST_V1", "846") | ||||
| 	t.Setenv("TEST_V2", "hello_world") | ||||
| 	t.Setenv("TEST_V3", "6") | ||||
| 	t.Setenv("TEST_V4", "333") | ||||
| 	t.Setenv("TEST_V5", "-937") | ||||
| 	t.Setenv("TEST_V6", "070") | ||||
| 	t.Setenv("TEST_V7", "AAAAAA") | ||||
| 	t.Setenv("TEST_V8", "1min4s") | ||||
| 	t.Setenv("TEST_V9", "2009-11-10T23:00:00Z") | ||||
|  | ||||
| 	err := ApplyEnvOverrides(&data) | ||||
| 	if err != nil { | ||||
| 		t.Errorf("%v", err) | ||||
| 		t.FailNow() | ||||
| 	} | ||||
|  | ||||
| 	assertEqual(t, data.V1, 846) | ||||
| 	assertEqual(t, data.V2, "hello_world") | ||||
| 	assertEqual(t, data.V3, 6) | ||||
| 	assertEqual(t, data.V4, 333) | ||||
| 	assertEqual(t, data.V5, -937) | ||||
| 	assertEqual(t, data.V6, 70) | ||||
| 	assertEqual(t, data.V7, "AAAAAA") | ||||
| 	assertEqual(t, data.V8, time.Second*64) | ||||
| 	assertEqual(t, data.V9, time.Unix(1257894000, 0).UTC()) | ||||
| } | ||||
|  | ||||
| func assertEqual[T comparable](t *testing.T, actual T, expected T) { | ||||
| 	if actual != expected { | ||||
| 		t.Errorf("values differ: Actual: '%v', Expected: '%v'", actual, expected) | ||||
| 	} | ||||
| } | ||||
| @@ -1,56 +1,157 @@ | ||||
| package dataext | ||||
|  | ||||
| import "io" | ||||
| import ( | ||||
| 	"errors" | ||||
| 	"io" | ||||
| ) | ||||
|  | ||||
| type brcMode int | ||||
|  | ||||
| const ( | ||||
| 	modeSourceReading  = 0 | ||||
| 	modeSourceFinished = 1 | ||||
| 	modeBufferReading  = 2 | ||||
| 	modeBufferFinished = 3 | ||||
| ) | ||||
|  | ||||
| type BufferedReadCloser interface { | ||||
| 	io.ReadCloser | ||||
| 	BufferedAll() ([]byte, error) | ||||
| 	Reset() error | ||||
| } | ||||
|  | ||||
| type bufferedReadCloser struct { | ||||
| 	buffer []byte | ||||
| 	inner  io.ReadCloser | ||||
| 	finished bool | ||||
| } | ||||
|  | ||||
| func (b *bufferedReadCloser) Read(p []byte) (int, error) { | ||||
|  | ||||
| 	n, err := b.inner.Read(p) | ||||
| 	if n > 0 { | ||||
| 		b.buffer = append(b.buffer, p[0:n]...) | ||||
| 	} | ||||
|  | ||||
| 	if err == io.EOF { | ||||
| 		b.finished = true | ||||
| 	} | ||||
|  | ||||
| 	return n, err | ||||
| 	mode   brcMode | ||||
| 	off    int | ||||
| } | ||||
|  | ||||
| func NewBufferedReadCloser(sub io.ReadCloser) BufferedReadCloser { | ||||
| 	return &bufferedReadCloser{ | ||||
| 		buffer: make([]byte, 0, 1024), | ||||
| 		inner:  sub, | ||||
| 		finished: false, | ||||
| 		mode:   modeSourceReading, | ||||
| 		off:    0, | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *bufferedReadCloser) Read(p []byte) (int, error) { | ||||
| 	switch b.mode { | ||||
| 	case modeSourceReading: | ||||
| 		n, err := b.inner.Read(p) | ||||
| 		if n > 0 { | ||||
| 			b.buffer = append(b.buffer, p[0:n]...) | ||||
| 		} | ||||
|  | ||||
| 		if err == io.EOF { | ||||
| 			b.mode = modeSourceFinished | ||||
| 		} | ||||
|  | ||||
| 		return n, err | ||||
|  | ||||
| 	case modeSourceFinished: | ||||
| 		return 0, io.EOF | ||||
|  | ||||
| 	case modeBufferReading: | ||||
|  | ||||
| 		if len(b.buffer) <= b.off { | ||||
| 			b.mode = modeBufferFinished | ||||
| 			if len(p) == 0 { | ||||
| 				return 0, nil | ||||
| 			} | ||||
| 			return 0, io.EOF | ||||
| 		} | ||||
|  | ||||
| 		n := copy(p, b.buffer[b.off:]) | ||||
| 		b.off += n | ||||
| 		return n, nil | ||||
|  | ||||
| 	case modeBufferFinished: | ||||
| 		return 0, io.EOF | ||||
|  | ||||
| 	default: | ||||
| 		return 0, errors.New("object in undefined status") | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *bufferedReadCloser) Close() error { | ||||
| 	err := b.inner.Close() | ||||
| 	switch b.mode { | ||||
| 	case modeSourceReading: | ||||
| 		_, err := b.BufferedAll() | ||||
| 		if err != nil { | ||||
| 		b.finished = true | ||||
| 	} | ||||
| 			return err | ||||
| 		} | ||||
| 		err = b.inner.Close() | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		b.mode = modeSourceFinished | ||||
| 		return nil | ||||
|  | ||||
| 	case modeSourceFinished: | ||||
| 		return nil | ||||
|  | ||||
| 	case modeBufferReading: | ||||
| 		b.mode = modeBufferFinished | ||||
| 		return nil | ||||
|  | ||||
| 	case modeBufferFinished: | ||||
| 		return nil | ||||
|  | ||||
| 	default: | ||||
| 		return errors.New("object in undefined status") | ||||
| 	} | ||||
|  | ||||
| } | ||||
|  | ||||
| func (b *bufferedReadCloser) BufferedAll() ([]byte, error) { | ||||
| 	switch b.mode { | ||||
| 	case modeSourceReading: | ||||
| 		arr := make([]byte, 1024) | ||||
| 	for !b.finished { | ||||
| 		for b.mode == modeSourceReading { | ||||
| 			_, err := b.Read(arr) | ||||
| 			if err != nil && err != io.EOF { | ||||
| 				return nil, err | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		return b.buffer, nil | ||||
|  | ||||
| 	case modeSourceFinished: | ||||
| 		return b.buffer, nil | ||||
|  | ||||
| 	case modeBufferReading: | ||||
| 		return b.buffer, nil | ||||
|  | ||||
| 	case modeBufferFinished: | ||||
| 		return b.buffer, nil | ||||
|  | ||||
| 	default: | ||||
| 		return nil, errors.New("object in undefined status") | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *bufferedReadCloser) Reset() error { | ||||
| 	switch b.mode { | ||||
| 	case modeSourceReading: | ||||
| 		fallthrough | ||||
| 	case modeSourceFinished: | ||||
| 		err := b.Close() | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		b.mode = modeBufferReading | ||||
| 		b.off = 0 | ||||
| 		return nil | ||||
|  | ||||
| 	case modeBufferReading: | ||||
| 		fallthrough | ||||
| 	case modeBufferFinished: | ||||
| 		b.mode = modeBufferReading | ||||
| 		b.off = 0 | ||||
| 		return nil | ||||
|  | ||||
| 	default: | ||||
| 		return errors.New("object in undefined status") | ||||
| 	} | ||||
| } | ||||
|   | ||||
							
								
								
									
										254
									
								
								dataext/structHash.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										254
									
								
								dataext/structHash.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,254 @@ | ||||
| package dataext | ||||
|  | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"crypto/sha256" | ||||
| 	"encoding/binary" | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"gogs.mikescher.com/BlackForestBytes/goext/langext" | ||||
| 	"hash" | ||||
| 	"io" | ||||
| 	"reflect" | ||||
| 	"sort" | ||||
| ) | ||||
|  | ||||
| type StructHashOptions struct { | ||||
| 	HashAlgo    hash.Hash | ||||
| 	Tag         *string | ||||
| 	SkipChannel bool | ||||
| 	SkipFunc    bool | ||||
| } | ||||
|  | ||||
| func StructHash(dat any, opt ...StructHashOptions) (r []byte, err error) { | ||||
| 	defer func() { | ||||
| 		if rec := recover(); rec != nil { | ||||
| 			r = nil | ||||
| 			err = errors.New(fmt.Sprintf("recovered panic: %v", rec)) | ||||
| 		} | ||||
| 	}() | ||||
|  | ||||
| 	shopt := StructHashOptions{} | ||||
| 	if len(opt) > 1 { | ||||
| 		return nil, errors.New("multiple options supplied") | ||||
| 	} else if len(opt) == 1 { | ||||
| 		shopt = opt[0] | ||||
| 	} | ||||
|  | ||||
| 	if shopt.HashAlgo == nil { | ||||
| 		shopt.HashAlgo = sha256.New() | ||||
| 	} | ||||
|  | ||||
| 	writer := new(bytes.Buffer) | ||||
|  | ||||
| 	if langext.IsNil(dat) { | ||||
| 		shopt.HashAlgo.Reset() | ||||
| 		shopt.HashAlgo.Write(writer.Bytes()) | ||||
| 		res := shopt.HashAlgo.Sum(nil) | ||||
| 		return res, nil | ||||
| 	} | ||||
|  | ||||
| 	err = binarize(writer, reflect.ValueOf(dat), shopt) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
|  | ||||
| 	shopt.HashAlgo.Reset() | ||||
| 	shopt.HashAlgo.Write(writer.Bytes()) | ||||
| 	res := shopt.HashAlgo.Sum(nil) | ||||
|  | ||||
| 	return res, nil | ||||
| } | ||||
|  | ||||
| func writeBinarized(writer io.Writer, dat any) error { | ||||
| 	tmp := bytes.Buffer{} | ||||
| 	err := binary.Write(&tmp, binary.LittleEndian, dat) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	err = binary.Write(writer, binary.LittleEndian, uint64(tmp.Len())) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	_, err = writer.Write(tmp.Bytes()) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func binarize(writer io.Writer, dat reflect.Value, opt StructHashOptions) error { | ||||
| 	var err error | ||||
|  | ||||
| 	err = binary.Write(writer, binary.LittleEndian, uint8(dat.Kind())) | ||||
| 	switch dat.Kind() { | ||||
| 	case reflect.Ptr, reflect.Map, reflect.Array, reflect.Chan, reflect.Slice, reflect.Interface: | ||||
| 		if dat.IsNil() { | ||||
| 			err = binary.Write(writer, binary.LittleEndian, uint64(0)) | ||||
| 			if err != nil { | ||||
| 				return err | ||||
| 			} | ||||
| 			return nil | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	err = binary.Write(writer, binary.LittleEndian, uint64(len(dat.Type().String()))) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	_, err = writer.Write([]byte(dat.Type().String())) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	switch dat.Type().Kind() { | ||||
| 	case reflect.Invalid: | ||||
| 		return errors.New("cannot binarize value of kind <Invalid>") | ||||
| 	case reflect.Bool: | ||||
| 		return writeBinarized(writer, dat.Bool()) | ||||
| 	case reflect.Int: | ||||
| 		return writeBinarized(writer, int64(dat.Int())) | ||||
| 	case reflect.Int8: | ||||
| 		fallthrough | ||||
| 	case reflect.Int16: | ||||
| 		fallthrough | ||||
| 	case reflect.Int32: | ||||
| 		fallthrough | ||||
| 	case reflect.Int64: | ||||
| 		return writeBinarized(writer, dat.Interface()) | ||||
| 	case reflect.Uint: | ||||
| 		return writeBinarized(writer, uint64(dat.Int())) | ||||
| 	case reflect.Uint8: | ||||
| 		fallthrough | ||||
| 	case reflect.Uint16: | ||||
| 		fallthrough | ||||
| 	case reflect.Uint32: | ||||
| 		fallthrough | ||||
| 	case reflect.Uint64: | ||||
| 		return writeBinarized(writer, dat.Interface()) | ||||
| 	case reflect.Uintptr: | ||||
| 		return errors.New("cannot binarize value of kind <Uintptr>") | ||||
| 	case reflect.Float32: | ||||
| 		fallthrough | ||||
| 	case reflect.Float64: | ||||
| 		return writeBinarized(writer, dat.Interface()) | ||||
| 	case reflect.Complex64: | ||||
| 		return errors.New("cannot binarize value of kind <Complex64>") | ||||
| 	case reflect.Complex128: | ||||
| 		return errors.New("cannot binarize value of kind <Complex128>") | ||||
| 	case reflect.Slice: | ||||
| 		fallthrough | ||||
| 	case reflect.Array: | ||||
| 		return binarizeArrayOrSlice(writer, dat, opt) | ||||
| 	case reflect.Chan: | ||||
| 		if opt.SkipChannel { | ||||
| 			return nil | ||||
| 		} | ||||
| 		return errors.New("cannot binarize value of kind <Chan>") | ||||
| 	case reflect.Func: | ||||
| 		if opt.SkipFunc { | ||||
| 			return nil | ||||
| 		} | ||||
| 		return errors.New("cannot binarize value of kind <Func>") | ||||
| 	case reflect.Interface: | ||||
| 		return binarize(writer, dat.Elem(), opt) | ||||
| 	case reflect.Map: | ||||
| 		return binarizeMap(writer, dat, opt) | ||||
| 	case reflect.Pointer: | ||||
| 		return binarize(writer, dat.Elem(), opt) | ||||
| 	case reflect.String: | ||||
| 		v := dat.String() | ||||
| 		err = binary.Write(writer, binary.LittleEndian, uint64(len(v))) | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		_, err = writer.Write([]byte(v)) | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		return nil | ||||
| 	case reflect.Struct: | ||||
| 		return binarizeStruct(writer, dat, opt) | ||||
| 	case reflect.UnsafePointer: | ||||
| 		return errors.New("cannot binarize value of kind <UnsafePointer>") | ||||
| 	default: | ||||
| 		return errors.New("cannot binarize value of unknown kind <" + dat.Type().Kind().String() + ">") | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func binarizeStruct(writer io.Writer, dat reflect.Value, opt StructHashOptions) error { | ||||
| 	err := binary.Write(writer, binary.LittleEndian, uint64(dat.NumField())) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	for i := 0; i < dat.NumField(); i++ { | ||||
|  | ||||
| 		if opt.Tag != nil { | ||||
| 			if _, ok := dat.Type().Field(i).Tag.Lookup(*opt.Tag); !ok { | ||||
| 				continue | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		err = binary.Write(writer, binary.LittleEndian, uint64(len(dat.Type().Field(i).Name))) | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		_, err = writer.Write([]byte(dat.Type().Field(i).Name)) | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
|  | ||||
| 		err = binarize(writer, dat.Field(i), opt) | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func binarizeArrayOrSlice(writer io.Writer, dat reflect.Value, opt StructHashOptions) error { | ||||
| 	err := binary.Write(writer, binary.LittleEndian, uint64(dat.Len())) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	for i := 0; i < dat.Len(); i++ { | ||||
| 		err := binarize(writer, dat.Index(i), opt) | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func binarizeMap(writer io.Writer, dat reflect.Value, opt StructHashOptions) error { | ||||
| 	err := binary.Write(writer, binary.LittleEndian, uint64(dat.Len())) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	sub := make([][]byte, 0, dat.Len()) | ||||
|  | ||||
| 	for _, k := range dat.MapKeys() { | ||||
| 		tmp := bytes.Buffer{} | ||||
| 		err = binarize(&tmp, dat.MapIndex(k), opt) | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		sub = append(sub, tmp.Bytes()) | ||||
| 	} | ||||
|  | ||||
| 	sort.Slice(sub, func(i1, i2 int) bool { return bytes.Compare(sub[i1], sub[i2]) < 0 }) | ||||
|  | ||||
| 	for _, v := range sub { | ||||
| 		_, err = writer.Write(v) | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
							
								
								
									
										143
									
								
								dataext/structHash_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										143
									
								
								dataext/structHash_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,143 @@ | ||||
| package dataext | ||||
|  | ||||
| import ( | ||||
| 	"encoding/hex" | ||||
| 	"gogs.mikescher.com/BlackForestBytes/goext/langext" | ||||
| 	"testing" | ||||
| ) | ||||
|  | ||||
| func noErrStructHash(t *testing.T, dat any, opt ...StructHashOptions) []byte { | ||||
| 	res, err := StructHash(dat, opt...) | ||||
| 	if err != nil { | ||||
| 		t.Error(err) | ||||
| 		t.FailNow() | ||||
| 		return nil | ||||
| 	} | ||||
| 	return res | ||||
| } | ||||
|  | ||||
| func TestStructHashSimple(t *testing.T) { | ||||
|  | ||||
| 	assertEqual(t, "209bf774af36cc3a045c152d9f1269ef3684ad819c1359ee73ff0283a308fefa", noErrStructHash(t, "Hello")) | ||||
| 	assertEqual(t, "c32f3626b981ae2997db656f3acad3f1dc9d30ef6b6d14296c023e391b25f71a", noErrStructHash(t, 0)) | ||||
| 	assertEqual(t, "01b781b03e9586b257d387057dfc70d9f06051e7d3c1e709a57e13cc8daf3e35", noErrStructHash(t, []byte{})) | ||||
| 	assertEqual(t, "93e1dcd45c732fe0079b0fb3204c7c803f0921835f6bfee2e6ff263e73eed53c", noErrStructHash(t, []int{})) | ||||
| 	assertEqual(t, "54f637a376aad55b3160d98ebbcae8099b70d91b9400df23fb3709855d59800a", noErrStructHash(t, []int{1, 2, 3})) | ||||
| 	assertEqual(t, "e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855", noErrStructHash(t, nil)) | ||||
| 	assertEqual(t, "349a7db91aa78fd30bbaa7c7f9c7bfb2fcfe72869b4861162a96713a852f60d3", noErrStructHash(t, []any{1, "", nil})) | ||||
| 	assertEqual(t, "ca51aab87808bf0062a4a024de6aac0c2bad54275cc857a4944569f89fd245ad", noErrStructHash(t, struct{}{})) | ||||
|  | ||||
| } | ||||
|  | ||||
| func TestStructHashSimpleStruct(t *testing.T) { | ||||
|  | ||||
| 	type t0 struct { | ||||
| 		F1 int | ||||
| 		F2 []string | ||||
| 		F3 *int | ||||
| 	} | ||||
|  | ||||
| 	assertEqual(t, "a90bff751c70c738bb5cfc9b108e783fa9c19c0bc9273458e0aaee6e74aa1b92", noErrStructHash(t, t0{ | ||||
| 		F1: 10, | ||||
| 		F2: []string{"1", "2", "3"}, | ||||
| 		F3: nil, | ||||
| 	})) | ||||
|  | ||||
| 	assertEqual(t, "5d09090dc34ac59dd645f197a255f653387723de3afa1b614721ea5a081c675f", noErrStructHash(t, t0{ | ||||
| 		F1: 10, | ||||
| 		F2: []string{"1", "2", "3"}, | ||||
| 		F3: langext.Ptr(99), | ||||
| 	})) | ||||
|  | ||||
| } | ||||
|  | ||||
| func TestStructHashLayeredStruct(t *testing.T) { | ||||
|  | ||||
| 	type t1_1 struct { | ||||
| 		F10 float32 | ||||
| 		F12 float64 | ||||
| 		F15 bool | ||||
| 	} | ||||
| 	type t1_2 struct { | ||||
| 		SV1 *t1_1 | ||||
| 		SV2 *t1_1 | ||||
| 		SV3 t1_1 | ||||
| 	} | ||||
|  | ||||
| 	assertEqual(t, "fd4ca071fb40a288fee4b7a3dfdaab577b30cb8f80f81ec511e7afd72dc3b469", noErrStructHash(t, t1_2{ | ||||
| 		SV1: nil, | ||||
| 		SV2: nil, | ||||
| 		SV3: t1_1{ | ||||
| 			F10: 1, | ||||
| 			F12: 2, | ||||
| 			F15: false, | ||||
| 		}, | ||||
| 	})) | ||||
| 	assertEqual(t, "3fbf7c67d8121deda075cc86319a4e32d71744feb2cebf89b43bc682f072a029", noErrStructHash(t, t1_2{ | ||||
| 		SV1: nil, | ||||
| 		SV2: &t1_1{}, | ||||
| 		SV3: t1_1{ | ||||
| 			F10: 3, | ||||
| 			F12: 4, | ||||
| 			F15: true, | ||||
| 		}, | ||||
| 	})) | ||||
| 	assertEqual(t, "b1791ccd1b346c3ede5bbffda85555adcd8216b93ffca23f14fe175ec47c5104", noErrStructHash(t, t1_2{ | ||||
| 		SV1: &t1_1{}, | ||||
| 		SV2: &t1_1{}, | ||||
| 		SV3: t1_1{ | ||||
| 			F10: 5, | ||||
| 			F12: 6, | ||||
| 			F15: false, | ||||
| 		}, | ||||
| 	})) | ||||
|  | ||||
| } | ||||
|  | ||||
| func TestStructHashMap(t *testing.T) { | ||||
|  | ||||
| 	type t0 struct { | ||||
| 		F1 int | ||||
| 		F2 map[string]int | ||||
| 	} | ||||
|  | ||||
| 	assertEqual(t, "d50c53ad1fafb448c33fddd5aca01a86a2edf669ce2ecab07ba6fe877951d824", noErrStructHash(t, t0{ | ||||
| 		F1: 10, | ||||
| 		F2: map[string]int{ | ||||
| 			"x": 1, | ||||
| 			"0": 2, | ||||
| 			"a": 99, | ||||
| 		}, | ||||
| 	})) | ||||
|  | ||||
| 	assertEqual(t, "d50c53ad1fafb448c33fddd5aca01a86a2edf669ce2ecab07ba6fe877951d824", noErrStructHash(t, t0{ | ||||
| 		F1: 10, | ||||
| 		F2: map[string]int{ | ||||
| 			"a": 99, | ||||
| 			"x": 1, | ||||
| 			"0": 2, | ||||
| 		}, | ||||
| 	})) | ||||
|  | ||||
| 	m3 := make(map[string]int, 99) | ||||
| 	m3["a"] = 0 | ||||
| 	m3["x"] = 0 | ||||
| 	m3["0"] = 0 | ||||
|  | ||||
| 	m3["0"] = 99 | ||||
| 	m3["x"] = 1 | ||||
| 	m3["a"] = 2 | ||||
|  | ||||
| 	assertEqual(t, "d50c53ad1fafb448c33fddd5aca01a86a2edf669ce2ecab07ba6fe877951d824", noErrStructHash(t, t0{ | ||||
| 		F1: 10, | ||||
| 		F2: m3, | ||||
| 	})) | ||||
|  | ||||
| } | ||||
|  | ||||
| func assertEqual(t *testing.T, expected string, actual []byte) { | ||||
| 	actualStr := hex.EncodeToString(actual) | ||||
| 	if actualStr != expected { | ||||
| 		t.Errorf("values differ: Actual: '%v', Expected: '%v'", actualStr, expected) | ||||
| 	} | ||||
| } | ||||
							
								
								
									
										2
									
								
								go.mod
									
									
									
									
									
								
							
							
						
						
									
										2
									
								
								go.mod
									
									
									
									
									
								
							| @@ -6,3 +6,5 @@ require ( | ||||
| 	golang.org/x/sys v0.1.0 | ||||
| 	golang.org/x/term v0.1.0 | ||||
| ) | ||||
|  | ||||
| require github.com/jmoiron/sqlx v1.3.5 // indirect | ||||
|   | ||||
							
								
								
									
										5
									
								
								go.sum
									
									
									
									
									
								
							
							
						
						
									
										5
									
								
								go.sum
									
									
									
									
									
								
							| @@ -1,3 +1,8 @@ | ||||
| github.com/go-sql-driver/mysql v1.6.0/go.mod h1:DCzpHaOWr8IXmIStZouvnhqoel9Qv2LBy8hT2VhHyBg= | ||||
| github.com/jmoiron/sqlx v1.3.5 h1:vFFPA71p1o5gAeqtEAwLU4dnX2napprKtHr7PYIcN3g= | ||||
| github.com/jmoiron/sqlx v1.3.5/go.mod h1:nRVWtLre0KfCLJvgxzCsLVMogSvQ1zNJtpYr2Ccp0mQ= | ||||
| github.com/lib/pq v1.2.0/go.mod h1:5WUZQaWbwv1U+lTReE5YruASi9Al49XbQIvNi/34Woo= | ||||
| github.com/mattn/go-sqlite3 v1.14.6/go.mod h1:NyWgC/yNuGj7Q9rpYnZvas74GogHl5/Z4A/KQRfk6bU= | ||||
| golang.org/x/sys v0.1.0 h1:kunALQeHf1/185U1i0GOB/fy1IPRDDpuoOOqRReG57U= | ||||
| golang.org/x/sys v0.1.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | ||||
| golang.org/x/term v0.1.0 h1:g6Z6vPFA9dYBAF7DWcH6sCcOntplXsDKcliusYijMlw= | ||||
|   | ||||
| @@ -70,7 +70,73 @@ func ArrEqualsExact[T comparable](arr1 []T, arr2 []T) bool { | ||||
| 	return true | ||||
| } | ||||
|  | ||||
| func ArrAll(arr interface{}, fn func(int) bool) bool { | ||||
| func ArrAll[T any](arr []T, fn func(T) bool) bool { | ||||
| 	for _, av := range arr { | ||||
| 		if !fn(av) { | ||||
| 			return false | ||||
| 		} | ||||
| 	} | ||||
| 	return true | ||||
| } | ||||
|  | ||||
| func ArrAllErr[T any](arr []T, fn func(T) (bool, error)) (bool, error) { | ||||
| 	for _, av := range arr { | ||||
| 		v, err := fn(av) | ||||
| 		if err != nil { | ||||
| 			return false, err | ||||
| 		} | ||||
| 		if !v { | ||||
| 			return false, nil | ||||
| 		} | ||||
| 	} | ||||
| 	return true, nil | ||||
| } | ||||
|  | ||||
| func ArrNone[T any](arr []T, fn func(T) bool) bool { | ||||
| 	for _, av := range arr { | ||||
| 		if fn(av) { | ||||
| 			return false | ||||
| 		} | ||||
| 	} | ||||
| 	return true | ||||
| } | ||||
|  | ||||
| func ArrNoneErr[T any](arr []T, fn func(T) (bool, error)) (bool, error) { | ||||
| 	for _, av := range arr { | ||||
| 		v, err := fn(av) | ||||
| 		if err != nil { | ||||
| 			return false, err | ||||
| 		} | ||||
| 		if v { | ||||
| 			return false, nil | ||||
| 		} | ||||
| 	} | ||||
| 	return true, nil | ||||
| } | ||||
|  | ||||
| func ArrAny[T any](arr []T, fn func(T) bool) bool { | ||||
| 	for _, av := range arr { | ||||
| 		if fn(av) { | ||||
| 			return true | ||||
| 		} | ||||
| 	} | ||||
| 	return false | ||||
| } | ||||
|  | ||||
| func ArrAnyErr[T any](arr []T, fn func(T) (bool, error)) (bool, error) { | ||||
| 	for _, av := range arr { | ||||
| 		v, err := fn(av) | ||||
| 		if err != nil { | ||||
| 			return false, err | ||||
| 		} | ||||
| 		if v { | ||||
| 			return true, nil | ||||
| 		} | ||||
| 	} | ||||
| 	return false, nil | ||||
| } | ||||
|  | ||||
| func ArrIdxAll(arr any, fn func(int) bool) bool { | ||||
| 	av := reflect.ValueOf(arr) | ||||
| 	for i := 0; i < av.Len(); i++ { | ||||
| 		if !fn(i) { | ||||
| @@ -80,7 +146,7 @@ func ArrAll(arr interface{}, fn func(int) bool) bool { | ||||
| 	return true | ||||
| } | ||||
|  | ||||
| func ArrAllErr(arr interface{}, fn func(int) (bool, error)) (bool, error) { | ||||
| func ArrIdxAllErr(arr any, fn func(int) (bool, error)) (bool, error) { | ||||
| 	av := reflect.ValueOf(arr) | ||||
| 	for i := 0; i < av.Len(); i++ { | ||||
| 		v, err := fn(i) | ||||
| @@ -94,7 +160,7 @@ func ArrAllErr(arr interface{}, fn func(int) (bool, error)) (bool, error) { | ||||
| 	return true, nil | ||||
| } | ||||
|  | ||||
| func ArrNone(arr interface{}, fn func(int) bool) bool { | ||||
| func ArrIdxNone(arr any, fn func(int) bool) bool { | ||||
| 	av := reflect.ValueOf(arr) | ||||
| 	for i := 0; i < av.Len(); i++ { | ||||
| 		if fn(i) { | ||||
| @@ -104,7 +170,7 @@ func ArrNone(arr interface{}, fn func(int) bool) bool { | ||||
| 	return true | ||||
| } | ||||
|  | ||||
| func ArrNoneErr(arr interface{}, fn func(int) (bool, error)) (bool, error) { | ||||
| func ArrIdxNoneErr(arr any, fn func(int) (bool, error)) (bool, error) { | ||||
| 	av := reflect.ValueOf(arr) | ||||
| 	for i := 0; i < av.Len(); i++ { | ||||
| 		v, err := fn(i) | ||||
| @@ -118,7 +184,7 @@ func ArrNoneErr(arr interface{}, fn func(int) (bool, error)) (bool, error) { | ||||
| 	return true, nil | ||||
| } | ||||
|  | ||||
| func ArrAny(arr interface{}, fn func(int) bool) bool { | ||||
| func ArrIdxAny(arr any, fn func(int) bool) bool { | ||||
| 	av := reflect.ValueOf(arr) | ||||
| 	for i := 0; i < av.Len(); i++ { | ||||
| 		if fn(i) { | ||||
| @@ -128,7 +194,7 @@ func ArrAny(arr interface{}, fn func(int) bool) bool { | ||||
| 	return false | ||||
| } | ||||
|  | ||||
| func ArrAnyErr(arr interface{}, fn func(int) (bool, error)) (bool, error) { | ||||
| func ArrIdxAnyErr(arr any, fn func(int) (bool, error)) (bool, error) { | ||||
| 	av := reflect.ValueOf(arr) | ||||
| 	for i := 0; i < av.Len(); i++ { | ||||
| 		v, err := fn(i) | ||||
| @@ -142,7 +208,7 @@ func ArrAnyErr(arr interface{}, fn func(int) (bool, error)) (bool, error) { | ||||
| 	return false, nil | ||||
| } | ||||
|  | ||||
| func ArrFirst[T comparable](arr []T, comp func(v T) bool) (T, bool) { | ||||
| func ArrFirst[T any](arr []T, comp func(v T) bool) (T, bool) { | ||||
| 	for _, v := range arr { | ||||
| 		if comp(v) { | ||||
| 			return v, true | ||||
| @@ -151,7 +217,7 @@ func ArrFirst[T comparable](arr []T, comp func(v T) bool) (T, bool) { | ||||
| 	return *new(T), false | ||||
| } | ||||
|  | ||||
| func ArrLast[T comparable](arr []T, comp func(v T) bool) (T, bool) { | ||||
| func ArrLast[T any](arr []T, comp func(v T) bool) (T, bool) { | ||||
| 	found := false | ||||
| 	result := *new(T) | ||||
| 	for _, v := range arr { | ||||
|   | ||||
							
								
								
									
										39
									
								
								langext/sort.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										39
									
								
								langext/sort.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,39 @@ | ||||
| package langext | ||||
|  | ||||
| import "sort" | ||||
|  | ||||
| func Sort[T OrderedConstraint](arr []T) { | ||||
| 	sort.Slice(arr, func(i1, i2 int) bool { | ||||
| 		return arr[i1] < arr[i2] | ||||
| 	}) | ||||
| } | ||||
|  | ||||
| func SortStable[T OrderedConstraint](arr []T) { | ||||
| 	sort.SliceStable(arr, func(i1, i2 int) bool { | ||||
| 		return arr[i1] < arr[i2] | ||||
| 	}) | ||||
| } | ||||
|  | ||||
| func IsSorted[T OrderedConstraint](arr []T) bool { | ||||
| 	return sort.SliceIsSorted(arr, func(i1, i2 int) bool { | ||||
| 		return arr[i1] < arr[i2] | ||||
| 	}) | ||||
| } | ||||
|  | ||||
| func SortSlice[T any](arr []T, less func(v1, v2 T) bool) { | ||||
| 	sort.Slice(arr, func(i1, i2 int) bool { | ||||
| 		return less(arr[i1], arr[i2]) | ||||
| 	}) | ||||
| } | ||||
|  | ||||
| func SortSliceStable[T any](arr []T, less func(v1, v2 T) bool) { | ||||
| 	sort.SliceStable(arr, func(i1, i2 int) bool { | ||||
| 		return less(arr[i1], arr[i2]) | ||||
| 	}) | ||||
| } | ||||
|  | ||||
| func IsSliceSorted[T any](arr []T, less func(v1, v2 T) bool) bool { | ||||
| 	return sort.SliceIsSorted(arr, func(i1, i2 int) bool { | ||||
| 		return less(arr[i1], arr[i2]) | ||||
| 	}) | ||||
| } | ||||
| @@ -107,3 +107,11 @@ func NumToStringOpt[V IntConstraint](v *V, fallback string) string { | ||||
| 		return fmt.Sprintf("%d", v) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func StrRepeat(val string, count int) string { | ||||
| 	r := "" | ||||
| 	for i := 0; i < count; i++ { | ||||
| 		r += val | ||||
| 	} | ||||
| 	return r | ||||
| } | ||||
|   | ||||
							
								
								
									
										89
									
								
								sq/database.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										89
									
								
								sq/database.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,89 @@ | ||||
| package sq | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| 	"database/sql" | ||||
| 	"github.com/jmoiron/sqlx" | ||||
| 	"sync" | ||||
| ) | ||||
|  | ||||
| type DB interface { | ||||
| 	Exec(ctx context.Context, sql string, prep PP) (sql.Result, error) | ||||
| 	Query(ctx context.Context, sql string, prep PP) (*sqlx.Rows, error) | ||||
| 	Ping(ctx context.Context) error | ||||
| 	BeginTransaction(ctx context.Context, iso sql.IsolationLevel) (Tx, error) | ||||
| 	SetListener(listener Listener) | ||||
| } | ||||
|  | ||||
| type database struct { | ||||
| 	db    *sqlx.DB | ||||
| 	txctr uint16 | ||||
| 	lock  sync.Mutex | ||||
| 	lstr  Listener | ||||
| } | ||||
|  | ||||
| func NewDB(db *sqlx.DB) DB { | ||||
| 	return &database{ | ||||
| 		db:    db, | ||||
| 		txctr: 0, | ||||
| 		lock:  sync.Mutex{}, | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (db *database) SetListener(listener Listener) { | ||||
| 	db.lstr = listener | ||||
| } | ||||
|  | ||||
| func (db *database) Exec(ctx context.Context, sql string, prep PP) (sql.Result, error) { | ||||
| 	if db.lstr != nil { | ||||
| 		db.lstr.OnExec(nil, &sql, &prep) | ||||
| 	} | ||||
|  | ||||
| 	res, err := db.db.NamedExecContext(ctx, sql, prep) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| 	return res, nil | ||||
| } | ||||
|  | ||||
| func (db *database) Query(ctx context.Context, sql string, prep PP) (*sqlx.Rows, error) { | ||||
| 	if db.lstr != nil { | ||||
| 		db.lstr.OnQuery(nil, &sql, &prep) | ||||
| 	} | ||||
|  | ||||
| 	rows, err := db.db.NamedQueryContext(ctx, sql, prep) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| 	return rows, nil | ||||
| } | ||||
|  | ||||
| func (db *database) Ping(ctx context.Context) error { | ||||
| 	if db.lstr != nil { | ||||
| 		db.lstr.OnPing() | ||||
| 	} | ||||
|  | ||||
| 	err := db.db.PingContext(ctx) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (db *database) BeginTransaction(ctx context.Context, iso sql.IsolationLevel) (Tx, error) { | ||||
| 	db.lock.Lock() | ||||
| 	txid := db.txctr | ||||
| 	db.txctr += 1 // with overflow ! | ||||
| 	db.lock.Unlock() | ||||
|  | ||||
| 	if db.lstr != nil { | ||||
| 		db.lstr.OnTxBegin(txid) | ||||
| 	} | ||||
|  | ||||
| 	xtx, err := db.db.BeginTxx(ctx, &sql.TxOptions{Isolation: iso}) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
|  | ||||
| 	return NewTransaction(xtx, txid, db.lstr), nil | ||||
| } | ||||
							
								
								
									
										10
									
								
								sq/listener.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										10
									
								
								sq/listener.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,10 @@ | ||||
| package sq | ||||
|  | ||||
| type Listener interface { | ||||
| 	OnQuery(txID *uint16, sql *string, params *PP) | ||||
| 	OnExec(txID *uint16, sql *string, params *PP) | ||||
| 	OnPing() | ||||
| 	OnTxBegin(txid uint16) | ||||
| 	OnTxCommit(txid uint16) | ||||
| 	OnTxRollback(txid uint16) | ||||
| } | ||||
							
								
								
									
										3
									
								
								sq/params.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										3
									
								
								sq/params.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,3 @@ | ||||
| package sq | ||||
|  | ||||
| type PP map[string]any | ||||
							
								
								
									
										12
									
								
								sq/queryable.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										12
									
								
								sq/queryable.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,12 @@ | ||||
| package sq | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| 	"database/sql" | ||||
| 	"github.com/jmoiron/sqlx" | ||||
| ) | ||||
|  | ||||
| type Queryable interface { | ||||
| 	Exec(ctx context.Context, sql string, prep PP) (sql.Result, error) | ||||
| 	Query(ctx context.Context, sql string, prep PP) (*sqlx.Rows, error) | ||||
| } | ||||
							
								
								
									
										52
									
								
								sq/scanner.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										52
									
								
								sq/scanner.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,52 @@ | ||||
| package sq | ||||
|  | ||||
| import ( | ||||
| 	"database/sql" | ||||
| 	"errors" | ||||
| 	"github.com/jmoiron/sqlx" | ||||
| ) | ||||
|  | ||||
| func ScanSingle[TData any](rows *sqlx.Rows, close bool) (TData, error) { | ||||
| 	if rows.Next() { | ||||
| 		var data TData | ||||
| 		err := rows.StructScan(&data) | ||||
| 		if err != nil { | ||||
| 			return *new(TData), err | ||||
| 		} | ||||
| 		if rows.Next() { | ||||
| 			_ = rows.Close() | ||||
| 			return *new(TData), errors.New("sql returned more than onw row") | ||||
| 		} | ||||
| 		if close { | ||||
| 			err = rows.Close() | ||||
| 			if err != nil { | ||||
| 				return *new(TData), err | ||||
| 			} | ||||
| 		} | ||||
| 		return data, nil | ||||
| 	} else { | ||||
| 		if close { | ||||
| 			_ = rows.Close() | ||||
| 		} | ||||
| 		return *new(TData), sql.ErrNoRows | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func ScanAll[TData any](rows *sqlx.Rows, close bool) ([]TData, error) { | ||||
| 	res := make([]TData, 0) | ||||
| 	for rows.Next() { | ||||
| 		var data TData | ||||
| 		err := rows.StructScan(&data) | ||||
| 		if err != nil { | ||||
| 			return nil, err | ||||
| 		} | ||||
| 		res = append(res, data) | ||||
| 	} | ||||
| 	if close { | ||||
| 		err := rows.Close() | ||||
| 		if err != nil { | ||||
| 			return nil, err | ||||
| 		} | ||||
| 	} | ||||
| 	return res, nil | ||||
| } | ||||
							
								
								
									
										69
									
								
								sq/transaction.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										69
									
								
								sq/transaction.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,69 @@ | ||||
| package sq | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| 	"database/sql" | ||||
| 	"github.com/jmoiron/sqlx" | ||||
| 	"gogs.mikescher.com/BlackForestBytes/goext/langext" | ||||
| ) | ||||
|  | ||||
| type Tx interface { | ||||
| 	Rollback() error | ||||
| 	Commit() error | ||||
| 	Exec(ctx context.Context, sql string, prep PP) (sql.Result, error) | ||||
| 	Query(ctx context.Context, sql string, prep PP) (*sqlx.Rows, error) | ||||
| } | ||||
|  | ||||
| type transaction struct { | ||||
| 	tx   *sqlx.Tx | ||||
| 	id   uint16 | ||||
| 	lstr Listener | ||||
| } | ||||
|  | ||||
| func NewTransaction(xtx *sqlx.Tx, txid uint16, lstr Listener) Tx { | ||||
| 	return &transaction{ | ||||
| 		tx:   xtx, | ||||
| 		id:   txid, | ||||
| 		lstr: lstr, | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (tx *transaction) Rollback() error { | ||||
| 	if tx.lstr != nil { | ||||
| 		tx.lstr.OnTxRollback(tx.id) | ||||
| 	} | ||||
|  | ||||
| 	return tx.tx.Rollback() | ||||
| } | ||||
|  | ||||
| func (tx *transaction) Commit() error { | ||||
| 	if tx.lstr != nil { | ||||
| 		tx.lstr.OnTxCommit(tx.id) | ||||
| 	} | ||||
|  | ||||
| 	return tx.tx.Commit() | ||||
| } | ||||
|  | ||||
| func (tx *transaction) Exec(ctx context.Context, sql string, prep PP) (sql.Result, error) { | ||||
| 	if tx.lstr != nil { | ||||
| 		tx.lstr.OnExec(langext.Ptr(tx.id), &sql, &prep) | ||||
| 	} | ||||
|  | ||||
| 	res, err := tx.tx.NamedExecContext(ctx, sql, prep) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| 	return res, nil | ||||
| } | ||||
|  | ||||
| func (tx *transaction) Query(ctx context.Context, sql string, prep PP) (*sqlx.Rows, error) { | ||||
| 	if tx.lstr != nil { | ||||
| 		tx.lstr.OnQuery(langext.Ptr(tx.id), &sql, &prep) | ||||
| 	} | ||||
|  | ||||
| 	rows, err := sqlx.NamedQueryContext(ctx, tx.tx, sql, prep) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| 	return rows, nil | ||||
| } | ||||
							
								
								
									
										152
									
								
								timeext/parser.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										152
									
								
								timeext/parser.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,152 @@ | ||||
| package timeext | ||||
|  | ||||
| import ( | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"strconv" | ||||
| 	"strings" | ||||
| 	"time" | ||||
| ) | ||||
|  | ||||
| var durationShortStringMap = map[string]time.Duration{ | ||||
| 	"ns":          time.Nanosecond, | ||||
| 	"nanosecond":  time.Nanosecond, | ||||
| 	"nanoseconds": time.Nanosecond, | ||||
|  | ||||
| 	"us":           time.Microsecond, | ||||
| 	"microsecond":  time.Microsecond, | ||||
| 	"microseconds": time.Microsecond, | ||||
|  | ||||
| 	"ms":           time.Millisecond, | ||||
| 	"millisecond":  time.Millisecond, | ||||
| 	"milliseconds": time.Millisecond, | ||||
|  | ||||
| 	"s":       time.Second, | ||||
| 	"sec":     time.Second, | ||||
| 	"second":  time.Second, | ||||
| 	"seconds": time.Second, | ||||
|  | ||||
| 	"m":       time.Minute, | ||||
| 	"min":     time.Minute, | ||||
| 	"minute":  time.Minute, | ||||
| 	"minutes": time.Minute, | ||||
|  | ||||
| 	"h":     time.Hour, | ||||
| 	"hour":  time.Hour, | ||||
| 	"hours": time.Hour, | ||||
|  | ||||
| 	"d":    24 * time.Hour, | ||||
| 	"day":  24 * time.Hour, | ||||
| 	"days": 24 * time.Hour, | ||||
|  | ||||
| 	"w":     7 * 24 * time.Hour, | ||||
| 	"wk":    7 * 24 * time.Hour, | ||||
| 	"week":  7 * 24 * time.Hour, | ||||
| 	"weeks": 7 * 24 * time.Hour, | ||||
| } | ||||
|  | ||||
| // ParseDurationShortString parses a duration in string format to a time.Duration | ||||
| // Examples for allowed formats: | ||||
| //   - '10m' | ||||
| //   - '10min' | ||||
| //   - '1minute' | ||||
| //   - '10minutes' | ||||
| //   - '10.5minutes' | ||||
| //   - '50s' | ||||
| //   - '50sec' | ||||
| //   - '1second' | ||||
| //   - '50seconds' | ||||
| //   - '100ms' | ||||
| //   - '100millisseconds' | ||||
| //   - '1h' | ||||
| //   - '1hour' | ||||
| //   - '2hours' | ||||
| //   - '1d' | ||||
| //   - '1day' | ||||
| //   - '10days' | ||||
| //   - '1d10m' | ||||
| //   - '1d10m200sec' | ||||
| //   - '1d:10m' | ||||
| //   - '1d 10m' | ||||
| //   - '1d,10m' | ||||
| func ParseDurationShortString(s string) (time.Duration, error) { | ||||
| 	s = strings.ToLower(s) | ||||
|  | ||||
| 	segments := make([]string, 0) | ||||
| 	collector := "" | ||||
|  | ||||
| 	prevWasNum := true | ||||
| 	for _, chr := range s { | ||||
| 		if chr >= '0' && chr <= '9' || chr == '.' { | ||||
| 			if prevWasNum { | ||||
| 				collector += string(chr) | ||||
| 			} else { | ||||
| 				segments = append(segments, collector) | ||||
| 				prevWasNum = true | ||||
| 				collector = string(chr) | ||||
| 			} | ||||
| 		} else if chr == ' ' || chr == ':' || chr == ',' { | ||||
| 			continue | ||||
| 		} else if chr >= 'a' && chr <= 'z' { | ||||
| 			prevWasNum = false | ||||
| 			collector += string(chr) | ||||
| 		} else { | ||||
| 			return 0, errors.New("unexpected character: " + string(chr)) | ||||
| 		} | ||||
| 	} | ||||
| 	if !prevWasNum { | ||||
| 		segments = append(segments, collector) | ||||
| 	} | ||||
|  | ||||
| 	result := time.Duration(0) | ||||
| 	for _, seg := range segments { | ||||
| 		segDur, err := parseDurationShortStringSegment(seg) | ||||
| 		if err != nil { | ||||
| 			return 0, err | ||||
| 		} | ||||
| 		result += segDur | ||||
| 	} | ||||
| 	return result, nil | ||||
| } | ||||
|  | ||||
| func parseDurationShortStringSegment(segment string) (time.Duration, error) { | ||||
| 	num := "" | ||||
| 	unit := "" | ||||
|  | ||||
| 	part0 := true | ||||
| 	for _, chr := range segment { | ||||
| 		if part0 { | ||||
|  | ||||
| 			if chr >= 'a' && chr <= 'z' { | ||||
| 				part0 = false | ||||
| 				unit += string(chr) | ||||
| 			} else if chr >= '0' && chr <= '9' || chr == '.' { | ||||
| 				num += string(chr) | ||||
| 			} else { | ||||
| 				return 0, errors.New(fmt.Sprintf("Unexpected character '%s' in segment [%s]", string(chr), segment)) | ||||
| 			} | ||||
|  | ||||
| 		} else { | ||||
|  | ||||
| 			if chr >= 'a' && chr <= 'z' { | ||||
| 				unit += string(chr) | ||||
| 			} else if chr >= '0' && chr <= '9' || chr == '.' { | ||||
| 				return 0, errors.New(fmt.Sprintf("Unexpected number '%s' in segment [%s]", string(chr), segment)) | ||||
| 			} else { | ||||
| 				return 0, errors.New(fmt.Sprintf("Unexpected character '%s' in segment [%s]", string(chr), segment)) | ||||
| 			} | ||||
|  | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	fpnum, err := strconv.ParseFloat(num, 64) | ||||
| 	if err != nil { | ||||
| 		return 0, errors.New(fmt.Sprintf("Failed to parse floating-point number '%s' in segment [%s]", num, segment)) | ||||
| 	} | ||||
|  | ||||
| 	if mult, ok := durationShortStringMap[unit]; ok { | ||||
| 		return time.Duration(int64(fpnum * float64(mult))), nil | ||||
| 	} else { | ||||
| 		return 0, errors.New(fmt.Sprintf("Unknown unit '%s' in segment [%s]", unit, segment)) | ||||
| 	} | ||||
| } | ||||
							
								
								
									
										70
									
								
								timeext/parser_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										70
									
								
								timeext/parser_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,70 @@ | ||||
| package timeext | ||||
|  | ||||
| import ( | ||||
| 	"testing" | ||||
| 	"time" | ||||
| ) | ||||
|  | ||||
| func TestParseDurationShortString(t *testing.T) { | ||||
|  | ||||
| 	assertPDSSEqual(t, FromSeconds(1), "1s") | ||||
| 	assertPDSSEqual(t, FromSeconds(1), "1sec") | ||||
| 	assertPDSSEqual(t, FromSeconds(1), "1second") | ||||
| 	assertPDSSEqual(t, FromSeconds(1), "1seconds") | ||||
| 	assertPDSSEqual(t, FromSeconds(100), "100second") | ||||
| 	assertPDSSEqual(t, FromSeconds(100), "100seconds") | ||||
| 	assertPDSSEqual(t, FromSeconds(1883639.77), "1883639.77second") | ||||
| 	assertPDSSEqual(t, FromSeconds(1883639.77), "1883639.77seconds") | ||||
| 	assertPDSSEqual(t, FromSeconds(50), "50s") | ||||
| 	assertPDSSEqual(t, FromSeconds(50), "50sec") | ||||
| 	assertPDSSEqual(t, FromSeconds(1), "1second") | ||||
| 	assertPDSSEqual(t, FromSeconds(50), "50seconds") | ||||
|  | ||||
| 	assertPDSSEqual(t, FromMinutes(10), "10m") | ||||
| 	assertPDSSEqual(t, FromMinutes(10), "10min") | ||||
| 	assertPDSSEqual(t, FromMinutes(1), "1minute") | ||||
| 	assertPDSSEqual(t, FromMinutes(10), "10minutes") | ||||
| 	assertPDSSEqual(t, FromMinutes(10.5), "10.5minutes") | ||||
|  | ||||
| 	assertPDSSEqual(t, FromMilliseconds(100), "100ms") | ||||
| 	assertPDSSEqual(t, FromMilliseconds(100), "100milliseconds") | ||||
| 	assertPDSSEqual(t, FromMilliseconds(100), "100millisecond") | ||||
|  | ||||
| 	assertPDSSEqual(t, FromNanoseconds(99235), "99235ns") | ||||
| 	assertPDSSEqual(t, FromNanoseconds(99235), "99235nanoseconds") | ||||
| 	assertPDSSEqual(t, FromNanoseconds(99235), "99235nanosecond") | ||||
|  | ||||
| 	assertPDSSEqual(t, FromMicroseconds(99235), "99235us") | ||||
| 	assertPDSSEqual(t, FromMicroseconds(99235), "99235microseconds") | ||||
| 	assertPDSSEqual(t, FromMicroseconds(99235), "99235microsecond") | ||||
|  | ||||
| 	assertPDSSEqual(t, FromHours(1), "1h") | ||||
| 	assertPDSSEqual(t, FromHours(1), "1hour") | ||||
| 	assertPDSSEqual(t, FromHours(2), "2hours") | ||||
|  | ||||
| 	assertPDSSEqual(t, FromDays(1), "1d") | ||||
| 	assertPDSSEqual(t, FromDays(1), "1day") | ||||
| 	assertPDSSEqual(t, FromDays(10), "10days") | ||||
| 	assertPDSSEqual(t, FromDays(1), "1days") | ||||
| 	assertPDSSEqual(t, FromDays(10), "10day") | ||||
|  | ||||
| 	assertPDSSEqual(t, FromDays(1)+FromMinutes(10), "1d10m") | ||||
| 	assertPDSSEqual(t, FromDays(1)+FromMinutes(10)+FromSeconds(200), "1d10m200sec") | ||||
| 	assertPDSSEqual(t, FromDays(1)+FromMinutes(10), "1d:10m") | ||||
| 	assertPDSSEqual(t, FromDays(1)+FromMinutes(10), "1d 10m") | ||||
| 	assertPDSSEqual(t, FromDays(1)+FromMinutes(10), "1d,10m") | ||||
| 	assertPDSSEqual(t, FromDays(1)+FromMinutes(10), "1d, 10m") | ||||
| 	assertPDSSEqual(t, FromDays(1)+FromSeconds(1000), "1d   1000seconds") | ||||
|  | ||||
| 	assertPDSSEqual(t, FromDays(1), "86400s") | ||||
| } | ||||
|  | ||||
| func assertPDSSEqual(t *testing.T, expected time.Duration, fmt string) { | ||||
| 	actual, err := ParseDurationShortString(fmt) | ||||
| 	if err != nil { | ||||
| 		t.Errorf("ParseDurationShortString('%s') failed with %v", fmt, err) | ||||
| 	} | ||||
| 	if actual != expected { | ||||
| 		t.Errorf("values differ: Actual: '%v', Expected: '%v'", actual.String(), expected.String()) | ||||
| 	} | ||||
| } | ||||
		Reference in New Issue
	
	Block a user