|
| 1 | +/* |
| 2 | + * Copyright The OpenTelemetry Authors |
| 3 | + * SPDX-License-Identifier: Apache-2.0 |
| 4 | + */ |
| 5 | + |
1 | 6 | package io.opentelemetry.javaagent.instrumentation.nacos.client.v2_0_3; |
2 | 7 |
|
3 | 8 | import com.alibaba.nacos.api.config.remote.request.ConfigChangeNotifyRequest; |
|
15 | 20 | import javax.annotation.Nonnull; |
16 | 21 |
|
17 | 22 | public class NacosClientHelper { |
18 | | - private static final NacosClientRequestOperator UNKNOWN_OPERATOR = new NacosClientRequestOperator( |
19 | | - request -> request.getClass().getSimpleName(), null); |
20 | | - private static final Map<Class<? extends Request>, NacosClientRequestOperator> KNOWN_OPERATOR_MAP = new HashMap<>(); |
| 23 | + private static final NacosClientRequestOperator UNKNOWN_OPERATOR = |
| 24 | + new NacosClientRequestOperator(request -> request.getClass().getSimpleName(), null); |
| 25 | + private static final Map<Class<? extends Request>, NacosClientRequestOperator> |
| 26 | + KNOWN_OPERATOR_MAP = new HashMap<>(); |
21 | 27 |
|
22 | 28 | private NacosClientHelper() {} |
23 | 29 |
|
24 | 30 | static { |
25 | | - KNOWN_OPERATOR_MAP.put(InstanceRequest.class, new NacosClientRequestOperator( |
26 | | - request -> ((InstanceRequest) request).getType(), |
27 | | - (attributesBuilder, request) -> { |
28 | | - InstanceRequest instanceRequest = (InstanceRequest) request; |
29 | | - attributesBuilder.put(NacosClientConstants.NACOS_NAME_SPACE_ATTR, |
30 | | - instanceRequest.getNamespace()); |
31 | | - attributesBuilder.put(NacosClientConstants.NACOS_GROUP_NAME_ATTR, |
32 | | - instanceRequest.getGroupName()); |
33 | | - attributesBuilder.put(NacosClientConstants.NACOS_SERVICE_NAME_ATTR, |
34 | | - instanceRequest.getServiceName()); |
35 | | - }) |
36 | | - ); |
| 31 | + KNOWN_OPERATOR_MAP.put( |
| 32 | + InstanceRequest.class, |
| 33 | + new NacosClientRequestOperator( |
| 34 | + request -> ((InstanceRequest) request).getType(), |
| 35 | + (attributesBuilder, request) -> { |
| 36 | + InstanceRequest instanceRequest = (InstanceRequest) request; |
| 37 | + attributesBuilder.put( |
| 38 | + NacosClientConstants.NACOS_NAME_SPACE_ATTR, instanceRequest.getNamespace()); |
| 39 | + attributesBuilder.put( |
| 40 | + NacosClientConstants.NACOS_GROUP_NAME_ATTR, instanceRequest.getGroupName()); |
| 41 | + attributesBuilder.put( |
| 42 | + NacosClientConstants.NACOS_SERVICE_NAME_ATTR, instanceRequest.getServiceName()); |
| 43 | + })); |
37 | 44 |
|
38 | | - KNOWN_OPERATOR_MAP.put(ServiceQueryRequest.class, new NacosClientRequestOperator( |
39 | | - request -> NacosClientConstants.QUERY_SERVICE, |
40 | | - (attributesBuilder, request) -> { |
41 | | - ServiceQueryRequest serviceQueryRequest = (ServiceQueryRequest) request; |
42 | | - attributesBuilder.put(NacosClientConstants.NACOS_NAME_SPACE_ATTR, |
43 | | - serviceQueryRequest.getNamespace()); |
44 | | - attributesBuilder.put(NacosClientConstants.NACOS_GROUP_NAME_ATTR, |
45 | | - serviceQueryRequest.getGroupName()); |
46 | | - attributesBuilder.put(NacosClientConstants.NACOS_SERVICE_NAME_ATTR, |
47 | | - serviceQueryRequest.getServiceName()); |
48 | | - }) |
49 | | - ); |
| 45 | + KNOWN_OPERATOR_MAP.put( |
| 46 | + ServiceQueryRequest.class, |
| 47 | + new NacosClientRequestOperator( |
| 48 | + request -> NacosClientConstants.QUERY_SERVICE, |
| 49 | + (attributesBuilder, request) -> { |
| 50 | + ServiceQueryRequest serviceQueryRequest = (ServiceQueryRequest) request; |
| 51 | + attributesBuilder.put( |
| 52 | + NacosClientConstants.NACOS_NAME_SPACE_ATTR, serviceQueryRequest.getNamespace()); |
| 53 | + attributesBuilder.put( |
| 54 | + NacosClientConstants.NACOS_GROUP_NAME_ATTR, serviceQueryRequest.getGroupName()); |
| 55 | + attributesBuilder.put( |
| 56 | + NacosClientConstants.NACOS_SERVICE_NAME_ATTR, |
| 57 | + serviceQueryRequest.getServiceName()); |
| 58 | + })); |
50 | 59 |
|
51 | | - KNOWN_OPERATOR_MAP.put(SubscribeServiceRequest.class, new NacosClientRequestOperator( |
52 | | - request -> ((SubscribeServiceRequest) request).isSubscribe() |
53 | | - ? NacosClientConstants.SUBSCRIBE_SERVICE |
54 | | - : NacosClientConstants.UNSUBSCRIBE_SERVICE, |
55 | | - (attributesBuilder, request) -> { |
56 | | - SubscribeServiceRequest subscribeServiceRequest = (SubscribeServiceRequest) request; |
57 | | - attributesBuilder.put(NacosClientConstants.NACOS_NAME_SPACE_ATTR, |
58 | | - subscribeServiceRequest.getNamespace()); |
59 | | - attributesBuilder.put(NacosClientConstants.NACOS_GROUP_NAME_ATTR, |
60 | | - subscribeServiceRequest.getGroupName()); |
61 | | - attributesBuilder.put(NacosClientConstants.NACOS_SERVICE_NAME_ATTR, |
62 | | - subscribeServiceRequest.getServiceName()); |
63 | | - }) |
64 | | - ); |
| 60 | + KNOWN_OPERATOR_MAP.put( |
| 61 | + SubscribeServiceRequest.class, |
| 62 | + new NacosClientRequestOperator( |
| 63 | + request -> |
| 64 | + ((SubscribeServiceRequest) request).isSubscribe() |
| 65 | + ? NacosClientConstants.SUBSCRIBE_SERVICE |
| 66 | + : NacosClientConstants.UNSUBSCRIBE_SERVICE, |
| 67 | + (attributesBuilder, request) -> { |
| 68 | + SubscribeServiceRequest subscribeServiceRequest = (SubscribeServiceRequest) request; |
| 69 | + attributesBuilder.put( |
| 70 | + NacosClientConstants.NACOS_NAME_SPACE_ATTR, |
| 71 | + subscribeServiceRequest.getNamespace()); |
| 72 | + attributesBuilder.put( |
| 73 | + NacosClientConstants.NACOS_GROUP_NAME_ATTR, |
| 74 | + subscribeServiceRequest.getGroupName()); |
| 75 | + attributesBuilder.put( |
| 76 | + NacosClientConstants.NACOS_SERVICE_NAME_ATTR, |
| 77 | + subscribeServiceRequest.getServiceName()); |
| 78 | + })); |
65 | 79 |
|
66 | | - KNOWN_OPERATOR_MAP.put(ServiceListRequest.class, new NacosClientRequestOperator( |
67 | | - request -> NacosClientConstants.GET_SERVICE_LIST, |
68 | | - (attributesBuilder, request) -> { |
69 | | - ServiceListRequest serviceListRequest = (ServiceListRequest) request; |
70 | | - attributesBuilder.put(NacosClientConstants.NACOS_NAME_SPACE_ATTR, |
71 | | - serviceListRequest.getNamespace()); |
72 | | - attributesBuilder.put(NacosClientConstants.NACOS_GROUP_NAME_ATTR, |
73 | | - serviceListRequest.getGroupName()); |
74 | | - attributesBuilder.put(NacosClientConstants.NACOS_SERVICE_NAME_ATTR, |
75 | | - serviceListRequest.getServiceName()); |
76 | | - }) |
77 | | - ); |
| 80 | + KNOWN_OPERATOR_MAP.put( |
| 81 | + ServiceListRequest.class, |
| 82 | + new NacosClientRequestOperator( |
| 83 | + request -> NacosClientConstants.GET_SERVICE_LIST, |
| 84 | + (attributesBuilder, request) -> { |
| 85 | + ServiceListRequest serviceListRequest = (ServiceListRequest) request; |
| 86 | + attributesBuilder.put( |
| 87 | + NacosClientConstants.NACOS_NAME_SPACE_ATTR, serviceListRequest.getNamespace()); |
| 88 | + attributesBuilder.put( |
| 89 | + NacosClientConstants.NACOS_GROUP_NAME_ATTR, serviceListRequest.getGroupName()); |
| 90 | + attributesBuilder.put( |
| 91 | + NacosClientConstants.NACOS_SERVICE_NAME_ATTR, |
| 92 | + serviceListRequest.getServiceName()); |
| 93 | + })); |
78 | 94 |
|
79 | | - KNOWN_OPERATOR_MAP.put(ConfigQueryRequest.class, new NacosClientRequestOperator( |
80 | | - request -> NacosClientConstants.QUERY_CONFIG, |
81 | | - (attributesBuilder, request) -> { |
82 | | - ConfigQueryRequest configQueryRequest = (ConfigQueryRequest) request; |
83 | | - attributesBuilder.put(NacosClientConstants.NACOS_DATA_ID_ATTR, |
84 | | - configQueryRequest.getDataId()); |
85 | | - attributesBuilder.put(NacosClientConstants.NACOS_GROUP_ATTR, |
86 | | - configQueryRequest.getGroup()); |
87 | | - attributesBuilder.put(NacosClientConstants.NACOS_TENANT_ATTR, |
88 | | - configQueryRequest.getTenant()); |
89 | | - }) |
90 | | - ); |
| 95 | + KNOWN_OPERATOR_MAP.put( |
| 96 | + ConfigQueryRequest.class, |
| 97 | + new NacosClientRequestOperator( |
| 98 | + request -> NacosClientConstants.QUERY_CONFIG, |
| 99 | + (attributesBuilder, request) -> { |
| 100 | + ConfigQueryRequest configQueryRequest = (ConfigQueryRequest) request; |
| 101 | + attributesBuilder.put( |
| 102 | + NacosClientConstants.NACOS_DATA_ID_ATTR, configQueryRequest.getDataId()); |
| 103 | + attributesBuilder.put( |
| 104 | + NacosClientConstants.NACOS_GROUP_ATTR, configQueryRequest.getGroup()); |
| 105 | + attributesBuilder.put( |
| 106 | + NacosClientConstants.NACOS_TENANT_ATTR, configQueryRequest.getTenant()); |
| 107 | + })); |
91 | 108 |
|
92 | | - KNOWN_OPERATOR_MAP.put(ConfigPublishRequest.class, new NacosClientRequestOperator( |
93 | | - request -> NacosClientConstants.PUBLISH_CONFIG, |
94 | | - (attributesBuilder, request) -> { |
95 | | - ConfigPublishRequest configPublishRequest = (ConfigPublishRequest) request; |
96 | | - attributesBuilder.put(NacosClientConstants.NACOS_DATA_ID_ATTR, |
97 | | - configPublishRequest.getDataId()); |
98 | | - attributesBuilder.put(NacosClientConstants.NACOS_GROUP_ATTR, |
99 | | - configPublishRequest.getGroup()); |
100 | | - attributesBuilder.put(NacosClientConstants.NACOS_TENANT_ATTR, |
101 | | - configPublishRequest.getTenant()); |
102 | | - }) |
103 | | - ); |
| 109 | + KNOWN_OPERATOR_MAP.put( |
| 110 | + ConfigPublishRequest.class, |
| 111 | + new NacosClientRequestOperator( |
| 112 | + request -> NacosClientConstants.PUBLISH_CONFIG, |
| 113 | + (attributesBuilder, request) -> { |
| 114 | + ConfigPublishRequest configPublishRequest = (ConfigPublishRequest) request; |
| 115 | + attributesBuilder.put( |
| 116 | + NacosClientConstants.NACOS_DATA_ID_ATTR, configPublishRequest.getDataId()); |
| 117 | + attributesBuilder.put( |
| 118 | + NacosClientConstants.NACOS_GROUP_ATTR, configPublishRequest.getGroup()); |
| 119 | + attributesBuilder.put( |
| 120 | + NacosClientConstants.NACOS_TENANT_ATTR, configPublishRequest.getTenant()); |
| 121 | + })); |
104 | 122 |
|
105 | | - KNOWN_OPERATOR_MAP.put(ConfigRemoveRequest.class, new NacosClientRequestOperator( |
106 | | - request -> NacosClientConstants.REMOVE_CONFIG, |
107 | | - (attributesBuilder, request) -> { |
108 | | - ConfigRemoveRequest configRemoveRequest = (ConfigRemoveRequest) request; |
109 | | - attributesBuilder.put(NacosClientConstants.NACOS_DATA_ID_ATTR, |
110 | | - configRemoveRequest.getDataId()); |
111 | | - attributesBuilder.put(NacosClientConstants.NACOS_GROUP_ATTR, |
112 | | - configRemoveRequest.getGroup()); |
113 | | - attributesBuilder.put(NacosClientConstants.NACOS_TENANT_ATTR, |
114 | | - configRemoveRequest.getTenant()); |
115 | | - }) |
116 | | - ); |
| 123 | + KNOWN_OPERATOR_MAP.put( |
| 124 | + ConfigRemoveRequest.class, |
| 125 | + new NacosClientRequestOperator( |
| 126 | + request -> NacosClientConstants.REMOVE_CONFIG, |
| 127 | + (attributesBuilder, request) -> { |
| 128 | + ConfigRemoveRequest configRemoveRequest = (ConfigRemoveRequest) request; |
| 129 | + attributesBuilder.put( |
| 130 | + NacosClientConstants.NACOS_DATA_ID_ATTR, configRemoveRequest.getDataId()); |
| 131 | + attributesBuilder.put( |
| 132 | + NacosClientConstants.NACOS_GROUP_ATTR, configRemoveRequest.getGroup()); |
| 133 | + attributesBuilder.put( |
| 134 | + NacosClientConstants.NACOS_TENANT_ATTR, configRemoveRequest.getTenant()); |
| 135 | + })); |
117 | 136 |
|
118 | | - KNOWN_OPERATOR_MAP.put(NotifySubscriberRequest.class, new NacosClientRequestOperator( |
119 | | - request -> NacosClientConstants.NOTIFY_SUBSCRIBE_CHANGE, |
120 | | - (attributesBuilder, request) -> { |
121 | | - NotifySubscriberRequest notifySubscriberRequest = (NotifySubscriberRequest) request; |
122 | | - attributesBuilder.put(NacosClientConstants.NACOS_NAME_SPACE_ATTR, |
123 | | - notifySubscriberRequest.getNamespace()); |
124 | | - attributesBuilder.put(NacosClientConstants.NACOS_GROUP_NAME_ATTR, |
125 | | - notifySubscriberRequest.getGroupName()); |
126 | | - attributesBuilder.put(NacosClientConstants.NACOS_SERVICE_NAME_ATTR, |
127 | | - notifySubscriberRequest.getServiceName()); |
128 | | - }) |
129 | | - ); |
| 137 | + KNOWN_OPERATOR_MAP.put( |
| 138 | + NotifySubscriberRequest.class, |
| 139 | + new NacosClientRequestOperator( |
| 140 | + request -> NacosClientConstants.NOTIFY_SUBSCRIBE_CHANGE, |
| 141 | + (attributesBuilder, request) -> { |
| 142 | + NotifySubscriberRequest notifySubscriberRequest = (NotifySubscriberRequest) request; |
| 143 | + attributesBuilder.put( |
| 144 | + NacosClientConstants.NACOS_NAME_SPACE_ATTR, |
| 145 | + notifySubscriberRequest.getNamespace()); |
| 146 | + attributesBuilder.put( |
| 147 | + NacosClientConstants.NACOS_GROUP_NAME_ATTR, |
| 148 | + notifySubscriberRequest.getGroupName()); |
| 149 | + attributesBuilder.put( |
| 150 | + NacosClientConstants.NACOS_SERVICE_NAME_ATTR, |
| 151 | + notifySubscriberRequest.getServiceName()); |
| 152 | + })); |
130 | 153 |
|
131 | | - KNOWN_OPERATOR_MAP.put(ConfigChangeNotifyRequest.class, new NacosClientRequestOperator( |
132 | | - request -> NacosClientConstants.NOTIFY_CONFIG_CHANGE, |
133 | | - (attributesBuilder, request) -> { |
134 | | - ConfigChangeNotifyRequest configChangeNotifyRequest = (ConfigChangeNotifyRequest) request; |
135 | | - attributesBuilder.put(NacosClientConstants.NACOS_DATA_ID_ATTR, |
136 | | - configChangeNotifyRequest.getDataId()); |
137 | | - attributesBuilder.put(NacosClientConstants.NACOS_GROUP_ATTR, |
138 | | - configChangeNotifyRequest.getGroup()); |
139 | | - attributesBuilder.put(NacosClientConstants.NACOS_TENANT_ATTR, |
140 | | - configChangeNotifyRequest.getTenant()); |
141 | | - }) |
142 | | - ); |
| 154 | + KNOWN_OPERATOR_MAP.put( |
| 155 | + ConfigChangeNotifyRequest.class, |
| 156 | + new NacosClientRequestOperator( |
| 157 | + request -> NacosClientConstants.NOTIFY_CONFIG_CHANGE, |
| 158 | + (attributesBuilder, request) -> { |
| 159 | + ConfigChangeNotifyRequest configChangeNotifyRequest = |
| 160 | + (ConfigChangeNotifyRequest) request; |
| 161 | + attributesBuilder.put( |
| 162 | + NacosClientConstants.NACOS_DATA_ID_ATTR, configChangeNotifyRequest.getDataId()); |
| 163 | + attributesBuilder.put( |
| 164 | + NacosClientConstants.NACOS_GROUP_ATTR, configChangeNotifyRequest.getGroup()); |
| 165 | + attributesBuilder.put( |
| 166 | + NacosClientConstants.NACOS_TENANT_ATTR, configChangeNotifyRequest.getTenant()); |
| 167 | + })); |
143 | 168 | } |
144 | 169 |
|
145 | 170 | @Nonnull |
146 | 171 | public static NacosClientRequestOperator getOperator(@Nonnull Request request) { |
147 | | - NacosClientRequestOperator nacosClientRequestOperator = KNOWN_OPERATOR_MAP.get( |
148 | | - request.getClass()); |
| 172 | + NacosClientRequestOperator nacosClientRequestOperator = |
| 173 | + KNOWN_OPERATOR_MAP.get(request.getClass()); |
149 | 174 | if (nacosClientRequestOperator != null) { |
150 | 175 | return nacosClientRequestOperator; |
151 | 176 | } |
152 | 177 | return UNKNOWN_OPERATOR; |
153 | 178 | } |
154 | | - |
155 | | - |
156 | 179 | } |
0 commit comments