metrics code optimization (#11352)

Author: wxbty

dubbo-metrics/dubbo-metrics-api/src/main/java/org/apache/dubbo/metrics/collector/stat/MetricsStatHandler.java

@@ -17,13 +17,17 @@
 
 package org.apache.dubbo.metrics.collector.stat;
 
+import org.apache.dubbo.metrics.event.MetricsEvent;
 import org.apache.dubbo.metrics.model.MethodMetric;
+import org.apache.dubbo.rpc.Invocation;
 
 import java.util.Map;
 import java.util.concurrent.atomic.AtomicLong;
 
 public interface MetricsStatHandler {
     Map<MethodMetric, AtomicLong> get();
-    void increase(String applicationName, String interfaceName, String methodName, String group, String version);
-    void decrease(String applicationName, String interfaceName, String methodName, String group, String version);
+
+    MetricsEvent increase(String applicationName, Invocation invocation);
+
+    MetricsEvent decrease(String applicationName, Invocation invocation);
 }

dubbo-metrics/dubbo-metrics-api/src/main/java/org/apache/dubbo/metrics/event/EmptyEvent.java

@@ -0,0 +1,34 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.dubbo.metrics.event;
+
+/**
+ * EmptyEvent, do nothing.
+ */
+public class EmptyEvent extends MetricsEvent {
+
+    private static final EmptyEvent empty = new EmptyEvent(new Object());
+
+    public EmptyEvent(Object source) {
+        super(source);
+    }
+
+    public static EmptyEvent instance() {
+        return empty;
+    }
+}

dubbo-metrics/dubbo-metrics-api/src/main/java/org/apache/dubbo/metrics/event/MetricsEventMulticaster.java

@@ -0,0 +1,27 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.dubbo.metrics.event;
+
+import org.apache.dubbo.metrics.listener.MetricsListener;
+
+public interface MetricsEventMulticaster {
+
+    void addListener(MetricsListener listener);
+
+    void publishEvent(MetricsEvent event);
+}

dubbo-metrics/dubbo-metrics-api/src/main/java/org/apache/dubbo/metrics/event/SimpleMetricsEventMulticaster.java

@@ -0,0 +1,61 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.dubbo.metrics.event;
+
+import org.apache.dubbo.metrics.listener.MetricsListener;
+
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+
+public final class SimpleMetricsEventMulticaster implements MetricsEventMulticaster {
+
+    @SuppressWarnings("java:S3077")
+    private static volatile SimpleMetricsEventMulticaster instance;
+
+    private SimpleMetricsEventMulticaster() {
+    }
+
+    private final List<MetricsListener> listeners = Collections.synchronizedList(new ArrayList<>());
+
+    public static SimpleMetricsEventMulticaster getInstance() {
+        if (instance == null) {
+            synchronized (SimpleMetricsEventMulticaster.class) {
+                if (instance == null) {
+                    instance = new SimpleMetricsEventMulticaster();
+                }
+            }
+        }
+        return instance;
+    }
+
+    @Override
+    public void addListener(MetricsListener listener) {
+        listeners.add(listener);
+    }
+
+    @Override
+    public void publishEvent(MetricsEvent event) {
+        if (event instanceof EmptyEvent) {
+            return;
+        }
+        for (MetricsListener listener : listeners) {
+            listener.onEvent(event);
+        }
+    }
+}

dubbo-metrics/dubbo-metrics-api/src/main/java/org/apache/dubbo/metrics/model/MethodMetric.java

@@ -17,6 +17,9 @@
 
 package org.apache.dubbo.metrics.model;
 
+import org.apache.dubbo.rpc.Invocation;
+import org.apache.dubbo.rpc.RpcInvocation;
+
 import java.util.HashMap;
 import java.util.Map;
 import java.util.Objects;
@@ -30,6 +33,7 @@
 import static org.apache.dubbo.common.constants.MetricsConstants.TAG_VERSION_KEY;
 import static org.apache.dubbo.common.utils.NetUtils.getLocalHost;
 import static org.apache.dubbo.common.utils.NetUtils.getLocalHostName;
+import static org.apache.dubbo.rpc.support.RpcUtils.isGenericCall;
 
 /**
  * Metric class for method.
@@ -45,12 +49,9 @@ public MethodMetric() {
 
     }
 
-    public MethodMetric(String applicationName, String interfaceName, String methodName, String group, String version) {
+    public MethodMetric(String applicationName, Invocation invocation) {
         this.applicationName = applicationName;
-        this.interfaceName = interfaceName;
-        this.methodName = methodName;
-        this.group = group;
-        this.version = version;
+        init(invocation);
     }
 
     public String getInterfaceName() {
@@ -121,4 +122,33 @@ public String toString() {
             ", version='" + version + '\'' +
             '}';
     }
+
+    private void init(Invocation invocation) {
+        String serviceUniqueName = invocation.getTargetServiceUniqueName();
+        String methodName = invocation.getMethodName();
+        if (invocation instanceof RpcInvocation
+            && isGenericCall(((RpcInvocation) invocation).getParameterTypesDesc(), methodName)
+            && invocation.getArguments() != null
+            && invocation.getArguments().length == 3) {
+            methodName = ((String) invocation.getArguments()[0]).trim();
+        }
+        String group = null;
+        String interfaceAndVersion;
+        String[] arr = serviceUniqueName.split("/");
+        if (arr.length == 2) {
+            group = arr[0];
+            interfaceAndVersion = arr[1];
+        } else {
+            interfaceAndVersion = arr[0];
+        }
+
+        String[] ivArr = interfaceAndVersion.split(":");
+        String interfaceName = ivArr[0];
+        String version = ivArr.length == 2 ? ivArr[1] : null;
+
+        this.interfaceName = interfaceName;
+        this.methodName = methodName;
+        this.group = group;
+        this.version = version;
+    }
 }