Skip to content

Commit 9cd901f

Browse files
committed
refactor:update polaris mock server.
1 parent 16a42a1 commit 9cd901f

File tree

1 file changed

+10
-12
lines changed
  • polaris-test/polaris-test-mock-discovery/src/main/java/com/tencent/polaris/test/mock/discovery

1 file changed

+10
-12
lines changed

polaris-test/polaris-test-mock-discovery/src/main/java/com/tencent/polaris/test/mock/discovery/NamingService.java

Lines changed: 10 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,6 @@
2929
import com.tencent.polaris.specification.api.v1.fault.tolerance.CircuitBreakerProto;
3030
import com.tencent.polaris.specification.api.v1.fault.tolerance.CircuitBreakerProto.CircuitBreaker;
3131
import com.tencent.polaris.specification.api.v1.fault.tolerance.FaultDetectorProto;
32-
import com.tencent.polaris.specification.api.v1.fault.tolerance.FaultDetectorProto.FaultDetectRule;
3332
import com.tencent.polaris.specification.api.v1.model.ModelProto;
3433
import com.tencent.polaris.specification.api.v1.model.ModelProto.Location;
3534
import com.tencent.polaris.specification.api.v1.service.manage.ClientProto.Client;
@@ -45,16 +44,11 @@
4544
import com.tencent.polaris.specification.api.v1.traffic.manage.RateLimitProto.RateLimit;
4645
import com.tencent.polaris.specification.api.v1.traffic.manage.RoutingProto;
4746
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;
5647
import org.slf4j.Logger;
5748

49+
import java.util.*;
50+
import java.util.concurrent.ConcurrentHashMap;
51+
5852
public class NamingService extends PolarisGRPCGrpc.PolarisGRPCImplBase {
5953

6054
private static final Logger LOG = LoggerFactory.getLogger(NamingService.class);
@@ -205,7 +199,7 @@ public void addInstance(ServiceKey svcKey, Node node, InstanceParameter paramete
205199
/**
206200
* 批量增加服务实例
207201
*
208-
* @param svcKey 服务名
202+
* @param svcKey 服务名
209203
* @param portStart 起始端口
210204
* @param instCount 实例数
211205
* @param parameter 实例参数
@@ -259,7 +253,7 @@ public void setInstanceHealthyStatus(
259253

260254
@Override
261255
public void registerInstance(ServiceProto.Instance request,
262-
StreamObserver<ResponseProto.Response> responseObserver) {
256+
StreamObserver<ResponseProto.Response> responseObserver) {
263257
ServiceKey serviceKey = new ServiceKey(request.getNamespace().getValue(), request.getService().getValue());
264258
if (!services.containsKey(serviceKey)) {
265259
services.put(serviceKey, new ArrayList<ServiceProto.Instance>());
@@ -306,7 +300,7 @@ private ResponseProto.Response buildResponse(int code, String info, ServiceProto
306300

307301
@Override
308302
public void deregisterInstance(ServiceProto.Instance request,
309-
StreamObserver<ResponseProto.Response> responseObserver) {
303+
StreamObserver<ResponseProto.Response> responseObserver) {
310304
ServiceKey serviceKey = new ServiceKey(request.getNamespace().getValue(), request.getService().getValue());
311305
if (!services.containsKey(serviceKey)) {
312306
responseObserver.onNext(
@@ -450,6 +444,10 @@ private ResponseProto.DiscoverResponse buildServiceResponse(
450444
break;
451445
case LANE:
452446
builder.setType(DiscoverResponseType.LANE);
447+
case LOSSLESS:
448+
builder.setType(DiscoverResponseType.LOSSLESS);
449+
case BLOCK_ALLOW_RULE:
450+
builder.setType(DiscoverResponseType.BLOCK_ALLOW_RULE);
453451
case UNRECOGNIZED:
454452
break;
455453
default:

0 commit comments

Comments
 (0)