Как передать traceIds в Hystrix Observables?
У меня есть несколько служб, некоторые из которых используют HystrixObservableCommand Hystrix для вызова других служб, а другие используют HystrixCommand. Как передать traceIds из вызывающей службы в Observables в HystrixObservableCommand, а также передать их, если вызван запасной вариант?
Все сервисы используют grpc-java.
Пример кода, который у меня есть:
WorldCommand worldCommand = new WorldCommand(greeterRequest, worldServiceStub);
String messageFromWorldService = "";
String idFromWorldService = "";
try {
Greeter.GreeterReply greeterReply = worldCommand.construct().toBlocking().toFuture().get();
messageFromWorldService = greeterReply.getMessage();
idFromWorldService = greeterReply.getId();
logger.info("Response from WorldService -- {}, id = {}", messageFromWorldService, idFromWorldService);
} catch (StatusRuntimeException | InterruptedException | ExecutionException e) {
logger.warn("Exception when calling WorldService\n" + e);
}
WorldCommand.java
public class WorldCommand extends HystrixObservableCommand<Greeter.GreeterReply> {
private static final Logger logger = LoggerFactory.getLogger(WorldCommand.class.getName());
private final Greeter.GreeterRequest greeterRequest;
private final WorldServiceGrpc.WorldServiceStub worldServiceStub;
public WorldCommand(Greeter.GreeterRequest greeterRequest, WorldServiceGrpc.WorldServiceStub worldServiceStub) {
super(HystrixCommandGroupKey.Factory.asKey("WorldService"));
this.greeterRequest = greeterRequest;
this.worldServiceStub = worldServiceStub;
}
@Override
protected Observable<Greeter.GreeterReply> construct() {
Context context = Context.current();
return Observable.create(new Observable.OnSubscribe<Greeter.GreeterReply>() {
@Override
public void call(Subscriber<? super Greeter.GreeterReply> observer) {
logger.info("In WorldCommand");
if (!observer.isUnsubscribed()) {
//pass on the context, if you want only certain headers to pass on then create a new Context and attach it.
context.attach();
logger.info("In WorldCommand after attach");
worldServiceStub.greetWithHelloOrWorld(greeterRequest, new StreamObserver<Greeter.GreeterReply>() {
@Override
public void onNext(Greeter.GreeterReply greeterReply) {
logger.info("Response from WorldService -- {}, id = {}", greeterReply.getMessage(), greeterReply.getId());
observer.onNext(greeterReply);
observer.onCompleted();
}
@Override
public void onError(Throwable t) {
logger.info("Exception from WorldService -- {}", t);
}
@Override
public void onCompleted() {
}
});
}
}
} ).subscribeOn(Schedulers.io());
}
@Override
protected Observable<Greeter.GreeterReply> resumeWithFallback() {
logger.info("Response from fallback");
Greeter.GreeterReply greeterReply = Greeter.GreeterReply.newBuilder().setMessage("teammate").setId("-1").build();
return Observable.just(greeterReply);
}
Я использую трассировку Zipkin grpc и MDCCurrentTraceContext для печати traceId и spanId в журналах.
Обе записи журнала в WorldCommand не выводят идентификаторы трассировки и диапазона, они вызываются в потоке RxIoScheduler.
РЕДАКТИРОВАТЬ
Добавлена ConcurrencyStrategy в соответствии с предложением Майка.
public class CustomHystrixConcurrencyStrategy extends HystrixConcurrencyStrategy {
private static final Logger log = LoggerFactory.getLogger(CustomHystrixConcurrencyStrategy.class);
public <T> Callable<T> wrapCallable(Callable<T> callable){
log.info("In CustomHystrixConcurrencyStrategy: callable="+ callable.toString());
return new ContextCallable<>(callable);
}
}
HelloService вызывает две службы World и Team. WorldCommand - это HystrixObservableCommand, а TeamCommand - это HystrixCommand.
logger.info("In the HelloService:greetWithHelloWorld");
Greeter.GreeterRequest greeterRequest = Greeter.GreeterRequest.newBuilder().setId(request.getId()).build();
//Call WorldService
ManagedChannel worldChannel = getChannel("localhost:8081", "helloService-world-client");
//Async stub instead of blockingStub
WorldServiceGrpc.WorldServiceStub worldServiceStub = WorldServiceGrpc.newStub(worldChannel);
WorldCommand worldCommand = new WorldCommand(greeterRequest, worldServiceStub);
String messageFromWorldService = "";
String idFromWorldService = "";
try {
Greeter.GreeterReply greeterReply = worldCommand.construct().toBlocking().toFuture().get();
messageFromWorldService = greeterReply.getMessage();
idFromWorldService = greeterReply.getId();
logger.info("Response from WorldService -- {}, id = {}", messageFromWorldService, idFromWorldService);
} catch (StatusRuntimeException | InterruptedException | ExecutionException e) {
logger.warn("Exception when calling WorldService\n" + e);
}
//Call TeamService
ManagedChannel teamChannel = getChannel("localhost:8082", "helloService-team-client");
TeamServiceGrpc.TeamServiceBlockingStub teamServiceStub = TeamServiceGrpc.newBlockingStub(teamChannel);
TeamCommand teamCommand = new TeamCommand(greeterRequest, teamServiceStub);
String messageFromTeamService = "";
String idFromTeamService = "";
try {
Greeter.GreeterReply greeterReply = teamCommand.construct().toBlocking().toFuture().get();
messageFromTeamService = greeterReply.getMessage();
idFromTeamService = greeterReply.getId();
logger.info("Response from TeamService -- {}, id = {}", messageFromTeamService, idFromTeamService);
} catch (StatusRuntimeException | InterruptedException | ExecutionException e) {
logger.warn("Exception when calling TeamService\n" + e);
}
assert(idFromWorldService.equals(idFromTeamService));
Greeter.GreeterReply greeterReply = Greeter.GreeterReply.newBuilder().setMessage("Hello" + messageFromWorldService + " from " + messageFromTeamService).setId(idFromWorldService).build();
responseObserver.onNext(greeterReply);
responseObserver.onCompleted();
Класс PreservableContext
public class PreservableContexts {
//private final TraceContext traceContext;
private static final Logger logger = LoggerFactory.getLogger(PreservableContexts.class.getName());
public PreservableContexts() {
logger.info("Creating new PreservableContexts");
//this.traceContext = TraceContextHolder.getContext();
}
public void set() {
// if (traceContext != null) {
//TraceContextHolder.setContext(traceContext);
// }
}
public void clear() {
//TraceContextHolder.clearContext();
}
Лог в PreservableContexts и CustomHystrixConcurrencyStrategy никогда не печатается. Я регистрирую стратегию, когда запускаю HelloServer.
HystrixConcurrencyStrategy strategy = new CustomHystrixConcurrencyStrategy();
HystrixPlugins.getInstance().registerConcurrencyStrategy(strategy);
context = HystrixRequestContext.initializeContext();
РЕДАКТИРОВАТЬ 2
Обновлен порядок настройки Observables:
ManagedChannel worldChannel = getChannel("localhost:8081", "helloService-world-client");
//Async stub instead of blockingStub
WorldServiceGrpc.WorldServiceStub worldServiceStub = WorldServiceGrpc.newStub(worldChannel);
WorldCommand worldCommand = new WorldCommand(greeterRequest, worldServiceStub);
//Call TeamService
ManagedChannel teamChannel = getChannel("localhost:8082", "helloService-team-client");
TeamServiceGrpc.TeamServiceStub teamServiceStub = TeamServiceGrpc.newStub(teamChannel);
//TeamServiceGrpc.TeamServiceBlockingStub teamServiceStub = TeamServiceGrpc.newBlockingStub(teamChannel);
TeamCommand teamCommand = new TeamCommand(greeterRequest, teamServiceStub);
try {
rx.Observable<Greeter.GreeterReply> worldReplyObservable = worldCommand.observe().subscribeOn(Schedulers.computation());
rx.Observable<Greeter.GreeterReply> teamReplyObservable = teamCommand.observe().subscribeOn(Schedulers.computation());
Observable.zip(worldReplyObservable, teamReplyObservable, new Func2<Greeter.GreeterReply, Greeter.GreeterReply, Object>() {
@Override
public Object call(Greeter.GreeterReply worldReply, Greeter.GreeterReply teamReply) {
String messageFromWorldService = worldReply.getMessage();
String idFromWorldService = worldReply.getId();
logger.info("Response from WorldService -- {}, id = {}", messageFromWorldService, idFromWorldService);
String messageFromTeamService = teamReply.getMessage();
String idFromTeamService = teamReply.getId();
logger.info("Response from TeamService -- {}, id = {}", messageFromTeamService, idFromTeamService);
assert(idFromWorldService.equals(idFromTeamService));
Greeter.GreeterReply greeterReply = Greeter.GreeterReply.newBuilder().setMessage("Hello" + messageFromWorldService + " from " + messageFromTeamService).setId(idFromWorldService).build();
logger.info("Final response=" + greeterReply.getMessage());
responseObserver.onNext(greeterReply);
responseObserver.onCompleted();
return null;
}
});
} catch (StatusRuntimeException e) {
logger.warn("Exception when calling WorldService and/or TeamService\n" + e);
}
У меня сейчас странная проблема, вызовы TeamCommand и WorldCommand не завершаются, так как в этом коде никогда не выполняется:
Observable.zip(worldReplyObservable, teamReplyObservable, new Func2<Greeter.GreeterReply, Greeter.GreeterReply, Object>() {
@Override
public Object call(Greeter.GreeterReply worldReply, Greeter.GreeterReply teamReply) {
String messageFromWorldService = worldReply.getMessage();
Кроме того, если есть запасной вариант, потоки Hystrix-timer больше не имеют MDC.
2 ответа
Я не очень разбираюсь в hysterix, но если вы пытаетесь передать некоторую контекстную информацию, такую как идентификаторы трассировки, то io.grpc.Context - это правильный класс для использования. Вам нужно будет позвонить context.withValue создать новый контекст с помощью traceID. В тех местах, где вы хотите данные, вы должны прикрепить контекст. Также не забудьте отключить контекст, когда это будет сделано, чего я не вижу в вашем фрагменте.
Вам нужно использовать...
HystrixPlugins.getInstance().registerConcurrencyStrategy(...)
... зарегистрировать кастом HystrixConcurrencyStrategy который использует свой собственный Callable...
public class ConcurrencyStrategy extends HystrixConcurrencyStrategy {
@Override
public <K> Callable<K> wrapCallable(Callable<K> c) {
return new ContextCallable<>(c);
}
}
... это применяет сохранение контекста вокруг схемы...
public class ContextCallable<K> implements Callable<K> {
private final Callable<K> callable;
private final PreservableContexts contexts;
public ContextCallable(Callable<K> actual) {
this.callable = actual;
this.contexts = new PreservableContexts();
}
@Override
public K call() throws Exception {
contexts.set();
try {
return callable.call();
} finally {
contexts.clear();
}
}
}
... via - вспомогательный класс, способный сохранять контекст Zipkin...
public class PreservableContexts {
private final TraceContext traceContext;
public PreservableContexts() {
this.traceContext = TraceContextHolder.getContext();
}
public void set() {
if (traceContext != null) {
TraceContextHolder.setContext(traceContext);
}
}
public void clear() {
TraceContextHolder.clearContext();
}
}
... и предоставляя простой способ добавления других контекстов, которые вы можете сохранить, например, MDC, SecurityContext и т. д....