apache#5550 add operationType

LiuQhahah · Nov 22, 2024 · abb2fe4 · abb2fe4
1 parent 5573d19
commit abb2fe4
Show file tree

Hide file tree

Showing 6 changed files with 58 additions and 0 deletions.
diff --git a/core/src/main/java/org/apache/gravitino/listener/api/event/AddPartitionPreEvent.java b/core/src/main/java/org/apache/gravitino/listener/api/event/AddPartitionPreEvent.java
@@ -44,4 +44,14 @@ public AddPartitionPreEvent(
   public PartitionInfo createdPartitionRequest() {
     return createPartitionRequest;
   }
+
+  /**
+   * Returns the type of operation.
+   *
+   * @return the operation type.
+   */
+  @Override
+  public OperationType operationType() {
+    return OperationType.ADD_PARTITION;
+  }
 }
diff --git a/core/src/main/java/org/apache/gravitino/listener/api/event/DropPartitionPreEvent.java b/core/src/main/java/org/apache/gravitino/listener/api/event/DropPartitionPreEvent.java
@@ -34,4 +34,14 @@ public DropPartitionPreEvent(String user, NameIdentifier identifier, String part
   public String partitionName() {
     return partitionName;
   }
+
+  /**
+   * Returns the type of operation.
+   *
+   * @return the operation type.
+   */
+  @Override
+  public OperationType operationType() {
+    return OperationType.DROP_PARTITION;
+  }
 }
diff --git a/core/src/main/java/org/apache/gravitino/listener/api/event/GetPartitionPreEvent.java b/core/src/main/java/org/apache/gravitino/listener/api/event/GetPartitionPreEvent.java
@@ -35,4 +35,14 @@ public GetPartitionPreEvent(String user, NameIdentifier identifier, String parti
   public String partitionName() {
     return partitionName;
   }
+
+  /**
+   * Returns the type of operation.
+   *
+   * @return the operation type.
+   */
+  @Override
+  public OperationType operationType() {
+    return OperationType.LOAD_PARTITION;
+  }
 }
diff --git a/core/src/main/java/org/apache/gravitino/listener/api/event/ListPartitionNamesPreEvent.java b/core/src/main/java/org/apache/gravitino/listener/api/event/ListPartitionNamesPreEvent.java
@@ -28,4 +28,14 @@ public class ListPartitionNamesPreEvent extends PartitionPreEvent {
   public ListPartitionNamesPreEvent(String user, NameIdentifier identifier) {
     super(user, identifier);
   }
+
+  /**
+   * Returns the type of operation.
+   *
+   * @return the operation type.
+   */
+  @Override
+  public OperationType operationType() {
+    return OperationType.LIST_PARTITION_NAMES;
+  }
 }
diff --git a/core/src/main/java/org/apache/gravitino/listener/api/event/ListPartitionPreEvent.java b/core/src/main/java/org/apache/gravitino/listener/api/event/ListPartitionPreEvent.java
@@ -28,4 +28,13 @@ public class ListPartitionPreEvent extends PartitionPreEvent {
   public ListPartitionPreEvent(String user, NameIdentifier identifier) {
     super(user, identifier);
   }
+  /**
+   * Returns the type of operation.
+   *
+   * @return the operation type.
+   */
+  @Override
+  public OperationType operationType() {
+    return OperationType.LIST_PARTITION;
+  }
 }
diff --git a/core/src/main/java/org/apache/gravitino/listener/api/event/PurgePartitionPreEvent.java b/core/src/main/java/org/apache/gravitino/listener/api/event/PurgePartitionPreEvent.java
@@ -35,4 +35,13 @@ public PurgePartitionPreEvent(String user, NameIdentifier identifier, String par
   public String partitionName() {
     return partitionName;
   }
+  /**
+   * Returns the type of operation.
+   *
+   * @return the operation type.
+   */
+  @Override
+  public OperationType operationType() {
+    return OperationType.PURGE_PARTITION;
+  }
 }