forked from dgraph-io/dgraph
-
Notifications
You must be signed in to change notification settings - Fork 0
/
scalar_types.go
150 lines (127 loc) · 3.06 KB
/
scalar_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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
/*
* Copyright 2016 DGraph Labs, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package types
import (
"time"
geom "github.com/twpayne/go-geom"
)
const (
nanoSecondsInSec = 1000000000
)
// Note: These ids are stored in the posting lists to indicate the type
// of the data. The order *cannot* be changed without breaking existing
// data. When adding a new type *always* add to the end of this list.
// Never delete anything from this list even if it becomes unused.
const (
BinaryID = TypeID(Posting_BINARY)
Int32ID = TypeID(Posting_INT32)
FloatID = TypeID(Posting_FLOAT)
BoolID = TypeID(Posting_BOOL)
DateTimeID = TypeID(Posting_DATETIME)
StringID = TypeID(Posting_STRING)
DateID = TypeID(Posting_DATE)
GeoID = TypeID(Posting_GEO)
UidID = TypeID(Posting_UID)
)
var typeNameMap = map[string]TypeID{
"int": Int32ID,
"float": FloatID,
"string": StringID,
"bool": BoolID,
"id": StringID,
"datetime": DateTimeID,
"date": DateID,
"geo": GeoID,
"uid": UidID,
}
type TypeID Posting_ValType
func (t TypeID) Name() string {
switch t {
case Int32ID:
return "int"
case FloatID:
return "float"
case BoolID:
return "bool"
case StringID:
return "string"
case DateID:
return "date"
case DateTimeID:
return "dateTime"
case GeoID:
return "geo"
case UidID:
return "uid"
}
return ""
}
type Val struct {
Tid TypeID
Value interface{}
}
// TypeForName returns the type corresponding to the given name.
// If name is not recognized, it returns nil.
func TypeForName(name string) (TypeID, bool) {
t, ok := typeNameMap[name]
return t, ok
}
func (t TypeID) IsScalar() bool {
return t != UidID
}
// ValueForType returns the zero value for a type id
func ValueForType(id TypeID) Val {
switch id {
case BinaryID:
var b []byte
return Val{BinaryID, &b}
case Int32ID:
var i int32
return Val{Int32ID, &i}
case FloatID:
var f float64
return Val{FloatID, &f}
case BoolID:
var b bool
return Val{BoolID, &b}
case DateTimeID:
var t time.Time
return Val{DateTimeID, &t}
case StringID:
var s string
return Val{StringID, &s}
case DateID:
var d time.Time
return Val{DateID, &d}
case GeoID:
var g geom.T
return Val{GeoID, &g}
case UidID:
var i uint64
return Val{UidID, &i}
default:
return Val{}
}
}
func createDate(y int, m time.Month, d int) time.Time {
var dt time.Time
dt = time.Date(y, m, d, 0, 0, 0, 0, time.UTC)
return dt
}
const dateFormatYMD = "2006-01-02"
const dateFormatYM = "2006-01"
const dateFormatY = "2006"
const dateTimeFormat = "2006-01-02T15:04:05"