Merge pull request #174 from xmidt-org/denopink/feat/qos-normifier

feat: add qos normifier

normify.go

@@ -195,6 +195,19 @@ func SetSessionID(sessionID string) NormifierOption {
 	})
 }
 
+// ClampQualityOfService clamps a wrp message's qos value between 0 and 99.
+func ClampQualityOfService() NormifierOption {
+	return optionFunc(func(m *Message) error {
+		if m.QualityOfService < 0 {
+			m.QualityOfService = 0
+		} else if m.QualityOfService > 99 {
+			m.QualityOfService = 99
+		}
+
+		return nil
+	})
+}
+
 // EnsureMetadataString ensures that the message has the given string metadata.
 // This will always set the metadata.
 func EnsureMetadataString(key, value string) NormifierOption {

normify_test.go

@@ -88,6 +88,24 @@ func TestNormifier(t *testing.T) {
 			want: Message{
 				SessionID: "session",
 			},
+		}, {
+			description: "ClampQualityOfService(), QualityOfService < 0",
+			opt:         ClampQualityOfService(),
+			msg: Message{
+				QualityOfService: -1,
+			},
+			want: Message{
+				QualityOfService: 0,
+			},
+		}, {
+			description: "ClampQualityOfService(), QualityOfService > 99",
+			opt:         ClampQualityOfService(),
+			msg: Message{
+				QualityOfService: 100,
+			},
+			want: Message{
+				QualityOfService: 99,
+			},
 		}, {
 			description: "EnsureMetadataString(key, value) add to empty",
 			opt:         EnsureMetadataString("key", "value"),