COPY TO STDOUT doesn't throw UnsupportedOperationException

TODO: put into separate loop?

Author: psolomin

vertx-pg-client/src/main/java/io/vertx/pgclient/impl/codec/PgDecoder.java

@@ -176,6 +176,20 @@ private void decodeMessage(ChannelHandlerContext ctx, byte id, ByteBuf in) {
         decodeNotificationResponse(ctx, in);
         break;
       }
+      // TODO: check if these handlers need to be at this level of loop
+      // TODO: check if COPY needs a separate loop
+      case PgProtocolConstants.MESSAGE_TYPE_COPY_OUT_RESPONSE: {
+        decodeCopyOutResponse(ctx, in);
+        break;
+      }
+      case PgProtocolConstants.MESSAGE_TYPE_COPY_DATA: {
+        decodeCopyData(ctx, in);
+        break;
+      }
+      case PgProtocolConstants.MESSAGE_TYPE_COPY_COMPLETION: {
+        decodeCopyCompletion(ctx, in);
+        break;
+      }
       default: {
         throw new UnsupportedOperationException();
       }
@@ -455,4 +469,10 @@ private void decodeBackendKeyData(ByteBuf in) {
   private void decodeNotificationResponse(ChannelHandlerContext ctx, ByteBuf in) {
     ctx.fireChannelRead(new Notification(in.readInt(), Util.readCStringUTF8(in), Util.readCStringUTF8(in)));
   }
+
+  private void decodeCopyOutResponse(ChannelHandlerContext ctx, ByteBuf in) {}
+
+  private void decodeCopyData(ChannelHandlerContext ctx, ByteBuf in) {}
+
+  private void decodeCopyCompletion(ChannelHandlerContext ctx, ByteBuf in) {}
 }

vertx-pg-client/src/main/java/io/vertx/pgclient/impl/codec/PgProtocolConstants.java

@@ -71,4 +71,9 @@ public class PgProtocolConstants {
   public static final byte MESSAGE_TYPE_FUNCTION_RESULT = 'V';
   public static final byte MESSAGE_TYPE_SSL_YES = 'S';
   public static final byte MESSAGE_TYPE_SSL_NO = 'N';
+
+  // COPY-related
+  public static final byte MESSAGE_TYPE_COPY_OUT_RESPONSE = 'H';
+  public static final byte MESSAGE_TYPE_COPY_DATA = 'd';
+  public static final byte MESSAGE_TYPE_COPY_COMPLETION = 'c';
 }

vertx-pg-client/src/test/java/io/vertx/pgclient/PgConnectionCopyTest.java

@@ -54,10 +54,10 @@ public void testSimpleQuery(TestContext ctx) {
     Async async = ctx.async();
     connector.accept(ctx.asyncAssertSuccess(conn -> {
       conn
-        .query("SELECT 1")
+        .query("COPY world TO STDOUT (FORMAT csv)")
         .execute()
         .onComplete(ctx.asyncAssertSuccess(result1 -> {
-          ctx.assertEquals(1, result1.size());
+          ctx.assertEquals(0, result1.size());
           async.complete();
         }));
     }));