[iotdb] 02/05: data node InternalServiceImpl

[ro...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

rong pushed a commit to branch IOTDB-3228
in repository https://gitbox.apache.org/repos/asf/iotdb.git

commit 6edc7c4c978d24d5f7ced71822d61c1941e1e7d0
Author: Steve Yurong Su <ro...@apache.org>
AuthorDate: Mon May 30 10:21:33 2022 +0800

    data node InternalServiceImpl
---
 .../iotdb/db/service/thrift/impl/InternalServiceImpl.java    | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/server/src/main/java/org/apache/iotdb/db/service/thrift/impl/InternalServiceImpl.java b/server/src/main/java/org/apache/iotdb/db/service/thrift/impl/InternalServiceImpl.java
index eff3a3ffa0..ab75ed7da9 100644
--- a/server/src/main/java/org/apache/iotdb/db/service/thrift/impl/InternalServiceImpl.java
+++ b/server/src/main/java/org/apache/iotdb/db/service/thrift/impl/InternalServiceImpl.java
@@ -76,6 +76,7 @@ import org.apache.iotdb.mpp.rpc.thrift.TCancelResp;
 import org.apache.iotdb.mpp.rpc.thrift.TCreateDataRegionReq;
 import org.apache.iotdb.mpp.rpc.thrift.TCreateFunctionRequest;
 import org.apache.iotdb.mpp.rpc.thrift.TCreateSchemaRegionReq;
+import org.apache.iotdb.mpp.rpc.thrift.TDropFunctionRequest;
 import org.apache.iotdb.mpp.rpc.thrift.TFetchFragmentInstanceStateReq;
 import org.apache.iotdb.mpp.rpc.thrift.TFragmentInstanceStateResp;
 import org.apache.iotdb.mpp.rpc.thrift.TInvalidateCacheReq;
@@ -398,5 +399,16 @@ public class InternalServiceImpl implements InternalService.Iface {
     }
   }
 
+  @Override
+  public TSStatus dropFunction(TDropFunctionRequest request) {
+    try {
+      UDFRegistrationService.getInstance().deregister(request.getUdfName());
+      return new TSStatus(TSStatusCode.SUCCESS_STATUS.getStatusCode());
+    } catch (Exception e) {
+      return new TSStatus(TSStatusCode.EXECUTE_STATEMENT_ERROR.getStatusCode())
+          .setMessage(e.getMessage());
+    }
+  }
+
   public void handleClientExit() {}
 }