/
server.go
49 lines (43 loc) · 1.05 KB
/
server.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
package runtime
import (
"bytes"
"context"
"fmt"
"google.golang.org/protobuf/proto"
)
type Server interface {
Selector() uint64
HandleRequestResponse(context.Context, *RequestWrapper) (proto.Message, error)
}
type Servers struct {
serverMap map[uint64]Server
}
func NewServers(servers ...Server) *Servers {
s := &Servers{
serverMap: make(map[uint64]Server),
}
for _, server := range servers {
s.serverMap[server.Selector()] = server
}
return s
}
func (servers *Servers) AddServer(server Server) {
servers.serverMap[server.Selector()] = server
}
func (servers *Servers) HandleRequestResponse(ctx context.Context, reqWrapperBytes []byte) ([]byte, error) {
inputBuffer := bytes.NewBuffer(reqWrapperBytes)
var reqWrapper RequestWrapper
err := reqWrapper.Unmarshal(inputBuffer)
if err != nil {
return nil, err
}
server, ok := servers.serverMap[reqWrapper.Selector]
if !ok {
return nil, fmt.Errorf("unknown selector")
}
rsp, err := server.HandleRequestResponse(ctx, &reqWrapper)
if err != nil {
return nil, err
}
return proto.Marshal(rsp)
}