refactor: make PFCP context child of app context

free5gc · Sep 25, 2024 · c8cf39f · c8cf39f
1 parent 0f052c7
commit c8cf39f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/cmd/main.go b/cmd/main.go
@@ -86,7 +86,7 @@ func action(cliCtx *cli.Context) error {
 	}
 	factory.UERoutingConfig = ueRoutingCfg
 
-	pfcpStart, pfcpTerminate := utils.InitPFCPFunc()
+	pfcpStart, pfcpTerminate := utils.InitPFCPFunc(ctx)
 	smf, err := service.NewApp(ctx, cfg, tlsKeyLogPath, pfcpStart, pfcpTerminate)
 	if err != nil {
 		sigCh <- nil

diff --git a/pkg/utils/pfcp_util.go b/pkg/utils/pfcp_util.go
@@ -11,12 +11,12 @@ import (
 	"github.com/free5gc/smf/pkg/service"
 )
 
-func InitPFCPFunc() (func(a *service.SmfApp), func()) {
+func InitPFCPFunc(pCtx context.Context) (func(a *service.SmfApp), func()) {
 	smfContext := smf_context.GetSelf()
 
 	pfcpStart := func(a *service.SmfApp) {
 		// Initialize PFCP server
-		smfContext.PfcpContext, smfContext.PfcpCancelFunc = context.WithCancel(context.Background())
+		smfContext.PfcpContext, smfContext.PfcpCancelFunc = context.WithCancel(pCtx)
 
 		udp.Run(pfcp.Dispatch)