From 79e52d1c6ed9a31e6e1d93a40ae01ba6eb15a510 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=88=98=E7=A5=A5=E8=B6=85?= Date: Wed, 19 Jan 2022 15:58:54 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96demo=E6=A8=A1=E5=BC=8F?= =?UTF-8?q?=E8=BF=9B=E5=85=A5=E5=91=BD=E4=BB=A4?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- cmd/edge-admin/main.go | 10 ++++++++-- internal/nodes/admin_node.go | 4 +++- 2 files changed, 11 insertions(+), 3 deletions(-) diff --git a/cmd/edge-admin/main.go b/cmd/edge-admin/main.go index df293f98..cdabc283 100644 --- a/cmd/edge-admin/main.go +++ b/cmd/edge-admin/main.go @@ -9,6 +9,7 @@ import ( "github.com/TeaOSLab/EdgeAdmin/internal/nodes" _ "github.com/TeaOSLab/EdgeAdmin/internal/web" _ "github.com/iwind/TeaGo/bootstrap" + "github.com/iwind/TeaGo/maps" "github.com/iwind/gosock/pkg/gosock" ) @@ -64,12 +65,17 @@ func main() { fmt.Println("[ERROR]the service not started yet, you should start the service first") return } - _, err := sock.Send(&gosock.Command{Code: "demo"}) + reply, err := sock.Send(&gosock.Command{Code: "demo"}) if err != nil { fmt.Println("[ERROR]change demo mode failed: " + err.Error()) return } - fmt.Println("change demo mode successfully") + var isDemo = maps.NewMap(reply.Params).GetBool("isDemo") + if isDemo { + fmt.Println("change demo mode to: on") + } else { + fmt.Println("change demo mode to: off") + } }) app.On("generate", func() { err := gen.Generate() diff --git a/internal/nodes/admin_node.go b/internal/nodes/admin_node.go index 4ef84eed..8443e365 100644 --- a/internal/nodes/admin_node.go +++ b/internal/nodes/admin_node.go @@ -325,7 +325,9 @@ func (this *AdminNode) listenSock() error { _ = cmd.ReplyOk() case "demo": teaconst.IsDemoMode = !teaconst.IsDemoMode - _ = cmd.ReplyOk() + _ = cmd.Reply(&gosock.Command{ + Params: map[string]interface{}{"isDemo": teaconst.IsDemoMode}, + }) case "info": exePath, _ := os.Executable() _ = cmd.Reply(&gosock.Command{