Skip to content
Snippets Groups Projects
Commit cf94b2f9 authored by Huxing Zhang's avatar Huxing Zhang Committed by GitHub
Browse files

Merge pull request #96 from divebomb/master

Fix: alexstocks/getty issue 18
parents 38b7dd8d 1573e98d
No related branches found
No related tags found
No related merge requests found
......@@ -24,7 +24,7 @@ import (
import (
"github.com/dubbogo/getty"
"github.com/dubbogo/hessian2"
hessian "github.com/dubbogo/hessian2"
perrors "github.com/pkg/errors"
)
import (
......@@ -89,12 +89,11 @@ func (p *RpcClientPackageHandler) Write(ss getty.Session, pkg interface{}) error
// RpcServerPackageHandler
////////////////////////////////////////////
type RpcServerPackageHandler struct {
}
var (
rpcServerPkgHandler = &RpcServerPackageHandler{}
)
func NewRpcServerPackageHandler() *RpcServerPackageHandler {
return &RpcServerPackageHandler{}
}
type RpcServerPackageHandler struct{}
func (p *RpcServerPackageHandler) Read(ss getty.Session, data []byte) (interface{}, int, error) {
pkg := &DubboPackage{
......
......@@ -79,6 +79,8 @@ type Server struct {
conf ServerConfig
tcpServer getty.Server
exporter protocol.Exporter
rpcHandler *RpcServerHandler
}
func NewServer(exporter protocol.Exporter) *Server {
......@@ -88,6 +90,8 @@ func NewServer(exporter protocol.Exporter) *Server {
conf: *srvConf,
}
s.rpcHandler = NewRpcServerHandler(s.exporter, s.conf.SessionNumber, s.conf.sessionTimeout)
return s
}
......@@ -116,8 +120,8 @@ func (s *Server) newSession(session getty.Session) error {
session.SetName(conf.GettySessionParam.SessionName)
session.SetMaxMsgLen(conf.GettySessionParam.MaxMsgLen)
session.SetPkgHandler(NewRpcServerPackageHandler())
session.SetEventListener(NewRpcServerHandler(s.exporter, conf.SessionNumber, conf.sessionTimeout))
session.SetPkgHandler(rpcServerPkgHandler)
session.SetEventListener(s.rpcHandler)
session.SetRQLen(conf.GettySessionParam.PkgRQSize)
session.SetWQLen(conf.GettySessionParam.PkgWQSize)
session.SetReadTimeout(conf.GettySessionParam.tcpReadTimeout)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment