eSlider
e04b037b33
stdlib flag dropped --hop after the query. One wrapper in internal/cli, source <(./bin/cli/complete.go bash). Gitea #34.
62 lines
1.4 KiB
Go
Executable File
62 lines
1.4 KiB
Go
Executable File
//usr/bin/env go run -tags=qa_stats "$0" "$@"; exit
|
|
//go:build qa_stats
|
|
//
|
|
// bin/qa/stats.go - DuckDB quantiles over a JSON number array or JSONL count.
|
|
//
|
|
// ./bin/qa/stats.go <<< '[1,2,3,4,5]'
|
|
// ./bin/qa/stats.go --jsonl rows.jsonl
|
|
//
|
|
// NOTE: never run `gofmt -w` on this file — it breaks the shebang.
|
|
// DuckDB CGO needs gcc/g++ (not Zig). After eval "$(bin/cgo/zig env)":
|
|
// CC=gcc CXX=g++ CGO_CFLAGS= CGO_LDFLAGS= ./bin/qa/stats.go
|
|
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
|
|
cliparse "github.com/eSlider/2dph/internal/cli"
|
|
"github.com/eSlider/2dph/internal/duckstats"
|
|
)
|
|
|
|
func main() {
|
|
os.Exit(run(os.Args[1:]))
|
|
}
|
|
|
|
func run(args []string) int {
|
|
c, err := cliparse.ParseQAStats(args)
|
|
if err != nil {
|
|
return cliparse.Fail(err)
|
|
}
|
|
jsonl := c.JSONL
|
|
if jsonl != "" {
|
|
n, err := duckstats.CountJSONL(jsonl)
|
|
if err != nil {
|
|
fmt.Fprintln(os.Stderr, err)
|
|
return 1
|
|
}
|
|
fmt.Printf("n: %d\n", n)
|
|
return 0
|
|
}
|
|
raw, err := io.ReadAll(os.Stdin)
|
|
if err != nil {
|
|
fmt.Fprintln(os.Stderr, err)
|
|
return 1
|
|
}
|
|
var samples []float64
|
|
if err := json.Unmarshal(raw, &samples); err != nil {
|
|
fmt.Fprintln(os.Stderr, err)
|
|
return 1
|
|
}
|
|
s, err := duckstats.Quantiles(samples)
|
|
if err != nil {
|
|
fmt.Fprintln(os.Stderr, err)
|
|
return 1
|
|
}
|
|
fmt.Printf("n: %d\nmin: %g\np50: %g\np95: %g\nmax: %g\navg: %g\n",
|
|
s.N, s.Min, s.P50, s.P95, s.Max, s.Avg)
|
|
return 0
|
|
}
|