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 index 6739c74ac27..cf0c43e8ae3 100644 --- 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 index dd5f6ea6f15..da677287e41 100644 --- 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 index a98e507e672..f373dfe6ee4 100644 --- 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 index d52a81261eb..c81db0ca108 100644 --- 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 index 7590a3fdf53..36b87e628ca 100644 --- 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 index 68af9c6f590..5b9f96fcc49 100644 --- 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; + } }