testlibpq3.c
上传用户:blenddy
上传日期:2007-01-07
资源大小:6495k
文件大小:4k
源码类别:

数据库系统

开发平台:

Unix_Linux

  1. /*
  2.  * testlibpq3.c
  3.  * Test the C version of LIBPQ, the POSTGRES frontend library.
  4.  *  tests the binary cursor interface
  5.  *
  6.  *
  7.  *
  8.  populate a database by doing the following:
  9. CREATE TABLE test1 (i int4, d float4, p polygon);
  10. INSERT INTO test1 values (1, 3.567, '(3.0, 4.0, 1.0, 2.0)'::polygon);
  11. INSERT INTO test1 values (2, 89.05, '(4.0, 3.0, 2.0, 1.0)'::polygon);
  12.  the expected output is:
  13. tuple 0: got
  14.  i = (4 bytes) 1,
  15.  d = (4 bytes) 3.567000,
  16.  p = (4 bytes) 2 points boundbox = (hi=3.000000/4.000000, lo = 1.000000,2.000000)
  17. tuple 1: got
  18.  i = (4 bytes) 2,
  19.  d = (4 bytes) 89.050003,
  20.  p = (4 bytes) 2 points boundbox = (hi=4.000000/3.000000, lo = 2.000000,1.000000)
  21.  *
  22.  */
  23. #include <stdio.h>
  24. #include <string.h>
  25. #include "postgres.h" /* -> "c.h" -> int16, in access/attnum.h */
  26. #include "libpq-fe.h"
  27. #include "utils/geo_decls.h" /* for the POLYGON type */
  28. static void
  29. exit_nicely(PGconn *conn)
  30. {
  31. PQfinish(conn);
  32. exit(1);
  33. }
  34. int
  35. main()
  36. {
  37. char    *pghost,
  38.    *pgport,
  39.    *pgoptions,
  40.    *pgtty;
  41. char    *dbName;
  42. /*
  43.  * int nFields; int   i, j;
  44.  */
  45. int i;
  46. int i_fnum,
  47. d_fnum,
  48. p_fnum;
  49. PGconn    *conn;
  50. PGresult   *res;
  51. /*
  52.  * begin, by setting the parameters for a backend connection if the
  53.  * parameters are null, then the system will try to use reasonable
  54.  * defaults by looking up environment variables or, failing that,
  55.  * using hardwired constants
  56.  */
  57. pghost = NULL; /* host name of the backend server */
  58. pgport = NULL; /* port of the backend server */
  59. pgoptions = NULL; /* special options to start up the backend
  60.  * server */
  61. pgtty = NULL; /* debugging tty for the backend server */
  62. dbName = getenv("USER"); /* change this to the name of your test
  63.  * database */
  64. /* make a connection to the database */
  65. conn = PQsetdb(pghost, pgport, pgoptions, pgtty, dbName);
  66. /* check to see that the backend connection was successfully made */
  67. if (PQstatus(conn) == CONNECTION_BAD)
  68. {
  69. fprintf(stderr, "Connection to database '%s' failed.n", dbName);
  70. fprintf(stderr, "%s", PQerrorMessage(conn));
  71. exit_nicely(conn);
  72. }
  73. /* start a transaction block */
  74. res = PQexec(conn, "BEGIN");
  75. if (PQresultStatus(res) != PGRES_COMMAND_OK)
  76. {
  77. fprintf(stderr, "BEGIN command failedn");
  78. PQclear(res);
  79. exit_nicely(conn);
  80. }
  81. /*
  82.  * should PQclear PGresult whenever it is no longer needed to avoid
  83.  * memory leaks
  84.  */
  85. PQclear(res);
  86. /*
  87.  * fetch instances from the pg_database, the system catalog of
  88.  * databases
  89.  */
  90. res = PQexec(conn, "DECLARE mycursor BINARY CURSOR FOR select * from test1");
  91. if (res == NULL ||
  92. PQresultStatus(res) != PGRES_COMMAND_OK)
  93. {
  94. fprintf(stderr, "DECLARE CURSOR command failedn");
  95. if (res)
  96. PQclear(res);
  97. exit_nicely(conn);
  98. }
  99. PQclear(res);
  100. res = PQexec(conn, "FETCH ALL in mycursor");
  101. if (res == NULL ||
  102. PQresultStatus(res) != PGRES_TUPLES_OK)
  103. {
  104. fprintf(stderr, "FETCH ALL command didn't return tuples properlyn");
  105. if (res)
  106. PQclear(res);
  107. exit_nicely(conn);
  108. }
  109. i_fnum = PQfnumber(res, "i");
  110. d_fnum = PQfnumber(res, "d");
  111. p_fnum = PQfnumber(res, "p");
  112. for (i = 0; i < 3; i++)
  113. {
  114. printf("type[%d] = %d, size[%d] = %dn",
  115.    i, PQftype(res, i),
  116.    i, PQfsize(res, i));
  117. }
  118. for (i = 0; i < PQntuples(res); i++)
  119. {
  120. int    *ival;
  121. float    *dval;
  122. int plen;
  123. POLYGON    *pval;
  124. /* we hard-wire this to the 3 fields we know about */
  125. ival = (int *) PQgetvalue(res, i, i_fnum);
  126. dval = (float *) PQgetvalue(res, i, d_fnum);
  127. plen = PQgetlength(res, i, p_fnum);
  128. /*
  129.  * plen doesn't include the length field so need to increment by
  130.  * VARHDSZ
  131.  */
  132. pval = (POLYGON *) malloc(plen + VARHDRSZ);
  133. pval->size = plen;
  134. memmove((char *) &pval->npts, PQgetvalue(res, i, p_fnum), plen);
  135. printf("tuple %d: gotn", i);
  136. printf(" i = (%d bytes) %d,n",
  137.    PQgetlength(res, i, i_fnum), *ival);
  138. printf(" d = (%d bytes) %f,n",
  139.    PQgetlength(res, i, d_fnum), *dval);
  140. printf(" p = (%d bytes) %d points tboundbox = (hi=%f/%f, lo = %f,%f)n",
  141.    PQgetlength(res, i, d_fnum),
  142.    pval->npts,
  143.    pval->boundbox.high.x,
  144.    pval->boundbox.high.y,
  145.    pval->boundbox.low.x,
  146.    pval->boundbox.low.y);
  147. }
  148. PQclear(res);
  149. /* close the portal */
  150. res = PQexec(conn, "CLOSE mycursor");
  151. PQclear(res);
  152. /* end the transaction */
  153. res = PQexec(conn, "END");
  154. PQclear(res);
  155. /* close the connection to the database and cleanup */
  156. PQfinish(conn);
  157. return 0; /* Though PQfinish(conn1) has called
  158.  * exit(1) */
  159. }