Skip to content

Commit

Permalink
review comments
Browse files Browse the repository at this point in the history
  • Loading branch information
bashokba committed Jan 14, 2021
1 parent 9461805 commit 9ffa38a
Show file tree
Hide file tree
Showing 2 changed files with 27 additions and 29 deletions.
29 changes: 0 additions & 29 deletions pkg/hostagent/agent.go
Original file line number Diff line number Diff line change
Expand Up @@ -17,8 +17,6 @@ package hostagent
import (
"fmt"
"net"
"os"
"path/filepath"
"sync"
"time"

Expand Down Expand Up @@ -387,14 +385,6 @@ func (agent *HostAgent) Run(stopCh <-chan struct{}) {
}
go agent.processSyncQueue(agent.syncQueue, stopCh)
}
if agent.config.OpFlexFaultDir == "" {
agent.log.Warn("OpFlex Faults directories not set")
} else {
err = removeAllFiles(agent.config.OpFlexFaultDir)
if err != nil {
agent.log.Warn("Not able to clear faults files on agent: ", err.Error())
}
}
agent.log.Info("Starting endpoint RPC")
err = agent.runEpRPC(stopCh)
if err != nil {
Expand All @@ -403,22 +393,3 @@ func (agent *HostAgent) Run(stopCh <-chan struct{}) {

agent.cleanupSetup()
}

func removeAllFiles(dir string) error {
d, err := os.Open(dir)
if err != nil {
return err
}
defer d.Close()
names, err := d.Readdirnames(-1)
if err != nil {
return err
}
for _, name := range names {
err = os.RemoveAll(filepath.Join(dir, name))
if err != nil {
return err
}
}
return nil
}
27 changes: 27 additions & 0 deletions pkg/hostagent/opflex.go
Original file line number Diff line number Diff line change
Expand Up @@ -327,6 +327,14 @@ func (agent *HostAgent) updateOpflexConfig() {
agent.log.Debug("OpFlex agent configuration path not set")
return
}
if agent.config.OpFlexFaultDir == "" {
agent.log.Warn("OpFlex Faults directories not set")
} else {
err := removeAllFiles(agent.config.OpFlexFaultDir)
if err != nil {
agent.log.Warn("Not able to clear faults files on agent: ", err.Error())
}
}

newNodeConfig := agent.discoverHostConfig()
if newNodeConfig == nil {
Expand Down Expand Up @@ -383,3 +391,22 @@ func (agent *HostAgent) writeOpflexConfig() error {
}
return nil
}

func removeAllFiles(dir string) error {
d, err := os.Open(dir)
if err != nil {
return err
}
defer d.Close()
names, err := d.Readdirnames(-1)
if err != nil {
return err
}
for _, name := range names {
err = os.RemoveAll(filepath.Join(dir, name))
if err != nil {
return err
}
}
return nil
}

0 comments on commit 9ffa38a

Please sign in to comment.