forked from marcboeker/go-duckdb
-
Notifications
You must be signed in to change notification settings - Fork 0
/
types.go
67 lines (54 loc) · 1.39 KB
/
types.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
package duckdb
/*
#include <duckdb.h>
*/
import "C"
import (
"encoding/binary"
"fmt"
"math"
"github.com/mitchellh/mapstructure"
)
// HugeInt are composed in a (lower, upper) component
// The value of the HugeInt is upper * 2^64 + lower
type HugeInt C.duckdb_hugeint
func (v HugeInt) UUID() []byte {
var uuid [16]byte
// We need to flip the `sign bit` of the signed `HugeInt` to transform it to UUID bytes
binary.BigEndian.PutUint64(uuid[:8], uint64(v.upper)^1<<63)
binary.BigEndian.PutUint64(uuid[8:], uint64(v.lower))
return uuid[:]
}
func (v HugeInt) Int64() (int64, error) {
if v.upper == 0 && v.lower <= math.MaxInt64 {
return int64(v.lower), nil
} else if v.upper == -1 {
return -int64(math.MaxUint64 - v.lower + 1), nil
} else {
return 0, fmt.Errorf("can not convert duckdb:hugeint to go:int64 (upper:%d, lower:%d)", v.upper, v.lower)
}
}
type Map map[any]any
func (m *Map) Scan(v any) error {
data, ok := v.(Map)
if !ok {
return fmt.Errorf("invalid type `%T` for scanning `Map`, expected `Map`", data)
}
*m = data
return nil
}
type Interval struct {
Days int32 `json:"days"`
Months int32 `json:"months"`
Micros int64 `json:"micros"`
}
// Use as the `Scanner` type for any composite types (maps, lists, structs)
type Composite[T any] struct {
t T
}
func (s Composite[T]) Get() T {
return s.t
}
func (s *Composite[T]) Scan(v any) error {
return mapstructure.Decode(v, &s.t)
}