sandbox-libpq/create_table/main.c

91 lines
2.4 KiB
C

#include <stdio.h>
#include <stdlib.h>
#include <libpq-fe.h>
void do_exit(PGconn *conn, PGresult *res) {
fprintf(stderr, "%s\n", PQerrorMessage(conn));
/*every statement result should be freed*/
PQclear(res);
/*close connection and free memory*/
PQfinish(conn);
exit(EXIT_FAILURE);
}
int main(int argc, char **argv) {
/*declaration*/
const char *conninfo;
printf("main() Started...\n");
/*read command line*/
if (argc > 1){
printf("main() argv[1]: %s\n", argv[1]);
conninfo = argv[1];
}else{
fprintf(stderr, "Usage: %s <connection info>\n", argv[0]);
return 1;
}
/*connect to database*/
PGconn *conn = PQconnectdb(conninfo);
if (PQstatus(conn) == CONNECTION_BAD) {
fprintf(stderr, "Connection to database failed: %s\n",
PQerrorMessage(conn));
PQfinish(conn);
exit(1);
}else if(PQstatus(conn)==CONNECTION_OK){
printf("main() connected to database\n");
}else{
printf("main() connection status NOT known\n");
}
char *user = PQuser(conn);
char *db_name = PQdb(conn);
char *pswd = PQpass(conn);
printf("User: %s\n", user);
printf("Database name: %s\n", db_name);
printf("Password: %s\n", pswd);
/*submit SQL statement to the server and wait for the result*/
PGresult *res = PQexec(conn, "DROP TABLE IF EXISTS cars_cleanup");
if (PQresultStatus(res) != PGRES_COMMAND_OK) {
do_exit(conn, res);
}else{
printf("main() database table cleaned\n");
}
/*every statement result should be freed*/
PQclear(res);
res = PQexec(conn, "CREATE TABLE cars_cleanup(Id INTEGER PRIMARY KEY, Name VARCHAR(20), Price INT)");
if (PQresultStatus(res) != PGRES_COMMAND_OK) {
do_exit(conn, res);
}else{
printf("main() database table created\n");
}
/*every statement result should be freed*/
PQclear(res);
res = PQexec(conn, "INSERT INTO cars_cleanup VALUES(1,'Audi',52642)");
if (PQresultStatus(res) != PGRES_COMMAND_OK){
do_exit(conn, res);
}else{
printf("main() record inserted into database table\n");
}
/*every statement result should be freed*/
PQclear(res);
res = PQexec(conn, "INSERT INTO cars_cleanup VALUES(2,'Mercedes',57127)");
if (PQresultStatus(res) != PGRES_COMMAND_OK) {
do_exit(conn, res);
}else{
printf("main() record inserted into database table\n");
}
/*every statement result should be freed*/
PQclear(res);
PQfinish(conn);
return 0;
}