控制台和节点间的通讯方式改为tcp通讯

Author: Your Name

admin/cmd/cmd.go

@@ -0,0 +1,47 @@
+package cmd
+
+import (
+	"encoding/json"
+	"errors"
+
+	"github.com/eolinker/goku-api-gateway/config"
+)
+
+var (
+	ErrorInvalidNodeInstance = errors.New("invalid instance value")
+	ErrorInvalidNodeConfig   = errors.New("invalid instance config")
+)
+
+type Code string
+
+const (
+	None               Code = "none"
+	NodeRegister       Code = "register"
+	NodeRegisterResult Code = "register-result"
+	NodeLevel          Code = "level"
+	Config             Code = "config"
+	Restart            Code = "restart"
+	Stop               Code = "stop"
+	Monitor            Code = "monitor"
+	EventClientLeave   Code = "leave"
+	Error              Code = "error"
+)
+
+func EncodeConfig(c *config.GokuConfig) ([]byte, error) {
+	if c == nil {
+		return nil, ErrorInvalidNodeConfig
+	}
+
+	return json.Marshal(c)
+}
+func DecodeConfig(data []byte) (*config.GokuConfig, error) {
+	if len(data) == 0 {
+		return nil, ErrorInvalidNodeConfig
+	}
+	c := new(config.GokuConfig)
+	err := json.Unmarshal(data, c)
+	if err != nil {
+		return nil, err
+	}
+	return c, nil
+}

admin/cmd/conn.go

@@ -0,0 +1,84 @@
+package cmd
+
+import (
+	"context"
+	"errors"
+	"net"
+	"sync"
+)
+
+var (
+	ErrorSendToClosedConnect = errors.New("send to closed connect")
+)
+
+type Connect struct {
+	conn net.Conn
+
+	//inputC chan _Frame
+	outputC chan []byte
+
+	doneC chan struct{}
+
+	ctx        context.Context
+	cancelFunc context.CancelFunc
+
+	once sync.Once
+}
+
+func NewConnect(conn net.Conn) *Connect {
+	ctx, cancel := context.WithCancel(context.Background())
+	c := &Connect{
+		conn: conn,
+
+		//inputC:     make(chan _Frame,10),
+		outputC: make(chan []byte, 10),
+
+		ctx:        ctx,
+		cancelFunc: cancel,
+	}
+
+	go c.r()
+
+	return c
+}
+func (c *Connect) r() {
+
+	for {
+		frame, err := ReadFrame(c.conn)
+		if err != nil {
+			break
+		}
+		c.outputC <- frame
+	}
+	close(c.outputC)
+}
+
+func (c *Connect) Close() error {
+
+	c.once.Do(func() {
+		c.cancelFunc()
+		c.conn.Close()
+
+	})
+
+	return nil
+}
+
+func (c *Connect) LocalAddr() net.Addr {
+	return c.conn.LocalAddr()
+}
+
+func (c *Connect) RemoteAddr() net.Addr {
+	return c.conn.RemoteAddr()
+}
+
+func (c *Connect) Send(code Code, data []byte) error {
+	return SendFrame(c.conn, code, data)
+}
+
+func (c *Connect) ReadC() <-chan []byte {
+	return c.outputC
+}
+func (c *Connect) Done() <-chan struct{} {
+	return c.ctx.Done()
+}

admin/cmd/error.go

@@ -0,0 +1,14 @@
+package cmd
+
+type ErrorInfo struct {
+	Error string `json:"error"`
+}
+
+func DecodeError(data []byte) (string, error) {
+
+	return string(data), nil
+}
+func EncodeError(err string) ([]byte, error) {
+
+	return []byte(err), nil
+}

admin/cmd/register.go

@@ -0,0 +1,52 @@
+package cmd
+
+import (
+	"encoding/json"
+	"github.com/eolinker/goku-api-gateway/config"
+)
+
+type RegisterResult struct {
+	Code   int
+	Error  string
+	Config *config.GokuConfig
+}
+
+func DecodeRegisterResult(data []byte) (*RegisterResult, error) {
+	r := new(RegisterResult)
+	err := json.Unmarshal(data, r)
+	if err != nil {
+		return nil, err
+	}
+	return r, nil
+}
+func EncodeRegisterResultConfig(c *config.GokuConfig) ([]byte, error) {
+	r := RegisterResult{
+		Code:   0,
+		Error:  "",
+		Config: c,
+	}
+	return json.Marshal(r)
+}
+func EncodeRegisterResultError(err string) ([]byte, error) {
+	r := RegisterResult{
+		Code:   -1,
+		Error:  err,
+		Config: nil,
+	}
+	return json.Marshal(r)
+}
+
+func DecodeRegister(data []byte) (string, error) {
+	if len(data) == 32 {
+		return string(data), nil
+	}
+	return "", ErrorInvalidNodeInstance
+}
+
+func EncodeRegister(nodeKey string) ([]byte, error) {
+	data := []byte(nodeKey)
+	if len(data) == 32 {
+		return data, nil
+	}
+	return nil, ErrorInvalidNodeInstance
+}

admin/cmd/server.go

@@ -0,0 +1,86 @@
+package cmd
+
+import (
+	"bytes"
+	"encoding/binary"
+	"errors"
+	"io"
+
+	"github.com/eolinker/goku-api-gateway/common/ioutils"
+)
+
+var (
+	ErrorEmptyFrame  = errors.New("empty frame")
+	ErrorInvalidCode = errors.New("invalid code")
+)
+
+func ReadFrame(reader io.Reader) ([]byte, error) {
+
+	sizeBuf := make([]byte, 4, 4)
+	// 获取报文头部信息
+	_, err := io.ReadFull(reader, sizeBuf)
+	if err != nil {
+		return nil, err
+	}
+	// 获取报文数据大小
+	size := binary.BigEndian.Uint32(sizeBuf)
+
+	data := make([]byte, size, size)
+
+	_, e := io.ReadFull(reader, data)
+	if e != nil {
+		return nil, err
+	}
+	return data, nil
+}
+func GetCmd(frame []byte) (Code, []byte, error) {
+	frameLen := len(frame)
+	if frameLen < 5 {
+		// 长度小于5时，报文没有数据
+		return "", nil, ErrorEmptyFrame
+	}
+	buf := bytes.NewBuffer(frame)
+	codeData, n, err := ioutils.ReadLField(buf, nil)
+	if err != nil {
+		return "", nil, err
+	}
+
+	return Code(codeData), frame[n:], nil
+}
+
+func SendError(w io.Writer, err error) {
+	if err == nil {
+		return
+	}
+	data := []byte(err.Error())
+	SendFrame(w, Error, data)
+}
+func SendFrame(w io.Writer, code Code, data []byte) error {
+	codeData := []byte(code)
+
+	size := uint32(len(data) + len(codeData) + 1)
+	sizeAll := size + 4
+	buf := bytes.NewBuffer(make([]byte, sizeAll, sizeAll))
+	buf.Reset()
+
+	err := binary.Write(buf, binary.BigEndian, size)
+	if err != nil {
+		return err
+	}
+
+	_, err = ioutils.WriteLField(buf, codeData)
+	if err != nil {
+		return err
+	}
+
+	if len(data) > 0 {
+		_, err := buf.Write(data)
+		if err != nil {
+			return err
+		}
+	}
+	//b:= buf.Bytes()
+	//_,err =w.Write(b)
+	_, err = buf.WriteTo(w)
+	return err
+}

admin/console/callback.go

@@ -0,0 +1,52 @@
+package console
+
+import (
+	"github.com/eolinker/goku-api-gateway/admin/cmd"
+	goku_log "github.com/eolinker/goku-api-gateway/goku-log"
+)
+
+type Callback func(code cmd.Code, data []byte,client *Client) error
+
+func (c Callback) ServerCode(code cmd.Code, data []byte, client *Client) error{
+	return c(code,data,client)
+}
+
+type CodeHandler interface {
+	ServerCode(code cmd.Code, data []byte,client *Client)error
+}
+
+//Register cmd 回调注册器
+type Register struct {
+	callbacks map[cmd.Code][]CodeHandler
+}
+//NewRegister create register
+func NewRegister() *Register {
+	return &Register{
+		callbacks: make(map[cmd.Code][]CodeHandler),
+	}
+}
+
+//Register 注册回调
+func (s *Register) Register(code cmd.Code,handler CodeHandler){
+	s.callbacks[code] = append(s.callbacks[code],handler)
+}
+//Register 注册回调
+func (s *Register) RegisterFunc(code cmd.Code,callback func(code cmd.Code, data []byte,client *Client) error){
+	s.callbacks[code] = append(s.callbacks[code],Callback(callback))
+}
+
+//Callback 调用回调
+func (s *Register)Callback(code cmd.Code,data []byte,client *Client)error  {
+	m:=s.callbacks
+	callbacks,has:=  m[code]
+	if !has{
+		goku_log.Info("not exists call for ",code)
+		return nil
+	}
+	for _,handler:=range callbacks{
+		if e:=handler.ServerCode(code,data,client);e!=nil{
+			return e
+		}
+	}
+	return nil
+}