Compare commits
30 Commits
Author | SHA1 | Date | |
---|---|---|---|
86c01659d7
|
|||
62acddda5e
|
|||
ee325f67fd
|
|||
dba0cd229e
|
|||
ec4dba173f
|
|||
22ce2d26f3
|
|||
4fd768e573
|
|||
bf16a8165f
|
|||
9f5612248a
|
|||
4a2b830252
|
|||
c492c80881
|
|||
26dd16d021
|
|||
b0b43de8ca
|
|||
94f72e4ddf
|
|||
df4388e6dc
|
|||
fd33b43f31
|
|||
be4de07eb8
|
|||
36ed474bfe
|
|||
fdc590c8c3
|
|||
1990e5d32d
|
|||
72883cf6bd
|
|||
ff08d5f180
|
|||
72d6b538f7
|
|||
48dd30fb94
|
|||
b7c5756f11
|
|||
2070a432a5
|
|||
34e6d1819d
|
|||
87fa6021e4
|
|||
297d6c52a8
|
|||
b9c46947d2
|
1
.idea/inspectionProfiles/Project_Default.xml
generated
1
.idea/inspectionProfiles/Project_Default.xml
generated
@@ -1,6 +1,7 @@
|
|||||||
<component name="InspectionProjectProfileManager">
|
<component name="InspectionProjectProfileManager">
|
||||||
<profile version="1.0">
|
<profile version="1.0">
|
||||||
<option name="myName" value="Project Default" />
|
<option name="myName" value="Project Default" />
|
||||||
|
<inspection_tool class="GoMixedReceiverTypes" enabled="false" level="WEAK WARNING" enabled_by_default="false" />
|
||||||
<inspection_tool class="LanguageDetectionInspection" enabled="false" level="WARNING" enabled_by_default="false" />
|
<inspection_tool class="LanguageDetectionInspection" enabled="false" level="WARNING" enabled_by_default="false" />
|
||||||
<inspection_tool class="SpellCheckingInspection" enabled="false" level="TYPO" enabled_by_default="false">
|
<inspection_tool class="SpellCheckingInspection" enabled="false" level="TYPO" enabled_by_default="false">
|
||||||
<option name="processCode" value="true" />
|
<option name="processCode" value="true" />
|
||||||
|
@@ -7,6 +7,20 @@ set -o pipefail # Return value of a pipeline is the value of the last (rightmos
|
|||||||
IFS=$'\n\t' # Set $IFS to only newline and tab.
|
IFS=$'\n\t' # Set $IFS to only newline and tab.
|
||||||
|
|
||||||
|
|
||||||
|
function black() { echo -e "\x1B[30m $1 \x1B[0m"; }
|
||||||
|
function red() { echo -e "\x1B[31m $1 \x1B[0m"; }
|
||||||
|
function green() { echo -e "\x1B[32m $1 \x1B[0m"; }
|
||||||
|
function yellow(){ echo -e "\x1B[33m $1 \x1B[0m"; }
|
||||||
|
function blue() { echo -e "\x1B[34m $1 \x1B[0m"; }
|
||||||
|
function purple(){ echo -e "\x1B[35m $1 \x1B[0m"; }
|
||||||
|
function cyan() { echo -e "\x1B[36m $1 \x1B[0m"; }
|
||||||
|
function white() { echo -e "\x1B[37m $1 \x1B[0m"; }
|
||||||
|
|
||||||
|
if [ "$( git rev-parse --abbrev-ref HEAD )" != "master" ]; then
|
||||||
|
>&2 red "[ERROR] Can only create versions of <master>"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
curr_vers=$(git describe --tags --abbrev=0 | sed 's/v//g')
|
curr_vers=$(git describe --tags --abbrev=0 | sed 's/v//g')
|
||||||
|
|
||||||
next_ver=$(echo "$curr_vers" | awk -F. -v OFS=. 'NF==1{print ++$NF}; NF>1{if(length($NF+1)>length($NF))$(NF-1)++; $NF=sprintf("%0*d", length($NF), ($NF+1)%(10^length($NF))); print}')
|
next_ver=$(echo "$curr_vers" | awk -F. -v OFS=. 'NF==1{print ++$NF}; NF>1{if(length($NF+1)>length($NF))$(NF-1)++; $NF=sprintf("%0*d", length($NF), ($NF+1)%(10^length($NF))); print}')
|
||||||
@@ -18,7 +32,13 @@ echo ""
|
|||||||
|
|
||||||
git add --verbose .
|
git add --verbose .
|
||||||
|
|
||||||
git commit -a -m "v${next_ver}"
|
msg="v${next_ver}"
|
||||||
|
|
||||||
|
if [ $# -gt 0 ]; then
|
||||||
|
msg="$1"
|
||||||
|
fi
|
||||||
|
|
||||||
|
git commit -a -m "${msg}"
|
||||||
|
|
||||||
git tag "v${next_ver}"
|
git tag "v${next_ver}"
|
||||||
|
|
||||||
|
93
cmdext/builder.go
Normal file
93
cmdext/builder.go
Normal file
@@ -0,0 +1,93 @@
|
|||||||
|
package cmdext
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"gogs.mikescher.com/BlackForestBytes/goext/langext"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type CommandRunner struct {
|
||||||
|
program string
|
||||||
|
args []string
|
||||||
|
timeout *time.Duration
|
||||||
|
env []string
|
||||||
|
listener []CommandListener
|
||||||
|
enforceExitCodes *[]int
|
||||||
|
enforceNoTimeout bool
|
||||||
|
}
|
||||||
|
|
||||||
|
func Runner(program string) *CommandRunner {
|
||||||
|
return &CommandRunner{
|
||||||
|
program: program,
|
||||||
|
args: make([]string, 0),
|
||||||
|
timeout: nil,
|
||||||
|
env: make([]string, 0),
|
||||||
|
listener: make([]CommandListener, 0),
|
||||||
|
enforceExitCodes: nil,
|
||||||
|
enforceNoTimeout: false,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *CommandRunner) Arg(arg string) *CommandRunner {
|
||||||
|
r.args = append(r.args, arg)
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *CommandRunner) Args(arg []string) *CommandRunner {
|
||||||
|
r.args = append(r.args, arg...)
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *CommandRunner) Timeout(timeout time.Duration) *CommandRunner {
|
||||||
|
r.timeout = &timeout
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *CommandRunner) Env(key, value string) *CommandRunner {
|
||||||
|
r.env = append(r.env, fmt.Sprintf("%s=%s", key, value))
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *CommandRunner) RawEnv(env string) *CommandRunner {
|
||||||
|
r.env = append(r.env, env)
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *CommandRunner) Envs(env []string) *CommandRunner {
|
||||||
|
r.env = append(r.env, env...)
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *CommandRunner) EnsureExitcode(arg ...int) *CommandRunner {
|
||||||
|
r.enforceExitCodes = langext.Ptr(langext.ForceArray(arg))
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *CommandRunner) FailOnExitCode() *CommandRunner {
|
||||||
|
r.enforceExitCodes = langext.Ptr([]int{0})
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *CommandRunner) FailOnTimeout() *CommandRunner {
|
||||||
|
r.enforceNoTimeout = true
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *CommandRunner) Listen(lstr CommandListener) *CommandRunner {
|
||||||
|
r.listener = append(r.listener, lstr)
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *CommandRunner) ListenStdout(lstr func(string)) *CommandRunner {
|
||||||
|
r.listener = append(r.listener, genericCommandListener{_readStdoutLine: &lstr})
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *CommandRunner) ListenStderr(lstr func(string)) *CommandRunner {
|
||||||
|
r.listener = append(r.listener, genericCommandListener{_readStderrLine: &lstr})
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *CommandRunner) Run() (CommandResult, error) {
|
||||||
|
return run(*r)
|
||||||
|
}
|
@@ -1,11 +1,17 @@
|
|||||||
package cmdext
|
package cmdext
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bufio"
|
"errors"
|
||||||
|
"gogs.mikescher.com/BlackForestBytes/goext/langext"
|
||||||
|
"gogs.mikescher.com/BlackForestBytes/goext/mathext"
|
||||||
|
"gogs.mikescher.com/BlackForestBytes/goext/syncext"
|
||||||
"os/exec"
|
"os/exec"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
var ErrExitCode = errors.New("process exited with an unexpected exitcode")
|
||||||
|
var ErrTimeout = errors.New("process did not exit after the specified timeout")
|
||||||
|
|
||||||
type CommandResult struct {
|
type CommandResult struct {
|
||||||
StdOut string
|
StdOut string
|
||||||
StdErr string
|
StdErr string
|
||||||
@@ -14,9 +20,10 @@ type CommandResult struct {
|
|||||||
CommandTimedOut bool
|
CommandTimedOut bool
|
||||||
}
|
}
|
||||||
|
|
||||||
func RunCommand(program string, args []string, timeout *time.Duration) (CommandResult, error) {
|
func run(opt CommandRunner) (CommandResult, error) {
|
||||||
|
cmd := exec.Command(opt.program, opt.args...)
|
||||||
|
|
||||||
cmd := exec.Command(program, args...)
|
cmd.Env = append(cmd.Env, opt.env...)
|
||||||
|
|
||||||
stdoutPipe, err := cmd.StdoutPipe()
|
stdoutPipe, err := cmd.StdoutPipe()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -28,115 +35,120 @@ func RunCommand(program string, args []string, timeout *time.Duration) (CommandR
|
|||||||
return CommandResult{}, err
|
return CommandResult{}, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
preader := pipeReader{
|
||||||
|
lineBufferSize: langext.Ptr(128 * 1024 * 1024), // 128MB max size of a single line, is hopefully enough....
|
||||||
|
stdout: stdoutPipe,
|
||||||
|
stderr: stderrPipe,
|
||||||
|
}
|
||||||
|
|
||||||
err = cmd.Start()
|
err = cmd.Start()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return CommandResult{}, err
|
return CommandResult{}, err
|
||||||
}
|
}
|
||||||
|
|
||||||
errch := make(chan error, 1)
|
type resultObj struct {
|
||||||
go func() { errch <- cmd.Wait() }()
|
stdout string
|
||||||
|
stderr string
|
||||||
|
stdcombined string
|
||||||
|
err error
|
||||||
|
}
|
||||||
|
|
||||||
combch := make(chan string, 32)
|
outputChan := make(chan resultObj)
|
||||||
stopCombch := make(chan bool)
|
|
||||||
|
|
||||||
stdout := ""
|
|
||||||
go func() {
|
go func() {
|
||||||
scanner := bufio.NewScanner(stdoutPipe)
|
// we need to first fully read the pipes and then call Wait
|
||||||
for scanner.Scan() {
|
// see https://pkg.go.dev/os/exec#Cmd.StdoutPipe
|
||||||
txt := scanner.Text()
|
|
||||||
stdout += txt
|
|
||||||
combch <- txt
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
stderr := ""
|
stdout, stderr, stdcombined, err := preader.Read(opt.listener)
|
||||||
go func() {
|
if err != nil {
|
||||||
scanner := bufio.NewScanner(stderrPipe)
|
outputChan <- resultObj{stdout, stderr, stdcombined, err}
|
||||||
for scanner.Scan() {
|
|
||||||
txt := scanner.Text()
|
|
||||||
stderr += txt
|
|
||||||
combch <- txt
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
defer func() {
|
|
||||||
stopCombch <- true
|
|
||||||
}()
|
|
||||||
|
|
||||||
stdcombined := ""
|
|
||||||
go func() {
|
|
||||||
for {
|
|
||||||
select {
|
|
||||||
case txt := <-combch:
|
|
||||||
stdcombined += txt
|
|
||||||
case <-stopCombch:
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
if timeout != nil {
|
|
||||||
|
|
||||||
select {
|
|
||||||
|
|
||||||
case <-time.After(*timeout):
|
|
||||||
_ = cmd.Process.Kill()
|
_ = cmd.Process.Kill()
|
||||||
return CommandResult{
|
return
|
||||||
StdOut: stdout,
|
}
|
||||||
StdErr: stderr,
|
|
||||||
StdCombined: stdcombined,
|
err = cmd.Wait()
|
||||||
|
if err != nil {
|
||||||
|
outputChan <- resultObj{stdout, stderr, stdcombined, err}
|
||||||
|
} else {
|
||||||
|
outputChan <- resultObj{stdout, stderr, stdcombined, nil}
|
||||||
|
}
|
||||||
|
|
||||||
|
}()
|
||||||
|
|
||||||
|
var timeoutChan <-chan time.Time = make(chan time.Time, 1)
|
||||||
|
if opt.timeout != nil {
|
||||||
|
timeoutChan = time.After(*opt.timeout)
|
||||||
|
}
|
||||||
|
|
||||||
|
select {
|
||||||
|
|
||||||
|
case <-timeoutChan:
|
||||||
|
_ = cmd.Process.Kill()
|
||||||
|
for _, lstr := range opt.listener {
|
||||||
|
lstr.Timeout()
|
||||||
|
}
|
||||||
|
|
||||||
|
if fallback, ok := syncext.ReadChannelWithTimeout(outputChan, mathext.Min(32*time.Millisecond, *opt.timeout)); ok {
|
||||||
|
// most of the time the cmd.Process.Kill() should also ahve finished the pipereader
|
||||||
|
// and we can at least return the already collected stdout, stderr, etc
|
||||||
|
res := CommandResult{
|
||||||
|
StdOut: fallback.stdout,
|
||||||
|
StdErr: fallback.stderr,
|
||||||
|
StdCombined: fallback.stdcombined,
|
||||||
ExitCode: -1,
|
ExitCode: -1,
|
||||||
CommandTimedOut: true,
|
CommandTimedOut: true,
|
||||||
}, nil
|
|
||||||
|
|
||||||
case err := <-errch:
|
|
||||||
if exiterr, ok := err.(*exec.ExitError); ok {
|
|
||||||
return CommandResult{
|
|
||||||
StdOut: stdout,
|
|
||||||
StdErr: stderr,
|
|
||||||
StdCombined: stdcombined,
|
|
||||||
ExitCode: exiterr.ExitCode(),
|
|
||||||
CommandTimedOut: false,
|
|
||||||
}, nil
|
|
||||||
} else if err != nil {
|
|
||||||
return CommandResult{}, err
|
|
||||||
} else {
|
|
||||||
return CommandResult{
|
|
||||||
StdOut: stdout,
|
|
||||||
StdErr: stderr,
|
|
||||||
StdCombined: stdcombined,
|
|
||||||
ExitCode: 0,
|
|
||||||
CommandTimedOut: false,
|
|
||||||
}, nil
|
|
||||||
}
|
}
|
||||||
|
if opt.enforceNoTimeout {
|
||||||
|
return res, ErrTimeout
|
||||||
|
}
|
||||||
|
return res, nil
|
||||||
|
} else {
|
||||||
|
res := CommandResult{
|
||||||
|
StdOut: "",
|
||||||
|
StdErr: "",
|
||||||
|
StdCombined: "",
|
||||||
|
ExitCode: -1,
|
||||||
|
CommandTimedOut: true,
|
||||||
|
}
|
||||||
|
if opt.enforceNoTimeout {
|
||||||
|
return res, ErrTimeout
|
||||||
|
}
|
||||||
|
return res, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
} else {
|
case outobj := <-outputChan:
|
||||||
|
if exiterr, ok := outobj.err.(*exec.ExitError); ok {
|
||||||
select {
|
excode := exiterr.ExitCode()
|
||||||
|
for _, lstr := range opt.listener {
|
||||||
case err := <-errch:
|
lstr.Finished(excode)
|
||||||
if exiterr, ok := err.(*exec.ExitError); ok {
|
|
||||||
return CommandResult{
|
|
||||||
StdOut: stdout,
|
|
||||||
StdErr: stderr,
|
|
||||||
StdCombined: stdcombined,
|
|
||||||
ExitCode: exiterr.ExitCode(),
|
|
||||||
CommandTimedOut: false,
|
|
||||||
}, nil
|
|
||||||
} else if err != nil {
|
|
||||||
return CommandResult{}, err
|
|
||||||
} else {
|
|
||||||
return CommandResult{
|
|
||||||
StdOut: stdout,
|
|
||||||
StdErr: stderr,
|
|
||||||
StdCombined: stdcombined,
|
|
||||||
ExitCode: 0,
|
|
||||||
CommandTimedOut: false,
|
|
||||||
}, nil
|
|
||||||
}
|
}
|
||||||
|
res := CommandResult{
|
||||||
|
StdOut: outobj.stdout,
|
||||||
|
StdErr: outobj.stderr,
|
||||||
|
StdCombined: outobj.stdcombined,
|
||||||
|
ExitCode: excode,
|
||||||
|
CommandTimedOut: false,
|
||||||
|
}
|
||||||
|
if opt.enforceExitCodes != nil && !langext.InArray(excode, *opt.enforceExitCodes) {
|
||||||
|
return res, ErrExitCode
|
||||||
|
}
|
||||||
|
return res, nil
|
||||||
|
} else if err != nil {
|
||||||
|
return CommandResult{}, err
|
||||||
|
} else {
|
||||||
|
for _, lstr := range opt.listener {
|
||||||
|
lstr.Finished(0)
|
||||||
|
}
|
||||||
|
res := CommandResult{
|
||||||
|
StdOut: outobj.stdout,
|
||||||
|
StdErr: outobj.stderr,
|
||||||
|
StdCombined: outobj.stdcombined,
|
||||||
|
ExitCode: 0,
|
||||||
|
CommandTimedOut: false,
|
||||||
|
}
|
||||||
|
if opt.enforceExitCodes != nil && !langext.InArray(0, *opt.enforceExitCodes) {
|
||||||
|
return res, ErrExitCode
|
||||||
|
}
|
||||||
|
return res, nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
323
cmdext/cmdrunner_test.go
Normal file
323
cmdext/cmdrunner_test.go
Normal file
@@ -0,0 +1,323 @@
|
|||||||
|
package cmdext
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestStdout(t *testing.T) {
|
||||||
|
|
||||||
|
res1, err := Runner("printf").Arg("hello").Run()
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("%v", err)
|
||||||
|
}
|
||||||
|
if res1.CommandTimedOut {
|
||||||
|
t.Errorf("Timeout")
|
||||||
|
}
|
||||||
|
if res1.ExitCode != 0 {
|
||||||
|
t.Errorf("res1.ExitCode == %v", res1.ExitCode)
|
||||||
|
}
|
||||||
|
if res1.StdErr != "" {
|
||||||
|
t.Errorf("res1.StdErr == '%v'", res1.StdErr)
|
||||||
|
}
|
||||||
|
if res1.StdOut != "hello" {
|
||||||
|
t.Errorf("res1.StdOut == '%v'", res1.StdOut)
|
||||||
|
}
|
||||||
|
if res1.StdCombined != "hello\n" {
|
||||||
|
t.Errorf("res1.StdCombined == '%v'", res1.StdCombined)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestStderr(t *testing.T) {
|
||||||
|
|
||||||
|
res1, err := Runner("python").Arg("-c").Arg("import sys; print(\"error\", file=sys.stderr, end='')").Run()
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("%v", err)
|
||||||
|
}
|
||||||
|
if res1.CommandTimedOut {
|
||||||
|
t.Errorf("Timeout")
|
||||||
|
}
|
||||||
|
if res1.ExitCode != 0 {
|
||||||
|
t.Errorf("res1.ExitCode == %v", res1.ExitCode)
|
||||||
|
}
|
||||||
|
if res1.StdErr != "error" {
|
||||||
|
t.Errorf("res1.StdErr == '%v'", res1.StdErr)
|
||||||
|
}
|
||||||
|
if res1.StdOut != "" {
|
||||||
|
t.Errorf("res1.StdOut == '%v'", res1.StdOut)
|
||||||
|
}
|
||||||
|
if res1.StdCombined != "error\n" {
|
||||||
|
t.Errorf("res1.StdCombined == '%v'", res1.StdCombined)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestStdcombined(t *testing.T) {
|
||||||
|
res1, err := Runner("python").
|
||||||
|
Arg("-c").
|
||||||
|
Arg("import sys; import time; print(\"1\", file=sys.stderr, flush=True); time.sleep(0.1); print(\"2\", file=sys.stdout, flush=True); time.sleep(0.1); print(\"3\", file=sys.stderr, flush=True)").
|
||||||
|
Run()
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("%v", err)
|
||||||
|
}
|
||||||
|
if res1.CommandTimedOut {
|
||||||
|
t.Errorf("Timeout")
|
||||||
|
}
|
||||||
|
if res1.ExitCode != 0 {
|
||||||
|
t.Errorf("res1.ExitCode == %v", res1.ExitCode)
|
||||||
|
}
|
||||||
|
if res1.StdErr != "1\n3\n" {
|
||||||
|
t.Errorf("res1.StdErr == '%v'", res1.StdErr)
|
||||||
|
}
|
||||||
|
if res1.StdOut != "2\n" {
|
||||||
|
t.Errorf("res1.StdOut == '%v'", res1.StdOut)
|
||||||
|
}
|
||||||
|
if res1.StdCombined != "1\n2\n3\n" {
|
||||||
|
t.Errorf("res1.StdCombined == '%v'", res1.StdCombined)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestPartialRead(t *testing.T) {
|
||||||
|
res1, err := Runner("python").
|
||||||
|
Arg("-c").
|
||||||
|
Arg("import sys; import time; print(\"first message\", flush=True); time.sleep(5); print(\"cant see me\", flush=True);").
|
||||||
|
Timeout(100 * time.Millisecond).
|
||||||
|
Run()
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("%v", err)
|
||||||
|
}
|
||||||
|
if !res1.CommandTimedOut {
|
||||||
|
t.Errorf("!CommandTimedOut")
|
||||||
|
}
|
||||||
|
if res1.StdErr != "" {
|
||||||
|
t.Errorf("res1.StdErr == '%v'", res1.StdErr)
|
||||||
|
}
|
||||||
|
if res1.StdOut != "first message\n" {
|
||||||
|
t.Errorf("res1.StdOut == '%v'", res1.StdOut)
|
||||||
|
}
|
||||||
|
if res1.StdCombined != "first message\n" {
|
||||||
|
t.Errorf("res1.StdCombined == '%v'", res1.StdCombined)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestPartialReadStderr(t *testing.T) {
|
||||||
|
res1, err := Runner("python").
|
||||||
|
Arg("-c").
|
||||||
|
Arg("import sys; import time; print(\"first message\", file=sys.stderr, flush=True); time.sleep(5); print(\"cant see me\", file=sys.stderr, flush=True);").
|
||||||
|
Timeout(100 * time.Millisecond).
|
||||||
|
Run()
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("%v", err)
|
||||||
|
}
|
||||||
|
if !res1.CommandTimedOut {
|
||||||
|
t.Errorf("!CommandTimedOut")
|
||||||
|
}
|
||||||
|
if res1.StdErr != "first message\n" {
|
||||||
|
t.Errorf("res1.StdErr == '%v'", res1.StdErr)
|
||||||
|
}
|
||||||
|
if res1.StdOut != "" {
|
||||||
|
t.Errorf("res1.StdOut == '%v'", res1.StdOut)
|
||||||
|
}
|
||||||
|
if res1.StdCombined != "first message\n" {
|
||||||
|
t.Errorf("res1.StdCombined == '%v'", res1.StdCombined)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestReadUnflushedStdout(t *testing.T) {
|
||||||
|
|
||||||
|
res1, err := Runner("python").Arg("-c").Arg("import sys; print(\"message101\", file=sys.stdout, end='')").Run()
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("%v", err)
|
||||||
|
}
|
||||||
|
if res1.CommandTimedOut {
|
||||||
|
t.Errorf("Timeout")
|
||||||
|
}
|
||||||
|
if res1.ExitCode != 0 {
|
||||||
|
t.Errorf("res1.ExitCode == %v", res1.ExitCode)
|
||||||
|
}
|
||||||
|
if res1.StdErr != "" {
|
||||||
|
t.Errorf("res1.StdErr == '%v'", res1.StdErr)
|
||||||
|
}
|
||||||
|
if res1.StdOut != "message101" {
|
||||||
|
t.Errorf("res1.StdOut == '%v'", res1.StdOut)
|
||||||
|
}
|
||||||
|
if res1.StdCombined != "message101\n" {
|
||||||
|
t.Errorf("res1.StdCombined == '%v'", res1.StdCombined)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestReadUnflushedStderr(t *testing.T) {
|
||||||
|
|
||||||
|
res1, err := Runner("python").Arg("-c").Arg("import sys; print(\"message101\", file=sys.stderr, end='')").Run()
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("%v", err)
|
||||||
|
}
|
||||||
|
if res1.CommandTimedOut {
|
||||||
|
t.Errorf("Timeout")
|
||||||
|
}
|
||||||
|
if res1.ExitCode != 0 {
|
||||||
|
t.Errorf("res1.ExitCode == %v", res1.ExitCode)
|
||||||
|
}
|
||||||
|
if res1.StdErr != "message101" {
|
||||||
|
t.Errorf("res1.StdErr == '%v'", res1.StdErr)
|
||||||
|
}
|
||||||
|
if res1.StdOut != "" {
|
||||||
|
t.Errorf("res1.StdOut == '%v'", res1.StdOut)
|
||||||
|
}
|
||||||
|
if res1.StdCombined != "message101\n" {
|
||||||
|
t.Errorf("res1.StdCombined == '%v'", res1.StdCombined)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestPartialReadUnflushed(t *testing.T) {
|
||||||
|
t.SkipNow()
|
||||||
|
|
||||||
|
res1, err := Runner("python").
|
||||||
|
Arg("-c").
|
||||||
|
Arg("import sys; import time; print(\"first message\", end=''); time.sleep(5); print(\"cant see me\", end='');").
|
||||||
|
Timeout(100 * time.Millisecond).
|
||||||
|
Run()
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("%v", err)
|
||||||
|
}
|
||||||
|
if !res1.CommandTimedOut {
|
||||||
|
t.Errorf("!CommandTimedOut")
|
||||||
|
}
|
||||||
|
if res1.StdErr != "" {
|
||||||
|
t.Errorf("res1.StdErr == '%v'", res1.StdErr)
|
||||||
|
}
|
||||||
|
if res1.StdOut != "first message" {
|
||||||
|
t.Errorf("res1.StdOut == '%v'", res1.StdOut)
|
||||||
|
}
|
||||||
|
if res1.StdCombined != "first message" {
|
||||||
|
t.Errorf("res1.StdCombined == '%v'", res1.StdCombined)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestPartialReadUnflushedStderr(t *testing.T) {
|
||||||
|
t.SkipNow()
|
||||||
|
|
||||||
|
res1, err := Runner("python").
|
||||||
|
Arg("-c").
|
||||||
|
Arg("import sys; import time; print(\"first message\", file=sys.stderr, end=''); time.sleep(5); print(\"cant see me\", file=sys.stderr, end='');").
|
||||||
|
Timeout(100 * time.Millisecond).
|
||||||
|
Run()
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("%v", err)
|
||||||
|
}
|
||||||
|
if !res1.CommandTimedOut {
|
||||||
|
t.Errorf("!CommandTimedOut")
|
||||||
|
}
|
||||||
|
if res1.StdErr != "first message" {
|
||||||
|
t.Errorf("res1.StdErr == '%v'", res1.StdErr)
|
||||||
|
}
|
||||||
|
if res1.StdOut != "" {
|
||||||
|
t.Errorf("res1.StdOut == '%v'", res1.StdOut)
|
||||||
|
}
|
||||||
|
if res1.StdCombined != "first message" {
|
||||||
|
t.Errorf("res1.StdCombined == '%v'", res1.StdCombined)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestListener(t *testing.T) {
|
||||||
|
|
||||||
|
res1, err := Runner("python").
|
||||||
|
Arg("-c").
|
||||||
|
Arg("import sys;" +
|
||||||
|
"import time;" +
|
||||||
|
"print(\"message 1\", flush=True);" +
|
||||||
|
"time.sleep(1);" +
|
||||||
|
"print(\"message 2\", flush=True);" +
|
||||||
|
"time.sleep(1);" +
|
||||||
|
"print(\"message 3\", flush=True);" +
|
||||||
|
"time.sleep(1);" +
|
||||||
|
"print(\"message 4\", file=sys.stderr, flush=True);" +
|
||||||
|
"time.sleep(1);" +
|
||||||
|
"print(\"message 5\", flush=True);" +
|
||||||
|
"time.sleep(1);" +
|
||||||
|
"print(\"final\");").
|
||||||
|
ListenStdout(func(s string) { fmt.Printf("@@STDOUT <<- %v (%v)\n", s, time.Now().Format(time.RFC3339Nano)) }).
|
||||||
|
ListenStderr(func(s string) { fmt.Printf("@@STDERR <<- %v (%v)\n", s, time.Now().Format(time.RFC3339Nano)) }).
|
||||||
|
Timeout(10 * time.Second).
|
||||||
|
Run()
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
if res1.CommandTimedOut {
|
||||||
|
t.Errorf("Timeout")
|
||||||
|
}
|
||||||
|
if res1.ExitCode != 0 {
|
||||||
|
t.Errorf("res1.ExitCode == %v", res1.ExitCode)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestLongStdout(t *testing.T) {
|
||||||
|
|
||||||
|
res1, err := Runner("python").
|
||||||
|
Arg("-c").
|
||||||
|
Arg("import sys; import time; print(\"X\" * 125001 + \"\\n\"); print(\"Y\" * 125001 + \"\\n\"); print(\"Z\" * 125001 + \"\\n\");").
|
||||||
|
Timeout(5000 * time.Millisecond).
|
||||||
|
Run()
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("%v", err)
|
||||||
|
}
|
||||||
|
if res1.CommandTimedOut {
|
||||||
|
t.Errorf("Timeout")
|
||||||
|
}
|
||||||
|
if res1.ExitCode != 0 {
|
||||||
|
t.Errorf("res1.ExitCode == %v", res1.ExitCode)
|
||||||
|
}
|
||||||
|
if res1.StdErr != "" {
|
||||||
|
t.Errorf("res1.StdErr == '%v'", res1.StdErr)
|
||||||
|
}
|
||||||
|
if len(res1.StdOut) != 375009 {
|
||||||
|
t.Errorf("len(res1.StdOut) == '%v'", len(res1.StdOut))
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestFailOnTimeout(t *testing.T) {
|
||||||
|
|
||||||
|
_, err := Runner("sleep").Arg("2").Timeout(200 * time.Millisecond).FailOnTimeout().Run()
|
||||||
|
if err != ErrTimeout {
|
||||||
|
t.Errorf("wrong err := %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestFailOnExitcode(t *testing.T) {
|
||||||
|
|
||||||
|
_, err := Runner("false").Timeout(200 * time.Millisecond).FailOnExitCode().Run()
|
||||||
|
if err != ErrExitCode {
|
||||||
|
t.Errorf("wrong err := %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestEnsureExitcode1(t *testing.T) {
|
||||||
|
|
||||||
|
_, err := Runner("false").Timeout(200 * time.Millisecond).EnsureExitcode(1).Run()
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("wrong err := %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestEnsureExitcode2(t *testing.T) {
|
||||||
|
|
||||||
|
_, err := Runner("false").Timeout(200*time.Millisecond).EnsureExitcode(0, 2, 3).Run()
|
||||||
|
if err != ErrExitCode {
|
||||||
|
t.Errorf("wrong err := %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
12
cmdext/helper.go
Normal file
12
cmdext/helper.go
Normal file
@@ -0,0 +1,12 @@
|
|||||||
|
package cmdext
|
||||||
|
|
||||||
|
import "time"
|
||||||
|
|
||||||
|
func RunCommand(program string, args []string, timeout *time.Duration) (CommandResult, error) {
|
||||||
|
b := Runner(program)
|
||||||
|
b = b.Args(args)
|
||||||
|
if timeout != nil {
|
||||||
|
b = b.Timeout(*timeout)
|
||||||
|
}
|
||||||
|
return b.Run()
|
||||||
|
}
|
57
cmdext/listener.go
Normal file
57
cmdext/listener.go
Normal file
@@ -0,0 +1,57 @@
|
|||||||
|
package cmdext
|
||||||
|
|
||||||
|
type CommandListener interface {
|
||||||
|
ReadRawStdout([]byte)
|
||||||
|
ReadRawStderr([]byte)
|
||||||
|
|
||||||
|
ReadStdoutLine(string)
|
||||||
|
ReadStderrLine(string)
|
||||||
|
|
||||||
|
Finished(int)
|
||||||
|
Timeout()
|
||||||
|
}
|
||||||
|
|
||||||
|
type genericCommandListener struct {
|
||||||
|
_readRawStdout *func([]byte)
|
||||||
|
_readRawStderr *func([]byte)
|
||||||
|
_readStdoutLine *func(string)
|
||||||
|
_readStderrLine *func(string)
|
||||||
|
_finished *func(int)
|
||||||
|
_timeout *func()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (g genericCommandListener) ReadRawStdout(v []byte) {
|
||||||
|
if g._readRawStdout != nil {
|
||||||
|
(*g._readRawStdout)(v)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (g genericCommandListener) ReadRawStderr(v []byte) {
|
||||||
|
if g._readRawStderr != nil {
|
||||||
|
(*g._readRawStderr)(v)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (g genericCommandListener) ReadStdoutLine(v string) {
|
||||||
|
if g._readStdoutLine != nil {
|
||||||
|
(*g._readStdoutLine)(v)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (g genericCommandListener) ReadStderrLine(v string) {
|
||||||
|
if g._readStderrLine != nil {
|
||||||
|
(*g._readStderrLine)(v)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (g genericCommandListener) Finished(v int) {
|
||||||
|
if g._finished != nil {
|
||||||
|
(*g._finished)(v)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (g genericCommandListener) Timeout() {
|
||||||
|
if g._timeout != nil {
|
||||||
|
(*g._timeout)()
|
||||||
|
}
|
||||||
|
}
|
158
cmdext/pipereader.go
Normal file
158
cmdext/pipereader.go
Normal file
@@ -0,0 +1,158 @@
|
|||||||
|
package cmdext
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bufio"
|
||||||
|
"gogs.mikescher.com/BlackForestBytes/goext/syncext"
|
||||||
|
"io"
|
||||||
|
"sync"
|
||||||
|
)
|
||||||
|
|
||||||
|
type pipeReader struct {
|
||||||
|
lineBufferSize *int
|
||||||
|
stdout io.ReadCloser
|
||||||
|
stderr io.ReadCloser
|
||||||
|
}
|
||||||
|
|
||||||
|
// Read ready stdout and stdin until finished
|
||||||
|
// also splits both pipes into lines and calld the listener
|
||||||
|
func (pr *pipeReader) Read(listener []CommandListener) (string, string, string, error) {
|
||||||
|
type combevt struct {
|
||||||
|
line string
|
||||||
|
stop bool
|
||||||
|
}
|
||||||
|
|
||||||
|
errch := make(chan error, 8)
|
||||||
|
|
||||||
|
wg := sync.WaitGroup{}
|
||||||
|
|
||||||
|
// [1] read raw stdout
|
||||||
|
|
||||||
|
wg.Add(1)
|
||||||
|
stdoutBufferReader, stdoutBufferWriter := io.Pipe()
|
||||||
|
stdout := ""
|
||||||
|
go func() {
|
||||||
|
buf := make([]byte, 128)
|
||||||
|
for true {
|
||||||
|
n, out := pr.stdout.Read(buf)
|
||||||
|
if n > 0 {
|
||||||
|
txt := string(buf[:n])
|
||||||
|
stdout += txt
|
||||||
|
_, _ = stdoutBufferWriter.Write(buf[:n])
|
||||||
|
for _, lstr := range listener {
|
||||||
|
lstr.ReadRawStdout(buf[:n])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if out == io.EOF {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
if out != nil {
|
||||||
|
errch <- out
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ = stdoutBufferWriter.Close()
|
||||||
|
wg.Done()
|
||||||
|
}()
|
||||||
|
|
||||||
|
// [2] read raw stderr
|
||||||
|
|
||||||
|
wg.Add(1)
|
||||||
|
stderrBufferReader, stderrBufferWriter := io.Pipe()
|
||||||
|
stderr := ""
|
||||||
|
go func() {
|
||||||
|
buf := make([]byte, 128)
|
||||||
|
for true {
|
||||||
|
n, err := pr.stderr.Read(buf)
|
||||||
|
|
||||||
|
if n > 0 {
|
||||||
|
txt := string(buf[:n])
|
||||||
|
stderr += txt
|
||||||
|
_, _ = stderrBufferWriter.Write(buf[:n])
|
||||||
|
for _, lstr := range listener {
|
||||||
|
lstr.ReadRawStderr(buf[:n])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if err == io.EOF {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
errch <- err
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ = stderrBufferWriter.Close()
|
||||||
|
wg.Done()
|
||||||
|
}()
|
||||||
|
|
||||||
|
combch := make(chan combevt, 32)
|
||||||
|
|
||||||
|
// [3] collect stdout line-by-line
|
||||||
|
|
||||||
|
wg.Add(1)
|
||||||
|
go func() {
|
||||||
|
scanner := bufio.NewScanner(stdoutBufferReader)
|
||||||
|
if pr.lineBufferSize != nil {
|
||||||
|
scanner.Buffer([]byte{}, *pr.lineBufferSize)
|
||||||
|
}
|
||||||
|
for scanner.Scan() {
|
||||||
|
txt := scanner.Text()
|
||||||
|
for _, lstr := range listener {
|
||||||
|
lstr.ReadStdoutLine(txt)
|
||||||
|
}
|
||||||
|
combch <- combevt{txt, false}
|
||||||
|
}
|
||||||
|
if err := scanner.Err(); err != nil {
|
||||||
|
errch <- err
|
||||||
|
}
|
||||||
|
combch <- combevt{"", true}
|
||||||
|
wg.Done()
|
||||||
|
}()
|
||||||
|
|
||||||
|
// [4] collect stderr line-by-line
|
||||||
|
|
||||||
|
wg.Add(1)
|
||||||
|
go func() {
|
||||||
|
scanner := bufio.NewScanner(stderrBufferReader)
|
||||||
|
if pr.lineBufferSize != nil {
|
||||||
|
scanner.Buffer([]byte{}, *pr.lineBufferSize)
|
||||||
|
}
|
||||||
|
for scanner.Scan() {
|
||||||
|
txt := scanner.Text()
|
||||||
|
for _, lstr := range listener {
|
||||||
|
lstr.ReadStderrLine(txt)
|
||||||
|
}
|
||||||
|
combch <- combevt{txt, false}
|
||||||
|
}
|
||||||
|
if err := scanner.Err(); err != nil {
|
||||||
|
errch <- err
|
||||||
|
}
|
||||||
|
combch <- combevt{"", true}
|
||||||
|
wg.Done()
|
||||||
|
}()
|
||||||
|
|
||||||
|
// [5] combine stdcombined
|
||||||
|
|
||||||
|
wg.Add(1)
|
||||||
|
stdcombined := ""
|
||||||
|
go func() {
|
||||||
|
stopctr := 0
|
||||||
|
for stopctr < 2 {
|
||||||
|
vvv := <-combch
|
||||||
|
if vvv.stop {
|
||||||
|
stopctr++
|
||||||
|
} else {
|
||||||
|
stdcombined += vvv.line + "\n" // this comes from bufio.Scanner and has no newlines...
|
||||||
|
}
|
||||||
|
}
|
||||||
|
wg.Done()
|
||||||
|
}()
|
||||||
|
|
||||||
|
// wait for all (5) goroutines to finish
|
||||||
|
wg.Wait()
|
||||||
|
|
||||||
|
if err, ok := syncext.ReadNonBlocking(errch); ok {
|
||||||
|
return "", "", "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
return stdout, stderr, stdcombined, nil
|
||||||
|
}
|
@@ -22,10 +22,10 @@ import (
|
|||||||
//
|
//
|
||||||
// sub-structs are recursively parsed (if they have an env tag) and the env-variable keys are delimited by the delim parameter
|
// sub-structs are recursively parsed (if they have an env tag) and the env-variable keys are delimited by the delim parameter
|
||||||
// sub-structs with `env:""` are also parsed, but the delimited is skipped (they are handled as if they were one level higher)
|
// sub-structs with `env:""` are also parsed, but the delimited is skipped (they are handled as if they were one level higher)
|
||||||
func ApplyEnvOverrides[T any](c *T, delim string) error {
|
func ApplyEnvOverrides[T any](prefix string, c *T, delim string) error {
|
||||||
rval := reflect.ValueOf(c).Elem()
|
rval := reflect.ValueOf(c).Elem()
|
||||||
|
|
||||||
return processEnvOverrides(rval, delim, "")
|
return processEnvOverrides(rval, delim, prefix)
|
||||||
}
|
}
|
||||||
|
|
||||||
func processEnvOverrides(rval reflect.Value, delim string, prefix string) error {
|
func processEnvOverrides(rval reflect.Value, delim string, prefix string) error {
|
||||||
@@ -70,103 +70,114 @@ func processEnvOverrides(rval reflect.Value, delim string, prefix string) error
|
|||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
if rvfield.Type() == reflect.TypeOf("") {
|
if rvfield.Type().Kind() == reflect.Pointer {
|
||||||
|
|
||||||
rvfield.Set(reflect.ValueOf(envval))
|
newval, err := parseEnvToValue(envval, fullEnvKey, rvfield.Type().Elem())
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// converts reflect.Value to pointer
|
||||||
|
ptrval := reflect.New(rvfield.Type().Elem())
|
||||||
|
ptrval.Elem().Set(newval)
|
||||||
|
|
||||||
|
rvfield.Set(ptrval)
|
||||||
|
|
||||||
fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", fullEnvKey, envval)
|
fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", fullEnvKey, 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')", fullEnvKey, envval))
|
|
||||||
}
|
|
||||||
|
|
||||||
rvfield.Set(reflect.ValueOf(int(envint)))
|
|
||||||
|
|
||||||
fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", fullEnvKey, 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')", fullEnvKey, envval))
|
|
||||||
}
|
|
||||||
|
|
||||||
rvfield.Set(reflect.ValueOf(int64(envint)))
|
|
||||||
|
|
||||||
fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", fullEnvKey, 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')", fullEnvKey, envval))
|
|
||||||
}
|
|
||||||
|
|
||||||
rvfield.Set(reflect.ValueOf(int32(envint)))
|
|
||||||
|
|
||||||
fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", fullEnvKey, 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')", fullEnvKey, envval))
|
|
||||||
}
|
|
||||||
|
|
||||||
rvfield.Set(reflect.ValueOf(int8(envint)))
|
|
||||||
|
|
||||||
fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", fullEnvKey, 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')", fullEnvKey, envval))
|
|
||||||
}
|
|
||||||
|
|
||||||
rvfield.Set(reflect.ValueOf(dur))
|
|
||||||
|
|
||||||
fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", fullEnvKey, 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')", fullEnvKey, envval))
|
|
||||||
}
|
|
||||||
|
|
||||||
rvfield.Set(reflect.ValueOf(tim))
|
|
||||||
|
|
||||||
fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", fullEnvKey, 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(), fullEnvKey, envval))
|
|
||||||
}
|
|
||||||
|
|
||||||
envcvl := reflect.ValueOf(envint).Convert(rvfield.Type())
|
|
||||||
|
|
||||||
rvfield.Set(envcvl)
|
|
||||||
|
|
||||||
fmt.Printf("[CONF] Overwrite config '%s' with '%v'\n", fullEnvKey, 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", fullEnvKey, envcvl.Interface())
|
|
||||||
|
|
||||||
} else {
|
} else {
|
||||||
return errors.New(fmt.Sprintf("Unknown kind/type in config: [ %s | %s ]", rvfield.Kind().String(), rvfield.Type().String()))
|
|
||||||
|
newval, err := parseEnvToValue(envval, fullEnvKey, rvfield.Type())
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
rvfield.Set(newval)
|
||||||
|
fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", fullEnvKey, envval)
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func parseEnvToValue(envval string, fullEnvKey string, rvtype reflect.Type) (reflect.Value, error) {
|
||||||
|
if rvtype == reflect.TypeOf("") {
|
||||||
|
|
||||||
|
return reflect.ValueOf(envval), nil
|
||||||
|
|
||||||
|
} else if rvtype == reflect.TypeOf(int(0)) {
|
||||||
|
|
||||||
|
envint, err := strconv.ParseInt(envval, 10, bits.UintSize)
|
||||||
|
if err != nil {
|
||||||
|
return reflect.Value{}, errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to int (value := '%s')", fullEnvKey, envval))
|
||||||
|
}
|
||||||
|
|
||||||
|
return reflect.ValueOf(int(envint)), nil
|
||||||
|
|
||||||
|
} else if rvtype == reflect.TypeOf(int64(0)) {
|
||||||
|
|
||||||
|
envint, err := strconv.ParseInt(envval, 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
return reflect.Value{}, errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to int64 (value := '%s')", fullEnvKey, envval))
|
||||||
|
}
|
||||||
|
|
||||||
|
return reflect.ValueOf(int64(envint)), nil
|
||||||
|
|
||||||
|
} else if rvtype == reflect.TypeOf(int32(0)) {
|
||||||
|
|
||||||
|
envint, err := strconv.ParseInt(envval, 10, 32)
|
||||||
|
if err != nil {
|
||||||
|
return reflect.Value{}, errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to int32 (value := '%s')", fullEnvKey, envval))
|
||||||
|
}
|
||||||
|
|
||||||
|
return reflect.ValueOf(int32(envint)), nil
|
||||||
|
|
||||||
|
} else if rvtype == reflect.TypeOf(int8(0)) {
|
||||||
|
|
||||||
|
envint, err := strconv.ParseInt(envval, 10, 8)
|
||||||
|
if err != nil {
|
||||||
|
return reflect.Value{}, errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to int32 (value := '%s')", fullEnvKey, envval))
|
||||||
|
}
|
||||||
|
|
||||||
|
return reflect.ValueOf(int8(envint)), nil
|
||||||
|
|
||||||
|
} else if rvtype == reflect.TypeOf(time.Duration(0)) {
|
||||||
|
|
||||||
|
dur, err := timeext.ParseDurationShortString(envval)
|
||||||
|
if err != nil {
|
||||||
|
return reflect.Value{}, errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to duration (value := '%s')", fullEnvKey, envval))
|
||||||
|
}
|
||||||
|
|
||||||
|
return reflect.ValueOf(dur), nil
|
||||||
|
|
||||||
|
} else if rvtype == reflect.TypeOf(time.UnixMilli(0)) {
|
||||||
|
|
||||||
|
tim, err := time.Parse(time.RFC3339Nano, envval)
|
||||||
|
if err != nil {
|
||||||
|
return reflect.Value{}, errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to time.time (value := '%s')", fullEnvKey, envval))
|
||||||
|
}
|
||||||
|
|
||||||
|
return reflect.ValueOf(tim), nil
|
||||||
|
|
||||||
|
} else if rvtype.ConvertibleTo(reflect.TypeOf(int(0))) {
|
||||||
|
|
||||||
|
envint, err := strconv.ParseInt(envval, 10, 8)
|
||||||
|
if err != nil {
|
||||||
|
return reflect.Value{}, errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to <%s, ,int> (value := '%s')", rvtype.Name(), fullEnvKey, envval))
|
||||||
|
}
|
||||||
|
|
||||||
|
envcvl := reflect.ValueOf(envint).Convert(rvtype)
|
||||||
|
|
||||||
|
return envcvl, nil
|
||||||
|
|
||||||
|
} else if rvtype.ConvertibleTo(reflect.TypeOf("")) {
|
||||||
|
|
||||||
|
envcvl := reflect.ValueOf(envval).Convert(rvtype)
|
||||||
|
return envcvl, nil
|
||||||
|
|
||||||
|
} else {
|
||||||
|
return reflect.Value{}, errors.New(fmt.Sprintf("Unknown kind/type in config: [ %s | %s ]", rvtype.Kind().String(), rvtype.String()))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
@@ -213,8 +213,65 @@ func TestApplyEnvOverridesRecursive(t *testing.T) {
|
|||||||
assertEqual(t, data.Sub4.V9, time.Unix(2335219200, 0).UTC())
|
assertEqual(t, data.Sub4.V9, time.Unix(2335219200, 0).UTC())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestApplyEnvOverridesPointer(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{}
|
||||||
|
|
||||||
|
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()
|
||||||
|
}
|
||||||
|
|
||||||
|
assertPtrEqual(t, data.V1, 846)
|
||||||
|
assertPtrEqual(t, data.V2, "hello_world")
|
||||||
|
assertPtrEqual(t, data.V3, 6)
|
||||||
|
assertPtrEqual(t, data.V4, 333)
|
||||||
|
assertPtrEqual(t, data.V5, -937)
|
||||||
|
assertPtrEqual(t, data.V6, 70)
|
||||||
|
assertPtrEqual(t, data.V7, "AAAAAA")
|
||||||
|
assertPtrEqual(t, data.V8, time.Second*64)
|
||||||
|
assertPtrEqual(t, data.V9, time.Unix(1257894000, 0).UTC())
|
||||||
|
}
|
||||||
|
|
||||||
func assertEqual[T comparable](t *testing.T, actual T, expected T) {
|
func assertEqual[T comparable](t *testing.T, actual T, expected T) {
|
||||||
if actual != expected {
|
if actual != expected {
|
||||||
t.Errorf("values differ: Actual: '%v', Expected: '%v'", actual, expected)
|
t.Errorf("values differ: Actual: '%v', Expected: '%v'", actual, expected)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func assertPtrEqual[T comparable](t *testing.T, actual *T, expected T) {
|
||||||
|
if actual == nil {
|
||||||
|
t.Errorf("values differ: Actual: NIL, Expected: '%v'", expected)
|
||||||
|
}
|
||||||
|
if *actual != expected {
|
||||||
|
t.Errorf("values differ: Actual: '%v', Expected: '%v'", actual, expected)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
132
cryptext/aes.go
Normal file
132
cryptext/aes.go
Normal file
@@ -0,0 +1,132 @@
|
|||||||
|
package cryptext
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"crypto/aes"
|
||||||
|
"crypto/cipher"
|
||||||
|
"crypto/rand"
|
||||||
|
"crypto/sha256"
|
||||||
|
"encoding/base32"
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"golang.org/x/crypto/scrypt"
|
||||||
|
"io"
|
||||||
|
)
|
||||||
|
|
||||||
|
// https://stackoverflow.com/a/18819040/1761622
|
||||||
|
|
||||||
|
type aesPayload struct {
|
||||||
|
Salt []byte `json:"s"`
|
||||||
|
IV []byte `json:"i"`
|
||||||
|
Data []byte `json:"d"`
|
||||||
|
Rounds int `json:"r"`
|
||||||
|
Version uint `json:"v"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func EncryptAESSimple(password []byte, data []byte, rounds int) (string, error) {
|
||||||
|
|
||||||
|
salt := make([]byte, 8)
|
||||||
|
_, err := io.ReadFull(rand.Reader, salt)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
key, err := scrypt.Key(password, salt, rounds, 8, 1, 32)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
block, err := aes.NewCipher(key)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
h := sha256.New()
|
||||||
|
h.Write(data)
|
||||||
|
checksum := h.Sum(nil)
|
||||||
|
if len(checksum) != 32 {
|
||||||
|
return "", errors.New("wrong cs size")
|
||||||
|
}
|
||||||
|
|
||||||
|
ciphertext := make([]byte, 32+len(data))
|
||||||
|
|
||||||
|
iv := make([]byte, aes.BlockSize)
|
||||||
|
_, err = io.ReadFull(rand.Reader, iv)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
combinedData := make([]byte, 0, 32+len(data))
|
||||||
|
combinedData = append(combinedData, checksum...)
|
||||||
|
combinedData = append(combinedData, data...)
|
||||||
|
|
||||||
|
cfb := cipher.NewCFBEncrypter(block, iv)
|
||||||
|
cfb.XORKeyStream(ciphertext, combinedData)
|
||||||
|
|
||||||
|
pl := aesPayload{
|
||||||
|
Salt: salt,
|
||||||
|
IV: iv,
|
||||||
|
Data: ciphertext,
|
||||||
|
Version: 1,
|
||||||
|
Rounds: rounds,
|
||||||
|
}
|
||||||
|
|
||||||
|
jbin, err := json.Marshal(pl)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
res := base32.StdEncoding.WithPadding(base32.NoPadding).EncodeToString(jbin)
|
||||||
|
|
||||||
|
return res, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func DecryptAESSimple(password []byte, encText string) ([]byte, error) {
|
||||||
|
|
||||||
|
jbin, err := base32.StdEncoding.WithPadding(base32.NoPadding).DecodeString(encText)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
var pl aesPayload
|
||||||
|
err = json.Unmarshal(jbin, &pl)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if pl.Version != 1 {
|
||||||
|
return nil, errors.New("unsupported version")
|
||||||
|
}
|
||||||
|
|
||||||
|
key, err := scrypt.Key(password, pl.Salt, pl.Rounds, 8, 1, 32) // this is not 100% correct, rounds too low and salt is missing
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
block, err := aes.NewCipher(key)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
dest := make([]byte, len(pl.Data))
|
||||||
|
|
||||||
|
cfb := cipher.NewCFBDecrypter(block, pl.IV)
|
||||||
|
cfb.XORKeyStream(dest, pl.Data)
|
||||||
|
|
||||||
|
if len(dest) < 32 {
|
||||||
|
return nil, errors.New("payload too small")
|
||||||
|
}
|
||||||
|
|
||||||
|
chck := dest[:32]
|
||||||
|
data := dest[32:]
|
||||||
|
|
||||||
|
h := sha256.New()
|
||||||
|
h.Write(data)
|
||||||
|
chck2 := h.Sum(nil)
|
||||||
|
|
||||||
|
if !bytes.Equal(chck, chck2) {
|
||||||
|
return nil, errors.New("checksum mismatch")
|
||||||
|
}
|
||||||
|
|
||||||
|
return data, nil
|
||||||
|
}
|
35
cryptext/aes_test.go
Normal file
35
cryptext/aes_test.go
Normal file
@@ -0,0 +1,35 @@
|
|||||||
|
package cryptext
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestEncryptAESSimple(t *testing.T) {
|
||||||
|
|
||||||
|
pw := []byte("hunter12")
|
||||||
|
|
||||||
|
str1 := []byte("Hello World")
|
||||||
|
|
||||||
|
str2, err := EncryptAESSimple(pw, str1, 512)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Printf("%s\n", str2)
|
||||||
|
|
||||||
|
str3, err := DecryptAESSimple(pw, str2)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
assertEqual(t, string(str1), string(str3))
|
||||||
|
|
||||||
|
str4, err := EncryptAESSimple(pw, str3, 512)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
assertNotEqual(t, string(str2), string(str4))
|
||||||
|
|
||||||
|
}
|
@@ -23,3 +23,9 @@ func assertEqual(t *testing.T, actual string, expected string) {
|
|||||||
t.Errorf("values differ: Actual: '%v', Expected: '%v'", actual, expected)
|
t.Errorf("values differ: Actual: '%v', Expected: '%v'", actual, expected)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func assertNotEqual(t *testing.T, actual string, expected string) {
|
||||||
|
if actual == expected {
|
||||||
|
t.Errorf("values do not differ: Actual: '%v', Expected: '%v'", actual, expected)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
@@ -12,7 +12,7 @@ func init() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func TestResultCache1(t *testing.T) {
|
func TestResultCache1(t *testing.T) {
|
||||||
cache := NewLRUMap[string](8)
|
cache := NewLRUMap[string, string](8)
|
||||||
verifyLRUList(cache, t)
|
verifyLRUList(cache, t)
|
||||||
|
|
||||||
key := randomKey()
|
key := randomKey()
|
||||||
@@ -50,7 +50,7 @@ func TestResultCache1(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func TestResultCache2(t *testing.T) {
|
func TestResultCache2(t *testing.T) {
|
||||||
cache := NewLRUMap[string](8)
|
cache := NewLRUMap[string, string](8)
|
||||||
verifyLRUList(cache, t)
|
verifyLRUList(cache, t)
|
||||||
|
|
||||||
key1 := "key1"
|
key1 := "key1"
|
||||||
@@ -150,7 +150,7 @@ func TestResultCache2(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func TestResultCache3(t *testing.T) {
|
func TestResultCache3(t *testing.T) {
|
||||||
cache := NewLRUMap[string](8)
|
cache := NewLRUMap[string, string](8)
|
||||||
verifyLRUList(cache, t)
|
verifyLRUList(cache, t)
|
||||||
|
|
||||||
key1 := "key1"
|
key1 := "key1"
|
||||||
@@ -173,7 +173,7 @@ func TestResultCache3(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// does a basic consistency check over the internal cache representation
|
// does a basic consistency check over the internal cache representation
|
||||||
func verifyLRUList[TData any](cache *LRUMap[TData], t *testing.T) {
|
func verifyLRUList[TKey comparable, TData any](cache *LRUMap[TKey, TData], t *testing.T) {
|
||||||
size := 0
|
size := 0
|
||||||
|
|
||||||
tailFound := false
|
tailFound := false
|
||||||
|
1
go.mod
1
go.mod
@@ -9,5 +9,6 @@ require (
|
|||||||
|
|
||||||
require (
|
require (
|
||||||
github.com/jmoiron/sqlx v1.3.5 // indirect
|
github.com/jmoiron/sqlx v1.3.5 // indirect
|
||||||
|
go.mongodb.org/mongo-driver v1.11.1 // indirect
|
||||||
golang.org/x/crypto v0.4.0 // indirect
|
golang.org/x/crypto v0.4.0 // indirect
|
||||||
)
|
)
|
||||||
|
35
go.sum
35
go.sum
@@ -1,15 +1,50 @@
|
|||||||
|
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
|
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
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/golang/snappy v0.0.1/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
|
||||||
|
github.com/google/go-cmp v0.5.2/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
|
||||||
github.com/jmoiron/sqlx v1.3.5 h1:vFFPA71p1o5gAeqtEAwLU4dnX2napprKtHr7PYIcN3g=
|
github.com/jmoiron/sqlx v1.3.5 h1:vFFPA71p1o5gAeqtEAwLU4dnX2napprKtHr7PYIcN3g=
|
||||||
github.com/jmoiron/sqlx v1.3.5/go.mod h1:nRVWtLre0KfCLJvgxzCsLVMogSvQ1zNJtpYr2Ccp0mQ=
|
github.com/jmoiron/sqlx v1.3.5/go.mod h1:nRVWtLre0KfCLJvgxzCsLVMogSvQ1zNJtpYr2Ccp0mQ=
|
||||||
|
github.com/klauspost/compress v1.13.6/go.mod h1:/3/Vjq9QcHkK5uEr5lBEmyoZ1iFhe47etQ6QUkpK6sk=
|
||||||
|
github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
|
||||||
|
github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
|
||||||
|
github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
|
||||||
github.com/lib/pq v1.2.0/go.mod h1:5WUZQaWbwv1U+lTReE5YruASi9Al49XbQIvNi/34Woo=
|
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=
|
github.com/mattn/go-sqlite3 v1.14.6/go.mod h1:NyWgC/yNuGj7Q9rpYnZvas74GogHl5/Z4A/KQRfk6bU=
|
||||||
|
github.com/montanaflynn/stats v0.0.0-20171201202039-1bf9dbcd8cbe/go.mod h1:wL8QJuTMNUDYhXwkmfOly8iTdp5TEcJFWZD2D7SIkUc=
|
||||||
|
github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
||||||
|
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||||
|
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||||
|
github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||||
|
github.com/tidwall/pretty v1.0.0/go.mod h1:XNkn88O1ChpSDQmQeStsy+sBenx6DDtFZJxhVysOjyk=
|
||||||
|
github.com/xdg-go/pbkdf2 v1.0.0/go.mod h1:jrpuAogTd400dnrH08LKmI/xc1MbPOebTwRqcT5RDeI=
|
||||||
|
github.com/xdg-go/scram v1.1.1/go.mod h1:RaEWvsqvNKKvBPvcKeFjrG2cJqOkHTiyTpzz23ni57g=
|
||||||
|
github.com/xdg-go/stringprep v1.0.3/go.mod h1:W3f5j4i+9rC0kuIEJL0ky1VpHXQU3ocBgklLGvcBnW8=
|
||||||
|
github.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d/go.mod h1:rHwXgn7JulP+udvsHwJoVG1YGAP6VLg4y9I5dyZdqmA=
|
||||||
|
go.mongodb.org/mongo-driver v1.11.1 h1:QP0znIRTuL0jf1oBQoAoM0C6ZJfBK4kx0Uumtv1A7w8=
|
||||||
|
go.mongodb.org/mongo-driver v1.11.1/go.mod h1:s7p5vEtfbeR1gYi6pnj3c3/urpbLv2T5Sfd6Rp2HBB8=
|
||||||
|
golang.org/x/crypto v0.0.0-20220622213112-05595931fe9d/go.mod h1:IxCIyHEi3zRg3s0A5j5BB6A9Jmi73HwBIUl50j+osU4=
|
||||||
golang.org/x/crypto v0.4.0 h1:UVQgzMY87xqpKNgb+kDsll2Igd33HszWHFLmpaRMq/8=
|
golang.org/x/crypto v0.4.0 h1:UVQgzMY87xqpKNgb+kDsll2Igd33HszWHFLmpaRMq/8=
|
||||||
golang.org/x/crypto v0.4.0/go.mod h1:3quD/ATkf6oY+rnes5c3ExXTbLc8mueNue5/DoinL80=
|
golang.org/x/crypto v0.4.0/go.mod h1:3quD/ATkf6oY+rnes5c3ExXTbLc8mueNue5/DoinL80=
|
||||||
|
golang.org/x/net v0.0.0-20211112202133-69e39bad7dc2/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=
|
||||||
|
golang.org/x/sync v0.0.0-20210220032951-036812b2e83c/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
|
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
|
golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
|
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
golang.org/x/sys v0.1.0 h1:kunALQeHf1/185U1i0GOB/fy1IPRDDpuoOOqRReG57U=
|
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/sys v0.1.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
golang.org/x/sys v0.3.0 h1:w8ZOecv6NaNa/zC8944JTU3vz4u6Lagfk4RPQxv92NQ=
|
golang.org/x/sys v0.3.0 h1:w8ZOecv6NaNa/zC8944JTU3vz4u6Lagfk4RPQxv92NQ=
|
||||||
golang.org/x/sys v0.3.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
golang.org/x/sys v0.3.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
|
||||||
golang.org/x/term v0.1.0 h1:g6Z6vPFA9dYBAF7DWcH6sCcOntplXsDKcliusYijMlw=
|
golang.org/x/term v0.1.0 h1:g6Z6vPFA9dYBAF7DWcH6sCcOntplXsDKcliusYijMlw=
|
||||||
golang.org/x/term v0.1.0/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
|
golang.org/x/term v0.1.0/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
|
||||||
golang.org/x/term v0.3.0 h1:qoo4akIqOcDME5bhc/NgxUdovd6BSS2uMsVjB56q1xI=
|
golang.org/x/term v0.3.0 h1:qoo4akIqOcDME5bhc/NgxUdovd6BSS2uMsVjB56q1xI=
|
||||||
golang.org/x/term v0.3.0/go.mod h1:q750SLmJuPmVoN1blW3UFBPREJfb1KmY3vwxfr+nFDA=
|
golang.org/x/term v0.3.0/go.mod h1:q750SLmJuPmVoN1blW3UFBPREJfb1KmY3vwxfr+nFDA=
|
||||||
|
golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
|
||||||
|
golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
|
||||||
|
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||||
|
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
|
gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
|
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
@@ -265,6 +265,46 @@ func ArrMap[T1 any, T2 any](arr []T1, conv func(v T1) T2) []T2 {
|
|||||||
return r
|
return r
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func ArrMapExt[T1 any, T2 any](arr []T1, conv func(idx int, v T1) T2) []T2 {
|
||||||
|
r := make([]T2, len(arr))
|
||||||
|
for i, v := range arr {
|
||||||
|
r[i] = conv(i, v)
|
||||||
|
}
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
|
||||||
|
func ArrMapErr[T1 any, T2 any](arr []T1, conv func(v T1) (T2, error)) ([]T2, error) {
|
||||||
|
var err error
|
||||||
|
r := make([]T2, len(arr))
|
||||||
|
for i, v := range arr {
|
||||||
|
r[i], err = conv(v)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return r, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func ArrFilterMap[T1 any, T2 any](arr []T1, filter func(v T1) bool, conv func(v T1) T2) []T2 {
|
||||||
|
r := make([]T2, 0, len(arr))
|
||||||
|
for _, v := range arr {
|
||||||
|
if filter(v) {
|
||||||
|
r = append(r, conv(v))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
|
||||||
|
func ArrFilter[T any](arr []T, filter func(v T) bool) []T {
|
||||||
|
r := make([]T, 0, len(arr))
|
||||||
|
for _, v := range arr {
|
||||||
|
if filter(v) {
|
||||||
|
r = append(r, v)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
|
||||||
func ArrSum[T NumberConstraint](arr []T) T {
|
func ArrSum[T NumberConstraint](arr []T) T {
|
||||||
var r T = 0
|
var r T = 0
|
||||||
for _, v := range arr {
|
for _, v := range arr {
|
||||||
@@ -272,3 +312,19 @@ func ArrSum[T NumberConstraint](arr []T) T {
|
|||||||
}
|
}
|
||||||
return r
|
return r
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func ArrFlatten[T1 any, T2 any](arr []T1, conv func(v T1) []T2) []T2 {
|
||||||
|
r := make([]T2, 0, len(arr))
|
||||||
|
for _, v1 := range arr {
|
||||||
|
r = append(r, conv(v1)...)
|
||||||
|
}
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
|
||||||
|
func ArrFlattenDirect[T1 any](arr [][]T1) []T1 {
|
||||||
|
r := make([]T1, 0, len(arr))
|
||||||
|
for _, v1 := range arr {
|
||||||
|
r = append(r, v1...)
|
||||||
|
}
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
178
langext/base58.go
Normal file
178
langext/base58.go
Normal file
@@ -0,0 +1,178 @@
|
|||||||
|
package langext
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"errors"
|
||||||
|
"math/big"
|
||||||
|
)
|
||||||
|
|
||||||
|
// shamelessly stolen from https://github.com/btcsuite/
|
||||||
|
|
||||||
|
type B58Encoding struct {
|
||||||
|
bigRadix [11]*big.Int
|
||||||
|
bigRadix10 *big.Int
|
||||||
|
alphabet string
|
||||||
|
alphabetIdx0 byte
|
||||||
|
b58 [256]byte
|
||||||
|
}
|
||||||
|
|
||||||
|
var Base58DefaultEncoding = newBase58Encoding("123456789ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijkmnopqrstuvwxyz")
|
||||||
|
var Base58FlickrEncoding = newBase58Encoding("123456789abcdefghijkmnopqrstuvwxyzABCDEFGHJKLMNPQRSTUVWXYZ")
|
||||||
|
var Base58RippleEncoding = newBase58Encoding("rpshnaf39wBUDNEGHJKLM4PQRST7VWXYZ2bcdeCg65jkm8oFqi1tuvAxyz")
|
||||||
|
var Base58BitcoinEncoding = newBase58Encoding("123456789ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijkmnopqrstuvwxyz")
|
||||||
|
|
||||||
|
func newBase58Encoding(alphabet string) *B58Encoding {
|
||||||
|
bigRadix10 := big.NewInt(58 * 58 * 58 * 58 * 58 * 58 * 58 * 58 * 58 * 58)
|
||||||
|
enc := &B58Encoding{
|
||||||
|
alphabet: alphabet,
|
||||||
|
alphabetIdx0: '1',
|
||||||
|
bigRadix: [...]*big.Int{
|
||||||
|
big.NewInt(0),
|
||||||
|
big.NewInt(58),
|
||||||
|
big.NewInt(58 * 58),
|
||||||
|
big.NewInt(58 * 58 * 58),
|
||||||
|
big.NewInt(58 * 58 * 58 * 58),
|
||||||
|
big.NewInt(58 * 58 * 58 * 58 * 58),
|
||||||
|
big.NewInt(58 * 58 * 58 * 58 * 58 * 58),
|
||||||
|
big.NewInt(58 * 58 * 58 * 58 * 58 * 58 * 58),
|
||||||
|
big.NewInt(58 * 58 * 58 * 58 * 58 * 58 * 58 * 58),
|
||||||
|
big.NewInt(58 * 58 * 58 * 58 * 58 * 58 * 58 * 58 * 58),
|
||||||
|
bigRadix10,
|
||||||
|
},
|
||||||
|
bigRadix10: bigRadix10,
|
||||||
|
}
|
||||||
|
|
||||||
|
b58 := make([]byte, 0, 256)
|
||||||
|
|
||||||
|
for i := byte(0); i < 32; i++ {
|
||||||
|
for j := byte(0); j < 8; j++ {
|
||||||
|
|
||||||
|
b := i*8 + j
|
||||||
|
|
||||||
|
idx := bytes.IndexByte([]byte(alphabet), b)
|
||||||
|
if idx == -1 {
|
||||||
|
b58 = append(b58, 255)
|
||||||
|
} else {
|
||||||
|
b58 = append(b58, byte(idx))
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
enc.b58 = *((*[256]byte)(b58))
|
||||||
|
|
||||||
|
return enc
|
||||||
|
}
|
||||||
|
|
||||||
|
func (enc *B58Encoding) EncodeString(src string) (string, error) {
|
||||||
|
v, err := enc.Encode([]byte(src))
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
return string(v), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (enc *B58Encoding) Encode(src []byte) ([]byte, error) {
|
||||||
|
x := new(big.Int)
|
||||||
|
x.SetBytes(src)
|
||||||
|
|
||||||
|
// maximum length of output is log58(2^(8*len(b))) == len(b) * 8 / log(58)
|
||||||
|
maxlen := int(float64(len(src))*1.365658237309761) + 1
|
||||||
|
answer := make([]byte, 0, maxlen)
|
||||||
|
mod := new(big.Int)
|
||||||
|
for x.Sign() > 0 {
|
||||||
|
// Calculating with big.Int is slow for each iteration.
|
||||||
|
// x, mod = x / 58, x % 58
|
||||||
|
//
|
||||||
|
// Instead we can try to do as much calculations on int64.
|
||||||
|
// x, mod = x / 58^10, x % 58^10
|
||||||
|
//
|
||||||
|
// Which will give us mod, which is 10 digit base58 number.
|
||||||
|
// We'll loop that 10 times to convert to the answer.
|
||||||
|
|
||||||
|
x.DivMod(x, enc.bigRadix10, mod)
|
||||||
|
if x.Sign() == 0 {
|
||||||
|
// When x = 0, we need to ensure we don't add any extra zeros.
|
||||||
|
m := mod.Int64()
|
||||||
|
for m > 0 {
|
||||||
|
answer = append(answer, enc.alphabet[m%58])
|
||||||
|
m /= 58
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
m := mod.Int64()
|
||||||
|
for i := 0; i < 10; i++ {
|
||||||
|
answer = append(answer, enc.alphabet[m%58])
|
||||||
|
m /= 58
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// leading zero bytes
|
||||||
|
for _, i := range src {
|
||||||
|
if i != 0 {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
answer = append(answer, enc.alphabetIdx0)
|
||||||
|
}
|
||||||
|
|
||||||
|
// reverse
|
||||||
|
alen := len(answer)
|
||||||
|
for i := 0; i < alen/2; i++ {
|
||||||
|
answer[i], answer[alen-1-i] = answer[alen-1-i], answer[i]
|
||||||
|
}
|
||||||
|
|
||||||
|
return answer, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (enc *B58Encoding) DecodeString(src string) (string, error) {
|
||||||
|
v, err := enc.Decode([]byte(src))
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
return string(v), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (enc *B58Encoding) Decode(src []byte) ([]byte, error) {
|
||||||
|
answer := big.NewInt(0)
|
||||||
|
scratch := new(big.Int)
|
||||||
|
|
||||||
|
for t := src; len(t) > 0; {
|
||||||
|
n := len(t)
|
||||||
|
if n > 10 {
|
||||||
|
n = 10
|
||||||
|
}
|
||||||
|
|
||||||
|
total := uint64(0)
|
||||||
|
for _, v := range t[:n] {
|
||||||
|
if v > 255 {
|
||||||
|
return []byte{}, errors.New("invalid char in input")
|
||||||
|
}
|
||||||
|
|
||||||
|
tmp := enc.b58[v]
|
||||||
|
if tmp == 255 {
|
||||||
|
return []byte{}, errors.New("invalid char in input")
|
||||||
|
}
|
||||||
|
total = total*58 + uint64(tmp)
|
||||||
|
}
|
||||||
|
|
||||||
|
answer.Mul(answer, enc.bigRadix[n])
|
||||||
|
scratch.SetUint64(total)
|
||||||
|
answer.Add(answer, scratch)
|
||||||
|
|
||||||
|
t = t[n:]
|
||||||
|
}
|
||||||
|
|
||||||
|
tmpval := answer.Bytes()
|
||||||
|
|
||||||
|
var numZeros int
|
||||||
|
for numZeros = 0; numZeros < len(src); numZeros++ {
|
||||||
|
if src[numZeros] != enc.alphabetIdx0 {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
flen := numZeros + len(tmpval)
|
||||||
|
val := make([]byte, flen)
|
||||||
|
copy(val[numZeros:], tmpval)
|
||||||
|
|
||||||
|
return val, nil
|
||||||
|
}
|
67
langext/base58_test.go
Normal file
67
langext/base58_test.go
Normal file
@@ -0,0 +1,67 @@
|
|||||||
|
package langext
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func _encStr(t *testing.T, enc *B58Encoding, v string) string {
|
||||||
|
v, err := enc.EncodeString(v)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
}
|
||||||
|
return v
|
||||||
|
}
|
||||||
|
|
||||||
|
func _decStr(t *testing.T, enc *B58Encoding, v string) string {
|
||||||
|
v, err := enc.DecodeString(v)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
}
|
||||||
|
return v
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestBase58DefaultEncoding(t *testing.T) {
|
||||||
|
assertEqual(t, _encStr(t, Base58DefaultEncoding, "Hello"), "9Ajdvzr")
|
||||||
|
assertEqual(t, _encStr(t, Base58DefaultEncoding, "If debugging is the process of removing software bugs, then programming must be the process of putting them in."), "48638SMcJuah5okqPx4kCVf5d8QAdgbdNf28g7ReY13prUENNbMyssjq5GjsrJHF5zeZfqs4uJMUJHr7VbrU4XBUZ2Fw9DVtqtn9N1eXucEWSEZahXV6w4ysGSWqGdpeYTJf1MdDzTg8vfcQViifJjZX")
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestBase58DefaultDecoding(t *testing.T) {
|
||||||
|
assertEqual(t, _decStr(t, Base58DefaultEncoding, "9Ajdvzr"), "Hello")
|
||||||
|
assertEqual(t, _decStr(t, Base58DefaultEncoding, "48638SMcJuah5okqPx4kCVf5d8QAdgbdNf28g7ReY13prUENNbMyssjq5GjsrJHF5zeZfqs4uJMUJHr7VbrU4XBUZ2Fw9DVtqtn9N1eXucEWSEZahXV6w4ysGSWqGdpeYTJf1MdDzTg8vfcQViifJjZX"), "If debugging is the process of removing software bugs, then programming must be the process of putting them in.")
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestBase58RippleEncoding(t *testing.T) {
|
||||||
|
assertEqual(t, _encStr(t, Base58RippleEncoding, "Hello"), "9wjdvzi")
|
||||||
|
assertEqual(t, _encStr(t, Base58RippleEncoding, "If debugging is the process of removing software bugs, then programming must be the process of putting them in."), "h3as3SMcJu26nokqPxhkUVCnd3Qwdgbd4Cp3gfReYrsFi7N44bMy11jqnGj1iJHEnzeZCq1huJM7JHifVbi7hXB7ZpEA9DVtqt894reXucNWSNZ26XVaAhy1GSWqGdFeYTJCrMdDzTg3vCcQV55CJjZX")
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestBase58RippleDecoding(t *testing.T) {
|
||||||
|
assertEqual(t, _decStr(t, Base58RippleEncoding, "9wjdvzi"), "Hello")
|
||||||
|
assertEqual(t, _decStr(t, Base58RippleEncoding, "h3as3SMcJu26nokqPxhkUVCnd3Qwdgbd4Cp3gfReYrsFi7N44bMy11jqnGj1iJHEnzeZCq1huJM7JHifVbi7hXB7ZpEA9DVtqt894reXucNWSNZ26XVaAhy1GSWqGdFeYTJCrMdDzTg3vCcQV55CJjZX"), "If debugging is the process of removing software bugs, then programming must be the process of putting them in.")
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestBase58BitcoinEncoding(t *testing.T) {
|
||||||
|
assertEqual(t, _encStr(t, Base58BitcoinEncoding, "Hello"), "9Ajdvzr")
|
||||||
|
assertEqual(t, _encStr(t, Base58BitcoinEncoding, "If debugging is the process of removing software bugs, then programming must be the process of putting them in."), "48638SMcJuah5okqPx4kCVf5d8QAdgbdNf28g7ReY13prUENNbMyssjq5GjsrJHF5zeZfqs4uJMUJHr7VbrU4XBUZ2Fw9DVtqtn9N1eXucEWSEZahXV6w4ysGSWqGdpeYTJf1MdDzTg8vfcQViifJjZX")
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestBase58BitcoinDecoding(t *testing.T) {
|
||||||
|
assertEqual(t, _decStr(t, Base58BitcoinEncoding, "9Ajdvzr"), "Hello")
|
||||||
|
assertEqual(t, _decStr(t, Base58BitcoinEncoding, "48638SMcJuah5okqPx4kCVf5d8QAdgbdNf28g7ReY13prUENNbMyssjq5GjsrJHF5zeZfqs4uJMUJHr7VbrU4XBUZ2Fw9DVtqtn9N1eXucEWSEZahXV6w4ysGSWqGdpeYTJf1MdDzTg8vfcQViifJjZX"), "If debugging is the process of removing software bugs, then programming must be the process of putting them in.")
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestBase58FlickrEncoding(t *testing.T) {
|
||||||
|
assertEqual(t, _encStr(t, Base58FlickrEncoding, "Hello"), "9aJCVZR")
|
||||||
|
assertEqual(t, _encStr(t, Base58FlickrEncoding, "If debugging is the process of removing software bugs, then programming must be the process of putting them in."), "48638rmBiUzG5NKQoX4KcuE5C8paCFACnE28F7qDx13PRtennAmYSSJQ5gJSRihf5ZDyEQS4UimtihR7uARt4wbty2fW9duTQTM9n1DwUBevreyzGwu6W4YSgrvQgCPDxsiE1mCdZsF8VEBpuHHEiJyw")
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestBase58FlickrDecoding(t *testing.T) {
|
||||||
|
assertEqual(t, _decStr(t, Base58FlickrEncoding, "9aJCVZR"), "Hello")
|
||||||
|
assertEqual(t, _decStr(t, Base58FlickrEncoding, "48638rmBiUzG5NKQoX4KcuE5C8paCFACnE28F7qDx13PRtennAmYSSJQ5gJSRihf5ZDyEQS4UimtihR7uARt4wbty2fW9duTQTM9n1DwUBevreyzGwu6W4YSgrvQgCPDxsiE1mCdZsF8VEBpuHHEiJyw"), "If debugging is the process of removing software bugs, then programming must be the process of putting them in.")
|
||||||
|
}
|
||||||
|
|
||||||
|
func assertEqual(t *testing.T, actual string, expected string) {
|
||||||
|
if actual != expected {
|
||||||
|
t.Errorf("values differ: Actual: '%v', Expected: '%v'", actual, expected)
|
||||||
|
}
|
||||||
|
}
|
@@ -7,3 +7,11 @@ func MapKeyArr[T comparable, V any](v map[T]V) []T {
|
|||||||
}
|
}
|
||||||
return result
|
return result
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func ArrToMap[T comparable, V any](a []V, keyfunc func(V) T) map[T]V {
|
||||||
|
result := make(map[T]V, len(a))
|
||||||
|
for _, v := range a {
|
||||||
|
result[keyfunc(v)] = v
|
||||||
|
}
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
@@ -37,3 +37,21 @@ func IsSliceSorted[T any](arr []T, less func(v1, v2 T) bool) bool {
|
|||||||
return less(arr[i1], arr[i2])
|
return less(arr[i1], arr[i2])
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func SortBy[TElem any, TSel OrderedConstraint](arr []TElem, selector func(v TElem) TSel) {
|
||||||
|
sort.Slice(arr, func(i1, i2 int) bool {
|
||||||
|
return selector(arr[i1]) < selector(arr[i2])
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func SortByStable[TElem any, TSel OrderedConstraint](arr []TElem, selector func(v TElem) TSel) {
|
||||||
|
sort.SliceStable(arr, func(i1, i2 int) bool {
|
||||||
|
return selector(arr[i1]) < selector(arr[i2])
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func IsSortedBy[TElem any, TSel OrderedConstraint](arr []TElem, selector func(v TElem) TSel) {
|
||||||
|
sort.SliceStable(arr, func(i1, i2 int) bool {
|
||||||
|
return selector(arr[i1]) < selector(arr[i2])
|
||||||
|
})
|
||||||
|
}
|
||||||
|
87
rfctime/interface.go
Normal file
87
rfctime/interface.go
Normal file
@@ -0,0 +1,87 @@
|
|||||||
|
package rfctime
|
||||||
|
|
||||||
|
import "time"
|
||||||
|
|
||||||
|
type RFCTime interface {
|
||||||
|
Time() time.Time
|
||||||
|
Serialize() string
|
||||||
|
|
||||||
|
UnmarshalJSON(bytes []byte) error
|
||||||
|
MarshalJSON() ([]byte, error)
|
||||||
|
|
||||||
|
MarshalBinary() ([]byte, error)
|
||||||
|
UnmarshalBinary(data []byte) error
|
||||||
|
|
||||||
|
GobEncode() ([]byte, error)
|
||||||
|
GobDecode(data []byte) error
|
||||||
|
|
||||||
|
MarshalText() ([]byte, error)
|
||||||
|
UnmarshalText(data []byte) error
|
||||||
|
|
||||||
|
After(u RFCTime) bool
|
||||||
|
Before(u RFCTime) bool
|
||||||
|
Equal(u RFCTime) bool
|
||||||
|
IsZero() bool
|
||||||
|
Date() (year int, month time.Month, day int)
|
||||||
|
Year() int
|
||||||
|
Month() time.Month
|
||||||
|
Day() int
|
||||||
|
Weekday() time.Weekday
|
||||||
|
ISOWeek() (year, week int)
|
||||||
|
Clock() (hour, min, sec int)
|
||||||
|
Hour() int
|
||||||
|
Minute() int
|
||||||
|
Second() int
|
||||||
|
Nanosecond() int
|
||||||
|
YearDay() int
|
||||||
|
Sub(u RFCTime) time.Duration
|
||||||
|
Unix() int64
|
||||||
|
UnixMilli() int64
|
||||||
|
UnixMicro() int64
|
||||||
|
UnixNano() int64
|
||||||
|
Format(layout string) string
|
||||||
|
GoString() string
|
||||||
|
String() string
|
||||||
|
}
|
||||||
|
|
||||||
|
type RFCDuration interface {
|
||||||
|
Time() time.Time
|
||||||
|
Serialize() string
|
||||||
|
|
||||||
|
UnmarshalJSON(bytes []byte) error
|
||||||
|
MarshalJSON() ([]byte, error)
|
||||||
|
|
||||||
|
MarshalBinary() ([]byte, error)
|
||||||
|
UnmarshalBinary(data []byte) error
|
||||||
|
|
||||||
|
GobEncode() ([]byte, error)
|
||||||
|
GobDecode(data []byte) error
|
||||||
|
|
||||||
|
MarshalText() ([]byte, error)
|
||||||
|
UnmarshalText(data []byte) error
|
||||||
|
|
||||||
|
After(u RFCTime) bool
|
||||||
|
Before(u RFCTime) bool
|
||||||
|
Equal(u RFCTime) bool
|
||||||
|
IsZero() bool
|
||||||
|
Date() (year int, month time.Month, day int)
|
||||||
|
Year() int
|
||||||
|
Month() time.Month
|
||||||
|
Day() int
|
||||||
|
Weekday() time.Weekday
|
||||||
|
ISOWeek() (year, week int)
|
||||||
|
Clock() (hour, min, sec int)
|
||||||
|
Hour() int
|
||||||
|
Minute() int
|
||||||
|
Second() int
|
||||||
|
Nanosecond() int
|
||||||
|
YearDay() int
|
||||||
|
Sub(u RFCTime) time.Duration
|
||||||
|
Unix() int64
|
||||||
|
UnixMilli() int64
|
||||||
|
UnixMicro() int64
|
||||||
|
UnixNano() int64
|
||||||
|
Format(layout string) string
|
||||||
|
GoString() string
|
||||||
|
String() string
|
||||||
|
}
|
203
rfctime/rfc3339.go
Normal file
203
rfctime/rfc3339.go
Normal file
@@ -0,0 +1,203 @@
|
|||||||
|
package rfctime
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"go.mongodb.org/mongo-driver/bson"
|
||||||
|
"go.mongodb.org/mongo-driver/bson/bsontype"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type RFC3339Time time.Time
|
||||||
|
|
||||||
|
func (t RFC3339Time) Time() time.Time {
|
||||||
|
return time.Time(t)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) MarshalBinary() ([]byte, error) {
|
||||||
|
return (time.Time)(t).MarshalBinary()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *RFC3339Time) UnmarshalBinary(data []byte) error {
|
||||||
|
return (*time.Time)(t).UnmarshalBinary(data)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) GobEncode() ([]byte, error) {
|
||||||
|
return (time.Time)(t).GobEncode()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *RFC3339Time) GobDecode(data []byte) error {
|
||||||
|
return (*time.Time)(t).GobDecode(data)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *RFC3339Time) UnmarshalJSON(data []byte) error {
|
||||||
|
str := ""
|
||||||
|
if err := json.Unmarshal(data, &str); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
t0, err := time.Parse(t.FormatStr(), str)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
*t = RFC3339Time(t0)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) MarshalJSON() ([]byte, error) {
|
||||||
|
str := t.Time().Format(t.FormatStr())
|
||||||
|
return json.Marshal(str)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) MarshalText() ([]byte, error) {
|
||||||
|
b := make([]byte, 0, len(t.FormatStr()))
|
||||||
|
return t.Time().AppendFormat(b, t.FormatStr()), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *RFC3339Time) UnmarshalText(data []byte) error {
|
||||||
|
var err error
|
||||||
|
v, err := time.Parse(t.FormatStr(), string(data))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
tt := RFC3339Time(v)
|
||||||
|
*t = tt
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *RFC3339Time) UnmarshalBSONValue(bt bsontype.Type, data []byte) error {
|
||||||
|
if bt != bsontype.DateTime {
|
||||||
|
return errors.New(fmt.Sprintf("cannot unmarshal %v into RFC3339Time", bt))
|
||||||
|
}
|
||||||
|
var tt time.Time
|
||||||
|
err := bson.Unmarshal(data, &tt)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
*t = RFC3339Time(tt)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) MarshalBSONValue() (bsontype.Type, []byte, error) {
|
||||||
|
return bson.MarshalValue(time.Time(t))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) Serialize() string {
|
||||||
|
return t.Time().Format(t.FormatStr())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) FormatStr() string {
|
||||||
|
return time.RFC3339
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) After(u RFCTime) bool {
|
||||||
|
return t.Time().After(u.Time())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) Before(u RFCTime) bool {
|
||||||
|
return t.Time().Before(u.Time())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) Equal(u RFCTime) bool {
|
||||||
|
return t.Time().Equal(u.Time())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) IsZero() bool {
|
||||||
|
return t.Time().IsZero()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) Date() (year int, month time.Month, day int) {
|
||||||
|
return t.Time().Date()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) Year() int {
|
||||||
|
return t.Time().Year()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) Month() time.Month {
|
||||||
|
return t.Time().Month()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) Day() int {
|
||||||
|
return t.Time().Day()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) Weekday() time.Weekday {
|
||||||
|
return t.Time().Weekday()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) ISOWeek() (year, week int) {
|
||||||
|
return t.Time().ISOWeek()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) Clock() (hour, min, sec int) {
|
||||||
|
return t.Time().Clock()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) Hour() int {
|
||||||
|
return t.Time().Hour()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) Minute() int {
|
||||||
|
return t.Time().Minute()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) Second() int {
|
||||||
|
return t.Time().Second()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) Nanosecond() int {
|
||||||
|
return t.Time().Nanosecond()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) YearDay() int {
|
||||||
|
return t.Time().YearDay()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) Add(d time.Duration) RFC3339Time {
|
||||||
|
return RFC3339Time(t.Time().Add(d))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) Sub(u RFCTime) time.Duration {
|
||||||
|
return t.Time().Sub(u.Time())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) AddDate(years int, months int, days int) RFC3339Time {
|
||||||
|
return RFC3339Time(t.Time().AddDate(years, months, days))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) Unix() int64 {
|
||||||
|
return t.Time().Unix()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) UnixMilli() int64 {
|
||||||
|
return t.Time().UnixMilli()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) UnixMicro() int64 {
|
||||||
|
return t.Time().UnixMicro()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) UnixNano() int64 {
|
||||||
|
return t.Time().UnixNano()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) Format(layout string) string {
|
||||||
|
return t.Time().Format(layout)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) GoString() string {
|
||||||
|
return t.Time().GoString()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339Time) String() string {
|
||||||
|
return t.Time().String()
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewRFC3339(t time.Time) RFC3339Time {
|
||||||
|
return RFC3339Time(t)
|
||||||
|
}
|
||||||
|
|
||||||
|
func NowRFC3339() RFC3339Time {
|
||||||
|
return RFC3339Time(time.Now())
|
||||||
|
}
|
203
rfctime/rfc3339Nano.go
Normal file
203
rfctime/rfc3339Nano.go
Normal file
@@ -0,0 +1,203 @@
|
|||||||
|
package rfctime
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"go.mongodb.org/mongo-driver/bson"
|
||||||
|
"go.mongodb.org/mongo-driver/bson/bsontype"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type RFC3339NanoTime time.Time
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) Time() time.Time {
|
||||||
|
return time.Time(t)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) MarshalBinary() ([]byte, error) {
|
||||||
|
return (time.Time)(t).MarshalBinary()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *RFC3339NanoTime) UnmarshalBinary(data []byte) error {
|
||||||
|
return (*time.Time)(t).UnmarshalBinary(data)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) GobEncode() ([]byte, error) {
|
||||||
|
return (time.Time)(t).GobEncode()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *RFC3339NanoTime) GobDecode(data []byte) error {
|
||||||
|
return (*time.Time)(t).GobDecode(data)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *RFC3339NanoTime) UnmarshalJSON(data []byte) error {
|
||||||
|
str := ""
|
||||||
|
if err := json.Unmarshal(data, &str); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
t0, err := time.Parse(t.FormatStr(), str)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
*t = RFC3339NanoTime(t0)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) MarshalJSON() ([]byte, error) {
|
||||||
|
str := t.Time().Format(t.FormatStr())
|
||||||
|
return json.Marshal(str)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) MarshalText() ([]byte, error) {
|
||||||
|
b := make([]byte, 0, len(t.FormatStr()))
|
||||||
|
return t.Time().AppendFormat(b, t.FormatStr()), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *RFC3339NanoTime) UnmarshalText(data []byte) error {
|
||||||
|
var err error
|
||||||
|
v, err := time.Parse(t.FormatStr(), string(data))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
tt := RFC3339NanoTime(v)
|
||||||
|
*t = tt
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *RFC3339NanoTime) UnmarshalBSONValue(bt bsontype.Type, data []byte) error {
|
||||||
|
if bt != bsontype.DateTime {
|
||||||
|
return errors.New(fmt.Sprintf("cannot unmarshal %v into RFC3339NanoTime", bt))
|
||||||
|
}
|
||||||
|
var tt time.Time
|
||||||
|
err := bson.RawValue{Type: bt, Value: data}.Unmarshal(&tt)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
*t = RFC3339NanoTime(tt)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) MarshalBSONValue() (bsontype.Type, []byte, error) {
|
||||||
|
return bson.MarshalValue(time.Time(t))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) Serialize() string {
|
||||||
|
return t.Time().Format(t.FormatStr())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) FormatStr() string {
|
||||||
|
return time.RFC3339Nano
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) After(u RFCTime) bool {
|
||||||
|
return t.Time().After(u.Time())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) Before(u RFCTime) bool {
|
||||||
|
return t.Time().Before(u.Time())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) Equal(u RFCTime) bool {
|
||||||
|
return t.Time().Equal(u.Time())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) IsZero() bool {
|
||||||
|
return t.Time().IsZero()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) Date() (year int, month time.Month, day int) {
|
||||||
|
return t.Time().Date()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) Year() int {
|
||||||
|
return t.Time().Year()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) Month() time.Month {
|
||||||
|
return t.Time().Month()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) Day() int {
|
||||||
|
return t.Time().Day()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) Weekday() time.Weekday {
|
||||||
|
return t.Time().Weekday()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) ISOWeek() (year, week int) {
|
||||||
|
return t.Time().ISOWeek()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) Clock() (hour, min, sec int) {
|
||||||
|
return t.Time().Clock()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) Hour() int {
|
||||||
|
return t.Time().Hour()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) Minute() int {
|
||||||
|
return t.Time().Minute()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) Second() int {
|
||||||
|
return t.Time().Second()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) Nanosecond() int {
|
||||||
|
return t.Time().Nanosecond()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) YearDay() int {
|
||||||
|
return t.Time().YearDay()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) Add(d time.Duration) RFC3339NanoTime {
|
||||||
|
return RFC3339NanoTime(t.Time().Add(d))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) Sub(u RFCTime) time.Duration {
|
||||||
|
return t.Time().Sub(u.Time())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) AddDate(years int, months int, days int) RFC3339NanoTime {
|
||||||
|
return RFC3339NanoTime(t.Time().AddDate(years, months, days))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) Unix() int64 {
|
||||||
|
return t.Time().Unix()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) UnixMilli() int64 {
|
||||||
|
return t.Time().UnixMilli()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) UnixMicro() int64 {
|
||||||
|
return t.Time().UnixMicro()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) UnixNano() int64 {
|
||||||
|
return t.Time().UnixNano()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) Format(layout string) string {
|
||||||
|
return t.Time().Format(layout)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) GoString() string {
|
||||||
|
return t.Time().GoString()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t RFC3339NanoTime) String() string {
|
||||||
|
return t.Time().String()
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewRFC3339Nano(t time.Time) RFC3339NanoTime {
|
||||||
|
return RFC3339NanoTime(t)
|
||||||
|
}
|
||||||
|
|
||||||
|
func NowRFC3339Nano() RFC3339NanoTime {
|
||||||
|
return RFC3339NanoTime(time.Now())
|
||||||
|
}
|
52
rfctime/rfc3339Nano_test.go
Normal file
52
rfctime/rfc3339Nano_test.go
Normal file
@@ -0,0 +1,52 @@
|
|||||||
|
package rfctime
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestRoundtrip(t *testing.T) {
|
||||||
|
|
||||||
|
type Wrap struct {
|
||||||
|
Value RFC3339NanoTime `json:"v"`
|
||||||
|
}
|
||||||
|
|
||||||
|
val1 := NewRFC3339Nano(time.Unix(0, 1675951556820915171))
|
||||||
|
w1 := Wrap{val1}
|
||||||
|
|
||||||
|
jstr1, err := json.Marshal(w1)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if string(jstr1) != "{\"v\":\"2023-02-09T15:05:56.820915171+01:00\"}" {
|
||||||
|
t.Errorf(string(jstr1))
|
||||||
|
t.Errorf("repr differs")
|
||||||
|
}
|
||||||
|
|
||||||
|
w2 := Wrap{}
|
||||||
|
|
||||||
|
err = json.Unmarshal(jstr1, &w2)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
jstr2, err := json.Marshal(w2)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
assertEqual(t, string(jstr1), string(jstr2))
|
||||||
|
|
||||||
|
if !w1.Value.Equal(&w2.Value) {
|
||||||
|
t.Errorf("time differs")
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func assertEqual(t *testing.T, actual string, expected string) {
|
||||||
|
if actual != expected {
|
||||||
|
t.Errorf("values differ: Actual: '%v', Expected: '%v'", actual, expected)
|
||||||
|
}
|
||||||
|
}
|
59
rfctime/seconds.go
Normal file
59
rfctime/seconds.go
Normal file
@@ -0,0 +1,59 @@
|
|||||||
|
package rfctime
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"gogs.mikescher.com/BlackForestBytes/goext/timeext"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type SecondsF64 time.Duration
|
||||||
|
|
||||||
|
func (d SecondsF64) Duration() time.Duration {
|
||||||
|
return time.Duration(d)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d SecondsF64) String() string {
|
||||||
|
return d.Duration().String()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d SecondsF64) Nanoseconds() int64 {
|
||||||
|
return d.Duration().Nanoseconds()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d SecondsF64) Microseconds() int64 {
|
||||||
|
return d.Duration().Microseconds()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d SecondsF64) Milliseconds() int64 {
|
||||||
|
return d.Duration().Milliseconds()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d SecondsF64) Seconds() float64 {
|
||||||
|
return d.Duration().Seconds()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d SecondsF64) Minutes() float64 {
|
||||||
|
return d.Duration().Minutes()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d SecondsF64) Hours() float64 {
|
||||||
|
return d.Duration().Hours()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *SecondsF64) UnmarshalJSON(data []byte) error {
|
||||||
|
var secs float64 = 0
|
||||||
|
if err := json.Unmarshal(data, &secs); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
*d = SecondsF64(timeext.FromSeconds(secs))
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d SecondsF64) MarshalJSON() ([]byte, error) {
|
||||||
|
secs := d.Seconds()
|
||||||
|
return json.Marshal(secs)
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewSecondsF64(t time.Duration) SecondsF64 {
|
||||||
|
return SecondsF64(t)
|
||||||
|
}
|
176
rfctime/unix.go
Normal file
176
rfctime/unix.go
Normal file
@@ -0,0 +1,176 @@
|
|||||||
|
package rfctime
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"strconv"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type UnixTime time.Time
|
||||||
|
|
||||||
|
func (t UnixTime) Time() time.Time {
|
||||||
|
return time.Time(t)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) MarshalBinary() ([]byte, error) {
|
||||||
|
return (time.Time)(t).MarshalBinary()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *UnixTime) UnmarshalBinary(data []byte) error {
|
||||||
|
return (*time.Time)(t).UnmarshalBinary(data)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) GobEncode() ([]byte, error) {
|
||||||
|
return (time.Time)(t).GobEncode()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *UnixTime) GobDecode(data []byte) error {
|
||||||
|
return (*time.Time)(t).GobDecode(data)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *UnixTime) UnmarshalJSON(data []byte) error {
|
||||||
|
str := ""
|
||||||
|
if err := json.Unmarshal(data, &str); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
t0, err := strconv.ParseInt(str, 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
*t = UnixTime(time.Unix(t0, 0))
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) MarshalJSON() ([]byte, error) {
|
||||||
|
str := strconv.FormatInt(t.Time().Unix(), 10)
|
||||||
|
return json.Marshal(str)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) MarshalText() ([]byte, error) {
|
||||||
|
return []byte(strconv.FormatInt(t.Time().Unix(), 10)), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *UnixTime) UnmarshalText(data []byte) error {
|
||||||
|
t0, err := strconv.ParseInt(string(data), 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
*t = UnixTime(time.Unix(t0, 0))
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) Serialize() string {
|
||||||
|
return strconv.FormatInt(t.Time().Unix(), 10)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) After(u RFCTime) bool {
|
||||||
|
return t.Time().After(u.Time())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) Before(u RFCTime) bool {
|
||||||
|
return t.Time().Before(u.Time())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) Equal(u RFCTime) bool {
|
||||||
|
return t.Time().Equal(u.Time())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) IsZero() bool {
|
||||||
|
return t.Time().IsZero()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) Date() (year int, month time.Month, day int) {
|
||||||
|
return t.Time().Date()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) Year() int {
|
||||||
|
return t.Time().Year()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) Month() time.Month {
|
||||||
|
return t.Time().Month()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) Day() int {
|
||||||
|
return t.Time().Day()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) Weekday() time.Weekday {
|
||||||
|
return t.Time().Weekday()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) ISOWeek() (year, week int) {
|
||||||
|
return t.Time().ISOWeek()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) Clock() (hour, min, sec int) {
|
||||||
|
return t.Time().Clock()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) Hour() int {
|
||||||
|
return t.Time().Hour()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) Minute() int {
|
||||||
|
return t.Time().Minute()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) Second() int {
|
||||||
|
return t.Time().Second()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) Nanosecond() int {
|
||||||
|
return t.Time().Nanosecond()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) YearDay() int {
|
||||||
|
return t.Time().YearDay()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) Add(d time.Duration) UnixTime {
|
||||||
|
return UnixTime(t.Time().Add(d))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) Sub(u RFCTime) time.Duration {
|
||||||
|
return t.Time().Sub(u.Time())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) AddDate(years int, months int, days int) UnixTime {
|
||||||
|
return UnixTime(t.Time().AddDate(years, months, days))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) Unix() int64 {
|
||||||
|
return t.Time().Unix()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) UnixMilli() int64 {
|
||||||
|
return t.Time().UnixMilli()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) UnixMicro() int64 {
|
||||||
|
return t.Time().UnixMicro()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) UnixNano() int64 {
|
||||||
|
return t.Time().UnixNano()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) Format(layout string) string {
|
||||||
|
return t.Time().Format(layout)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) GoString() string {
|
||||||
|
return t.Time().GoString()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixTime) String() string {
|
||||||
|
return t.Time().String()
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewUnix(t time.Time) UnixTime {
|
||||||
|
return UnixTime(t)
|
||||||
|
}
|
||||||
|
|
||||||
|
func NowUnix() UnixTime {
|
||||||
|
return UnixTime(time.Now())
|
||||||
|
}
|
176
rfctime/unixMilli.go
Normal file
176
rfctime/unixMilli.go
Normal file
@@ -0,0 +1,176 @@
|
|||||||
|
package rfctime
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"strconv"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type UnixMilliTime time.Time
|
||||||
|
|
||||||
|
func (t UnixMilliTime) Time() time.Time {
|
||||||
|
return time.Time(t)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) MarshalBinary() ([]byte, error) {
|
||||||
|
return (time.Time)(t).MarshalBinary()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *UnixMilliTime) UnmarshalBinary(data []byte) error {
|
||||||
|
return (*time.Time)(t).UnmarshalBinary(data)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) GobEncode() ([]byte, error) {
|
||||||
|
return (time.Time)(t).GobEncode()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *UnixMilliTime) GobDecode(data []byte) error {
|
||||||
|
return (*time.Time)(t).GobDecode(data)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *UnixMilliTime) UnmarshalJSON(data []byte) error {
|
||||||
|
str := ""
|
||||||
|
if err := json.Unmarshal(data, &str); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
t0, err := strconv.ParseInt(str, 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
*t = UnixMilliTime(time.UnixMilli(t0))
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) MarshalJSON() ([]byte, error) {
|
||||||
|
str := strconv.FormatInt(t.Time().UnixMilli(), 10)
|
||||||
|
return json.Marshal(str)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) MarshalText() ([]byte, error) {
|
||||||
|
return []byte(strconv.FormatInt(t.Time().UnixMilli(), 10)), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *UnixMilliTime) UnmarshalText(data []byte) error {
|
||||||
|
t0, err := strconv.ParseInt(string(data), 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
*t = UnixMilliTime(time.UnixMilli(t0))
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) Serialize() string {
|
||||||
|
return strconv.FormatInt(t.Time().UnixMilli(), 10)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) After(u RFCTime) bool {
|
||||||
|
return t.Time().After(u.Time())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) Before(u RFCTime) bool {
|
||||||
|
return t.Time().Before(u.Time())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) Equal(u RFCTime) bool {
|
||||||
|
return t.Time().Equal(u.Time())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) IsZero() bool {
|
||||||
|
return t.Time().IsZero()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) Date() (year int, month time.Month, day int) {
|
||||||
|
return t.Time().Date()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) Year() int {
|
||||||
|
return t.Time().Year()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) Month() time.Month {
|
||||||
|
return t.Time().Month()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) Day() int {
|
||||||
|
return t.Time().Day()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) Weekday() time.Weekday {
|
||||||
|
return t.Time().Weekday()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) ISOWeek() (year, week int) {
|
||||||
|
return t.Time().ISOWeek()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) Clock() (hour, min, sec int) {
|
||||||
|
return t.Time().Clock()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) Hour() int {
|
||||||
|
return t.Time().Hour()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) Minute() int {
|
||||||
|
return t.Time().Minute()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) Second() int {
|
||||||
|
return t.Time().Second()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) Nanosecond() int {
|
||||||
|
return t.Time().Nanosecond()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) YearDay() int {
|
||||||
|
return t.Time().YearDay()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) Add(d time.Duration) UnixMilliTime {
|
||||||
|
return UnixMilliTime(t.Time().Add(d))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) Sub(u RFCTime) time.Duration {
|
||||||
|
return t.Time().Sub(u.Time())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) AddDate(years int, months int, days int) UnixMilliTime {
|
||||||
|
return UnixMilliTime(t.Time().AddDate(years, months, days))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) Unix() int64 {
|
||||||
|
return t.Time().Unix()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) UnixMilli() int64 {
|
||||||
|
return t.Time().UnixMilli()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) UnixMicro() int64 {
|
||||||
|
return t.Time().UnixMicro()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) UnixNano() int64 {
|
||||||
|
return t.Time().UnixNano()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) Format(layout string) string {
|
||||||
|
return t.Time().Format(layout)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) GoString() string {
|
||||||
|
return t.Time().GoString()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixMilliTime) String() string {
|
||||||
|
return t.Time().String()
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewUnixMilli(t time.Time) UnixMilliTime {
|
||||||
|
return UnixMilliTime(t)
|
||||||
|
}
|
||||||
|
|
||||||
|
func NowUnixMilli() UnixMilliTime {
|
||||||
|
return UnixMilliTime(time.Now())
|
||||||
|
}
|
176
rfctime/unixNano.go
Normal file
176
rfctime/unixNano.go
Normal file
@@ -0,0 +1,176 @@
|
|||||||
|
package rfctime
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"strconv"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type UnixNanoTime time.Time
|
||||||
|
|
||||||
|
func (t UnixNanoTime) Time() time.Time {
|
||||||
|
return time.Time(t)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) MarshalBinary() ([]byte, error) {
|
||||||
|
return (time.Time)(t).MarshalBinary()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *UnixNanoTime) UnmarshalBinary(data []byte) error {
|
||||||
|
return (*time.Time)(t).UnmarshalBinary(data)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) GobEncode() ([]byte, error) {
|
||||||
|
return (time.Time)(t).GobEncode()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *UnixNanoTime) GobDecode(data []byte) error {
|
||||||
|
return (*time.Time)(t).GobDecode(data)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *UnixNanoTime) UnmarshalJSON(data []byte) error {
|
||||||
|
str := ""
|
||||||
|
if err := json.Unmarshal(data, &str); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
t0, err := strconv.ParseInt(str, 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
*t = UnixNanoTime(time.Unix(0, t0))
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) MarshalJSON() ([]byte, error) {
|
||||||
|
str := strconv.FormatInt(t.Time().UnixNano(), 10)
|
||||||
|
return json.Marshal(str)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) MarshalText() ([]byte, error) {
|
||||||
|
return []byte(strconv.FormatInt(t.Time().UnixNano(), 10)), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *UnixNanoTime) UnmarshalText(data []byte) error {
|
||||||
|
t0, err := strconv.ParseInt(string(data), 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
*t = UnixNanoTime(time.Unix(0, t0))
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) Serialize() string {
|
||||||
|
return strconv.FormatInt(t.Time().UnixNano(), 10)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) After(u RFCTime) bool {
|
||||||
|
return t.Time().After(u.Time())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) Before(u RFCTime) bool {
|
||||||
|
return t.Time().Before(u.Time())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) Equal(u RFCTime) bool {
|
||||||
|
return t.Time().Equal(u.Time())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) IsZero() bool {
|
||||||
|
return t.Time().IsZero()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) Date() (year int, month time.Month, day int) {
|
||||||
|
return t.Time().Date()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) Year() int {
|
||||||
|
return t.Time().Year()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) Month() time.Month {
|
||||||
|
return t.Time().Month()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) Day() int {
|
||||||
|
return t.Time().Day()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) Weekday() time.Weekday {
|
||||||
|
return t.Time().Weekday()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) ISOWeek() (year, week int) {
|
||||||
|
return t.Time().ISOWeek()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) Clock() (hour, min, sec int) {
|
||||||
|
return t.Time().Clock()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) Hour() int {
|
||||||
|
return t.Time().Hour()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) Minute() int {
|
||||||
|
return t.Time().Minute()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) Second() int {
|
||||||
|
return t.Time().Second()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) Nanosecond() int {
|
||||||
|
return t.Time().Nanosecond()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) YearDay() int {
|
||||||
|
return t.Time().YearDay()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) Add(d time.Duration) UnixNanoTime {
|
||||||
|
return UnixNanoTime(t.Time().Add(d))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) Sub(u RFCTime) time.Duration {
|
||||||
|
return t.Time().Sub(u.Time())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) AddDate(years int, months int, days int) UnixNanoTime {
|
||||||
|
return UnixNanoTime(t.Time().AddDate(years, months, days))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) Unix() int64 {
|
||||||
|
return t.Time().Unix()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) UnixMilli() int64 {
|
||||||
|
return t.Time().UnixNano()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) UnixMicro() int64 {
|
||||||
|
return t.Time().UnixMicro()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) UnixNano() int64 {
|
||||||
|
return t.Time().UnixNano()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) Format(layout string) string {
|
||||||
|
return t.Time().Format(layout)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) GoString() string {
|
||||||
|
return t.Time().GoString()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t UnixNanoTime) String() string {
|
||||||
|
return t.Time().String()
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewUnixNano(t time.Time) UnixNanoTime {
|
||||||
|
return UnixNanoTime(t)
|
||||||
|
}
|
||||||
|
|
||||||
|
func NowUnixNano() UnixNanoTime {
|
||||||
|
return UnixNanoTime(time.Now())
|
||||||
|
}
|
@@ -2,58 +2,50 @@ package syncext
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"sync/atomic"
|
"gogs.mikescher.com/BlackForestBytes/goext/langext"
|
||||||
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
type AtomicBool struct {
|
type AtomicBool struct {
|
||||||
v int32
|
v bool
|
||||||
waiter chan bool // unbuffered
|
listener map[string]chan bool
|
||||||
|
lock sync.Mutex
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewAtomicBool(value bool) *AtomicBool {
|
func NewAtomicBool(value bool) *AtomicBool {
|
||||||
if value {
|
return &AtomicBool{
|
||||||
return &AtomicBool{v: 1, waiter: make(chan bool)}
|
v: value,
|
||||||
} else {
|
listener: make(map[string]chan bool),
|
||||||
return &AtomicBool{v: 0, waiter: make(chan bool)}
|
lock: sync.Mutex{},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *AtomicBool) Get() bool {
|
func (a *AtomicBool) Get() bool {
|
||||||
return atomic.LoadInt32(&a.v) == 1
|
a.lock.Lock()
|
||||||
|
defer a.lock.Unlock()
|
||||||
|
return a.v
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *AtomicBool) Set(value bool) {
|
func (a *AtomicBool) Set(value bool) {
|
||||||
if value {
|
a.lock.Lock()
|
||||||
atomic.StoreInt32(&a.v, 1)
|
defer a.lock.Unlock()
|
||||||
} else {
|
|
||||||
atomic.StoreInt32(&a.v, 0)
|
|
||||||
}
|
|
||||||
|
|
||||||
select {
|
a.v = value
|
||||||
case a.waiter <- value:
|
|
||||||
// message sent
|
for k, v := range a.listener {
|
||||||
default:
|
select {
|
||||||
// no receiver on channel
|
case v <- value:
|
||||||
|
// message sent
|
||||||
|
default:
|
||||||
|
// no receiver on channel
|
||||||
|
delete(a.listener, k)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *AtomicBool) Wait(waitFor bool) {
|
func (a *AtomicBool) Wait(waitFor bool) {
|
||||||
if a.Get() == waitFor {
|
_ = a.WaitWithContext(context.Background(), waitFor)
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
for {
|
|
||||||
if v, ok := ReadChannelWithTimeout(a.waiter, 128*time.Millisecond); ok {
|
|
||||||
if v == waitFor {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
if a.Get() == waitFor {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *AtomicBool) WaitWithTimeout(timeout time.Duration, waitFor bool) error {
|
func (a *AtomicBool) WaitWithTimeout(timeout time.Duration, waitFor bool) error {
|
||||||
@@ -71,12 +63,25 @@ func (a *AtomicBool) WaitWithContext(ctx context.Context, waitFor bool) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
uuid, _ := langext.NewHexUUID()
|
||||||
|
|
||||||
|
waitchan := make(chan bool)
|
||||||
|
|
||||||
|
a.lock.Lock()
|
||||||
|
a.listener[uuid] = waitchan
|
||||||
|
a.lock.Unlock()
|
||||||
|
defer func() {
|
||||||
|
a.lock.Lock()
|
||||||
|
delete(a.listener, uuid)
|
||||||
|
a.lock.Unlock()
|
||||||
|
}()
|
||||||
|
|
||||||
for {
|
for {
|
||||||
if err := ctx.Err(); err != nil {
|
if err := ctx.Err(); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
timeOut := 128 * time.Millisecond
|
timeOut := 1024 * time.Millisecond
|
||||||
|
|
||||||
if dl, ok := ctx.Deadline(); ok {
|
if dl, ok := ctx.Deadline(); ok {
|
||||||
timeOutMax := dl.Sub(time.Now())
|
timeOutMax := dl.Sub(time.Now())
|
||||||
@@ -87,7 +92,7 @@ func (a *AtomicBool) WaitWithContext(ctx context.Context, waitFor bool) error {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if v, ok := ReadChannelWithTimeout(a.waiter, timeOut); ok {
|
if v, ok := ReadChannelWithTimeout(waitchan, timeOut); ok {
|
||||||
if v == waitFor {
|
if v == waitFor {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user