فهرست منبع

fix: pass context to Program function for improved goroutine handling

Jacky 1 روز پیش
والد
کامیت
836e25f805
1فایلهای تغییر یافته به همراه2 افزوده شده و 5 حذف شده
  1. 2 5
      main.go

+ 2 - 5
main.go

@@ -27,11 +27,8 @@ import (
 	cSettings "github.com/uozi-tech/cosy/settings"
 )
 
-func Program(confPath string) func(l []net.Listener) error {
+func Program(ctx context.Context, confPath string) func(l []net.Listener) error {
 	return func(l []net.Listener) error {
-		ctx, cancel := context.WithCancel(context.Background())
-		defer cancel()
-
 		listener := l[0]
 
 		cosy.RegisterMigrationsBeforeAutoMigrate(migrate.BeforeAutoMigrate)
@@ -112,7 +109,7 @@ func main() {
 	defer cancel()
 
 	err := risefront.New(ctx, risefront.Config{
-		Run:       Program(confPath),
+		Run:       Program(ctx, confPath),
 		Name:      "nginx-ui",
 		Addresses: []string{fmt.Sprintf("%s:%d", cSettings.ServerSettings.Host, cSettings.ServerSettings.Port)},
 		ErrorHandler: func(kind string, err error) {