fix(cli): yomo run sfn exit if init error (#828)

yomorun · May 30, 2024 · e7e4ff8 · e7e4ff8
1 parent 965b8d6
commit e7e4ff8
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/cli/serverless/golang/templates/main.tmpl b/cli/serverless/golang/templates/main.tmpl
@@ -34,7 +34,10 @@ func run(_ *cobra.Command, _ []string) {
 	)
 	{{if .WithInitFunc}}
 	// init
-	sfn.Init(Init)
+	if err := sfn.Init(Init); err != nil {
+		log.Printf("[sfn] init error[%s], %v\n", addr, err)
+		os.Exit(1)
+	}
 	{{end}}
 	// set observe data tags
 	sfn.SetObserveDataTags(DataTags()...)

diff --git a/cli/serverless/golang/templates/main_rx.tmpl b/cli/serverless/golang/templates/main_rx.tmpl
@@ -33,7 +33,10 @@ func run(_ *cobra.Command, _ []string) {
 	)
 	{{if .WithInitFunc}}
 	// init
-	sfn.Init(Init)
+	if err := sfn.Init(Init); err != nil {
+		log.Printf("[sfn] init error[%s], %v\n", addr, err)
+		os.Exit(1)
+	}
 	{{end}}
 	// set observe data tags
 	sfn.SetObserveDataTags(DataTags()...)