Blame view

common/components/http.go 1.66 KB
0e5d52de   zhangqijia   reactor: 重构底层框架1.0
1
  package components
252ca9a0   zhangqijia   解释目录结构
2
3
  
  import (
252ca9a0   zhangqijia   解释目录结构
4
  	"github.com/gin-gonic/gin"
62d5d847   zhangqijia   http服务只用写actionSt...
5
  	"net/http"
252ca9a0   zhangqijia   解释目录结构
6
7
8
9
10
  	"reflect"
  	"strings"
  )
  
  type HttpServer struct {
0e5d52de   zhangqijia   reactor: 重构底层框架1.0
11
  	IHttp
252ca9a0   zhangqijia   解释目录结构
12
  	version string
b3d79719   zhangqijia   fix: 返回值如果code=0则...
13
  	port    []string
252ca9a0   zhangqijia   解释目录结构
14
15
16
  	Handler interface{}
  }
  
b3d79719   zhangqijia   fix: 返回值如果code=0则...
17
  func Pong(c *gin.Context) {
252ca9a0   zhangqijia   解释目录结构
18
19
20
21
22
23
24
25
26
  	c.JSON(200, gin.H{
  		"message": "pong",
  	})
  }
  
  func NewHttpServer(version string, port ...string) *HttpServer {
  	return &HttpServer{version: version, port: port}
  }
  
b3d79719   zhangqijia   fix: 返回值如果code=0则...
27
  func GetRoutePath(objName, objFunc string) string {
0e5d52de   zhangqijia   reactor: 重构底层框架1.0
28
29
30
  	return strings.ToLower(objName + "/" + objFunc)
  }
  
b3d79719   zhangqijia   fix: 返回值如果code=0则...
31
  func (h *HttpServer) HandlerFuncObj(tvl, obj reflect.Value) gin.HandlerFunc {
252ca9a0   zhangqijia   解释目录结构
32
33
  	return func(c *gin.Context) {
  		v := tvl.Call([]reflect.Value{obj, reflect.ValueOf(c)})
62d5d847   zhangqijia   http服务只用写actionSt...
34
  		if len(v) != 2 {
b3d79719   zhangqijia   fix: 返回值如果code=0则...
35
  			c.JSON(http.StatusNotFound, gin.H{"code": -100, "message": "request param len is error", "data": ""})
62d5d847   zhangqijia   http服务只用写actionSt...
36
37
  			return
  		}
b3d79719   zhangqijia   fix: 返回值如果code=0则...
38
39
40
41
42
43
  		code := v[0].Int()
  		if code == 0 {
  			c.JSON(http.StatusOK, gin.H{"code": v[0].Interface(), "message": "success", "data": v[1].Interface()})
  		} else {
  			c.JSON(http.StatusOK, gin.H{"code": v[0].Interface(), "message": v[1].Interface()})
  		}
252ca9a0   zhangqijia   解释目录结构
44
45
46
  	}
  }
  
b3d79719   zhangqijia   fix: 返回值如果code=0则...
47
  func (h *HttpServer) BindHandler(handler interface{}) {
252ca9a0   zhangqijia   解释目录结构
48
49
50
51
  	h.Handler = handler
  }
  
  func (h *HttpServer) Start() error {
252ca9a0   zhangqijia   解释目录结构
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
  	//gin初始化
  	r := gin.Default()
  	r.GET("/ping", Pong)
  	typ := reflect.TypeOf(h.Handler)
  	val := reflect.ValueOf(h.Handler)
  	//t := reflect.Indirect(val).Type()
  	//objectName := t.Name()
  
  	numOfMethod := val.NumMethod()
  	for i := 0; i < numOfMethod; i++ {
  		method := typ.Method(i)
  		r.GET(GetRoutePath(h.version, method.Name), h.HandlerFuncObj(method.Func, val))
  		r.POST(GetRoutePath(h.version, method.Name), h.HandlerFuncObj(method.Func, val))
  	}
  	return r.Run(h.port...) // listen and serve on 0.0.0.0:8080 (for windows "localhost:8080")
b3d79719   zhangqijia   fix: 返回值如果code=0则...
67
  }