ing
This commit is contained in:
parent
8411acee46
commit
eac45ad70e
10
config.json
10
config.json
|
@ -24,6 +24,16 @@
|
||||||
"redis": {
|
"redis": {
|
||||||
"network": "tcp4",
|
"network": "tcp4",
|
||||||
"address": "192.168.1.50:6379"
|
"address": "192.168.1.50:6379"
|
||||||
|
},
|
||||||
|
"kafka": {
|
||||||
|
"producers": {
|
||||||
|
"metric": {
|
||||||
|
"brokers": [
|
||||||
|
"192.168.1.50:9092"
|
||||||
|
],
|
||||||
|
"topic": "overflow-metric-topic"
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,17 +1,23 @@
|
||||||
package server
|
package server
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
crpj "git.loafle.net/commons/rpc-go/protocol/json"
|
||||||
cssw "git.loafle.net/commons/server-go/socket/web"
|
cssw "git.loafle.net/commons/server-go/socket/web"
|
||||||
"git.loafle.net/overflow/probe_gateway_rpc/config"
|
"git.loafle.net/overflow/probe_gateway_rpc/config"
|
||||||
"git.loafle.net/overflow/probe_gateway_rpc/servlet"
|
"git.loafle.net/overflow/probe_gateway_rpc/servlet"
|
||||||
)
|
)
|
||||||
|
|
||||||
func New(_config *config.Config) *cssw.Server {
|
func New(_config *config.Config) *cssw.Server {
|
||||||
|
rpcServerCodec := crpj.NewServerCodec()
|
||||||
|
|
||||||
as := &servlet.AuthServlets{}
|
as := &servlet.AuthServlets{}
|
||||||
|
as.RPCServerCodec = rpcServerCodec
|
||||||
as.UseSession = true
|
as.UseSession = true
|
||||||
|
|
||||||
ps := &servlet.ProbeServlets{}
|
ps := &servlet.ProbeServlets{}
|
||||||
|
ps.RPCServerCodec = rpcServerCodec
|
||||||
ps.UseSession = true
|
ps.UseSession = true
|
||||||
|
|
||||||
ds := &servlet.DataServlets{}
|
ds := &servlet.DataServlets{}
|
||||||
|
|
||||||
sh := &ServerHandlers{
|
sh := &ServerHandlers{
|
||||||
|
|
|
@ -2,35 +2,33 @@ package servlet
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"crypto/rsa"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"sync"
|
|
||||||
|
|
||||||
"github.com/valyala/fasthttp"
|
"github.com/valyala/fasthttp"
|
||||||
|
|
||||||
|
logging "git.loafle.net/commons/logging-go"
|
||||||
|
crp "git.loafle.net/commons/rpc-go/protocol"
|
||||||
"git.loafle.net/commons/server-go"
|
"git.loafle.net/commons/server-go"
|
||||||
"git.loafle.net/commons/server-go/socket"
|
"git.loafle.net/commons/server-go/socket"
|
||||||
|
cssw "git.loafle.net/commons/server-go/socket/web"
|
||||||
occp "git.loafle.net/overflow/commons-go/config/probe"
|
occp "git.loafle.net/overflow/commons-go/config/probe"
|
||||||
og "git.loafle.net/overflow/gateway"
|
og "git.loafle.net/overflow/gateway"
|
||||||
"git.loafle.net/overflow/gateway/external/grpc"
|
"git.loafle.net/overflow/gateway/external/grpc"
|
||||||
ogrs "git.loafle.net/overflow/gateway_rpc/servlet"
|
"git.loafle.net/overflow/gateway/external/kafka"
|
||||||
)
|
)
|
||||||
|
|
||||||
type DataServlet interface {
|
type DataServlet interface {
|
||||||
ogrs.RPCServlet
|
cssw.Servlet
|
||||||
}
|
}
|
||||||
|
|
||||||
type DataServlets struct {
|
type DataServlets struct {
|
||||||
ogrs.RPCServlets
|
cssw.Servlets
|
||||||
|
|
||||||
VerifyKey *rsa.PublicKey
|
RPCServerCodec crp.ServerCodec
|
||||||
SignKey *rsa.PrivateKey
|
|
||||||
|
|
||||||
connections sync.Map
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *DataServlets) Init(serverCtx server.ServerCtx) error {
|
func (s *DataServlets) Init(serverCtx server.ServerCtx) error {
|
||||||
if err := s.RPCServlets.Init(serverCtx); nil != err {
|
if err := s.Servlets.Init(serverCtx); nil != err {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -38,7 +36,7 @@ func (s *DataServlets) Init(serverCtx server.ServerCtx) error {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *DataServlets) OnStart(serverCtx server.ServerCtx) error {
|
func (s *DataServlets) OnStart(serverCtx server.ServerCtx) error {
|
||||||
if err := s.RPCServlets.OnStart(serverCtx); nil != err {
|
if err := s.Servlets.OnStart(serverCtx); nil != err {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -47,12 +45,12 @@ func (s *DataServlets) OnStart(serverCtx server.ServerCtx) error {
|
||||||
|
|
||||||
func (s *DataServlets) OnStop(serverCtx server.ServerCtx) {
|
func (s *DataServlets) OnStop(serverCtx server.ServerCtx) {
|
||||||
|
|
||||||
s.RPCServlets.OnStop(serverCtx)
|
s.Servlets.OnStop(serverCtx)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *DataServlets) Destroy(serverCtx server.ServerCtx) {
|
func (s *DataServlets) Destroy(serverCtx server.ServerCtx) {
|
||||||
|
|
||||||
s.RPCServlets.Destroy(serverCtx)
|
s.Servlets.Destroy(serverCtx)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *DataServlets) Handshake(servletCtx server.ServletCtx, ctx *fasthttp.RequestCtx) (*fasthttp.ResponseHeader, error) {
|
func (s *DataServlets) Handshake(servletCtx server.ServletCtx, ctx *fasthttp.RequestCtx) (*fasthttp.ResponseHeader, error) {
|
||||||
|
@ -91,9 +89,57 @@ func (s *DataServlets) Handshake(servletCtx server.ServletCtx, ctx *fasthttp.Req
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *DataServlets) OnConnect(servletCtx server.ServletCtx, conn socket.Conn) {
|
func (s *DataServlets) OnConnect(servletCtx server.ServletCtx, conn socket.Conn) {
|
||||||
s.RPCServlets.OnConnect(servletCtx, conn)
|
s.Servlets.OnConnect(servletCtx, conn)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *DataServlets) OnDisconnect(servletCtx server.ServletCtx) {
|
func (s *DataServlets) OnDisconnect(servletCtx server.ServletCtx) {
|
||||||
s.RPCServlets.OnDisconnect(servletCtx)
|
s.Servlets.OnDisconnect(servletCtx)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *DataServlets) Handle(servletCtx server.ServletCtx,
|
||||||
|
stopChan <-chan struct{}, doneChan chan<- struct{},
|
||||||
|
readChan <-chan []byte, writeChan chan<- []byte) {
|
||||||
|
defer func() {
|
||||||
|
doneChan <- struct{}{}
|
||||||
|
}()
|
||||||
|
|
||||||
|
var (
|
||||||
|
src crp.ServerRequestCodec
|
||||||
|
params []string
|
||||||
|
err error
|
||||||
|
)
|
||||||
|
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case msg, ok := <-readChan:
|
||||||
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
src, err = s.RPCServerCodec.NewRequest(msg)
|
||||||
|
if nil != err {
|
||||||
|
logging.Logger().Error(err)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
|
params, err = src.Params()
|
||||||
|
if nil != err {
|
||||||
|
logging.Logger().Error(err)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
if nil == params || 2 != len(params) {
|
||||||
|
logging.Logger().Errorf("metric is not valid %v", params)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
err = kafka.Write("metric", []byte(params[0]), []byte(params[1]))
|
||||||
|
if nil != err {
|
||||||
|
logging.Logger().Error(err)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
|
case <-stopChan:
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue
Block a user