|
29 | 29 | import com.tencent.polaris.specification.api.v1.fault.tolerance.CircuitBreakerProto; |
30 | 30 | import com.tencent.polaris.specification.api.v1.fault.tolerance.CircuitBreakerProto.CircuitBreaker; |
31 | 31 | import com.tencent.polaris.specification.api.v1.fault.tolerance.FaultDetectorProto; |
32 | | -import com.tencent.polaris.specification.api.v1.fault.tolerance.FaultDetectorProto.FaultDetectRule; |
33 | 32 | import com.tencent.polaris.specification.api.v1.model.ModelProto; |
34 | 33 | import com.tencent.polaris.specification.api.v1.model.ModelProto.Location; |
35 | 34 | import com.tencent.polaris.specification.api.v1.service.manage.ClientProto.Client; |
|
45 | 44 | import com.tencent.polaris.specification.api.v1.traffic.manage.RateLimitProto.RateLimit; |
46 | 45 | import com.tencent.polaris.specification.api.v1.traffic.manage.RoutingProto; |
47 | 46 | import io.grpc.stub.StreamObserver; |
48 | | -import java.util.ArrayList; |
49 | | -import java.util.HashMap; |
50 | | -import java.util.List; |
51 | | -import java.util.Map; |
52 | | -import java.util.Objects; |
53 | | -import java.util.Set; |
54 | | -import java.util.UUID; |
55 | | -import java.util.concurrent.ConcurrentHashMap; |
56 | 47 | import org.slf4j.Logger; |
57 | 48 |
|
| 49 | +import java.util.*; |
| 50 | +import java.util.concurrent.ConcurrentHashMap; |
| 51 | + |
58 | 52 | public class NamingService extends PolarisGRPCGrpc.PolarisGRPCImplBase { |
59 | 53 |
|
60 | 54 | private static final Logger LOG = LoggerFactory.getLogger(NamingService.class); |
@@ -205,7 +199,7 @@ public void addInstance(ServiceKey svcKey, Node node, InstanceParameter paramete |
205 | 199 | /** |
206 | 200 | * 批量增加服务实例 |
207 | 201 | * |
208 | | - * @param svcKey 服务名 |
| 202 | + * @param svcKey 服务名 |
209 | 203 | * @param portStart 起始端口 |
210 | 204 | * @param instCount 实例数 |
211 | 205 | * @param parameter 实例参数 |
@@ -259,7 +253,7 @@ public void setInstanceHealthyStatus( |
259 | 253 |
|
260 | 254 | @Override |
261 | 255 | public void registerInstance(ServiceProto.Instance request, |
262 | | - StreamObserver<ResponseProto.Response> responseObserver) { |
| 256 | + StreamObserver<ResponseProto.Response> responseObserver) { |
263 | 257 | ServiceKey serviceKey = new ServiceKey(request.getNamespace().getValue(), request.getService().getValue()); |
264 | 258 | if (!services.containsKey(serviceKey)) { |
265 | 259 | services.put(serviceKey, new ArrayList<ServiceProto.Instance>()); |
@@ -306,7 +300,7 @@ private ResponseProto.Response buildResponse(int code, String info, ServiceProto |
306 | 300 |
|
307 | 301 | @Override |
308 | 302 | public void deregisterInstance(ServiceProto.Instance request, |
309 | | - StreamObserver<ResponseProto.Response> responseObserver) { |
| 303 | + StreamObserver<ResponseProto.Response> responseObserver) { |
310 | 304 | ServiceKey serviceKey = new ServiceKey(request.getNamespace().getValue(), request.getService().getValue()); |
311 | 305 | if (!services.containsKey(serviceKey)) { |
312 | 306 | responseObserver.onNext( |
@@ -450,6 +444,10 @@ private ResponseProto.DiscoverResponse buildServiceResponse( |
450 | 444 | break; |
451 | 445 | case LANE: |
452 | 446 | builder.setType(DiscoverResponseType.LANE); |
| 447 | + case LOSSLESS: |
| 448 | + builder.setType(DiscoverResponseType.LOSSLESS); |
| 449 | + case BLOCK_ALLOW_RULE: |
| 450 | + builder.setType(DiscoverResponseType.BLOCK_ALLOW_RULE); |
453 | 451 | case UNRECOGNIZED: |
454 | 452 | break; |
455 | 453 | default: |
|
0 commit comments