Add some modicum of error detection...

diff --git a/src/test/examples/testlo.c b/src/test/examples/testlo.c
index 7700ce78833d90c975b8f096388be5b1e155b2e7..181216bf7830b1d89346299187343ae399f59737 100644 (file)
--- a/src/test/examples/testlo.c
+++ b/src/test/examples/testlo.c
@@ -8,7 +8,7 @@
  *
  *
  * IDENTIFICATION
- *   $Header: /cvsroot/pgsql/src/test/examples/testlo.c,v 1.15 2000/04/25 16:39:07 momjian Exp $
+ *   $Header: /cvsroot/pgsql/src/test/examples/testlo.c,v 1.16 2000/10/24 00:08:02 tgl Exp $
  *
  *-------------------------------------------------------------------------
  */
@@ -96,6 +96,8 @@ pickout(PGconn *conn, Oid lobjId, int start, int len)
        buf[nbytes] = '\0';
        fprintf(stderr, ">>> %s", buf);
        nread += nbytes;
+       if (nbytes <= 0)
+           break;              /* no more data? */
    }
    fprintf(stderr, "\n");
    lo_close(conn, lobj_fd);
@@ -126,6 +128,11 @@ overwrite(PGconn *conn, Oid lobjId, int start, int len)
    {
        nbytes = lo_write(conn, lobj_fd, buf + nwritten, len - nwritten);
        nwritten += nbytes;
+       if (nbytes <= 0)
+       {
+           fprintf(stderr, "\nWRITE FAILED!\n");
+           break;
+       }
    }
    fprintf(stderr, "\n");
    lo_close(conn, lobj_fd);
@@ -229,21 +236,24 @@ main(int argc, char **argv)
 /* lobjOid = importFile(conn, in_filename); */
    lobjOid = lo_import(conn, in_filename);
    if (lobjOid == 0)
+   {
        fprintf(stderr, "%s\n", PQerrorMessage(conn));
+   }
+   else
+   {
+       printf("\tas large object %u.\n", lobjOid);
 
-   printf("\tas large object %u.\n", lobjOid);
-
-   printf("picking out bytes 1000-2000 of the large object\n");
-   pickout(conn, lobjOid, 1000, 1000);
-
-   printf("overwriting bytes 1000-2000 of the large object with X's\n");
-   overwrite(conn, lobjOid, 1000, 1000);
+       printf("picking out bytes 1000-2000 of the large object\n");
+       pickout(conn, lobjOid, 1000, 1000);
 
+       printf("overwriting bytes 1000-2000 of the large object with X's\n");
+       overwrite(conn, lobjOid, 1000, 1000);
 
-   printf("exporting large object to file \"%s\" ...\n", out_filename);
-/*   exportFile(conn, lobjOid, out_filename); */
-   if (!lo_export(conn, lobjOid, out_filename))
-       fprintf(stderr, "%s\n", PQerrorMessage(conn));
+       printf("exporting large object to file \"%s\" ...\n", out_filename);
+/*     exportFile(conn, lobjOid, out_filename); */
+       if (!lo_export(conn, lobjOid, out_filename))
+           fprintf(stderr, "%s\n", PQerrorMessage(conn));
+   }
 
    res = PQexec(conn, "end");
    PQclear(res);