rpc/server/server.go

package server

import (
	"log"
	"net"

	"git.loafle.net/commons_go/server"
)

func New(sh ServerHandler) server.Server {
	lsh := &serverHandlers{}
	lsh.ServerHandler = sh
	lsh.lsh = sh
	s := server.New(lsh)

	return s
}

type serverHandlers struct {
	server.ServerHandler
	lsh ServerHandler
}

func (sh *serverHandlers) Handle(conn net.Conn, stopChan <-chan struct{}, doneChan chan<- struct{}) {
	contentType := sh.lsh.GetContentType(conn)
	codec, err := sh.lsh.getCodec(contentType)
	if nil != err {
		log.Printf("RPC Handle: %v", err)
		doneChan <- struct{}{}
		return
	}

	for {
		sh.lsh.OnPreRead(conn)
		// Create a new codec request.
		codecReq, errNew := codec.NewRequest(conn)
		if nil != errNew {
			if sh.IsClientDisconnect(errNew) {
				doneChan <- struct{}{}
				return
			}
			log.Printf("RPC Handle: %v", errNew)
			doneChan <- struct{}{}
			return
		}
		sh.lsh.OnPostRead(conn)

		result, err := sh.lsh.invoke(codecReq)

		if nil != err {
			sh.lsh.OnPreWriteError(conn, err)
			codecReq.WriteError(conn, 400, err)
			sh.lsh.OnPostWriteError(conn, err)
		} else {
			sh.lsh.OnPreWriteResult(conn, result)
			codecReq.WriteResponse(conn, result)
			sh.lsh.OnPostWriteResult(conn, result)
		}

		select {
		case <-stopChan:
			return
		default:
		}
	}

}