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.
91 lines
1.9 KiB
Go
91 lines
1.9 KiB
Go
package column
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/hanzo-ds/native/proto"
|
|
"strings"
|
|
)
|
|
|
|
type Nested struct {
|
|
Interface
|
|
name string
|
|
}
|
|
|
|
func (col *Nested) Reset() {
|
|
col.Interface.Reset()
|
|
}
|
|
|
|
func asDDL(cols []namedCol) string {
|
|
sCols := make([]string, len(cols), len(cols))
|
|
for i := range cols {
|
|
sCols[i] = fmt.Sprintf("%s %s", cols[i].name, cols[i].colType)
|
|
}
|
|
return strings.Join(sCols, ", ")
|
|
}
|
|
|
|
func (col *Nested) parse(t Type, sc *ServerContext) (_ Interface, err error) {
|
|
columns := fmt.Sprintf("Array(Tuple(%s))", asDDL(nestedColumns(t.params())))
|
|
if col.Interface, err = (&Array{name: col.name}).parse(Type(columns), sc); err != nil {
|
|
return nil, err
|
|
}
|
|
return col, nil
|
|
}
|
|
|
|
func nestedColumns(raw string) (columns []namedCol) {
|
|
var (
|
|
nBegin int
|
|
begin int
|
|
brackets int
|
|
)
|
|
for i, r := range raw + "," {
|
|
switch r {
|
|
case '(':
|
|
brackets++
|
|
case ')':
|
|
brackets--
|
|
case ' ':
|
|
if brackets == 0 {
|
|
begin = i + 1
|
|
}
|
|
case ',':
|
|
if brackets == 0 {
|
|
columns, begin = append(columns, namedCol{
|
|
name: strings.TrimSpace(raw[nBegin:begin]),
|
|
colType: Type(raw[begin:i]),
|
|
}), i+1
|
|
nBegin = i + 1
|
|
continue
|
|
}
|
|
}
|
|
}
|
|
for i, column := range columns {
|
|
if strings.HasPrefix(string(column.colType), "Nested(") {
|
|
columns[i] = namedCol{
|
|
colType: Type(fmt.Sprintf("Array(Tuple(%s))", asDDL(nestedColumns(column.colType.params())))),
|
|
name: column.name,
|
|
}
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func (col *Nested) ReadStatePrefix(reader *proto.Reader) error {
|
|
if serialize, ok := col.Interface.(CustomSerialization); ok {
|
|
if err := serialize.ReadStatePrefix(reader); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (col *Nested) WriteStatePrefix(buffer *proto.Buffer) error {
|
|
if serialize, ok := col.Interface.(CustomSerialization); ok {
|
|
if err := serialize.WriteStatePrefix(buffer); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
var _ Interface = (*Nested)(nil)
|