mirror of
https://github.com/hanzo-ds/go.git
synced 2026-10-30 19:45:01 +00:00
zeekay
22b3f5ce2b
Feature-complete v2.44.0 (lib/churl, lib/col, BFloat16, QBit, scanTypeDuration, Options.Logger *slog.Logger) — fixes the [email protected].0 downgrade RED caught. Package datastore, module github.com/hanzo-ds/go, depends only on hanzo-ds/native. Wire strings preserved VERBATIM (X-ClickHouse-* headers, sql.Register("clickhouse"), clickhouse:// DSN, ClientName="clickhouse-go") — consistent with native.
75 lines
2.0 KiB
Go
75 lines
2.0 KiB
Go
package datastore
|
|
|
|
import (
|
|
"errors"
|
|
"regexp"
|
|
"time"
|
|
|
|
"github.com/hanzo-ds/go/lib/driver"
|
|
)
|
|
|
|
var (
|
|
ErrInvalidValueInNamedDateValue = errors.New("invalid value in NamedDateValue for query parameter")
|
|
ErrUnsupportedQueryParameter = errors.New("unsupported query parameter type")
|
|
|
|
hasQueryParamsRe = regexp.MustCompile("{.+:.+}")
|
|
)
|
|
|
|
func bindQueryOrAppendParameters(paramsProtocolSupport bool, options *QueryOptions, query string, timezone *time.Location, args ...any) (string, error) {
|
|
// prefer native query parameters over legacy bind if query parameters provided explicit
|
|
if len(options.parameters) > 0 {
|
|
return query, nil
|
|
}
|
|
|
|
// validate if query contains a {<name>:<data type>} syntax, so it's intentional use of query parameters
|
|
// parameter values will be loaded from `args ...any` for compatibility
|
|
if paramsProtocolSupport &&
|
|
len(args) > 0 &&
|
|
hasQueryParamsRe.MatchString(query) {
|
|
options.parameters = make(Parameters, len(args))
|
|
for _, a := range args {
|
|
switch p := a.(type) {
|
|
case driver.NamedValue:
|
|
if str, ok := p.Value.(string); ok {
|
|
options.parameters[p.Name] = str
|
|
continue
|
|
}
|
|
// using the same format logic for NamedValue typed value in function bindNamed
|
|
strVal, err := format(timezone, Seconds, p.Value)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
options.parameters[p.Name] = strVal
|
|
|
|
case driver.NamedDateValue:
|
|
if !p.Value.IsZero() && p.Name != "" {
|
|
formatted := formatTimeWithScale(p.Value, TimeUnit(p.Scale))
|
|
options.parameters[p.Name] = formatted
|
|
continue
|
|
}
|
|
return "", ErrInvalidValueInNamedDateValue
|
|
|
|
default:
|
|
return "", ErrUnsupportedQueryParameter
|
|
}
|
|
}
|
|
|
|
return query, nil
|
|
}
|
|
|
|
return bind(timezone, query, args...)
|
|
}
|
|
|
|
func formatTimeWithScale(t time.Time, scale TimeUnit) string {
|
|
switch scale {
|
|
case MicroSeconds:
|
|
return t.Format("2006-01-02 15:04:05.000000")
|
|
case MilliSeconds:
|
|
return t.Format("2006-01-02 15:04:05.000")
|
|
case NanoSeconds:
|
|
return t.Format("2006-01-02 15:04:05.000000000")
|
|
default:
|
|
return t.Format("2006-01-02 15:04:05")
|
|
}
|
|
}
|