mirror of
https://github.com/duanhf2012/origin.git
synced 2026-02-10 20:44:41 +08:00
127 lines
2.8 KiB
Go
127 lines
2.8 KiB
Go
package sysservice
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"reflect"
|
|
"strings"
|
|
|
|
"github.com/duanhf2012/origin/network"
|
|
"github.com/duanhf2012/origin/service"
|
|
)
|
|
|
|
type ControllerMapsType map[string]reflect.Value
|
|
|
|
type HttpServerService struct {
|
|
service.BaseService
|
|
httpserver network.HttpServer
|
|
port uint16
|
|
|
|
controllerMaps ControllerMapsType
|
|
}
|
|
|
|
func (slf *HttpServerService) OnInit() error {
|
|
|
|
slf.httpserver.Init(slf.port)
|
|
slf.httpserver.HandleFunc("/{server:[a-zA-Z0-9]+}/{method:[a-zA-Z0-9]+}", func(w http.ResponseWriter, r *http.Request) {
|
|
slf.httpHandler(w, r)
|
|
})
|
|
|
|
return nil
|
|
}
|
|
|
|
func (slf *HttpServerService) OnRun() error {
|
|
|
|
slf.httpserver.Start()
|
|
return nil
|
|
}
|
|
|
|
func NewHttpServerService(port uint16) *HttpServerService {
|
|
http := new(HttpServerService)
|
|
|
|
http.port = port
|
|
http.Init(http, 0)
|
|
return http
|
|
}
|
|
|
|
func (slf *HttpServerService) OnDestory() error {
|
|
return nil
|
|
}
|
|
|
|
func (slf *HttpServerService) OnSetupService(iservice service.IService) {
|
|
//
|
|
|
|
//反射
|
|
vf := reflect.ValueOf(iservice)
|
|
vft := vf.Type()
|
|
//读取方法数量
|
|
mNum := vf.NumMethod()
|
|
for i := 0; i < mNum; i++ {
|
|
mName := vft.Method(i).Name
|
|
startPos := strings.Index(mName, "Http_")
|
|
if startPos != 0 {
|
|
continue
|
|
}
|
|
fmt.Println("index:", i, " MethodName:", mName)
|
|
slf.controllerMaps[mName] = vf.Method(i)
|
|
}
|
|
}
|
|
|
|
func (slf *HttpServerService) OnRemoveService(iservice service.IService) {
|
|
return
|
|
}
|
|
|
|
func (slf *HttpServerService) httpHandler(w http.ResponseWriter, r *http.Request) {
|
|
writeError := func(status int, msg string) {
|
|
w.Header().Set("Content-Type", "text/plain; charset=utf-8")
|
|
w.WriteHeader(status)
|
|
w.Write([]byte(msg))
|
|
}
|
|
if r.Method != "POST" {
|
|
writeError(http.StatusMethodNotAllowed, "rpc: POST method required, received "+r.Method)
|
|
return
|
|
}
|
|
defer r.Body.Close()
|
|
msg, err := ioutil.ReadAll(r.Body)
|
|
if err != nil {
|
|
writeError(http.StatusBadRequest, "rpc: ioutil.ReadAll "+err.Error())
|
|
return
|
|
}
|
|
fmt.Printf("PATH: %s\n%s\n", r.URL.Path, string(msg))
|
|
// 在这儿处理例外路由接口
|
|
|
|
// 拼接得到rpc服务的名称
|
|
vstr := strings.Split(r.URL.Path, "/")
|
|
if len(vstr) != 3 {
|
|
writeError(http.StatusBadRequest, "rpc: ioutil.ReadAll "+err.Error())
|
|
return
|
|
}
|
|
strCallPath := vstr[1] + "." + vstr[2]
|
|
|
|
method, err := slf.GetMethod(strCallPath)
|
|
var respone string
|
|
if err != nil {
|
|
respone = fmt.Sprintf("http respone => %v\n", err)
|
|
|
|
} else {
|
|
params := make([]reflect.Value, 0) // 参数
|
|
params = append(params, reflect.ValueOf(msg))
|
|
method.Call(params)
|
|
//respone = method.Call(params)
|
|
}
|
|
|
|
w.Header().Set("Content-Type", "application/json;charset=utf-8")
|
|
w.Write([]byte(respone))
|
|
}
|
|
|
|
func (slf *HttpServerService) GetMethod(strCallPath string) (*reflect.Value, error) {
|
|
value, ok := slf.controllerMaps[strCallPath]
|
|
if ok == false {
|
|
err := fmt.Errorf("not find api")
|
|
return nil, err
|
|
}
|
|
|
|
return &value, nil
|
|
}
|