Set error code in grpc response

Author: zhicwu

clickhouse-grpc-client/src/main/java/com/clickhouse/client/grpc/ClickHouseGrpcClient.java

@@ -57,19 +57,19 @@ protected static Compression getResultCompression(ClickHouseConfig config) {
         CompressionAlgorithm algorithm = CompressionAlgorithm.DEFLATE;
         CompressionLevel level = CompressionLevel.COMPRESSION_MEDIUM;
         switch (config.getDecompressAlgorithmForClientRequest()) {
-        case NONE:
-            algorithm = CompressionAlgorithm.NO_COMPRESSION;
-            break;
-        case DEFLATE:
-            break;
-        case GZIP:
-            algorithm = CompressionAlgorithm.GZIP;
-            break;
-        // case STREAM_GZIP:
-        default:
-            log.warn("Unsupported algorithm [%s], change to [%s]", config.getDecompressAlgorithmForClientRequest(),
-                    algorithm);
-            break;
+            case NONE:
+                algorithm = CompressionAlgorithm.NO_COMPRESSION;
+                break;
+            case DEFLATE:
+                break;
+            case GZIP:
+                algorithm = CompressionAlgorithm.GZIP;
+                break;
+            // case STREAM_GZIP:
+            default:
+                log.warn("Unsupported algorithm [%s], change to [%s]", config.getDecompressAlgorithmForClientRequest(),
+                        algorithm);
+                break;
         }
 
         int l = config.getDecompressLevelForClientRequest();
@@ -269,7 +269,8 @@ protected CompletableFuture<ClickHouseResponse> executeSync(ClickHouseRequest<?>
                     sealedRequest.getSettings(), result);
 
             return result.hasException()
-                    ? failedResponse(ClickHouseException.of(result.getException().getDisplayText(), server))
+                    ? failedResponse(new ClickHouseException(result.getException().getCode(),
+                            result.getException().getDisplayText(), server))
                     : CompletableFuture.completedFuture(response);
         } catch (IOException e) {
             throw new CompletionException(ClickHouseException.of(e, server));

clickhouse-grpc-client/src/test/java/com/clickhouse/client/grpc/ClickHouseGrpcClientTest.java

@@ -277,6 +277,19 @@ public void testReadWriteDateTimeTypes() throws Exception {
         }
     }
 
+    @Test(groups = "integration")
+    public void testDropNonExistDb() throws Exception {
+        ClickHouseNode server = getServer(ClickHouseProtocol.GRPC);
+
+        try {
+            ClickHouseClient.send(server, "drop database non_exist_db").get();
+            Assert.fail("Exception is excepted");
+        } catch (ExecutionException e) {
+            ClickHouseException ce = (ClickHouseException) e.getCause();
+            Assert.assertEquals(ce.getErrorCode(), 81);
+        }
+    }
+
     @Test(groups = "integration")
     public void testReadWriteDomains() throws Exception {
         ClickHouseNode server = getServer(ClickHouseProtocol.GRPC);