Add PFCPEntity.Stop()

close #28
nextmn · Sep 4, 2024 · 0e5f627 · 0e5f627
1 parent 1e3e40d
commit 0e5f627
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -12,6 +12,7 @@
 ```golang
 upNode := NewPFCPEntityUP(UPF_NODE_ID, UPF_IP_ADDR) // node id can be an IP Address or a FQDN
 upNode.Start()
+defer upNode.Close()
 // Access list of associations
 associations := upNode.GetPFCPAssociations()
 // Access list of sessions
@@ -23,6 +24,7 @@ sessions := upNode.GetPFCPSessions()
 ```golang
 cpNode := NewPFCPEntityCP(SMF_NODE_ID, SMF_IP_ADDR) // node id can be an IP Address or a FQDN
 cpNode.Start()
+defer cpNode.Close()
 association, _ := cpNode.NewEstablishedPFCPAssociation(ie.NewNodeIDHeuristic(UPFADDR))
 session, _ := a.CreateSession(pdrs, fars)
 

diff --git a/pfcp/entity.go b/pfcp/entity.go
@@ -206,6 +206,10 @@ func (e *PFCPEntity) Start() error {
 	return nil
 }
 
+func (e *PFCPEntity) Stop() error {
+	return e.conn.Close()
+}
+
 func (e *PFCPEntity) IsUserPlane() bool {
 	return e.kind == "UP"
 }