feat: circle mode

LilithGames · Jul 6, 2021 · 22fae3c · 22fae3c
1 parent e6f3aa3
commit 22fae3c
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 19 deletions.
diff --git a/pkg/agent/manager.go b/pkg/agent/manager.go
@@ -75,6 +75,7 @@ func (m *manager) startAgentEngine(content []byte) error {
 
 func (m *manager) startReadyService() {
 	m.stream.setPlanCount(len(m.engine.plans))
+	var executors []*executor
 	for _, plan := range m.engine.plans {
 		treeCfg := m.engine.trees[plan.TreeName]
 		tree := loader.CreateBevTreeFromConfig(treeCfg, m.engine.registerMap)
@@ -84,6 +85,7 @@ func (m *manager) startReadyService() {
 			tree:     tree,
 			metadata: m.engine.metadata,
 		}
+		executors = append(executors, executor)
 		m.startExecutor(executor)
 	}
 }

diff --git a/pkg/transfer/transfer.pb.go b/pkg/transfer/transfer.pb.go
diff --git a/pkg/transfer/transfer.proto b/pkg/transfer/transfer.proto
@@ -7,6 +7,7 @@ enum ACTION {
   STOP_AGENT = 1;
   REPORT_DATA = 2;
   FINISH_PLAN = 3;
+  START_CIRCLE = 4;
 }
 
 message Mail {