List of fields with default units prepared

Graylog2 · May 9, 2024 · d936ce2 · d936ce2
1 parent 9c8a2a3
commit d936ce2
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/...n/java/org/graylog/plugins/formatting/units/fields/HardcodedFieldUnitObtainingMethod.java b/...n/java/org/graylog/plugins/formatting/units/fields/HardcodedFieldUnitObtainingMethod.java
@@ -23,6 +23,12 @@
 
 import java.util.Optional;
 
+import static org.graylog.schema.HttpFields.HTTP_BYTES;
+import static org.graylog.schema.HttpFields.HTTP_RESPONSE_BYTES;
+import static org.graylog.schema.NetworkFields.NETWORK_BYTES;
+import static org.graylog.schema.NetworkFields.NETWORK_DATA_BYTES;
+import static org.graylog.schema.NetworkFields.NETWORK_HEADER_BYTES;
+import static org.graylog.schema.SourceFields.SOURCE_BYTES_SENT;
 import static org.graylog2.plugin.Message.FIELD_GL2_ACCOUNTED_MESSAGE_SIZE;
 import static org.graylog2.plugin.Message.FIELD_GL2_PROCESSING_DURATION_MS;
 
@@ -38,10 +44,15 @@ public HardcodedFieldUnitObtainingMethod(final SupportedUnits supportedUnits) {
 
     @Override
     public Optional<Unit> obtainUnit(final String fieldName) {
-        //TODO: complete list
         return switch (fieldName) {
             case FIELD_GL2_PROCESSING_DURATION_MS -> supportedUnits.getUnit(new UnitId("time", "ms"));
-            case FIELD_GL2_ACCOUNTED_MESSAGE_SIZE -> supportedUnits.getUnit(new UnitId("size", "B"));
+            case FIELD_GL2_ACCOUNTED_MESSAGE_SIZE,
+                    HTTP_BYTES,
+                    HTTP_RESPONSE_BYTES,
+                    NETWORK_BYTES,
+                    NETWORK_DATA_BYTES,
+                    NETWORK_HEADER_BYTES,
+                    SOURCE_BYTES_SENT -> supportedUnits.getUnit(new UnitId("size", "B"));
             default -> Optional.empty();
         };