|
20 | 20 | import org.apache.dubbo.qos.command.CommandContext;
|
21 | 21 | import org.apache.dubbo.qos.command.annotation.Cmd;
|
22 | 22 | import org.apache.dubbo.qos.textui.TTable;
|
23 |
| -import org.apache.dubbo.registry.support.ConsumerInvokerWrapper; |
24 |
| -import org.apache.dubbo.registry.support.ProviderConsumerRegTable; |
25 |
| -import org.apache.dubbo.registry.support.ProviderInvokerWrapper; |
26 | 23 | import org.apache.dubbo.rpc.model.ApplicationModel;
|
27 | 24 | import org.apache.dubbo.rpc.model.ConsumerModel;
|
28 | 25 | import org.apache.dubbo.rpc.model.ProviderModel;
|
29 | 26 |
|
30 | 27 | import java.util.Collection;
|
31 |
| -import java.util.Set; |
| 28 | + |
| 29 | +import static org.apache.dubbo.registry.support.ProviderConsumerRegTable.getConsumerAddressNum; |
| 30 | +import static org.apache.dubbo.registry.support.ProviderConsumerRegTable.isRegistered; |
32 | 31 |
|
33 | 32 | @Cmd(name = "ls", summary = "ls service", example = {
|
34 | 33 | "ls"
|
@@ -58,7 +57,7 @@ public String listProvider() {
|
58 | 57 |
|
59 | 58 | //Content
|
60 | 59 | for (ProviderModel providerModel : ProviderModelList) {
|
61 |
| - tTable.addRow(providerModel.getServiceName(), isReg(providerModel.getServiceName()) ? "Y" : "N"); |
| 60 | + tTable.addRow(providerModel.getServiceName(), isRegistered(providerModel.getServiceName()) ? "Y" : "N"); |
62 | 61 | }
|
63 | 62 | stringBuilder.append(tTable.rendering());
|
64 | 63 |
|
@@ -88,29 +87,4 @@ public String listConsumer() {
|
88 | 87 |
|
89 | 88 | return stringBuilder.toString();
|
90 | 89 | }
|
91 |
| - |
92 |
| - private boolean isReg(String serviceUniqueName) { |
93 |
| - Set<ProviderInvokerWrapper> providerInvokerWrapperSet = ProviderConsumerRegTable.getProviderInvoker(serviceUniqueName); |
94 |
| - for (ProviderInvokerWrapper providerInvokerWrapper : providerInvokerWrapperSet) { |
95 |
| - if (providerInvokerWrapper.isReg()) { |
96 |
| - return true; |
97 |
| - } |
98 |
| - } |
99 |
| - |
100 |
| - return false; |
101 |
| - } |
102 |
| - |
103 |
| - private int getConsumerAddressNum(String serviceUniqueName) { |
104 |
| - int count = 0; |
105 |
| - Set<ConsumerInvokerWrapper> providerInvokerWrapperSet = ProviderConsumerRegTable.getConsumerInvoker(serviceUniqueName); |
106 |
| - for (ConsumerInvokerWrapper consumerInvokerWrapper : providerInvokerWrapperSet) { |
107 |
| - //TODO not thread safe,fixme |
108 |
| - int addNum = 0; |
109 |
| - if (consumerInvokerWrapper.getRegistryDirectory().getUrlInvokerMap() != null) { |
110 |
| - addNum = consumerInvokerWrapper.getRegistryDirectory().getUrlInvokerMap().size(); |
111 |
| - } |
112 |
| - count += addNum; |
113 |
| - } |
114 |
| - return count; |
115 |
| - } |
116 | 90 | }
|
0 commit comments