Examples with BaseKVStoreClosure io.dingodb.store.row.storage.BaseKVStoreClosure used on opensource projects

Example 6 with BaseKVStoreClosure

use of io.dingodb.store.row.storage.BaseKVStoreClosure in project dingo by dingodb.

the class DefaultRegionKVService method handleResetSequence.

@Override
public void handleResetSequence(final ResetSequenceRequest request, final RequestProcessClosure<BaseRequest, BaseResponse<?>> closure) {
    final ResetSequenceResponse response = new ResetSequenceResponse();
    response.setRegionId(getRegionId());
    response.setRegionEpoch(getRegionEpoch());
    try {
        KVParameterRequires.requireSameEpoch(request, getRegionEpoch());
        final byte[] seqKey = KVParameterRequires.requireNonNull(request.getSeqKey(), "sequence.seqKey");
        this.rawKVStore.resetSequence(seqKey, new BaseKVStoreClosure() {

            @Override
            public void run(final Status status) {
                if (status.isOk()) {
                    response.setValue((Boolean) getData());
                } else {
                    setFailure(request, response, status, getError());
                }
                closure.sendResponse(response);
            }
        });
    } catch (final Throwable t) {
        LOG.error("Failed to handle: {}, {}.", request, StackTraceUtil.stackTrace(t));
        response.setError(Errors.forException(t));
        closure.sendResponse(response);
    }
}

Example 7 with BaseKVStoreClosure

use of io.dingodb.store.row.storage.BaseKVStoreClosure in project dingo by dingodb.

the class DefaultRegionKVService method handleCompareAndPutRequest.

@Override
public void handleCompareAndPutRequest(final CompareAndPutRequest request, final RequestProcessClosure<BaseRequest, BaseResponse<?>> closure) {
    final CompareAndPutResponse response = new CompareAndPutResponse();
    response.setRegionId(getRegionId());
    response.setRegionEpoch(getRegionEpoch());
    try {
        KVParameterRequires.requireSameEpoch(request, getRegionEpoch());
        final byte[] key = KVParameterRequires.requireNonNull(request.getKey(), "compareAndPut.key");
        final byte[] expect = KVParameterRequires.requireNonNull(request.getExpect(), "compareAndPut.expect");
        final byte[] update = KVParameterRequires.requireNonNull(request.getUpdate(), "compareAndPut.update");
        this.rawKVStore.compareAndPut(key, expect, update, new BaseKVStoreClosure() {

            @Override
            public void run(final Status status) {
                if (status.isOk()) {
                    response.setValue((Boolean) getData());
                } else {
                    setFailure(request, response, status, getError());
                }
                closure.sendResponse(response);
            }
        });
    } catch (final Throwable t) {
        LOG.error("Failed to handle: {}, {}.", request, StackTraceUtil.stackTrace(t));
        response.setError(Errors.forException(t));
        closure.sendResponse(response);
    }
}

Example 8 with BaseKVStoreClosure

use of io.dingodb.store.row.storage.BaseKVStoreClosure in project dingo by dingodb.

the class DefaultRegionKVService method handlePutIfAbsentRequest.

@Override
public void handlePutIfAbsentRequest(final PutIfAbsentRequest request, final RequestProcessClosure<BaseRequest, BaseResponse<?>> closure) {
    final PutIfAbsentResponse response = new PutIfAbsentResponse();
    response.setRegionId(getRegionId());
    response.setRegionEpoch(getRegionEpoch());
    try {
        KVParameterRequires.requireSameEpoch(request, getRegionEpoch());
        final byte[] key = KVParameterRequires.requireNonNull(request.getKey(), "putIfAbsent.key");
        final byte[] value = KVParameterRequires.requireNonNull(request.getValue(), "putIfAbsent.value");
        this.rawKVStore.putIfAbsent(key, value, new BaseKVStoreClosure() {

            @Override
            public void run(final Status status) {
                if (status.isOk()) {
                    response.setValue((byte[]) getData());
                } else {
                    setFailure(request, response, status, getError());
                }
                closure.sendResponse(response);
            }
        });
    } catch (final Throwable t) {
        LOG.error("Failed to handle: {}, {}.", request, StackTraceUtil.stackTrace(t));
        response.setError(Errors.forException(t));
        closure.sendResponse(response);
    }
}

Example 9 with BaseKVStoreClosure

use of io.dingodb.store.row.storage.BaseKVStoreClosure in project dingo by dingodb.

the class DefaultRegionKVService method handleGetSequence.

@Override
public void handleGetSequence(final GetSequenceRequest request, final RequestProcessClosure<BaseRequest, BaseResponse<?>> closure) {
    final GetSequenceResponse response = new GetSequenceResponse();
    response.setRegionId(getRegionId());
    response.setRegionEpoch(getRegionEpoch());
    try {
        KVParameterRequires.requireSameEpoch(request, getRegionEpoch());
        final byte[] seqKey = KVParameterRequires.requireNonNull(request.getSeqKey(), "sequence.seqKey");
        final int step = KVParameterRequires.requireNonNegative(request.getStep(), "sequence.step");
        this.rawKVStore.getSequence(seqKey, step, new BaseKVStoreClosure() {

            @Override
            public void run(final Status status) {
                if (status.isOk()) {
                    response.setValue((Sequence) getData());
                } else {
                    setFailure(request, response, status, getError());
                }
                closure.sendResponse(response);
            }
        });
    } catch (final Throwable t) {
        LOG.error("Failed to handle: {}, {}.", request, StackTraceUtil.stackTrace(t));
        response.setError(Errors.forException(t));
        closure.sendResponse(response);
    }
}

Example 10 with BaseKVStoreClosure

use of io.dingodb.store.row.storage.BaseKVStoreClosure in project dingo by dingodb.

the class DefaultRegionKVService method handleCompareAndPutAll.

@Override
public void handleCompareAndPutAll(final CASAllRequest request, final RequestProcessClosure<BaseRequest, BaseResponse<?>> closure) {
    final CASAllResponse response = new CASAllResponse();
    response.setRegionId(getRegionId());
    response.setRegionEpoch(getRegionEpoch());
    try {
        KVParameterRequires.requireSameEpoch(request, getRegionEpoch());
        final List<CASEntry> casEntries = KVParameterRequires.requireNonEmpty(request.getCasEntries(), "casAll.casEntries");
        this.rawKVStore.compareAndPutAll(casEntries, new BaseKVStoreClosure() {

            @Override
            public void run(final Status status) {
                if (status.isOk()) {
                    response.setValue((Boolean) getData());
                } else {
                    setFailure(request, response, status, getError());
                }
                closure.sendResponse(response);
            }
        });
    } catch (final Throwable t) {
        LOG.error("Failed to handle: {}, {}.", request, StackTraceUtil.stackTrace(t));
        response.setError(Errors.forException(t));
        closure.sendResponse(response);
    }
}