Add server.GetPFCPSessions

nextmn · Feb 3, 2022 · 092afe0 · 092afe0
1 parent 331e350
commit 092afe0
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/pfcp_server_entity.go b/pfcp_server_entity.go
@@ -102,3 +102,14 @@ func (e *PFCPServerEntity) Start() error {
 	}()
 	return nil
 }
+
+func (e *PFCPServerEntity) GetPFCPSessions() []*PFCPSession {
+	sessions := make([]*PFCPSession, 0)
+	for _, a := range e.associations {
+		as := a.GetSessions()
+		for _, s := range as {
+			sessions = append(sessions, s)
+		}
+	}
+	return sessions
+}
diff --git a/pfcp_session.go b/pfcp_session.go
@@ -11,8 +11,6 @@ import (
 	"github.com/wmnsk/go-pfcp/message"
 )
 
-const MAX_PDR = 1024
-
 type PFCPSession struct {
 	fseid     *ie.IE
 	rseid     uint64
@@ -28,7 +26,7 @@ func NewPFCPSession(fseid *ie.IE, rseid uint64) PFCPSession {
 		rseid:     rseid, // SEID present in FSEID ie send by remote peer
 		pdr:       make(map[uint16]*pfcprule.PDR),
 		far:       make(map[uint32]*pfcprule.FAR),
-		sortedPDR: make(pfcprule.PDRs, MAX_PDR),
+		sortedPDR: make(pfcprule.PDRs, 0),
 		mu:        sync.Mutex{},
 	}
 }