Java StreamObserver.onNext方法代码示例

import io.grpc.stub.StreamObserver; //导入方法依赖的package包/类
@Test
public void manyToOne() {
    AtomicBoolean called = new AtomicBoolean(false);
    GreeterGrpc.GreeterStub stub = GreeterGrpc.newStub(channel);

    StreamObserver<HelloRequest> requestStream = stub.sayHelloReqStream(new LambdaStreamObserver<>(
            response -> {
                assertThat(response.getMessage()).isEqualTo("Hello A and B and C");
                called.set(true);
            }
    ));

    requestStream.onNext(HelloRequest.newBuilder().setName("A").build());
    requestStream.onNext(HelloRequest.newBuilder().setName("B").build());
    requestStream.onNext(HelloRequest.newBuilder().setName("C").build());
    requestStream.onCompleted();

    await().atMost(1, TimeUnit.SECONDS).untilTrue(called);
}
 

import io.grpc.stub.StreamObserver; //导入方法依赖的package包/类
@Override
public void downloadProductImage(DownloadProductImageRequest request, StreamObserver<DataChunk> responseObserver) {
  try {
    BufferedInputStream imageStream = new BufferedInputStream(
        productImageSeeker.seekProductImage(request.getProductId())
    );
    int bufferSize = 256 * 1024;// 256k
    byte[] buffer = new byte[bufferSize];
    int length;
    while ((length = imageStream.read(buffer, 0, bufferSize)) != -1) {
      responseObserver.onNext(
          DataChunk.newBuilder().setData(ByteString.copyFrom(buffer, 0, length)).build()
      );
    }
    responseObserver.onCompleted();
    imageStream.close();
    counter.labels("downloadProductImage", "success");
  } catch (Exception e) {
    counter.labels("downloadProductImage", "failed");
    log.error("error on read product image", e);
    responseObserver.onError(e);
  }

}
 

import io.grpc.stub.StreamObserver; //导入方法依赖的package包/类
@Override
public void createTblPriv(MetaProto.TblPrivParam tblPriv,
                          StreamObserver<StatusProto.ResponseStatus> responseStreamObserver)
{
    TransactionController txController = null;
    try {
        txController = ConnectionPool.INSTANCE().getTxController();
        ActionResponse input = new ActionResponse();
        input.setParam(tblPriv);
        input.setProperties("dbName", tblPriv.getDbName());
        input.setProperties("tblName", tblPriv.getTblName());
        input.setProperties("userName", tblPriv.getUserName());
        txController.setAutoCommit(false);
        txController.addAction(new GetDatabaseIdAction());
        txController.addAction(new GetTableIdAction());
        txController.addAction(new GetUserIdAction());
        txController.addAction(new CreateTblPrivAction());
        txController.commit(input);
        responseStreamObserver.onNext(MetaConstants.OKStatus);
        responseStreamObserver.onCompleted();
    }
    catch (ParaFlowException e) {
        responseStreamObserver.onNext(e.getResponseStatus());
        responseStreamObserver.onCompleted();
        e.handle();
    }
    finally {
        if (txController != null) {
            txController.close();
        }
    }
}
 

import io.grpc.stub.StreamObserver; //导入方法依赖的package包/类
@Override
public void failPlease(FailWithProbabilityOrSucceedEchoRequest request,
                       StreamObserver<EchoResponse> responseObserver) {
    EchoRequest echoRequest = request.getEchoRequest();
    int failProbability = request.getFailProbability();
    Preconditions.checkArgument(failProbability >= 0 && failProbability <= 100,
            "fail probability not [" + failProbability + "] not in range [0, 100] inclusive");
    logger.info("fail please - p(" + failProbability + " / 100)" + " echo: " + echoRequest.getEcho()
            + " with " + "repetitions: " + echoRequest.getRepeatEcho() + " received at "
            + DATE_FORMAT.format(new Date(System.currentTimeMillis())));

    int randomFail = RANDOM.nextInt(100);
    if (randomFail < failProbability) {
        Status status = Status.INTERNAL;
        status = status.withCause(new FailPleaseException("Looks like you hit jackpot - we failed!"));
        responseObserver.onError(status.asRuntimeException());
    } else {
        responseObserver.onNext(buildEchoResponseFromEchoRequest(echoRequest));
        responseObserver.onCompleted();
    }
}
 

import io.grpc.stub.StreamObserver; //导入方法依赖的package包/类
@Override
public void refresh(RefreshRequest request, StreamObserver<Empty> responseObserver) {
	// TODO Auto-generated method stub
	String serviceName=request.getServiceName();
	ArrayList<HostInfo> hostInfos=new ArrayList<HostInfo>();
	logger.info("Refresh host infos:"+serviceName+" Begin");
	for(Object object:request.getHostInfosList())
	{
		HostInfo hostInfo=(HostInfo) object;
		hostInfos.add(hostInfo);
		logger.info(HostInfoFormatter.getFormatString(hostInfo));
	}
	LoadBalance.refreshHostInfos(serviceName, hostInfos);
	logger.info("Refresh host infos:"+serviceName+" End");
	Empty result=Empty.newBuilder().build();
	responseObserver.onNext(result);
	responseObserver.onCompleted();
}
 

import io.grpc.stub.StreamObserver; //导入方法依赖的package包/类
@RequestMapping("/bodistream")
public HelloReply bodistream(@RequestParam(value = "name", required = false) String name) {
  HelloRequest request = new HelloRequest();
  request.setName(name);
  StreamObserver<com.quancheng.examples.model.hello.HelloRequest> requestObserver =
      helloService.sayHelloBidiStream(responseObserver());
  try {
    for (int i = 0; i < 10; i++) {
      requestObserver.onNext(request);
    }
  } catch (Exception e) {
    requestObserver.onError(e);
  }
  requestObserver.onCompleted();
  return null;
}
 

import io.grpc.stub.StreamObserver; //导入方法依赖的package包/类
@Override
public void register(Application request, StreamObserver<ApplicationMapping> responseObserver) {
    logger.debug("receive application register.");
    ApplicationMapping.Builder builder = ApplicationMapping.newBuilder();
    for (String applicationCode : request.getApplicationCodeList()) {
        if (applicationCode.startsWith("localhost") || applicationCode.startsWith("127.0.0.1")){
            continue;
        }
        Integer applicationId = applicationMapping.get(applicationCode);
        if (applicationId == null) {
            applicationId = currentId.incrementAndGet();
            applicationMapping.put(applicationCode, applicationId);
            ValidateData.INSTANCE.getRegistryItem().registryApplication(new RegistryItem.Application(applicationCode,
                applicationId));
        }

        builder.addApplication(KeyWithIntegerValue.newBuilder().setKey(applicationCode).setValue(applicationId));
    }
    responseObserver.onNext(builder.build());
    responseObserver.onCompleted();
}
 

import io.grpc.stub.StreamObserver; //导入方法依赖的package包/类
public void getStore(GetStoreRequest request, StreamObserver<GetStoreResponse> resp) {
  try {
    resp.onNext(getStoreResp.removeFirst().get());
    resp.onCompleted();
  } catch (Exception e) {
    resp.onError(Status.INTERNAL.asRuntimeException());
  }
}
 

import io.grpc.stub.StreamObserver; //导入方法依赖的package包/类
@Override
    public void createTable(MetaProto.TblParam tblParam,
                            StreamObserver<StatusProto.ResponseStatus> responseStreamObserver)
    {
        TransactionController txController = null;
        try {
            txController = ConnectionPool.INSTANCE().getTxController();
            ActionResponse input = new ActionResponse();
            input.setParam(tblParam);
            input.setProperties("userName", tblParam.getUserName());
            input.setProperties("dbName", tblParam.getDbName());
            input.setProperties("sfName", tblParam.getStorageFormatName());
            input.setProperties("funcName", tblParam.getFuncName());
            txController.setAutoCommit(false);
            txController.addAction(new GetUserIdAction());
            txController.addAction(new GetDatabaseIdAction());
            txController.addAction(new GetStorageFormatIdAction());
            txController.addAction(new GetFuncIdAction());
            txController.addAction(new CreateTableAction());
            txController.addAction(new GetTableIdAction());
            txController.addAction(new CreateColumnAction());
//            txController.addAction(new CreateTblFuncAction());
            txController.commit(input);
            responseStreamObserver.onNext(MetaConstants.OKStatus);
            responseStreamObserver.onCompleted();
        }
        catch (ParaFlowException e) {
            responseStreamObserver.onNext(e.getResponseStatus());
            responseStreamObserver.onCompleted();
            e.handle();
        }
        finally {
            if (txController != null) {
                txController.close();
            }
        }
    }
 

import io.grpc.stub.StreamObserver; //导入方法依赖的package包/类
@Test
public void contextTransfersOneHopSync() throws Exception {
    Metadata.Key<String> ctxKey = Metadata.Key.of("ctx-context-key", Metadata.ASCII_STRING_MARSHALLER);
    String expectedCtxValue = "context-value";
    AtomicReference<String> ctxValue = new AtomicReference<>();

    // Service
    GreeterGrpc.GreeterImplBase svc = new GreeterGrpc.GreeterImplBase() {
        @Override
        public void sayHello(HelloRequest request, StreamObserver<HelloResponse> responseObserver) {
            ctxValue.set(AmbientContext.current().get(ctxKey));
            responseObserver.onNext(HelloResponse.newBuilder().setMessage("Hello " + request.getName()).build());
            responseObserver.onCompleted();
        }
    };

    // Plumbing
    serverRule1.getServiceRegistry().addService(ServerInterceptors
            .intercept(svc, new AmbientContextServerInterceptor("ctx-")));

    GreeterGrpc.GreeterBlockingStub stub = GreeterGrpc
            .newBlockingStub(serverRule1.getChannel())
            .withInterceptors(new AmbientContextClientInterceptor("ctx-"));

    // Test
    AmbientContext.initialize(Context.current()).run(() -> {
        AmbientContext.current().put(ctxKey, expectedCtxValue);
        stub.sayHello(HelloRequest.newBuilder().setName("world").build());
    });

    assertThat(ctxValue.get()).isEqualTo(expectedCtxValue);
}
 

import io.grpc.stub.StreamObserver; //导入方法依赖的package包/类
@Override
public void requestReply(
    SimpleRequest request, StreamObserver<SimpleResponse> responseObserver) {
  SimpleResponse response =
      SimpleResponse.newBuilder().setResponseMessage(request.getRequestMessage()).build();
  responseObserver.onNext(response);
  responseObserver.onCompleted();
}
 

import io.grpc.stub.StreamObserver; //导入方法依赖的package包/类
@Test
public void perSessionShouldFailMissingTransportFilter() throws Exception {
    class TestService extends GreeterGrpc.GreeterImplBase {
        @Override
        public void sayHello(HelloRequest request, StreamObserver<HelloResponse> responseObserver) {
            responseObserver.onNext(HelloResponse.newBuilder().setMessage(Integer.toString(System.identityHashCode(this))).build());
            responseObserver.onCompleted();
        }
    }

    ClientSessionTransportFilter tf = new ClientSessionTransportFilter();
    Server server = InProcessServerBuilder.forName("perSessionShouldInstantiateOneInstancePerSession")
            .addService(new PerSessionService<>(() -> new TestService(), tf))
            .build()
            .start();

    ManagedChannel channel = InProcessChannelBuilder.forName("perSessionShouldInstantiateOneInstancePerSession")
            .usePlaintext(true)
            .build();
    GreeterGrpc.GreeterBlockingStub stub = GreeterGrpc.newBlockingStub(channel);

    try {
        assertThatThrownBy(() -> stub.sayHello(HelloRequest.getDefaultInstance()).getMessage()).isInstanceOf(StatusRuntimeException.class);
    } finally {
        channel.shutdown();
        server.shutdown();
    }
}
 

import io.grpc.stub.StreamObserver; //导入方法依赖的package包/类
@Override
public synchronized void listBooks(
    ListBooksRequest request, StreamObserver<ListBooksResponse> responseObserver) {
  NavigableMap<String, Book> cursor = booksById;

  // Resume iteration from the page token.
  if (!request.getPageToken().isEmpty()) {
    String pageToken = decodePageToken(request.getPageToken());
    cursor = cursor.tailMap(pageToken, false);
  }

  ImmutableList<Book> books =
      cursor
          .values()
          .stream()
          .limit(request.getPageSize() > 0 ? request.getPageSize() : DEFAULT_PAGE_SIZE)
          .collect(ImmutableList.toImmutableList());

  // Return one page of results.
  ListBooksResponse.Builder responseBuilder = ListBooksResponse.newBuilder().addAllBooks(books);
  // Set next page token to resume iteration in the next request.
  if (cursor.values().size() > books.size()) {
    String nextPageToken = encodePageToken(books.get(books.size() - 1).getId());
    responseBuilder.setNextPageToken(nextPageToken);
  }

  responseObserver.onNext(responseBuilder.build());
  responseObserver.onCompleted();
}
 

import io.grpc.stub.StreamObserver; //导入方法依赖的package包/类
@Override
public void createDatabase(MetaProto.DbParam dbParam,
                           StreamObserver<StatusProto.ResponseStatus> responseStreamObserver)
{
    TransactionController txController = null;
    try {
        txController = ConnectionPool.INSTANCE().getTxController();
        ActionResponse input = new ActionResponse();
        input.setParam(dbParam);
        input.setProperties("userName", dbParam.getUserName());
        txController.setAutoCommit(false);
        txController.addAction(new GetUserIdAction());
        txController.addAction(new CreateDatabaseAction());
        txController.commit(input);
        responseStreamObserver.onNext(MetaConstants.OKStatus);
        responseStreamObserver.onCompleted();
    }
    catch (ParaFlowException e) {
        responseStreamObserver.onNext(e.getResponseStatus());
        responseStreamObserver.onCompleted();
        e.handle();
    }
    finally {
        if (txController != null) {
            txController.close();
        }
    }
}
 

import io.grpc.stub.StreamObserver; //导入方法依赖的package包/类
@Override
public void linkVanished(LinkVanishedMsg request,
                         StreamObserver<Void> responseObserver) {
    try {
        onLinksVanished(request, responseObserver);
        // If onNext call was not mandatory, it can be removed.
        responseObserver.onNext(Void.getDefaultInstance());
        responseObserver.onCompleted();
    } catch (Exception e) {
        log.error("Exception caught", e);
        responseObserver.onError(e);
    }
}