新增系统服务

sysservice/httpserverervice.go

@@ -0,0 +1,126 @@
+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
+}

sysservice/wsserverservice.go

@@ -0,0 +1,43 @@
+package sysservice
+
+import (
+	"github.com/duanhf2012/origin/network"
+	"github.com/duanhf2012/origin/service"
+)
+
+type WSServerService struct {
+	service.BaseService
+	wsserver network.WebsocketServer
+
+	pattern            string
+	port               uint16
+	messageReciver     network.IMessageReceiver
+	bEnableCompression bool
+}
+
+func (ws *WSServerService) OnInit() error {
+
+	ws.wsserver.Init(ws.pattern, ws.port, ws.messageReciver, ws.bEnableCompression)
+	return nil
+}
+
+func (ws *WSServerService) OnRun() error {
+	ws.wsserver.Start()
+
+	return nil
+}
+
+func NewWSServerService(pattern string, port uint16, messageReciver network.IMessageReceiver, bEnableCompression bool) *WSServerService {
+	wss := new(WSServerService)
+	wss.pattern = pattern
+	wss.port = port
+	wss.messageReciver = messageReciver
+	wss.bEnableCompression = bEnableCompression
+
+	wss.Init(wss, 0)
+	return wss
+}
+
+func (ws *WSServerService) OnDestory() error {
+	return nil
+}