Examples with ScatterRegionRequest org.tikv.kvproto.Pdpb.ScatterRegionRequest used on opensource projects

Example 1 with ScatterRegionRequest

use of org.tikv.kvproto.Pdpb.ScatterRegionRequest in project tispark by pingcap.

the class PDClient method scatterRegion.

/**
 * Sends request to pd to scatter region.
 *
 * @param region represents a region info
 */
void scatterRegion(TiRegion region, BackOffer backOffer) {
    Supplier<ScatterRegionRequest> request = () -> ScatterRegionRequest.newBuilder().setHeader(header).setRegionId(region.getId()).build();
    PDErrorHandler<ScatterRegionResponse> handler = new PDErrorHandler<>(r -> r.getHeader().hasError() ? buildFromPdpbError(r.getHeader().getError()) : null, this);
    ScatterRegionResponse resp = callWithRetry(backOffer, PDGrpc.getScatterRegionMethod(), request, handler);
    // TODO: maybe we should retry here, need dig into pd's codebase.
    if (resp.hasHeader() && resp.getHeader().hasError()) {
        throw new TiClientInternalException(String.format("failed to scatter region because %s", resp.getHeader().getError()));
    }
}

Example 2 with ScatterRegionRequest

use of org.tikv.kvproto.Pdpb.ScatterRegionRequest in project client-java by tikv.

the class PDClient method scatterRegion.

/**
 * Sends request to pd to scatter region.
 *
 * @param region represents a region info
 */
void scatterRegion(Metapb.Region region, BackOffer backOffer) {
    Supplier<ScatterRegionRequest> request = () -> ScatterRegionRequest.newBuilder().setHeader(header).setRegionId(region.getId()).build();
    PDErrorHandler<ScatterRegionResponse> handler = new PDErrorHandler<>(r -> r.getHeader().hasError() ? buildFromPdpbError(r.getHeader().getError()) : null, this);
    ScatterRegionResponse resp = callWithRetry(backOffer, PDGrpc.getScatterRegionMethod(), request, handler);
    // TODO: maybe we should retry here, need dig into pd's codebase.
    if (resp.hasHeader() && resp.getHeader().hasError()) {
        throw new TiClientInternalException(String.format("failed to scatter region because %s", resp.getHeader().getError()));
    }
}