fix: empty config

LilithGames · Mar 8, 2022 · 8db206a · 8db206a
1 parent 090d4d8
commit 8db206a
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/examples/hello/main.go b/examples/hello/main.go
@@ -87,7 +87,7 @@ func HelloD(tick agent.Ticker) (behavior3go.Status, error) {
 }
 
 func HelloE(tick agent.Ticker) (behavior3go.Status, error) {
-	return behavior3go.FAILURE, fmt.Errorf("test error essage.##")
+	return behavior3go.FAILURE, fmt.Errorf("test error essage.#123456789#")
 	// return behavior3go.SUCCESS, nil
 }
 

diff --git a/pkg/agent/agent.go b/pkg/agent/agent.go
@@ -3,7 +3,9 @@ package agent
 import (
 	"context"
 	"net/http"
+	"os"
 	"strconv"
+	"strings"
 
 	"github.com/LilithGames/agent-go/tools/metric"
 	"github.com/spf13/viper"
@@ -35,7 +37,15 @@ func NewAgent(engine *Engine, cfg *viper.Viper, opts ...Option) *Agent {
 		log.Panic("absent plans")
 	}
 	if cfg == nil {
-		cfg = viper.New()
+		cfg := viper.New()
+		cfg.SetConfigType("prop")
+		envs := os.Environ()
+		for _, env := range envs {
+			parts := strings.SplitN(env, "=", 2)
+			if len(parts) == 2 {
+				cfg.Set(parts[0], parts[1])
+			}
+		}
 	}
 	id := cfg.GetString("ID")
 	endpoint := cfg.GetString(masterAddr)