overflow_cloud/probe
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ing

Browse Source
This commit is contained in:
crusader
2018-05-10 19:18:00 +09:00
parent c479442b6f
commit 6b52a1ef57
5 changed files with 42 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
auth/service/NoAuthProbeService.go
View File

@@ -35,7 +35,7 @@ type NoAuthProbeService struct {
}
func (s *NoAuthProbeService) InitService() error {
client, err := auth.New(s.HandleTempKey, s)
client, err := auth.New(s.HandleResponse, s)
if nil != err {
return err
}
@@ -98,12 +98,21 @@ func (s *NoAuthProbeService) Deny() error {
return nil
}
func (s *NoAuthProbeService) HandleTempKey(tempKey string) {
logging.Logger().Infof("registered by central")
s.AuthConfig.TempKey = &tempKey
if err := configuration.Save(s.AuthConfig, config.NoAuthProbeConfigFilePath(), true); nil != err {
logging.Logger().Error(err)
s.AuthDoneChan <- err
return
func (s *NoAuthProbeService) HandleResponse(method string, param string) {
switch method {
case occn.HTTPResponseHeaderValue_NoAuthProbe_Method_TempProbeKey:
logging.Logger().Infof("registered by central")
s.AuthConfig.TempKey = &param
if err := configuration.Save(s.AuthConfig, config.NoAuthProbeConfigFilePath(), true); nil != err {
logging.Logger().Error(err)
s.AuthDoneChan <- err
return
}
case occn.HTTPResponseHeaderValue_NoAuthProbe_Method_Accept:
s.Accept(param)
case occn.HTTPResponseHeaderValue_NoAuthProbe_Method_Deny:
s.Deny()
}
}