gered/vault-sidekick
Archived
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Refactor main.go loop. Fix panic.

Browse source
This commit is contained in:
James Munnelly 2017-07-18 00:41:11 +01:00
parent 6ac7073f27
commit d11da90402
1 changed files with 17 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
main.go
View file

@ -82,24 +82,29 @@ func main() {
case evt := <-updates:
glog.V(10).Infof("recieved an update from the resource: %s", evt.Resource)
go func(r VaultEvent) {
if err := processResource(evt.Resource, evt.Secret); err != nil {
glog.Errorf("failed to write out the update, error: %s", err)
}
toProcessLock.Lock()
defer toProcessLock.Unlock()
for i, r := range toProcess {
if evt.Resource == r {
switch {
case evt.Resource.maxRetries > 0 && evt.Resource.maxRetries < evt.Resource.retries:
toProcess = append(toProcess[:i], toProcess[i+1:]...)
failedResource = true
break
case options.oneShot:
if evt.Type == EventTypeSuccess {
switch r.Type {
case EventTypeSuccess:
if err := processResource(evt.Resource, evt.Secret); err != nil {
glog.Errorf("failed to write out the update, error: %s", err)
}
if options.oneShot {
for i, r := range toProcess {
if evt.Resource == r {
toProcess = append(toProcess[:i], toProcess[i+1:]...)
}
}
}
case EventTypeFailure:
if evt.Resource.maxRetries > 0 && evt.Resource.maxRetries < evt.Resource.retries {
for i, r := range toProcess {
if evt.Resource == r {
toProcess = append(toProcess[:i], toProcess[i+1:]...)
failedResource = true
}
}
}
}
if len(toProcess) == 0 {
glog.Infof("no resources left to process. exiting...")