schema.go
691 Bytes
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
package db
import (
"reflect"
"strings"
)
type Schema struct {
reflect.Type
pri interface{}
schema interface{}
}
func NewSchema(pri, schema interface{}) *Schema {
s := reflect.TypeOf(schema)
if s.Kind() == reflect.Ptr {
s = reflect.TypeOf(s).Elem()
}
return &Schema{
Type: s,
pri: pri,
schema: schema,
}
}
func (s *Schema)GetSchemaType() reflect.Type {
return s.Type
}
func (s *Schema)GetCollName() string {
return strings.ToLower(s.GetSchemaType().Name())
}
func (s *Schema)GetPriKey() string {
var pri string
for i := 0; i < s.NumField(); i++ {
if s.Field(i).Tag.Get("pri") == "1" {
pri = strings.ToLower(s.Field(i).Name)
break
}
}
return pri
}