feat: use environments

examples/hello/task.yaml

@@ -10,5 +10,5 @@ plans:
     parallel: 100
     interval: 1
 
-metadata:
-  org: lilithgame
+environments:
+  branch: master

pkg/agent/engine.go

@@ -15,7 +15,7 @@ import (
 
 type Config struct {
 	Plans    []*transfer.Plan
-	Metadata map[string]string
+	Environments map[string]string
 }
 
 func (c *Config) MarshalBinary() (data []byte, err error) {
@@ -93,7 +93,7 @@ func (b *Behavior) BuildEngineFromConfig(conf []byte) *Engine {
 		log.Panic("unmarshal plan config", zap.Error(err))
 	}
 	engine := &Engine{Behavior: b}
-	engine.metadata = cfg.Metadata
+	engine.metadata = cfg.Environments
 	for _, plan := range cfg.Plans {
 		if _, ok := b.trees[plan.TreeName]; !ok {
 			log.Panic("plan name not found")
@@ -126,6 +126,12 @@ func (e *Engine) setPlans(plans []*transfer.Plan) {
 	e.plans = plans
 }
 
-func (e *Engine) setMetadata(metadata map[string]string) {
-	e.metadata = metadata
+func (e *Engine) setMetadata(envs map[string]string) {
+	if len(e.metadata) == 0 {
+		e.metadata = envs
+		return
+	}
+	for name, value := range envs {
+		e.metadata[name] = value
+	}
 }

pkg/agent/manager.go

@@ -61,16 +61,13 @@ func (m *manager) reviewMail(mail *transfer.Mail) error {
 }
 
 func (m *manager) startAgentEngine(content []byte) error {
-	var conf Config
-	err := json.Unmarshal(content, &conf)
+	var envs map[string]string
+	err := json.Unmarshal(content, &envs)
 	if err != nil {
 		return err
 	}
-	if conf.Plans != nil {
-		m.engine.setPlans(conf.Plans)
-	}
-	if len(conf.Metadata) != 0 {
-		m.engine.setMetadata(conf.Metadata)
+	if len(envs) != 0 {
+		m.engine.setMetadata(envs)
 	}
 	m.startReadyService()
 	return nil

pkg/transfer/transfer.proto

@@ -24,7 +24,7 @@ message Plan {
 message Event {
   string image = 1;
   int32 agentNum = 2;
-  bytes rawCfg = 3;
+  map<string, string> environments = 3;
 }
 
 message EventReply {