mirror of
https://gitee.com/johng/gf.git
synced 2024-12-06 06:08:38 +08:00
111 lines
3.8 KiB
Go
111 lines
3.8 KiB
Go
// Copyright 2018 gf Author(https://gitee.com/johng/gf). All Rights Reserved.
|
||
//
|
||
// This Source Code Form is subject to the terms of the MIT License.
|
||
// If a copy of the MIT was not distributed with this file,
|
||
// You can obtain one at https://gitee.com/johng/gf.
|
||
// 服务注册.
|
||
|
||
package ghttp
|
||
|
||
import (
|
||
"errors"
|
||
"strings"
|
||
"reflect"
|
||
)
|
||
|
||
// 绑定对象到URI请求处理中,会自动识别方法名称,并附加到对应的URI地址后面
|
||
// 需要注意对象方法的定义必须按照ghttp.HandlerFunc来定义
|
||
func (s *Server)BindObject(pattern string, obj interface{}, methods...string) error {
|
||
if len(methods) > 0 {
|
||
return s.BindObjectMethod(pattern, obj, strings.Join(methods, ","))
|
||
}
|
||
m := make(handlerMap)
|
||
v := reflect.ValueOf(obj)
|
||
t := v.Type()
|
||
sname := t.Elem().Name()
|
||
for i := 0; i < v.NumMethod(); i++ {
|
||
method := t.Method(i).Name
|
||
key := s.mergeBuildInNameToPattern(pattern, sname, method)
|
||
m[key] = &handlerItem {
|
||
ctype : nil,
|
||
fname : "",
|
||
faddr : v.Method(i).Interface().(func(*Request)),
|
||
}
|
||
// 如果方法中带有Index方法,那么额外自动增加一个路由规则匹配主URI
|
||
if strings.EqualFold(method, "Index") {
|
||
p := key
|
||
if strings.EqualFold(p[len(p) - 6:], "/index") {
|
||
p = p[0 : len(p) - 6]
|
||
if len(p) == 0 {
|
||
p = "/"
|
||
}
|
||
}
|
||
m[p] = &handlerItem {
|
||
ctype : nil,
|
||
fname : "",
|
||
faddr : v.Method(i).Interface().(func(*Request)),
|
||
}
|
||
}
|
||
}
|
||
return s.bindHandlerByMap(m)
|
||
}
|
||
|
||
// 绑定对象到URI请求处理中,会自动识别方法名称,并附加到对应的URI地址后面
|
||
// 第三个参数methods支持多个方法注册,多个方法以英文“,”号分隔,区分大小写
|
||
func (s *Server)BindObjectMethod(pattern string, obj interface{}, methods string) error {
|
||
m := make(handlerMap)
|
||
v := reflect.ValueOf(obj)
|
||
t := v.Type()
|
||
sname := t.Elem().Name()
|
||
for _, method := range strings.Split(methods, ",") {
|
||
mname := strings.TrimSpace(method)
|
||
fval := v.MethodByName(mname)
|
||
if !fval.IsValid() {
|
||
return errors.New("invalid method name:" + mname)
|
||
}
|
||
key := s.mergeBuildInNameToPattern(pattern, sname, mname)
|
||
m[key] = &handlerItem{
|
||
ctype : nil,
|
||
fname : "",
|
||
faddr : fval.Interface().(func(*Request)),
|
||
}
|
||
// 如果方法中带有Index方法,那么额外自动增加一个路由规则匹配主URI
|
||
if strings.EqualFold(mname, "Index") {
|
||
p := key
|
||
if strings.EqualFold(p[len(p) - 6:], "/index") {
|
||
p = p[0 : len(p) - 6]
|
||
if len(p) == 0 {
|
||
p = "/"
|
||
}
|
||
}
|
||
m[p] = &handlerItem {
|
||
ctype : nil,
|
||
fname : "",
|
||
faddr : fval.Interface().(func(*Request)),
|
||
}
|
||
}
|
||
}
|
||
return s.bindHandlerByMap(m)
|
||
}
|
||
|
||
// 绑定对象到URI请求处理中,会自动识别方法名称,并附加到对应的URI地址后面
|
||
// 需要注意对象方法的定义必须按照ghttp.HandlerFunc来定义
|
||
func (s *Server)BindObjectRest(pattern string, obj interface{}) error {
|
||
m := make(handlerMap)
|
||
v := reflect.ValueOf(obj)
|
||
t := v.Type()
|
||
for i := 0; i < v.NumMethod(); i++ {
|
||
name := t.Method(i).Name
|
||
method := strings.ToUpper(name)
|
||
if _, ok := s.methodsMap[method]; !ok {
|
||
continue
|
||
}
|
||
key := name + ":" + pattern
|
||
m[key] = &handlerItem {
|
||
ctype : nil,
|
||
fname : "",
|
||
faddr : v.Method(i).Interface().(func(*Request)),
|
||
}
|
||
}
|
||
return s.bindHandlerByMap(m)
|
||
} |