1 files changed, 26 insertions, 0 deletions
diff --git a/vespaclient-container-plugin/src/main/java/com/yahoo/documentapi/metrics/DocumentOperationType.java b/vespaclient-container-plugin/src/main/java/com/yahoo/documentapi/metrics/DocumentOperationType.java
new file mode 100644
index 00000000000..56849b9db8f
--- /dev/null
+++ b/vespaclient-container-plugin/src/main/java/com/yahoo/documentapi/metrics/DocumentOperationType.java
@@ -0,0 +1,26 @@
+// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.documentapi.metrics;
+
+import com.yahoo.documentapi.messagebus.protocol.DocumentProtocol;
+import com.yahoo.messagebus.Message;
+
+/**
+ * @author freva
+ */
+public enum DocumentOperationType {
+    PUT, REMOVE, UPDATE;
+
+    public static DocumentOperationType fromMessage(Message msg) {
+        switch (msg.getType()) {
+            case DocumentProtocol.MESSAGE_PUTDOCUMENT:
+                return PUT;
+            case DocumentProtocol.MESSAGE_UPDATEDOCUMENT:
+                return UPDATE;
+            case DocumentProtocol.MESSAGE_REMOVEDOCUMENT:
+                return REMOVE;
+            default:
+                return null;
+        }
+    }
+}
+