fix agent data directory resolution (#991)

This commit is contained in:
henrygd
2025-07-30 14:34:36 -04:00
parent 39dbe0eac5
commit ce93d54aa7
2 changed files with 3 additions and 3 deletions

View File

@@ -111,7 +111,7 @@ func main() {
serverConfig.Addr = addr serverConfig.Addr = addr
serverConfig.Network = agent.GetNetwork(addr) serverConfig.Network = agent.GetNetwork(addr)
agent, err := agent.NewAgent("") agent, err := agent.NewAgent()
if err != nil { if err != nil {
log.Fatal("Failed to create agent: ", err) log.Fatal("Failed to create agent: ", err)
} }

View File

@@ -40,13 +40,13 @@ type Agent struct {
// NewAgent creates a new agent with the given data directory for persisting data. // NewAgent creates a new agent with the given data directory for persisting data.
// If the data directory is not set, it will attempt to find the optimal directory. // If the data directory is not set, it will attempt to find the optimal directory.
func NewAgent(dataDir string) (agent *Agent, err error) { func NewAgent(dataDir ...string) (agent *Agent, err error) {
agent = &Agent{ agent = &Agent{
fsStats: make(map[string]*system.FsStats), fsStats: make(map[string]*system.FsStats),
cache: NewSessionCache(69 * time.Second), cache: NewSessionCache(69 * time.Second),
} }
agent.dataDir, err = getDataDir(dataDir) agent.dataDir, err = getDataDir(dataDir...)
if err != nil { if err != nil {
slog.Warn("Data directory not found") slog.Warn("Data directory not found")
} else { } else {