add strict flag for assert (#329)

resonatehq · May 20, 2024 · d5a4a3d · d5a4a3d
1 parent 4875c84
commit d5a4a3d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/cmd/serve/serve.go b/cmd/serve/serve.go
@@ -184,6 +184,10 @@ func ServeCmd() *cobra.Command {
 		},
 	}
 
+	// assert
+	cmd.Flags().Bool("ignore-asserts", false, "ignore-asserts mode")
+	_ = viper.BindPFlag("ignore-asserts", cmd.Flags().Lookup("ignore-asserts"))
+
 	// api
 	cmd.Flags().Int("api-size", 100, "size of the submission queue buffered channel")
 	cmd.Flags().String("api-http-addr", "0.0.0.0:8001", "http server address")

diff --git a/internal/util/util.go b/internal/util/util.go
@@ -6,10 +6,12 @@ import (
 	"time"
 
 	"github.com/robfig/cron/v3"
+	"github.com/spf13/viper"
 )
 
 func Assert(cond bool, msg string) {
-	if !cond {
+	ignoreAsserts := viper.GetBool("ignore-asserts")
+	if !ignoreAsserts && !cond {
 		panic(msg)
 	}
 }