Fix gate status

omec-project · Mar 10, 2022 · 78a8373 · 78a8373
1 parent 0100962
commit 78a8373
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git a/pkg/pfcpsim/session/ies.go b/pkg/pfcpsim/session/ies.go
@@ -8,7 +8,6 @@ type IEMethod uint8
 // Definitions for session rules
 
 type direction int
-type gateStatus int
 
 const (
 	notSet direction = iota
@@ -24,8 +23,5 @@ const (
 	ActionBuffer  uint8 = 0x4
 	ActionNotify  uint8 = 0x8
 
-	GateOpen gateStatus = iota
-	GateClosed
-
 	S_TAG = 0x100 // Refer to table 8.2.56-1 in PFCP specs Release 16
 )
diff --git a/pkg/pfcpsim/session/qer_builder.go b/pkg/pfcpsim/session/qer_builder.go
@@ -14,8 +14,8 @@ type qerBuilder struct {
 	dlMbr    uint64
 	isGbrSet bool
 	ulGbr    uint64
-	dlGbr    uint64
-	gateStatus gateStatus
+	dlGbr      uint64
+	gateStatus GateStatus
 
 	isIDSet bool
 }
@@ -65,7 +65,7 @@ func (b *qerBuilder) WithDownlinkGBR(dlGbr uint64) *qerBuilder {
 	return b
 }
 
-func (b *qerBuilder) WithGateStatus(status gateStatus) *qerBuilder {
+func (b *qerBuilder) WithGateStatus(status uint8) *qerBuilder {
 	b.gateStatus = status
 
 	return b
@@ -91,7 +91,7 @@ func (b *qerBuilder) Build() *ie.IE {
 	}
 
 	gate := ie.NewGateStatus(0, 0) //Open
-	if b.gateStatus == GateClosed {
+	if b.gateStatus == ie.GateStatusClosed {
 		gate = ie.NewGateStatus(1, 1)
 	}