record meta space

sofa-serverless-runtime/arklet-core/src/main/java/com/alipay/sofa/serverless/arklet/core/command/builtin/handler/InstallBizHandler.java

@@ -30,29 +30,49 @@
 import lombok.Getter;
 import lombok.Setter;
 
+import java.lang.management.ManagementFactory;
+import java.lang.management.MemoryPoolMXBean;
+import java.util.List;
+
 /**
  * @author mingmen
  * @date 2023/6/8
  */
-public class InstallBizHandler extends
-                              AbstractCommandHandler<InstallBizHandler.Input, ClientResponse>
-                                                                                             implements
-                                                                                             ArkBizOps {
+public class InstallBizHandler
+                              extends
+                              AbstractCommandHandler<InstallBizHandler.Input, InstallBizHandler.InstallBizClientResponse>
+                                                                                                                         implements
+                                                                                                                         ArkBizOps {
 
     @Override
-    public Output<ClientResponse> handle(Input input) {
+    public Output<InstallBizClientResponse> handle(Input input) {
+        MemoryPoolMXBean metaSpaceMXBean = getMetaSpaceMXBean();
+        long startSpace = metaSpaceMXBean.getUsage().getUsed();
+        InstallBizClientResponse installBizClientResponse;
         try {
             ClientResponse res = getOperationService().install(input.getBizUrl());
+            installBizClientResponse = (InstallBizClientResponse) res;
+            installBizClientResponse.setElapsedSpace(metaSpaceMXBean.getUsage().getUsed()
+                                                     - startSpace);
             if (ResponseCode.SUCCESS.equals(res.getCode())) {
-                return Output.ofSuccess(res);
+                return Output.ofSuccess(installBizClientResponse);
             } else {
-                return Output.ofFailed(res, "install biz not success!");
+                return Output.ofFailed(installBizClientResponse, "install biz not success!");
             }
         } catch (Throwable e) {
             throw new ArkletRuntimeException(e);
         }
     }
 
+    private MemoryPoolMXBean getMetaSpaceMXBean() {
+        MemoryPoolMXBean metaSpaceMXBean = null;
+        List<MemoryPoolMXBean> memoryPoolMXBeans = ManagementFactory.getMemoryPoolMXBeans();
+        for (MemoryPoolMXBean memoryPoolMXBean : memoryPoolMXBeans)
+            if (memoryPoolMXBean.getName().equals("Metaspace"))
+                metaSpaceMXBean = memoryPoolMXBean;
+        return metaSpaceMXBean;
+    }
+
     @Override
     public Command command() {
         return BuiltinCommand.INSTALL_BIZ;
@@ -73,4 +93,10 @@
         private String bizUrl;
     }
 
+    @Getter
+    @Setter
+    public static class InstallBizClientResponse extends ClientResponse {
+        private long elapsedSpace;
+    }
+
 }

sofa-serverless-runtime/arklet-core/src/test/java/com/alipay/sofa/serverless/arklet/core/command/handler/BaseHandlerTest.java

@@ -17,10 +17,10 @@
 package com.alipay.sofa.serverless.arklet.core.command.handler;
 
 import com.alipay.sofa.ark.api.ArkClient;
-import com.alipay.sofa.ark.api.ClientResponse;
 import com.alipay.sofa.ark.api.ResponseCode;
 import com.alipay.sofa.ark.spi.model.BizOperation;
 import com.alipay.sofa.serverless.arklet.core.BaseTest;
+import com.alipay.sofa.serverless.arklet.core.command.builtin.handler.InstallBizHandler;
 import com.alipay.sofa.serverless.arklet.core.health.custom.CustomBizManagerService;
 import com.alipay.sofa.serverless.arklet.core.health.custom.CustomPluginManagerService;
 import org.junit.After;
@@ -35,9 +35,9 @@
  */
 public class BaseHandlerTest extends BaseTest {
 
-    public final ClientResponse    success = new ClientResponse();
-    public final ClientResponse    failed  = new ClientResponse();
-    public MockedStatic<ArkClient> arkClient;
+    public final InstallBizHandler.InstallBizClientResponse success = new InstallBizHandler.InstallBizClientResponse();
+    public final InstallBizHandler.InstallBizClientResponse failed  = new InstallBizHandler.InstallBizClientResponse();
+    public MockedStatic<ArkClient>                          arkClient;
 
     @Before
     public void setupHandler() {

sofa-serverless-runtime/arklet-core/src/test/java/com/alipay/sofa/serverless/arklet/core/command/handler/InstallBizHandlerTests.java

@@ -16,7 +16,6 @@
  */
 package com.alipay.sofa.serverless.arklet.core.command.handler;
 
-import com.alipay.sofa.ark.api.ClientResponse;
 import com.alipay.sofa.serverless.arklet.core.command.builtin.BuiltinCommand;
 import com.alipay.sofa.serverless.arklet.core.command.builtin.handler.InstallBizHandler;
 import com.alipay.sofa.serverless.arklet.core.command.builtin.handler.InstallBizHandler.Input;
@@ -48,7 +47,7 @@ public void testHandle_Success() throws Throwable {
 
         when(handler.getOperationService().install(input.getBizUrl())).thenReturn(success);
 
-        Output<ClientResponse> result = handler.handle(input);
+        Output<?> result = handler.handle(input);
 
         Assert.assertEquals(success, result.getData());
         Assert.assertTrue(result.success());
@@ -61,7 +60,7 @@ public void testHandle_Failure() throws Throwable {
 
         when(handler.getOperationService().install(input.getBizUrl())).thenReturn(failed);
 
-        Output<ClientResponse> result = handler.handle(input);
+        Output<InstallBizHandler.InstallBizClientResponse> result = handler.handle(input);
 
         Assert.assertSame(failed, result.getData());
         Assert.assertTrue(result.failed());