ShpDumperConnectDatabase()

int ShpDumperConnectDatabase

(

SHPDUMPERSTATE *

state

)

Definition at line 1253 of file pgsql2shp-core.c.

References _, shp_dumper_state::config, shp_dumper_config::conn, shp_dumper_state::conn, free(), shp_dumper_state::geog_oid, shp_dumper_state::geom_oid, shp_dumper_state::message, shp_dumper_state::pgis_major_version, window::res, SHPDUMPERERR, ShpDumperGetConnectionStringFromConn(), SHPDUMPERMSGLEN, and SHPDUMPEROK.

Referenced by main(), and pgui_action_export().

{
        PGresult *res;

        char *connstring, *tmpvalue;

        /* Generate the PostgreSQL connection string */
        connstring = ShpDumperGetConnectionStringFromConn(state->config->conn);

        /* Connect to the database */
        state->conn = PQconnectdb(connstring);
        if (PQstatus(state->conn) == CONNECTION_BAD)
        {
                snprintf(state->message, SHPDUMPERMSGLEN, "%s", PQerrorMessage(state->conn));
                free(connstring);
                return SHPDUMPERERR;
        }

        /* Set datestyle to ISO */
        res = PQexec(state->conn, "SET DATESTYLE='ISO'");
        if (PQresultStatus(res) != PGRES_COMMAND_OK)
        {
                snprintf(state->message, SHPDUMPERMSGLEN, "%s", PQresultErrorMessage(res));
                PQclear(res);
                free(connstring);
                return SHPDUMPERERR;
        }

        PQclear(res);

        /* Retrieve PostGIS major version */
        res = PQexec(state->conn, "SELECT postgis_version()");
        if (PQresultStatus(res) != PGRES_TUPLES_OK)
        {
                snprintf(state->message, SHPDUMPERMSGLEN, "%s", PQresultErrorMessage(res));
                PQclear(res);
                free(connstring);
                return SHPDUMPERERR;
        }

        tmpvalue = PQgetvalue(res, 0, 0);
        state->pgis_major_version = atoi(tmpvalue);

        PQclear(res);

        /* Find the OID for the geometry type */
        res = PQexec(state->conn, "SELECT oid FROM pg_type WHERE typname = 'geometry'");
        if (PQresultStatus(res) != PGRES_TUPLES_OK)
        {
                snprintf(state->message, SHPDUMPERMSGLEN, _("Error looking up geometry oid: %s"), PQresultErrorMessage(res));
                PQclear(res);
                free(connstring);
                return SHPDUMPERERR;
        }

        if (PQntuples(res) > 0)
        {
                tmpvalue = PQgetvalue(res, 0, 0);
                state->geom_oid = atoi(tmpvalue);
        }
        else
        {
                snprintf(state->message, SHPDUMPERMSGLEN, _("Geometry type unknown (have you enabled postgis?)"));
                PQclear(res);
                free(connstring);
                return SHPDUMPERERR;
        }

        PQclear(res);

        /* Find the OID for the geography type */
        res = PQexec(state->conn, "SELECT oid FROM pg_type WHERE typname = 'geography'");
        if (PQresultStatus(res) != PGRES_TUPLES_OK)
        {
                snprintf(state->message, SHPDUMPERMSGLEN, _("Error looking up geography oid: %s"), PQresultErrorMessage(res));
                PQclear(res);
                free(connstring);
                return SHPDUMPERERR;
        }

        if (PQntuples(res) > 0)
        {
                /* Old databases don't have a geography type, so don't fail if we don't find it */
                tmpvalue = PQgetvalue(res, 0, 0);
                state->geog_oid = atoi(tmpvalue);
        }

        PQclear(res);

        free(connstring);

        return SHPDUMPEROK;
}

Here is the call graph for this function:

Here is the caller graph for this function: