优化demo模式进入命令

This commit is contained in:
GoEdgeLab
2022-01-19 15:58:54 +08:00
parent 60da62ce5e
commit db951b5344
2 changed files with 11 additions and 3 deletions

View File

@@ -9,6 +9,7 @@ import (
"github.com/TeaOSLab/EdgeAdmin/internal/nodes" "github.com/TeaOSLab/EdgeAdmin/internal/nodes"
_ "github.com/TeaOSLab/EdgeAdmin/internal/web" _ "github.com/TeaOSLab/EdgeAdmin/internal/web"
_ "github.com/iwind/TeaGo/bootstrap" _ "github.com/iwind/TeaGo/bootstrap"
"github.com/iwind/TeaGo/maps"
"github.com/iwind/gosock/pkg/gosock" "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") fmt.Println("[ERROR]the service not started yet, you should start the service first")
return return
} }
_, err := sock.Send(&gosock.Command{Code: "demo"}) reply, err := sock.Send(&gosock.Command{Code: "demo"})
if err != nil { if err != nil {
fmt.Println("[ERROR]change demo mode failed: " + err.Error()) fmt.Println("[ERROR]change demo mode failed: " + err.Error())
return 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() { app.On("generate", func() {
err := gen.Generate() err := gen.Generate()

View File

@@ -325,7 +325,9 @@ func (this *AdminNode) listenSock() error {
_ = cmd.ReplyOk() _ = cmd.ReplyOk()
case "demo": case "demo":
teaconst.IsDemoMode = !teaconst.IsDemoMode teaconst.IsDemoMode = !teaconst.IsDemoMode
_ = cmd.ReplyOk() _ = cmd.Reply(&gosock.Command{
Params: map[string]interface{}{"isDemo": teaconst.IsDemoMode},
})
case "info": case "info":
exePath, _ := os.Executable() exePath, _ := os.Executable()
_ = cmd.Reply(&gosock.Command{ _ = cmd.Reply(&gosock.Command{