rpc/server/servlet_handlers.go

package server

import (
	"fmt"

	"git.loafle.net/commons_go/logging"
	"git.loafle.net/commons_go/rpc"
	"git.loafle.net/commons_go/rpc/protocol"
	"git.loafle.net/commons_go/rpc/registry"
)

type ServletHandlers struct {
	rpc.ServletHandlers

	RPCInvoker registry.RPCInvoker
}

func (sh *ServletHandlers) Invoke(servletCTX rpc.ServletContext, requestCodec protocol.RegistryCodec) (result interface{}, err error) {
	if !sh.RPCInvoker.HasMethod(requestCodec.Method()) {
		return nil, fmt.Errorf("RPC Servlet Handler: Method[%s] is not exist", requestCodec.Method())
	}

	result, err = sh.RPCInvoker.Invoke(requestCodec)
	if nil != err {
		if params, pErr := requestCodec.Params(); nil != err {
			logging.Logger().Error(fmt.Sprintf("RPC Servlet Handler: Read Param of Method[%s] returns error %v", requestCodec.Method(), pErr))
		} else {
			logging.Logger().Error(fmt.Sprintf("RPC Servlet Handler: Method[%s] params[%v] returns error %v", requestCodec.Method(), params, err))
		}
	}

	return
}

func (sh *ServletHandlers) GetRPCInvoker() registry.RPCInvoker {
	return sh.RPCInvoker
}

func (sh *ServletHandlers) Validate() {
	sh.ServletHandlers.Validate()

	if nil == sh.RPCInvoker {
		logging.Logger().Panic(fmt.Sprintf("RPC Servlet Handler: RPC Invoker must be specified"))
	}
}