(arrow-flight-sql-postgresql) branch main updated: Add COPY benchmark (#152)

[kou]

This is an automated email from the ASF dual-hosted git repository.

kou pushed a commit to branch main
in repository 
https://gitbox.apache.org/repos/asf/arrow-flight-sql-postgresql.git


The following commit(s) were added to refs/heads/main by this push:
     new 0176b4c  Add COPY benchmark (#152)
0176b4c is described below

commit 0176b4c9831cdb25bb06b51132f33a9ae6ab23d4
Author: Sutou Kouhei <k...@clear-code.com>
AuthorDate: Sat Oct 28 17:33:44 2023 +0900

    Add COPY benchmark (#152)
    
    Closes GH-151
---
 benchmark/integer/{select.c => copy.c} | 32 +++++++++++++++++++-------------
 benchmark/integer/meson.build          |  1 +
 benchmark/integer/select.c             |  4 ++--
 3 files changed, 22 insertions(+), 15 deletions(-)

diff --git a/benchmark/integer/select.c b/benchmark/integer/copy.c
similarity index 70%
copy from benchmark/integer/select.c
copy to benchmark/integer/copy.c
index 4cca75d..9a1ca04 100644
--- a/benchmark/integer/select.c
+++ b/benchmark/integer/copy.c
@@ -17,6 +17,7 @@
  * under the License.
  */
 
+#include <stdbool.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <sys/time.h>
@@ -30,36 +31,41 @@ main(int argc, char** argv)
        PGresult* result;
        struct timeval before;
        struct timeval after;
-       int nFields;
-       int iField;
-       int nTuples;
-       int iTuple;
 
        if (PQstatus(connection) != CONNECTION_OK)
        {
-               fprintf(stderr, "failed to connect: %s", 
PQerrorMessage(connection));
+               fprintf(stderr, "failed to connect: %s\n", 
PQerrorMessage(connection));
                PQfinish(connection);
                return EXIT_FAILURE;
        }
 
        gettimeofday(&before, NULL);
-       result = PQexec(connection, "SELECT * FROM data");
-       if (PQresultStatus(result) != PGRES_TUPLES_OK)
+       result = PQexec(connection, "COPY data TO STDOUT (FORMAT binary)");
+       if (PQresultStatus(result) != PGRES_COPY_OUT)
        {
-               fprintf(stderr, "failed to select: %s", 
PQerrorMessage(connection));
+               fprintf(stderr, "failed to copy: %s\n", 
PQerrorMessage(connection));
                PQclear(result);
                PQfinish(connection);
                return EXIT_FAILURE;
        }
 
-       nTuples = PQntuples(result);
-       nFields = PQnfields(result);
-       for (iTuple = 0; iTuple < nTuples; iTuple++)
+       while (true)
        {
-               for (iField = 0; iField < nFields; iField++)
+               char* buffer;
+               int size = PQgetCopyData(connection, &buffer, 0);
+               if (size == -1)
                {
-                       PQgetvalue(result, iTuple, iField);
+                       break;
                }
+               if (size == -2)
+               {
+                       fprintf(stderr, "failed to read copy data: %s\n", 
PQerrorMessage(connection));
+                       PQclear(result);
+                       PQfinish(connection);
+                       return EXIT_FAILURE;
+               }
+               /* printf("%.*s\n", size, buffer); */
+               free(buffer);
        }
        gettimeofday(&after, NULL);
        printf("%.3fsec\n",
diff --git a/benchmark/integer/meson.build b/benchmark/integer/meson.build
index 93561b2..4bfdea4 100644
--- a/benchmark/integer/meson.build
+++ b/benchmark/integer/meson.build
@@ -16,4 +16,5 @@
 # under the License.
 
 libpq = dependency('libpq')
+executable('copy', 'copy.c', dependencies: [libpq])
 executable('select', 'select.c', dependencies: [libpq])
diff --git a/benchmark/integer/select.c b/benchmark/integer/select.c
index 4cca75d..bf3ad9d 100644
--- a/benchmark/integer/select.c
+++ b/benchmark/integer/select.c
@@ -37,7 +37,7 @@ main(int argc, char** argv)
 
        if (PQstatus(connection) != CONNECTION_OK)
        {
-               fprintf(stderr, "failed to connect: %s", 
PQerrorMessage(connection));
+               fprintf(stderr, "failed to connect: %s\n", 
PQerrorMessage(connection));
                PQfinish(connection);
                return EXIT_FAILURE;
        }
@@ -46,7 +46,7 @@ main(int argc, char** argv)
        result = PQexec(connection, "SELECT * FROM data");
        if (PQresultStatus(result) != PGRES_TUPLES_OK)
        {
-               fprintf(stderr, "failed to select: %s", 
PQerrorMessage(connection));
+               fprintf(stderr, "failed to select: %s\n", 
PQerrorMessage(connection));
                PQclear(result);
                PQfinish(connection);
                return EXIT_FAILURE;