overflow_backup_2018_0455/overflow_discovery
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ing

Browse Source
This commit is contained in:
crusader 2017-10-26 22:18:20 +09:00
parent 44bc4f51d8
commit 97acdddce5
2 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
server/server.go
View File

@ -14,6 +14,7 @@ func New(addr string) server.Server {
registry.RegisterService(new(dRPC.DiscoveryService), "") registry.RegisterService(new(dRPC.DiscoveryService), "")
sh := NewServerHandler(addr, registry) sh := NewServerHandler(addr, registry)
sh.workersChan = make(chan struct{}, 10)
s := server.NewServer(sh) s := server.NewServer(sh)

18
server/server_handlers.go
View File

@ -4,11 +4,10 @@ import (
"io" "io"
"git.loafle.net/commons_go/rpc" "git.loafle.net/commons_go/rpc"
"git.loafle.net/commons_go/server"
"git.loafle.net/commons_go/server/ipc" "git.loafle.net/commons_go/server/ipc"
) )
func NewServerHandler(addr string, registry rpc.Registry) ServerHandler { func NewServerHandler(addr string, registry rpc.Registry) *ServerHandlers {
sh := &ServerHandlers{ sh := &ServerHandlers{
registry: registry, registry: registry,
} }
@ -17,16 +16,23 @@ func NewServerHandler(addr string, registry rpc.Registry) ServerHandler {
return sh return sh
} }
type ServerHandler interface {
server.ServerHandler
}
type ServerHandlers struct { type ServerHandlers struct {
ipc.ServerHandlers ipc.ServerHandlers
registry rpc.Registry registry rpc.Registry
workersChan chan struct{}
} }
func (sh *ServerHandlers) Handle(remoteAddr string, rwc io.ReadWriteCloser, stopChan chan struct{}) { func (sh *ServerHandlers) Handle(remoteAddr string, rwc io.ReadWriteCloser, stopChan chan struct{}) {
contentType := "json"
for {
sh.registry.Invoke(contentType, rwc, rwc, nil, nil)
select {
case <-stopChan:
return
}
}
} }