Blame view

src/components/db/schema.go 691 Bytes
98b0736d   zhangqijia   添加定时器, 检查心跳
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
  }