2018-04-19 11:36:56 +00:00
|
|
|
package service
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
"reflect"
|
|
|
|
"sync"
|
|
|
|
|
2018-04-23 10:24:00 +00:00
|
|
|
"git.loafle.net/commons/logging-go"
|
|
|
|
|
2018-04-19 11:36:56 +00:00
|
|
|
cda "git.loafle.net/commons/di-go/annotation"
|
|
|
|
cdr "git.loafle.net/commons/di-go/registry"
|
2018-05-11 02:44:47 +00:00
|
|
|
occu "git.loafle.net/overflow/commons-go/core/util"
|
2018-04-26 09:00:24 +00:00
|
|
|
ocmd "git.loafle.net/overflow/commons-go/model/discovery"
|
|
|
|
ocscd "git.loafle.net/overflow/commons-go/service/container/discovery"
|
2018-04-19 11:36:56 +00:00
|
|
|
ocs "git.loafle.net/overflow/container-go/service"
|
|
|
|
"git.loafle.net/overflow/container_discovery/internal/discoverer"
|
|
|
|
|
|
|
|
// For annotation
|
|
|
|
_ "git.loafle.net/overflow/commons-go/core/annotation"
|
|
|
|
)
|
|
|
|
|
|
|
|
var DiscoveryServiceType = reflect.TypeOf((*DiscoveryService)(nil))
|
|
|
|
|
|
|
|
func init() {
|
|
|
|
cdr.RegisterType(DiscoveryServiceType)
|
|
|
|
}
|
|
|
|
|
|
|
|
type DiscoveryService struct {
|
2018-04-26 09:00:24 +00:00
|
|
|
ocscd.DiscoveryService
|
2018-04-19 11:36:56 +00:00
|
|
|
cda.TypeAnnotation `annotation:"@overflow:RPCService()"`
|
|
|
|
|
|
|
|
ProbeService *ocs.ProbeService `annotation:"@Inject()"`
|
|
|
|
|
|
|
|
pendingDiscovery sync.Map
|
|
|
|
discoverer discoverer.Discoverer
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *DiscoveryService) InitService() error {
|
|
|
|
s.discoverer = discoverer.GetDiscoverer()
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *DiscoveryService) StartService() error {
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *DiscoveryService) StopService() {
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *DiscoveryService) DestroyService() {
|
|
|
|
|
|
|
|
}
|
|
|
|
|
2018-04-27 16:20:01 +00:00
|
|
|
func (s *DiscoveryService) DiscoverZone(requesterID string, dz *ocmd.DiscoverZone) error {
|
2018-04-19 14:45:39 +00:00
|
|
|
go s.handleDiscovery(requesterID, func(dataChan chan *discoverer.DiscoveryData) {
|
2018-04-19 11:36:56 +00:00
|
|
|
s.discoverer.DiscoverZone(dataChan, dz)
|
|
|
|
})
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2018-04-27 16:20:01 +00:00
|
|
|
func (s *DiscoveryService) DiscoverHost(requesterID string, zone *ocmd.Zone, dh *ocmd.DiscoverHost) error {
|
2018-04-19 14:45:39 +00:00
|
|
|
go s.handleDiscovery(requesterID, func(dataChan chan *discoverer.DiscoveryData) {
|
2018-04-19 11:36:56 +00:00
|
|
|
s.discoverer.DiscoverHost(dataChan, zone, dh)
|
|
|
|
})
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2018-04-27 16:20:01 +00:00
|
|
|
func (s *DiscoveryService) DiscoverPort(requesterID string, host *ocmd.Host, dp *ocmd.DiscoverPort) error {
|
2018-04-19 14:45:39 +00:00
|
|
|
go s.handleDiscovery(requesterID, func(dataChan chan *discoverer.DiscoveryData) {
|
2018-04-19 11:36:56 +00:00
|
|
|
s.discoverer.DiscoverPort(dataChan, host, dp)
|
|
|
|
})
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2018-04-27 16:20:01 +00:00
|
|
|
func (s *DiscoveryService) DiscoverService(requesterID string, port *ocmd.Port, ds *ocmd.DiscoverService) error {
|
2018-04-19 14:45:39 +00:00
|
|
|
go s.handleDiscovery(requesterID, func(dataChan chan *discoverer.DiscoveryData) {
|
2018-04-19 11:36:56 +00:00
|
|
|
s.discoverer.DiscoverSerice(dataChan, port, ds)
|
|
|
|
})
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *DiscoveryService) StopRequest(requesterID string) error {
|
|
|
|
_stopChan, ok := s.pendingDiscovery.Load(requesterID)
|
|
|
|
if !ok {
|
|
|
|
return fmt.Errorf("discovery request for [%s] is not exist", requesterID)
|
|
|
|
}
|
|
|
|
stopChan := _stopChan.(chan struct{})
|
|
|
|
|
|
|
|
close(stopChan)
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *DiscoveryService) handleDiscovery(requesterID string, discoveryFunc func(dataChan chan *discoverer.DiscoveryData)) error {
|
|
|
|
stopChan := make(chan struct{})
|
|
|
|
s.pendingDiscovery.Store(requesterID, stopChan)
|
|
|
|
defer func() {
|
|
|
|
s.pendingDiscovery.Delete(requesterID)
|
|
|
|
}()
|
|
|
|
|
|
|
|
var dataChan chan *discoverer.DiscoveryData
|
|
|
|
retainChan := make(chan struct{})
|
|
|
|
|
|
|
|
go func() {
|
|
|
|
dataChan = s.discoverer.Retain()
|
|
|
|
close(retainChan)
|
|
|
|
}()
|
|
|
|
|
|
|
|
select {
|
|
|
|
case <-stopChan:
|
|
|
|
return nil
|
|
|
|
case <-retainChan:
|
|
|
|
}
|
|
|
|
defer func() {
|
|
|
|
s.discoverer.Release(dataChan)
|
|
|
|
}()
|
|
|
|
|
2018-04-19 14:45:39 +00:00
|
|
|
go discoveryFunc(dataChan)
|
2018-04-19 11:36:56 +00:00
|
|
|
|
|
|
|
for {
|
|
|
|
select {
|
|
|
|
case data, ok := <-dataChan:
|
|
|
|
if !ok {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
switch data.Type {
|
|
|
|
case discoverer.DiscoveryDataTypeStart:
|
2018-04-23 10:24:00 +00:00
|
|
|
logging.Logger().Debugf("DiscoveryService.DiscoveryStart")
|
2018-05-11 02:44:47 +00:00
|
|
|
s.ProbeService.Send("DiscoveryService.DiscoveryStart", requesterID, occu.Now())
|
2018-04-19 11:36:56 +00:00
|
|
|
case discoverer.DiscoveryDataTypeStop:
|
2018-04-23 10:24:00 +00:00
|
|
|
logging.Logger().Debugf("DiscoveryService.DiscoveryStop")
|
2018-05-11 02:44:47 +00:00
|
|
|
s.ProbeService.Send("DiscoveryService.DiscoveryStop", requesterID, occu.Now())
|
2018-04-19 11:36:56 +00:00
|
|
|
data.Release()
|
|
|
|
return nil
|
|
|
|
case discoverer.DiscoveryDataTypeError:
|
2018-04-27 16:59:23 +00:00
|
|
|
// s.ProbeService.Send("DiscoveryService.DiscoveryError", requesterID, data.Error)
|
|
|
|
logging.Logger().Debugf("DiscoveryService.DiscoveryError %v", data.Error)
|
2018-04-19 11:36:56 +00:00
|
|
|
case discoverer.DiscoveryDataTypeZone:
|
|
|
|
s.ProbeService.Send("DiscoveryService.DiscoveredZone", requesterID, data.Result)
|
|
|
|
case discoverer.DiscoveryDataTypeHost:
|
|
|
|
s.ProbeService.Send("DiscoveryService.DiscoveredHost", requesterID, data.Result)
|
|
|
|
case discoverer.DiscoveryDataTypePort:
|
|
|
|
s.ProbeService.Send("DiscoveryService.DiscoveredPort", requesterID, data.Result)
|
|
|
|
case discoverer.DiscoveryDataTypeService:
|
|
|
|
s.ProbeService.Send("DiscoveryService.DiscoveredService", requesterID, data.Result)
|
|
|
|
}
|
|
|
|
data.Release()
|
|
|
|
case <-stopChan:
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|