mirror of
https://gitee.com/johng/gf.git
synced 2024-12-04 21:28:22 +08:00
188 lines
6.6 KiB
Go
188 lines
6.6 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"
|
||
"gitee.com/johng/gf/g/os/glog"
|
||
"strings"
|
||
"reflect"
|
||
"fmt"
|
||
"gitee.com/johng/gf/g/util/gstr"
|
||
"gitee.com/johng/gf/g/os/gfile"
|
||
)
|
||
|
||
// 绑定对象到URI请求处理中，会自动识别方法名称，并附加到对应的URI地址后面
|
||
// 第三个参数methods用以指定需要注册的方法，支持多个方法名称，多个方法以英文“,”号分隔，区分大小写
|
||
func (s *Server)BindObject(pattern string, obj interface{}, methods...string) error {
|
||
methodMap := (map[string]bool)(nil)
|
||
if len(methods) > 0 {
|
||
methodMap = make(map[string]bool)
|
||
for _, v := range strings.Split(methods[0], ",") {
|
||
methodMap[strings.TrimSpace(v)] = true
|
||
}
|
||
}
|
||
m := make(handlerMap)
|
||
v := reflect.ValueOf(obj)
|
||
t := v.Type()
|
||
sname := t.Elem().Name()
|
||
finit := (func(*Request))(nil)
|
||
fshut := (func(*Request))(nil)
|
||
if v.MethodByName("Init").IsValid() {
|
||
finit = v.MethodByName("Init").Interface().(func(*Request))
|
||
}
|
||
if v.MethodByName("Shut").IsValid() {
|
||
fshut = v.MethodByName("Shut").Interface().(func(*Request))
|
||
}
|
||
pkgPath := t.Elem().PkgPath()
|
||
pkgName := gfile.Basename(pkgPath)
|
||
for i := 0; i < v.NumMethod(); i++ {
|
||
mname := t.Method(i).Name
|
||
if methodMap != nil && !methodMap[mname] {
|
||
continue
|
||
}
|
||
if mname == "Init" || mname == "Shut" {
|
||
continue
|
||
}
|
||
faddr, ok := v.Method(i).Interface().(func(*Request))
|
||
if !ok {
|
||
if methodMap != nil {
|
||
s := fmt.Sprintf(`invalid medthod definition "%s", while "func(*Request))" is required`, v.Method(i).Type().String())
|
||
glog.Error(s)
|
||
return errors.New(s)
|
||
}
|
||
continue
|
||
}
|
||
objName := gstr.Replace(t.String(), fmt.Sprintf(`%s.`, pkgName), "")
|
||
if objName[0] == '*' {
|
||
objName = fmt.Sprintf(`(%s)`, objName)
|
||
}
|
||
key := s.mergeBuildInNameToPattern(pattern, sname, mname, true)
|
||
m[key] = &handlerItem {
|
||
name : fmt.Sprintf(`%s.%s.%s`, pkgPath, objName, mname),
|
||
rtype : gROUTE_REGISTER_OBJECT,
|
||
ctype : nil,
|
||
fname : "",
|
||
faddr : faddr,
|
||
finit : finit,
|
||
fshut : fshut,
|
||
}
|
||
// 如果方法中带有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 {
|
||
name : fmt.Sprintf(`%s.%s.%s`, pkgPath, objName, mname),
|
||
rtype : gROUTE_REGISTER_OBJECT,
|
||
ctype : nil,
|
||
fname : "",
|
||
faddr : faddr,
|
||
finit : finit,
|
||
fshut : fshut,
|
||
}
|
||
}
|
||
}
|
||
return s.bindHandlerByMap(m)
|
||
}
|
||
|
||
// 绑定对象到URI请求处理中，会自动识别方法名称，并附加到对应的URI地址后面
|
||
// 第三个参数methods支持多个方法注册，多个方法以英文“,”号分隔，区分大小写
|
||
func (s *Server)BindObjectMethod(pattern string, obj interface{}, method string) error {
|
||
m := make(handlerMap)
|
||
v := reflect.ValueOf(obj)
|
||
t := v.Type()
|
||
sname := t.Elem().Name()
|
||
mname := strings.TrimSpace(method)
|
||
fval := v.MethodByName(mname)
|
||
if !fval.IsValid() {
|
||
return errors.New("invalid method name:" + mname)
|
||
}
|
||
faddr, ok := fval.Interface().(func(*Request))
|
||
if !ok {
|
||
s := fmt.Sprintf(`invalid medthod definition "%s", while "func(*Request)" is required`, fval.Type().String())
|
||
glog.Error(s)
|
||
return errors.New(s)
|
||
}
|
||
finit := (func(*Request))(nil)
|
||
fshut := (func(*Request))(nil)
|
||
if v.MethodByName("Init").IsValid() {
|
||
finit = v.MethodByName("Init").Interface().(func(*Request))
|
||
}
|
||
if v.MethodByName("Shut").IsValid() {
|
||
fshut = v.MethodByName("Shut").Interface().(func(*Request))
|
||
}
|
||
pkgPath := t.Elem().PkgPath()
|
||
pkgName := gfile.Basename(pkgPath)
|
||
objName := gstr.Replace(t.String(), fmt.Sprintf(`%s.`, pkgName), "")
|
||
if objName[0] == '*' {
|
||
objName = fmt.Sprintf(`(%s)`, objName)
|
||
}
|
||
key := s.mergeBuildInNameToPattern(pattern, sname, mname, false)
|
||
m[key] = &handlerItem{
|
||
name : fmt.Sprintf(`%s.%s.%s`, pkgPath, objName, mname),
|
||
rtype : gROUTE_REGISTER_OBJECT,
|
||
ctype : nil,
|
||
fname : "",
|
||
faddr : faddr,
|
||
finit : finit,
|
||
fshut : fshut,
|
||
}
|
||
|
||
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()
|
||
finit := (func(*Request))(nil)
|
||
fshut := (func(*Request))(nil)
|
||
if v.MethodByName("Init").IsValid() {
|
||
finit = v.MethodByName("Init").Interface().(func(*Request))
|
||
}
|
||
if v.MethodByName("Shut").IsValid() {
|
||
fshut = v.MethodByName("Shut").Interface().(func(*Request))
|
||
}
|
||
pkgPath := t.Elem().PkgPath()
|
||
for i := 0; i < v.NumMethod(); i++ {
|
||
mname := t.Method(i).Name
|
||
method := strings.ToUpper(mname)
|
||
if _, ok := s.methodsMap[method]; !ok {
|
||
continue
|
||
}
|
||
faddr, ok := v.Method(i).Interface().(func(*Request))
|
||
if !ok {
|
||
s := fmt.Sprintf(`invalid medthod definition "%s", while "func()" is required`, v.Method(i).Type().String())
|
||
glog.Error(s)
|
||
return errors.New(s)
|
||
}
|
||
pkgName := gfile.Basename(pkgPath)
|
||
objName := gstr.Replace(t.String(), fmt.Sprintf(`%s.`, pkgName), "")
|
||
if objName[0] == '*' {
|
||
objName = fmt.Sprintf(`(%s)`, objName)
|
||
}
|
||
key := mname + ":" + pattern
|
||
m[key] = &handlerItem {
|
||
name : fmt.Sprintf(`%s.%s.%s`, pkgPath, objName, mname),
|
||
rtype : gROUTE_REGISTER_OBJECT,
|
||
ctype : nil,
|
||
fname : "",
|
||
faddr : faddr,
|
||
finit : finit,
|
||
fshut : fshut,
|
||
}
|
||
}
|
||
return s.bindHandlerByMap(m)
|
||
} |