ing
This commit is contained in:
crusader
parent
d938e396f9
commit
6bcb527545
128
probe/_probe.go
Normal file
128
probe/_probe.go
Normal file
|
|
@ -0,0 +1,128 @@
|
||||||
|
package probe
|
||||||
|
|
||||||
|
// import (
|
||||||
|
// "context"
|
||||||
|
// "fmt"
|
||||||
|
// "net/http"
|
||||||
|
|
||||||
|
// "git.loafle.net/commons_go/logging"
|
||||||
|
// "git.loafle.net/overflow/overflow_probes/central/api/module"
|
||||||
|
// "git.loafle.net/overflow/overflow_probes/central/client"
|
||||||
|
// "git.loafle.net/overflow/overflow_probes/commons"
|
||||||
|
// "git.loafle.net/overflow/overflow_probes/config"
|
||||||
|
// opuu "git.loafle.net/overflow/overflow_probes/util/url"
|
||||||
|
// )
|
||||||
|
|
||||||
|
// const (
|
||||||
|
// probeEntryPoint = "/probe"
|
||||||
|
// fileEntryPoint = "/file"
|
||||||
|
// metricEntryPoint = "/metric"
|
||||||
|
// )
|
||||||
|
|
||||||
|
// type Prober interface {
|
||||||
|
// commons.Starter
|
||||||
|
// commons.Shutdowner
|
||||||
|
// }
|
||||||
|
|
||||||
|
// type probe struct {
|
||||||
|
// probeEntryURL string
|
||||||
|
// fileEntryURL string
|
||||||
|
// metricEntryURL string
|
||||||
|
|
||||||
|
// probeClient client.Client
|
||||||
|
// fileClient client.Client
|
||||||
|
// metricClient client.Client
|
||||||
|
|
||||||
|
// shutdown chan bool
|
||||||
|
// }
|
||||||
|
|
||||||
|
// func New() (Prober, error) {
|
||||||
|
// p := &probe{
|
||||||
|
// shutdown: make(chan bool),
|
||||||
|
// }
|
||||||
|
|
||||||
|
// var err error
|
||||||
|
|
||||||
|
// if p.probeEntryURL, err = opuu.Join(config.Config.Central.URL, probeEntryPoint); nil != err {
|
||||||
|
// return nil, err
|
||||||
|
// }
|
||||||
|
// if p.fileEntryURL, err = opuu.Join(config.Config.Central.URL, fileEntryPoint); nil != err {
|
||||||
|
// return nil, err
|
||||||
|
// }
|
||||||
|
// if p.metricEntryURL, err = opuu.Join(config.Config.Central.URL, metricEntryPoint); nil != err {
|
||||||
|
// return nil, err
|
||||||
|
// }
|
||||||
|
|
||||||
|
// p.probeClient = client.New()
|
||||||
|
// p.fileClient = client.New()
|
||||||
|
// p.metricClient = client.New()
|
||||||
|
|
||||||
|
// return p, nil
|
||||||
|
// }
|
||||||
|
|
||||||
|
// func (p *probe) Start() error {
|
||||||
|
// if err := p.connectToCentral(); nil != err {
|
||||||
|
// return err
|
||||||
|
// }
|
||||||
|
|
||||||
|
// p.listen()
|
||||||
|
|
||||||
|
// return nil
|
||||||
|
// }
|
||||||
|
|
||||||
|
// func (p *probe) listen() {
|
||||||
|
// go func() {
|
||||||
|
// for {
|
||||||
|
// select {
|
||||||
|
// case <-p.shutdown:
|
||||||
|
// break
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
// }()
|
||||||
|
|
||||||
|
// }
|
||||||
|
|
||||||
|
// func (p *probe) connectToCentral() error {
|
||||||
|
// var err error
|
||||||
|
// var res *http.Response
|
||||||
|
// if res, err = client.ConnectToCentralAsProbe(p.probeClient, p.probeEntryURL); nil != err {
|
||||||
|
// return err
|
||||||
|
// }
|
||||||
|
|
||||||
|
// encryptionKey := res.Header.Get(module.ProbeHeader_Probe_EncryptionKey)
|
||||||
|
// config.EncryptionKey = &encryptionKey
|
||||||
|
|
||||||
|
// p.probeClient.OnNotify(p.onNotify)
|
||||||
|
|
||||||
|
// // if _, err = client.ConnectToCentralAsProbe(p.metricClient, p.metricEntryURL); nil != err {
|
||||||
|
// // return err
|
||||||
|
// // }
|
||||||
|
|
||||||
|
// return nil
|
||||||
|
// }
|
||||||
|
|
||||||
|
// func (p *probe) sendNotifyToCentral(method string, params ...string) {
|
||||||
|
// if err := p.probeClient.Notify(method, params); nil != err {
|
||||||
|
// logging.Logger().Error(fmt.Sprintf("Probe notify: %v", err))
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
|
||||||
|
// func (p *probe) Shutdown(ctx context.Context) error {
|
||||||
|
// for {
|
||||||
|
// p.stop(fmt.Errorf("Shutdown"))
|
||||||
|
// select {
|
||||||
|
// case <-ctx.Done():
|
||||||
|
// return ctx.Err()
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
|
||||||
|
// func (p *probe) stop(err error) {
|
||||||
|
// defer close(p.shutdown)
|
||||||
|
|
||||||
|
// ctx := context.Background()
|
||||||
|
// if err := p.probeClient.Shutdown(ctx); nil != err {
|
||||||
|
// logging.Logger().Error(fmt.Sprintf("Client of Probe: %v", err))
|
||||||
|
// }
|
||||||
|
|
||||||
|
// }
|
||||||
11
probe/client/client.go
Normal file
11
probe/client/client.go
Normal file
|
|
@ -0,0 +1,11 @@
|
||||||
|
package client
|
||||||
|
|
||||||
|
import (
|
||||||
|
cwfc "git.loafle.net/commons_go/websocket_fasthttp/client"
|
||||||
|
oogwc "git.loafle.net/overflow/overflow_gateway_websocket/client"
|
||||||
|
)
|
||||||
|
|
||||||
|
func NewClient(ch oogwc.ClientHandler, sb cwfc.SocketBuilder) oogwc.Client {
|
||||||
|
|
||||||
|
return oogwc.New(ch, sb)
|
||||||
|
}
|
||||||
31
probe/client/client_handlers.go
Normal file
31
probe/client/client_handlers.go
Normal file
|
|
@ -0,0 +1,31 @@
|
||||||
|
package client
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"git.loafle.net/commons_go/logging"
|
||||||
|
crc "git.loafle.net/commons_go/rpc/client"
|
||||||
|
crr "git.loafle.net/commons_go/rpc/registry"
|
||||||
|
oogwc "git.loafle.net/overflow/overflow_gateway_websocket/client"
|
||||||
|
)
|
||||||
|
|
||||||
|
func NewClientHandler(rpcInvoker crr.RPCInvoker) oogwc.ClientHandler {
|
||||||
|
ch := &ClientHandlers{}
|
||||||
|
ch.ClientHandler = oogwc.NewClientHandler(rpcInvoker)
|
||||||
|
|
||||||
|
return ch
|
||||||
|
}
|
||||||
|
|
||||||
|
type ClientHandlers struct {
|
||||||
|
oogwc.ClientHandler
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ch *ClientHandlers) Init(clientCTX crc.ClientContext) error {
|
||||||
|
logging.Logger().Info(fmt.Sprintf("Probe: client has been initialized"))
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ch *ClientHandlers) Destroy(clientCTX crc.ClientContext) {
|
||||||
|
logging.Logger().Info(fmt.Sprintf("Probe: client has been destroyed"))
|
||||||
|
}
|
||||||
51
probe/client/socket_builders.go
Normal file
51
probe/client/socket_builders.go
Normal file
|
|
@ -0,0 +1,51 @@
|
||||||
|
package client
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"git.loafle.net/commons_go/logging"
|
||||||
|
cwfc "git.loafle.net/commons_go/websocket_fasthttp/client"
|
||||||
|
oocmn "git.loafle.net/overflow/overflow_commons_go/modules/noauthprobe"
|
||||||
|
oopar "git.loafle.net/overflow/overflow_probes/auth/rpc"
|
||||||
|
oopcc "git.loafle.net/overflow/overflow_probes/central/client"
|
||||||
|
)
|
||||||
|
|
||||||
|
func NewSocketBuilder(napService *oopar.NoAuthProbeService) cwfc.SocketBuilder {
|
||||||
|
sb := &SocketBuilders{
|
||||||
|
napService: napService,
|
||||||
|
}
|
||||||
|
sb.SocketBuilders = oopcc.NewSocketBuilder(oocmn.HTTPEntry_NoAuthProbe)
|
||||||
|
if nil == sb.SocketBuilders {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return sb
|
||||||
|
}
|
||||||
|
|
||||||
|
type SocketBuilders struct {
|
||||||
|
*oopcc.SocketBuilders
|
||||||
|
|
||||||
|
napService *oopar.NoAuthProbeService
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sb *SocketBuilders) SocketHandler() cwfc.SocketHandler {
|
||||||
|
return newSocketHandler(sb.napService)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sb *SocketBuilders) GetRequestHeader() http.Header {
|
||||||
|
h := sb.napService.GetRequestHeader()
|
||||||
|
header := http.Header{}
|
||||||
|
for k, v := range h {
|
||||||
|
header[k] = v
|
||||||
|
}
|
||||||
|
|
||||||
|
return header
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sb *SocketBuilders) Validate() {
|
||||||
|
sb.SocketBuilders.Validate()
|
||||||
|
|
||||||
|
if nil == sb.napService {
|
||||||
|
logging.Logger().Panic("Auth: NoAuthProbeService must be specified")
|
||||||
|
}
|
||||||
|
}
|
||||||
41
probe/client/socket_handlers.go
Normal file
41
probe/client/socket_handlers.go
Normal file
|
|
@ -0,0 +1,41 @@
|
||||||
|
package client
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"git.loafle.net/commons_go/logging"
|
||||||
|
cwfc "git.loafle.net/commons_go/websocket_fasthttp/client"
|
||||||
|
ooccn "git.loafle.net/overflow/overflow_commons_go/config/noauthprobe"
|
||||||
|
oocmn "git.loafle.net/overflow/overflow_commons_go/modules/noauthprobe"
|
||||||
|
oopar "git.loafle.net/overflow/overflow_probes/auth/rpc"
|
||||||
|
)
|
||||||
|
|
||||||
|
type SocketHandlers struct {
|
||||||
|
cwfc.SocketHandlers
|
||||||
|
|
||||||
|
napService *oopar.NoAuthProbeService
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sh *SocketHandlers) OnConnect(socketContext cwfc.SocketContext, res *http.Response) {
|
||||||
|
logging.Logger().Info(fmt.Sprintf("Auth: client has been connected res[%v]", res))
|
||||||
|
|
||||||
|
switch sh.napService.Config.State() {
|
||||||
|
case ooccn.NoAuthProbeStateTypeNotRegisterd:
|
||||||
|
tempProbeKey := res.Header.Get(oocmn.HTTPResponseHeaderKey_NoAuthProbe_SetTempProbeKey)
|
||||||
|
sh.napService.SetTempProbeKey(tempProbeKey)
|
||||||
|
case ooccn.NoAuthProbeStateTypeRegisterd:
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sh *SocketHandlers) OnDisconnect(soc cwfc.Socket) {
|
||||||
|
logging.Logger().Info(fmt.Sprintf("Auth: client has been disconnected soc[%v]", soc))
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func newSocketHandler(napService *oopar.NoAuthProbeService) cwfc.SocketHandler {
|
||||||
|
return &SocketHandlers{
|
||||||
|
napService: napService,
|
||||||
|
}
|
||||||
|
}
|
||||||
127
probe/probe.go
127
probe/probe.go
|
|
@ -1,128 +1 @@
|
||||||
package probe
|
package probe
|
||||||
|
|
||||||
// import (
|
|
||||||
// "context"
|
|
||||||
// "fmt"
|
|
||||||
// "net/http"
|
|
||||||
|
|
||||||
// "git.loafle.net/commons_go/logging"
|
|
||||||
// "git.loafle.net/overflow/overflow_probes/central/api/module"
|
|
||||||
// "git.loafle.net/overflow/overflow_probes/central/client"
|
|
||||||
// "git.loafle.net/overflow/overflow_probes/commons"
|
|
||||||
// "git.loafle.net/overflow/overflow_probes/config"
|
|
||||||
// opuu "git.loafle.net/overflow/overflow_probes/util/url"
|
|
||||||
// )
|
|
||||||
|
|
||||||
// const (
|
|
||||||
// probeEntryPoint = "/probe"
|
|
||||||
// fileEntryPoint = "/file"
|
|
||||||
// metricEntryPoint = "/metric"
|
|
||||||
// )
|
|
||||||
|
|
||||||
// type Prober interface {
|
|
||||||
// commons.Starter
|
|
||||||
// commons.Shutdowner
|
|
||||||
// }
|
|
||||||
|
|
||||||
// type probe struct {
|
|
||||||
// probeEntryURL string
|
|
||||||
// fileEntryURL string
|
|
||||||
// metricEntryURL string
|
|
||||||
|
|
||||||
// probeClient client.Client
|
|
||||||
// fileClient client.Client
|
|
||||||
// metricClient client.Client
|
|
||||||
|
|
||||||
// shutdown chan bool
|
|
||||||
// }
|
|
||||||
|
|
||||||
// func New() (Prober, error) {
|
|
||||||
// p := &probe{
|
|
||||||
// shutdown: make(chan bool),
|
|
||||||
// }
|
|
||||||
|
|
||||||
// var err error
|
|
||||||
|
|
||||||
// if p.probeEntryURL, err = opuu.Join(config.Config.Central.URL, probeEntryPoint); nil != err {
|
|
||||||
// return nil, err
|
|
||||||
// }
|
|
||||||
// if p.fileEntryURL, err = opuu.Join(config.Config.Central.URL, fileEntryPoint); nil != err {
|
|
||||||
// return nil, err
|
|
||||||
// }
|
|
||||||
// if p.metricEntryURL, err = opuu.Join(config.Config.Central.URL, metricEntryPoint); nil != err {
|
|
||||||
// return nil, err
|
|
||||||
// }
|
|
||||||
|
|
||||||
// p.probeClient = client.New()
|
|
||||||
// p.fileClient = client.New()
|
|
||||||
// p.metricClient = client.New()
|
|
||||||
|
|
||||||
// return p, nil
|
|
||||||
// }
|
|
||||||
|
|
||||||
// func (p *probe) Start() error {
|
|
||||||
// if err := p.connectToCentral(); nil != err {
|
|
||||||
// return err
|
|
||||||
// }
|
|
||||||
|
|
||||||
// p.listen()
|
|
||||||
|
|
||||||
// return nil
|
|
||||||
// }
|
|
||||||
|
|
||||||
// func (p *probe) listen() {
|
|
||||||
// go func() {
|
|
||||||
// for {
|
|
||||||
// select {
|
|
||||||
// case <-p.shutdown:
|
|
||||||
// break
|
|
||||||
// }
|
|
||||||
// }
|
|
||||||
// }()
|
|
||||||
|
|
||||||
// }
|
|
||||||
|
|
||||||
// func (p *probe) connectToCentral() error {
|
|
||||||
// var err error
|
|
||||||
// var res *http.Response
|
|
||||||
// if res, err = client.ConnectToCentralAsProbe(p.probeClient, p.probeEntryURL); nil != err {
|
|
||||||
// return err
|
|
||||||
// }
|
|
||||||
|
|
||||||
// encryptionKey := res.Header.Get(module.ProbeHeader_Probe_EncryptionKey)
|
|
||||||
// config.EncryptionKey = &encryptionKey
|
|
||||||
|
|
||||||
// p.probeClient.OnNotify(p.onNotify)
|
|
||||||
|
|
||||||
// // if _, err = client.ConnectToCentralAsProbe(p.metricClient, p.metricEntryURL); nil != err {
|
|
||||||
// // return err
|
|
||||||
// // }
|
|
||||||
|
|
||||||
// return nil
|
|
||||||
// }
|
|
||||||
|
|
||||||
// func (p *probe) sendNotifyToCentral(method string, params ...string) {
|
|
||||||
// if err := p.probeClient.Notify(method, params); nil != err {
|
|
||||||
// logging.Logger().Error(fmt.Sprintf("Probe notify: %v", err))
|
|
||||||
// }
|
|
||||||
// }
|
|
||||||
|
|
||||||
// func (p *probe) Shutdown(ctx context.Context) error {
|
|
||||||
// for {
|
|
||||||
// p.stop(fmt.Errorf("Shutdown"))
|
|
||||||
// select {
|
|
||||||
// case <-ctx.Done():
|
|
||||||
// return ctx.Err()
|
|
||||||
// }
|
|
||||||
// }
|
|
||||||
// }
|
|
||||||
|
|
||||||
// func (p *probe) stop(err error) {
|
|
||||||
// defer close(p.shutdown)
|
|
||||||
|
|
||||||
// ctx := context.Background()
|
|
||||||
// if err := p.probeClient.Shutdown(ctx); nil != err {
|
|
||||||
// logging.Logger().Error(fmt.Sprintf("Client of Probe: %v", err))
|
|
||||||
// }
|
|
||||||
|
|
||||||
// }
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue
Block a user