From db6c60ab7b161be80e84dae2e647072c7d311ac7 Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期四, 12 十二月 2019 10:43:38 +0800
Subject: [PATCH] update
---
main.go | 15 +++++++++------
1 files changed, 9 insertions(+), 6 deletions(-)
diff --git a/main.go b/main.go
index a0d3545..156b723 100644
--- a/main.go
+++ b/main.go
@@ -143,18 +143,21 @@
logo.Infoln(os.Args)
ctx, cancel := context.WithCancel(context.Background())
+ ret := false
if role == single {
- proc.SingleRole(ctx, runType, id, gpu, shm)
+ ret = proc.SingleRole(ctx, runType, id, gpu, shm)
} else if role == master {
setParamters()
- proc.MasterRole(ctx)
+ ret = proc.MasterRole(ctx)
} else if role == slave {
- proc.SingleRole(ctx, runType, id, gpu, shm)
+ ret = proc.SingleRole(ctx, runType, id, gpu, shm)
}
- c := make(chan os.Signal, 1)
- signal.Notify(c, os.Interrupt, os.Kill, unix.SIGTERM)
- <-c
+ if ret {
+ c := make(chan os.Signal, 1)
+ signal.Notify(c, os.Interrupt, os.Kill, unix.SIGTERM)
+ <-c
+ }
cancel()
--
Gitblit v1.8.0