andreacavalli/rockserver
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Translate rocksdberrors

Browse Source
This commit is contained in:
Andrea Cavalli 2024-10-24 02:31:45 +02:00
parent 60b3ff5a58
commit 384382ce34
1 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/main/java/it/cavallium/rockserver/core/client/GrpcConnection.java
View File

@ -8,6 +8,8 @@ import com.google.common.util.concurrent.MoreExecutors;
import com.google.protobuf.ByteString; import com.google.protobuf.ByteString;
import com.google.protobuf.UnsafeByteOperations; import com.google.protobuf.UnsafeByteOperations;
import io.grpc.ManagedChannel; import io.grpc.ManagedChannel;
import io.grpc.Status;
import io.grpc.StatusRuntimeException;
import io.grpc.netty.NettyChannelBuilder; import io.grpc.netty.NettyChannelBuilder;
import io.netty.channel.epoll.EpollDomainSocketChannel; import io.netty.channel.epoll.EpollDomainSocketChannel;
import io.netty.channel.epoll.EpollEventLoopGroup; import io.netty.channel.epoll.EpollEventLoopGroup;
@ -28,6 +30,7 @@ import it.cavallium.rockserver.core.common.RequestType.RequestNothing;
import it.cavallium.rockserver.core.common.RequestType.RequestPrevious; import it.cavallium.rockserver.core.common.RequestType.RequestPrevious;
import it.cavallium.rockserver.core.common.RequestType.RequestPreviousPresence; import it.cavallium.rockserver.core.common.RequestType.RequestPreviousPresence;
import it.cavallium.rockserver.core.common.RequestType.RequestPut; import it.cavallium.rockserver.core.common.RequestType.RequestPut;
import it.cavallium.rockserver.core.common.RocksDBException.RocksDBErrorType;
import it.cavallium.rockserver.core.common.Utils.HostAndPort; import it.cavallium.rockserver.core.common.Utils.HostAndPort;
import it.cavallium.rockserver.core.common.api.proto.*; import it.cavallium.rockserver.core.common.api.proto.*;
import it.cavallium.rockserver.core.common.api.proto.ColumnHashType; import it.cavallium.rockserver.core.common.api.proto.ColumnHashType;
@ -566,13 +569,30 @@ public class GrpcConnection extends BaseConnection implements RocksDBAPI {
@Override @Override
public void onFailure(@NotNull Throwable t) { public void onFailure(@NotNull Throwable t) {
cf.completeExceptionally(t); cf.completeExceptionally(mapGrpcStatusError(t));
} }
}, DIRECT_EXECUTOR); }, DIRECT_EXECUTOR);
return cf; return cf;
} }
private static final String grpcRocksDbErrorPrefixString = "RocksDBError: [uid:";
private static Throwable mapGrpcStatusError(@NotNull Throwable t) {
if (t instanceof StatusRuntimeException statusRuntimeException
&& statusRuntimeException.getStatus() == Status.INTERNAL
&& statusRuntimeException.getStatus().getDescription() != null
&& statusRuntimeException.getStatus().getDescription().startsWith(grpcRocksDbErrorPrefixString)) {
var desc = statusRuntimeException.getStatus().getDescription();
var closeIndex = desc.indexOf(']');
var errorCode = desc.substring(grpcRocksDbErrorPrefixString.length(), closeIndex);
var errorDescription = desc.substring(closeIndex + 1);
return RocksDBException.of(RocksDBErrorType.valueOf(errorCode), errorDescription);
} else {
return t;
}
}
private static <T> CompletableFuture<T> toResponse(ListenableFuture<T> listenableFuture) { private static <T> CompletableFuture<T> toResponse(ListenableFuture<T> listenableFuture) {
var cf = new CompletableFuture<T>() { var cf = new CompletableFuture<T>() {
@Override @Override