EVOLUTION-MANAGER
Edit File: ogrpgdatasource.cpp
/****************************************************************************** * $Id: ogrpgdatasource.cpp 27741 2014-09-26 19:20:02Z goatbar $ * * Project: OpenGIS Simple Features Reference Implementation * Purpose: Implements OGRPGDataSource class. * Author: Frank Warmerdam, warmerdam@pobox.com * ****************************************************************************** * Copyright (c) 2000, Frank Warmerdam * Copyright (c) 2008-2014, Even Rouault <even dot rouault at mines-paris dot org> * * Permission is hereby granted, free of charge, to any person obtaining a * copy of this software and associated documentation files (the "Software"), * to deal in the Software without restriction, including without limitation * the rights to use, copy, modify, merge, publish, distribute, sublicense, * and/or sell copies of the Software, and to permit persons to whom the * Software is furnished to do so, subject to the following conditions: * * The above copyright notice and this permission notice shall be included * in all copies or substantial portions of the Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER * DEALINGS IN THE SOFTWARE. ****************************************************************************/ #include <string.h> #include "ogr_pg.h" #include "cpl_conv.h" #include "cpl_string.h" #include "cpl_hash_set.h" #include <set> #define PQexec this_is_an_error CPL_CVSID("$Id: ogrpgdatasource.cpp 27741 2014-09-26 19:20:02Z goatbar $"); static void OGRPGNoticeProcessor( void *arg, const char * pszMessage ); /************************************************************************/ /* OGRPGDataSource() */ /************************************************************************/ OGRPGDataSource::OGRPGDataSource() { pszName = NULL; pszDBName = NULL; papoLayers = NULL; nLayers = 0; hPGConn = NULL; bHavePostGIS = FALSE; bHaveGeography = FALSE; bUseBinaryCursor = FALSE; nSoftTransactionLevel = 0; bBinaryTimeFormatIsInt8 = FALSE; bUseEscapeStringSyntax = FALSE; nGeometryOID = (Oid) 0; nGeographyOID = (Oid) 0; nKnownSRID = 0; panSRID = NULL; papoSRS = NULL; poLayerInCopyMode = NULL; nUndefinedSRID = -1; /* actual value will be autotected if PostGIS >= 2.0 detected */ } /************************************************************************/ /* ~OGRPGDataSource() */ /************************************************************************/ OGRPGDataSource::~OGRPGDataSource() { int i; FlushSoftTransaction(); CPLFree( pszName ); CPLFree( pszDBName ); for( i = 0; i < nLayers; i++ ) delete papoLayers[i]; CPLFree( papoLayers ); if( hPGConn != NULL ) { /* XXX - mloskot: After the connection is closed, valgrind still * reports 36 bytes definitely lost, somewhere in the libpq. */ PQfinish( hPGConn ); hPGConn = NULL; } for( i = 0; i < nKnownSRID; i++ ) { if( papoSRS[i] != NULL ) papoSRS[i]->Release(); } CPLFree( panSRID ); CPLFree( papoSRS ); } /************************************************************************/ /* GetCurrentSchema() */ /************************************************************************/ CPLString OGRPGDataSource::GetCurrentSchema() { /* -------------------------------------------- */ /* Get the current schema */ /* -------------------------------------------- */ PGresult *hResult = OGRPG_PQexec(hPGConn,"SELECT current_schema()"); if ( hResult && PQntuples(hResult) == 1 && !PQgetisnull(hResult,0,0) ) { osCurrentSchema = PQgetvalue(hResult,0,0); } OGRPGClearResult( hResult ); return osCurrentSchema; } /************************************************************************/ /* OGRPGDecodeVersionString() */ /************************************************************************/ void OGRPGDataSource::OGRPGDecodeVersionString(PGver* psVersion, const char* pszVer) { GUInt32 iLen; const char* ptr; char szNum[25]; char szVer[10]; while ( *pszVer == ' ' ) pszVer++; ptr = pszVer; // get Version string while (*ptr && *ptr != ' ') ptr++; iLen = ptr-pszVer; if ( iLen > sizeof(szVer) - 1 ) iLen = sizeof(szVer) - 1; strncpy(szVer,pszVer,iLen); szVer[iLen] = '\0'; ptr = pszVer = szVer; // get Major number while (*ptr && *ptr != '.') ptr++; iLen = ptr-pszVer; if ( iLen > sizeof(szNum) - 1) iLen = sizeof(szNum) - 1; strncpy(szNum,pszVer,iLen); szNum[iLen] = '\0'; psVersion->nMajor = atoi(szNum); if (*ptr == 0) return; pszVer = ++ptr; // get Minor number while (*ptr && *ptr != '.') ptr++; iLen = ptr-pszVer; if ( iLen > sizeof(szNum) - 1) iLen = sizeof(szNum) - 1; strncpy(szNum,pszVer,iLen); szNum[iLen] = '\0'; psVersion->nMinor = atoi(szNum); if ( *ptr ) { pszVer = ++ptr; // get Release number while (*ptr && *ptr != '.') ptr++; iLen = ptr-pszVer; if ( iLen > sizeof(szNum) - 1) iLen = sizeof(szNum) - 1; strncpy(szNum,pszVer,iLen); szNum[iLen] = '\0'; psVersion->nRelease = atoi(szNum); } } /************************************************************************/ /* One entry for each PG table */ /************************************************************************/ typedef struct { char* pszTableName; char* pszSchemaName; int nGeomColumnCount; PGGeomColumnDesc* pasGeomColumns; /* list of geometry columns */ int bDerivedInfoAdded; /* set to TRUE if it derives from another table */ } PGTableEntry; static unsigned long OGRPGHashTableEntry(const void * _psTableEntry) { const PGTableEntry* psTableEntry = (PGTableEntry*)_psTableEntry; return CPLHashSetHashStr(CPLString().Printf("%s.%s", psTableEntry->pszSchemaName, psTableEntry->pszTableName)); } static int OGRPGEqualTableEntry(const void* _psTableEntry1, const void* _psTableEntry2) { const PGTableEntry* psTableEntry1 = (PGTableEntry*)_psTableEntry1; const PGTableEntry* psTableEntry2 = (PGTableEntry*)_psTableEntry2; return strcmp(psTableEntry1->pszTableName, psTableEntry2->pszTableName) == 0 && strcmp(psTableEntry1->pszSchemaName, psTableEntry2->pszSchemaName) == 0; } static void OGRPGTableEntryAddGeomColumn(PGTableEntry* psTableEntry, const char* pszName, const char* pszGeomType = NULL, int nCoordDimension = 0, int nSRID = UNDETERMINED_SRID, PostgisType ePostgisType = GEOM_TYPE_UNKNOWN) { psTableEntry->pasGeomColumns = (PGGeomColumnDesc*) CPLRealloc(psTableEntry->pasGeomColumns, sizeof(PGGeomColumnDesc) * (psTableEntry->nGeomColumnCount + 1)); psTableEntry->pasGeomColumns[psTableEntry->nGeomColumnCount].pszName = CPLStrdup(pszName); psTableEntry->pasGeomColumns[psTableEntry->nGeomColumnCount].pszGeomType = (pszGeomType) ? CPLStrdup(pszGeomType) : NULL; psTableEntry->pasGeomColumns[psTableEntry->nGeomColumnCount].nCoordDimension = nCoordDimension; /* With PostGIS 2.0, querying geometry_columns can return 0, not only when */ /* the SRID is truly set to 0, but also when there's no constraint */ psTableEntry->pasGeomColumns[psTableEntry->nGeomColumnCount].nSRID = nSRID > 0 ? nSRID : UNDETERMINED_SRID; psTableEntry->pasGeomColumns[psTableEntry->nGeomColumnCount].ePostgisType = ePostgisType; psTableEntry->nGeomColumnCount ++; } static void OGRPGTableEntryAddGeomColumn(PGTableEntry* psTableEntry, const PGGeomColumnDesc* psGeomColumnDesc) { OGRPGTableEntryAddGeomColumn(psTableEntry, psGeomColumnDesc->pszName, psGeomColumnDesc->pszGeomType, psGeomColumnDesc->nCoordDimension, psGeomColumnDesc->nSRID, psGeomColumnDesc->ePostgisType); } static void OGRPGFreeTableEntry(void * _psTableEntry) { PGTableEntry* psTableEntry = (PGTableEntry*)_psTableEntry; CPLFree(psTableEntry->pszTableName); CPLFree(psTableEntry->pszSchemaName); int i; for(i=0;i<psTableEntry->nGeomColumnCount;i++) { CPLFree(psTableEntry->pasGeomColumns[i].pszName); CPLFree(psTableEntry->pasGeomColumns[i].pszGeomType); } CPLFree(psTableEntry->pasGeomColumns); CPLFree(psTableEntry); } static PGTableEntry* OGRPGFindTableEntry(CPLHashSet* hSetTables, const char* pszTableName, const char* pszSchemaName) { PGTableEntry sEntry; sEntry.pszTableName = (char*) pszTableName; sEntry.pszSchemaName = (char*) pszSchemaName; return (PGTableEntry*) CPLHashSetLookup(hSetTables, &sEntry); } static PGTableEntry* OGRPGAddTableEntry(CPLHashSet* hSetTables, const char* pszTableName, const char* pszSchemaName) { PGTableEntry* psEntry = (PGTableEntry*) CPLCalloc(1, sizeof(PGTableEntry)); psEntry->pszTableName = CPLStrdup(pszTableName); psEntry->pszSchemaName = CPLStrdup(pszSchemaName); CPLHashSetInsert(hSetTables, psEntry); return psEntry; } /************************************************************************/ /* Open() */ /************************************************************************/ int OGRPGDataSource::Open( const char * pszNewName, int bUpdate, int bTestOpen ) { CPLAssert( nLayers == 0 ); /* -------------------------------------------------------------------- */ /* Verify postgresql prefix. */ /* -------------------------------------------------------------------- */ if( EQUALN(pszNewName,"PGB:",4) ) { bUseBinaryCursor = TRUE; CPLDebug("PG","BINARY cursor is used for geometry fetching"); } else if( !EQUALN(pszNewName,"PG:",3) ) { if( !bTestOpen ) CPLError( CE_Failure, CPLE_AppDefined, "%s does not conform to PostgreSQL naming convention," " PG:*\n", pszNewName ); return FALSE; } pszName = CPLStrdup( pszNewName ); char* pszConnectionName = CPLStrdup(pszName); /* -------------------------------------------------------------------- */ /* Determine if the connection string contains an optional */ /* ACTIVE_SCHEMA portion. If so, parse it out. */ /* -------------------------------------------------------------------- */ char *pszActiveSchemaStart; CPLString osActiveSchema; pszActiveSchemaStart = strstr(pszConnectionName, "active_schema="); if (pszActiveSchemaStart == NULL) pszActiveSchemaStart = strstr(pszConnectionName, "ACTIVE_SCHEMA="); if (pszActiveSchemaStart != NULL) { char *pszActiveSchema; const char *pszEnd = NULL; pszActiveSchema = CPLStrdup( pszActiveSchemaStart + strlen("active_schema=") ); pszEnd = strchr(pszActiveSchemaStart, ' '); if( pszEnd == NULL ) pszEnd = pszConnectionName + strlen(pszConnectionName); // Remove ACTIVE_SCHEMA=xxxxx from pszConnectionName string memmove( pszActiveSchemaStart, pszEnd, strlen(pszEnd) + 1 ); pszActiveSchema[pszEnd - pszActiveSchemaStart - strlen("active_schema=")] = '\0'; osActiveSchema = pszActiveSchema; CPLFree(pszActiveSchema); } else { osActiveSchema = "public"; } /* -------------------------------------------------------------------- */ /* Determine if the connection string contains an optional */ /* SCHEMAS portion. If so, parse it out. */ /* -------------------------------------------------------------------- */ char *pszSchemasStart; char **papszSchemaList = NULL; pszSchemasStart = strstr(pszConnectionName, "schemas="); if (pszSchemasStart == NULL) pszSchemasStart = strstr(pszConnectionName, "SCHEMAS="); if (pszSchemasStart != NULL) { char *pszSchemas; const char *pszEnd = NULL; pszSchemas = CPLStrdup( pszSchemasStart + strlen("schemas=") ); pszEnd = strchr(pszSchemasStart, ' '); if( pszEnd == NULL ) pszEnd = pszConnectionName + strlen(pszConnectionName); // Remove SCHEMAS=xxxxx from pszConnectionName string memmove( pszSchemasStart, pszEnd, strlen(pszEnd) + 1 ); pszSchemas[pszEnd - pszSchemasStart - strlen("schemas=")] = '\0'; papszSchemaList = CSLTokenizeString2( pszSchemas, ",", 0 ); CPLFree(pszSchemas); /* If there is only one schema specified, make it the active schema */ if (CSLCount(papszSchemaList) == 1) { osActiveSchema = papszSchemaList[0]; } } /* -------------------------------------------------------------------- */ /* Determine if the connection string contains an optional */ /* TABLES portion. If so, parse it out. The expected */ /* connection string in this case will be, e.g.: */ /* */ /* 'PG:dbname=warmerda user=warmerda tables=s1.t1,[s2.t2,...] */ /* - where sN is schema and tN is table name */ /* We must also strip this information from the connection */ /* string; PQconnectdb() does not like unknown directives */ /* -------------------------------------------------------------------- */ PGTableEntry **papsTables = NULL; int nTableCount = 0; char *pszTableStart; pszTableStart = strstr(pszConnectionName, "tables="); if (pszTableStart == NULL) pszTableStart = strstr(pszConnectionName, "TABLES="); if( pszTableStart != NULL ) { char **papszTableList; char *pszTableSpec; const char *pszEnd = NULL; int i; pszTableSpec = CPLStrdup( pszTableStart + 7 ); pszEnd = strchr(pszTableStart, ' '); if( pszEnd == NULL ) pszEnd = pszConnectionName + strlen(pszConnectionName); // Remove TABLES=xxxxx from pszConnectionName string memmove( pszTableStart, pszEnd, strlen(pszEnd) + 1 ); pszTableSpec[pszEnd - pszTableStart - 7] = '\0'; papszTableList = CSLTokenizeString2( pszTableSpec, ",", 0 ); for( i = 0; i < CSLCount(papszTableList); i++ ) { char **papszQualifiedParts; // Get schema and table name papszQualifiedParts = CSLTokenizeString2( papszTableList[i], ".", 0 ); int nParts = CSLCount( papszQualifiedParts ); if( nParts == 1 || nParts == 2 ) { /* Find the geometry column name if specified */ char* pszGeomColumnName = NULL; char* pos = strchr(papszQualifiedParts[CSLCount( papszQualifiedParts ) - 1], '('); if (pos != NULL) { *pos = '\0'; pszGeomColumnName = pos+1; int len = strlen(pszGeomColumnName); if (len > 0) pszGeomColumnName[len - 1] = '\0'; } papsTables = (PGTableEntry**)CPLRealloc(papsTables, sizeof(PGTableEntry*) * (nTableCount + 1)); papsTables[nTableCount] = (PGTableEntry*) CPLCalloc(1, sizeof(PGTableEntry)); if (pszGeomColumnName) OGRPGTableEntryAddGeomColumn(papsTables[nTableCount], pszGeomColumnName); if( nParts == 2 ) { papsTables[nTableCount]->pszSchemaName = CPLStrdup( papszQualifiedParts[0] ); papsTables[nTableCount]->pszTableName = CPLStrdup( papszQualifiedParts[1] ); } else { papsTables[nTableCount]->pszSchemaName = CPLStrdup( osActiveSchema.c_str()); papsTables[nTableCount]->pszTableName = CPLStrdup( papszQualifiedParts[0] ); } nTableCount ++; } CSLDestroy(papszQualifiedParts); } CSLDestroy(papszTableList); CPLFree(pszTableSpec); } CPLString osCurrentSchema; CPLHashSet *hSetTables = NULL; int bRet = FALSE; int bListAllTables = CSLTestBoolean(CPLGetConfigOption("PG_LIST_ALL_TABLES", "NO")); PGresult *hResult = NULL; std::set<CPLString> osRegisteredLayers; /* -------------------------------------------------------------------- */ /* Try to establish connection. */ /* -------------------------------------------------------------------- */ hPGConn = PQconnectdb( pszConnectionName + (bUseBinaryCursor ? 4 : 3) ); CPLFree(pszConnectionName); pszConnectionName = NULL; if( hPGConn == NULL || PQstatus(hPGConn) == CONNECTION_BAD ) { CPLError( CE_Failure, CPLE_AppDefined, "PQconnectdb failed.\n%s", PQerrorMessage(hPGConn) ); PQfinish(hPGConn); hPGConn = NULL; goto end; } bDSUpdate = bUpdate; /* -------------------------------------------------------------------- */ /* Set the encoding to UTF8 as the driver advertizes UTF8 */ /* unless PGCLIENTENCODING is defined */ /* -------------------------------------------------------------------- */ if (CPLGetConfigOption("PGCLIENTENCODING", NULL) == NULL) { const char* encoding = "UNICODE"; if (PQsetClientEncoding(hPGConn, encoding) == -1) { CPLError( CE_Warning, CPLE_AppDefined, "PQsetClientEncoding(%s) failed.\n%s", encoding, PQerrorMessage( hPGConn ) ); } } /* -------------------------------------------------------------------- */ /* Install a notice processor. */ /* -------------------------------------------------------------------- */ PQsetNoticeProcessor( hPGConn, OGRPGNoticeProcessor, this ); /* -------------------------------------------------------------------- */ /* Try to establish the database name from the connection */ /* string passed. */ /* -------------------------------------------------------------------- */ if( strstr(pszNewName, "dbname=") != NULL ) { pszDBName = CPLStrdup( strstr(pszNewName, "dbname=") + 7 ); for( int i = 0; pszDBName[i] != '\0'; i++ ) { if( pszDBName[i] == ' ' ) { pszDBName[i] = '\0'; break; } } } else if( getenv( "USER" ) != NULL ) pszDBName = CPLStrdup( getenv("USER") ); else pszDBName = CPLStrdup( "unknown_dbname" ); CPLDebug( "PG", "DBName=\"%s\"", pszDBName ); /* -------------------------------------------------------------------- */ /* Set active schema if different from 'public' */ /* -------------------------------------------------------------------- */ if (strcmp(osActiveSchema, "public") != 0) { CPLString osCommand; osCommand.Printf("SET search_path='%s',public", osActiveSchema.c_str()); PGresult *hResult = OGRPG_PQexec(hPGConn, osCommand ); if( !hResult || PQresultStatus(hResult) != PGRES_COMMAND_OK ) { OGRPGClearResult( hResult ); CPLDebug("PG","Command \"%s\" failed. Trying without 'public'.",osCommand.c_str()); osCommand.Printf("SET search_path='%s'", osActiveSchema.c_str()); PGresult *hResult2 = OGRPG_PQexec(hPGConn, osCommand ); if( !hResult2 || PQresultStatus(hResult2) != PGRES_COMMAND_OK ) { OGRPGClearResult( hResult2 ); CPLError( CE_Failure, CPLE_AppDefined, "%s", PQerrorMessage(hPGConn) ); goto end; } } OGRPGClearResult(hResult); } /* -------------------------------------------------------------------- */ /* Find out PostgreSQL version */ /* -------------------------------------------------------------------- */ sPostgreSQLVersion.nMajor = -1; sPostgreSQLVersion.nMinor = -1; sPostgreSQLVersion.nRelease = -1; hResult = OGRPG_PQexec(hPGConn, "SELECT version()" ); if( hResult && PQresultStatus(hResult) == PGRES_TUPLES_OK && PQntuples(hResult) > 0 ) { const char* pszSpace; char * pszVer = PQgetvalue(hResult,0,0); CPLDebug("PG","PostgreSQL version string : '%s'", pszVer); /* Should work with "PostgreSQL X.Y.Z ..." or "EnterpriseDB X.Y.Z ..." */ pszSpace = strchr(pszVer, ' '); if( pszSpace != NULL && isdigit(pszSpace[1]) ) { OGRPGDecodeVersionString(&sPostgreSQLVersion, pszSpace + 1); if (sPostgreSQLVersion.nMajor == 7 && sPostgreSQLVersion.nMinor < 4) { /* We don't support BINARY CURSOR for PostgreSQL < 7.4. */ /* The binary protocol for arrays seems to be different from later versions */ CPLDebug("PG","BINARY cursor will finally NOT be used because version < 7.4"); bUseBinaryCursor = FALSE; } } } OGRPGClearResult(hResult); CPLAssert(NULL == hResult); /* Test if safe PQclear has not been broken */ /* -------------------------------------------------------------------- */ /* Test if standard_conforming_strings is recognized */ /* -------------------------------------------------------------------- */ hResult = OGRPG_PQexec(hPGConn, "SHOW standard_conforming_strings" ); if( hResult && PQresultStatus(hResult) == PGRES_TUPLES_OK && PQntuples(hResult) == 1 ) { /* Whatever the value is, it means that we can use the E'' */ /* syntax */ bUseEscapeStringSyntax = TRUE; } OGRPGClearResult(hResult); /* -------------------------------------------------------------------- */ /* Test if time binary format is int8 or float8 */ /* -------------------------------------------------------------------- */ #if !defined(PG_PRE74) if (bUseBinaryCursor) { SoftStartTransaction(); hResult = OGRPG_PQexec(hPGConn, "DECLARE gettimebinaryformat BINARY CURSOR FOR SELECT CAST ('00:00:01' AS time)"); if( hResult && PQresultStatus(hResult) == PGRES_COMMAND_OK ) { OGRPGClearResult( hResult ); hResult = OGRPG_PQexec(hPGConn, "FETCH ALL IN gettimebinaryformat" ); if( hResult && PQresultStatus(hResult) == PGRES_TUPLES_OK && PQntuples(hResult) == 1 ) { if ( PQfformat( hResult, 0 ) == 1 ) // Binary data representation { CPLAssert(PQgetlength(hResult, 0, 0) == 8); double dVal; unsigned int nVal[2]; memcpy( nVal, PQgetvalue( hResult, 0, 0 ), 8 ); CPL_MSBPTR32(&nVal[0]); CPL_MSBPTR32(&nVal[1]); memcpy( &dVal, PQgetvalue( hResult, 0, 0 ), 8 ); CPL_MSBPTR64(&dVal); if (nVal[0] == 0 && nVal[1] == 1000000) { bBinaryTimeFormatIsInt8 = TRUE; CPLDebug( "PG", "Time binary format is int8"); } else if (dVal == 1.) { bBinaryTimeFormatIsInt8 = FALSE; CPLDebug( "PG", "Time binary format is float8"); } else { bBinaryTimeFormatIsInt8 = FALSE; CPLDebug( "PG", "Time binary format is unknown"); } } } } OGRPGClearResult( hResult ); hResult = OGRPG_PQexec(hPGConn, "CLOSE gettimebinaryformat"); OGRPGClearResult( hResult ); SoftCommit(); } #endif #ifdef notdef /* This would be the quickest fix... instead, ogrpglayer has been updated to support */ /* bytea hex format */ if (sPostgreSQLVersion.nMajor >= 9) { /* Starting with PostgreSQL 9.0, the default output format for values of type bytea */ /* is hex, whereas we traditionnaly expect escape */ hResult = OGRPG_PQexec(hPGConn, "SET bytea_output TO escape"); OGRPGClearResult( hResult ); } #endif /* -------------------------------------------------------------------- */ /* Test to see if this database instance has support for the */ /* PostGIS Geometry type. If so, disable sequential scanning */ /* so we will get the value of the gist indexes. */ /* -------------------------------------------------------------------- */ hResult = OGRPG_PQexec(hPGConn, "SELECT oid, typname FROM pg_type WHERE typname IN ('geometry', 'geography')" ); if( hResult && PQresultStatus(hResult) == PGRES_TUPLES_OK && PQntuples(hResult) > 0 && CSLTestBoolean(CPLGetConfigOption("PG_USE_POSTGIS", "YES"))) { for( int iRecord = 0; iRecord < PQntuples(hResult); iRecord++ ) { const char *pszOid = PQgetvalue(hResult, iRecord, 0); const char *pszTypname = PQgetvalue(hResult, iRecord, 1); if( EQUAL(pszTypname, "geometry") ) { bHavePostGIS = TRUE; nGeometryOID = atoi(pszOid); } else if( CSLTestBoolean(CPLGetConfigOption("PG_USE_GEOGRAPHY", "YES")) ) { bHaveGeography = TRUE; nGeographyOID = atoi(pszOid); } } } OGRPGClearResult( hResult ); /* -------------------------------------------------------------------- */ /* Find out PostGIS version */ /* -------------------------------------------------------------------- */ sPostGISVersion.nMajor = -1; sPostGISVersion.nMinor = -1; sPostGISVersion.nRelease = -1; if( bHavePostGIS ) { hResult = OGRPG_PQexec(hPGConn, "SELECT postgis_version()" ); if( hResult && PQresultStatus(hResult) == PGRES_TUPLES_OK && PQntuples(hResult) > 0 ) { char * pszVer = PQgetvalue(hResult,0,0); CPLDebug("PG","PostGIS version string : '%s'", pszVer); OGRPGDecodeVersionString(&sPostGISVersion, pszVer); } OGRPGClearResult(hResult); if (sPostGISVersion.nMajor == 0 && sPostGISVersion.nMinor < 8) { // Turning off sequential scans for PostGIS < 0.8 hResult = OGRPG_PQexec(hPGConn, "SET ENABLE_SEQSCAN = OFF"); CPLDebug( "PG", "SET ENABLE_SEQSCAN=OFF" ); } else { // PostGIS >=0.8 is correctly integrated with query planner, // thus PostgreSQL will use indexes whenever appropriate. hResult = OGRPG_PQexec(hPGConn, "SET ENABLE_SEQSCAN = ON"); } OGRPGClearResult( hResult ); } /* -------------------------------------------------------------------- */ /* Find out "unknown SRID" value */ /* -------------------------------------------------------------------- */ if (sPostGISVersion.nMajor >= 2) { hResult = OGRPG_PQexec(hPGConn, "SELECT ST_Srid('POINT EMPTY'::GEOMETRY)" ); if( hResult && PQresultStatus(hResult) == PGRES_TUPLES_OK && PQntuples(hResult) > 0) { nUndefinedSRID = atoi(PQgetvalue(hResult,0,0)); } OGRPGClearResult( hResult ); } else nUndefinedSRID = -1; /* -------------------------------------------------------------------- */ /* Get a list of available tables if they have not been */ /* specified through the TABLES connection string param */ /* -------------------------------------------------------------------- */ hSetTables = CPLHashSetNew(OGRPGHashTableEntry, OGRPGEqualTableEntry, OGRPGFreeTableEntry); if (nTableCount == 0) { CPLString osCommand; const char* pszAllowedRelations; if( CSLTestBoolean(CPLGetConfigOption("PG_SKIP_VIEWS", "NO")) ) pszAllowedRelations = "'r'"; else pszAllowedRelations = "'r','v'"; hResult = OGRPG_PQexec(hPGConn, "BEGIN"); if( hResult && PQresultStatus(hResult) == PGRES_COMMAND_OK ) { OGRPGClearResult( hResult ); /* Caution : in PostGIS case, the result has 3 columns, whereas in the */ /* non-PostGIS case it has only 2 columns */ if ( bHavePostGIS && !bListAllTables ) { osCommand.Printf("DECLARE mycursor CURSOR for " "SELECT c.relname, n.nspname, c.relkind, g.f_geometry_column, g.type, g.coord_dimension, g.srid, %d, c.oid as oid, a.attnum as attnum FROM pg_class c, pg_namespace n, geometry_columns g, pg_attribute a " "WHERE (c.relkind in (%s) AND c.relname !~ '^pg_' AND c.relnamespace=n.oid " "AND c.relname::TEXT = g.f_table_name::TEXT AND n.nspname = g.f_table_schema AND a.attname = g.f_geometry_column AND a.attrelid = c.oid) ", GEOM_TYPE_GEOMETRY, pszAllowedRelations); if (bHaveGeography) osCommand += CPLString().Printf( "UNION SELECT c.relname, n.nspname, c.relkind, g.f_geography_column, g.type, g.coord_dimension, g.srid, %d, c.oid as oid, a.attnum as attnum FROM pg_class c, pg_namespace n, geography_columns g, pg_attribute a " "WHERE (c.relkind in (%s) AND c.relname !~ '^pg_' AND c.relnamespace=n.oid " "AND c.relname::TEXT = g.f_table_name::TEXT AND n.nspname = g.f_table_schema AND a.attname = g.f_geography_column AND a.attrelid = c.oid)", GEOM_TYPE_GEOGRAPHY, pszAllowedRelations); osCommand += " ORDER BY oid, attnum"; } else osCommand.Printf("DECLARE mycursor CURSOR for " "SELECT c.relname, n.nspname, c.relkind FROM pg_class c, pg_namespace n " "WHERE (c.relkind in (%s) AND c.relname !~ '^pg_' AND c.relnamespace=n.oid)", pszAllowedRelations); hResult = OGRPG_PQexec(hPGConn, osCommand.c_str()); } if( hResult && PQresultStatus(hResult) == PGRES_COMMAND_OK ) { OGRPGClearResult( hResult ); hResult = OGRPG_PQexec(hPGConn, "FETCH ALL in mycursor" ); } if( !hResult || PQresultStatus(hResult) != PGRES_TUPLES_OK ) { OGRPGClearResult( hResult ); CPLError( CE_Failure, CPLE_AppDefined, "%s", PQerrorMessage(hPGConn) ); goto end; } /* -------------------------------------------------------------------- */ /* Parse the returned table list */ /* -------------------------------------------------------------------- */ for( int iRecord = 0; iRecord < PQntuples(hResult); iRecord++ ) { const char *pszTable = PQgetvalue(hResult, iRecord, 0); const char *pszSchemaName = PQgetvalue(hResult, iRecord, 1); const char *pszRelkind = PQgetvalue(hResult, iRecord, 2); const char *pszGeomColumnName = NULL; const char *pszGeomType = NULL; int nGeomCoordDimension = 0; int nSRID = 0; PostgisType ePostgisType = GEOM_TYPE_UNKNOWN; if (bHavePostGIS && !bListAllTables) { pszGeomColumnName = PQgetvalue(hResult, iRecord, 3); pszGeomType = PQgetvalue(hResult, iRecord, 4); nGeomCoordDimension = atoi(PQgetvalue(hResult, iRecord, 5)); nSRID = atoi(PQgetvalue(hResult, iRecord, 6)); ePostgisType = (PostgisType) atoi(PQgetvalue(hResult, iRecord, 7)); /* We cannot reliably find geometry columns of a view that is */ /* based on a table that inherits from another one, wit that */ /* method, so give up, and let OGRPGTableLayer::ReadTableDefinition() */ /* do the job */ if( pszRelkind[0] == 'v' && sPostGISVersion.nMajor < 2 ) pszGeomColumnName = NULL; } if( EQUAL(pszTable,"spatial_ref_sys") || EQUAL(pszTable,"geometry_columns") || EQUAL(pszTable,"geography_columns") ) continue; if( EQUAL(pszSchemaName,"information_schema") ) continue; papsTables = (PGTableEntry**)CPLRealloc(papsTables, sizeof(PGTableEntry*) * (nTableCount + 1)); papsTables[nTableCount] = (PGTableEntry*) CPLCalloc(1, sizeof(PGTableEntry)); papsTables[nTableCount]->pszTableName = CPLStrdup( pszTable ); papsTables[nTableCount]->pszSchemaName = CPLStrdup( pszSchemaName ); if (pszGeomColumnName) OGRPGTableEntryAddGeomColumn(papsTables[nTableCount], pszGeomColumnName, pszGeomType, nGeomCoordDimension, nSRID, ePostgisType); nTableCount ++; PGTableEntry* psEntry = OGRPGFindTableEntry(hSetTables, pszTable, pszSchemaName); if (psEntry == NULL) psEntry = OGRPGAddTableEntry(hSetTables, pszTable, pszSchemaName); if (pszGeomColumnName) OGRPGTableEntryAddGeomColumn(psEntry, pszGeomColumnName, pszGeomType, nGeomCoordDimension, nSRID, ePostgisType); } /* -------------------------------------------------------------------- */ /* Cleanup */ /* -------------------------------------------------------------------- */ OGRPGClearResult( hResult ); hResult = OGRPG_PQexec(hPGConn, "CLOSE mycursor"); OGRPGClearResult( hResult ); hResult = OGRPG_PQexec(hPGConn, "COMMIT"); OGRPGClearResult( hResult ); /* With PostGIS 2.0, we don't need to query base tables of inherited */ /* tables */ if ( bHavePostGIS && !bListAllTables && sPostGISVersion.nMajor < 2 ) { hResult = OGRPG_PQexec(hPGConn, "BEGIN"); OGRPGClearResult( hResult ); /* -------------------------------------------------------------------- */ /* Fetch inherited tables */ /* -------------------------------------------------------------------- */ hResult = OGRPG_PQexec(hPGConn, "DECLARE mycursor CURSOR for " "SELECT c1.relname AS derived, c2.relname AS parent, n.nspname " "FROM pg_class c1, pg_class c2, pg_namespace n, pg_inherits i " "WHERE i.inhparent = c2.oid AND i.inhrelid = c1.oid AND c1.relnamespace=n.oid " "AND c1.relkind in ('r', 'v') AND c1.relnamespace=n.oid AND c2.relkind in ('r','v') " "AND c2.relname !~ '^pg_' AND c2.relnamespace=n.oid"); if( hResult && PQresultStatus(hResult) == PGRES_COMMAND_OK ) { OGRPGClearResult( hResult ); hResult = OGRPG_PQexec(hPGConn, "FETCH ALL in mycursor" ); } if( !hResult || PQresultStatus(hResult) != PGRES_TUPLES_OK ) { OGRPGClearResult( hResult ); CPLError( CE_Failure, CPLE_AppDefined, "%s", PQerrorMessage(hPGConn) ); goto end; } /* -------------------------------------------------------------------- */ /* Parse the returned table list */ /* -------------------------------------------------------------------- */ int bHasDoneSomething; do { /* Iterate over the tuples while we have managed to resolved at least one */ /* table to its table parent with a geometry */ /* For example if we have C inherits B and B inherits A, where A is a base table with a geometry */ /* The first pass will add B to the set of tables */ /* The second pass will add C to the set of tables */ bHasDoneSomething = FALSE; for( int iRecord = 0; iRecord < PQntuples(hResult); iRecord++ ) { const char *pszTable = PQgetvalue(hResult, iRecord, 0); const char *pszParentTable = PQgetvalue(hResult, iRecord, 1); const char *pszSchemaName = PQgetvalue(hResult, iRecord, 2); PGTableEntry* psEntry = OGRPGFindTableEntry(hSetTables, pszTable, pszSchemaName); /* We must be careful that a derived table can have its own geometry column(s) */ /* and some inherited from another table */ if (psEntry == NULL || psEntry->bDerivedInfoAdded == FALSE) { PGTableEntry* psParentEntry = OGRPGFindTableEntry(hSetTables, pszParentTable, pszSchemaName); if (psParentEntry != NULL) { /* The parent table of this table is already in the set, so we */ /* can now add the table in the set if it was not in it already */ bHasDoneSomething = TRUE; if (psEntry == NULL) psEntry = OGRPGAddTableEntry(hSetTables, pszTable, pszSchemaName); int iGeomColumn; for(iGeomColumn = 0; iGeomColumn < psParentEntry->nGeomColumnCount; iGeomColumn++) { papsTables = (PGTableEntry**)CPLRealloc(papsTables, sizeof(PGTableEntry*) * (nTableCount + 1)); papsTables[nTableCount] = (PGTableEntry*) CPLCalloc(1, sizeof(PGTableEntry)); papsTables[nTableCount]->pszTableName = CPLStrdup( pszTable ); papsTables[nTableCount]->pszSchemaName = CPLStrdup( pszSchemaName ); OGRPGTableEntryAddGeomColumn(papsTables[nTableCount], &psParentEntry->pasGeomColumns[iGeomColumn]); nTableCount ++; OGRPGTableEntryAddGeomColumn(psEntry, &psParentEntry->pasGeomColumns[iGeomColumn]); } psEntry->bDerivedInfoAdded = TRUE; } } } } while(bHasDoneSomething); /* -------------------------------------------------------------------- */ /* Cleanup */ /* -------------------------------------------------------------------- */ OGRPGClearResult( hResult ); hResult = OGRPG_PQexec(hPGConn, "CLOSE mycursor"); OGRPGClearResult( hResult ); hResult = OGRPG_PQexec(hPGConn, "COMMIT"); OGRPGClearResult( hResult ); } } osCurrentSchema = GetCurrentSchema(); /* -------------------------------------------------------------------- */ /* Register the available tables. */ /* -------------------------------------------------------------------- */ for( int iRecord = 0; iRecord < nTableCount; iRecord++ ) { PGTableEntry* psEntry; CPLString osDefnName; psEntry = (PGTableEntry* )CPLHashSetLookup(hSetTables, papsTables[iRecord]); /* If SCHEMAS= is specified, only take into account tables inside */ /* one of the specified schemas */ if (papszSchemaList != NULL && CSLFindString(papszSchemaList, papsTables[iRecord]->pszSchemaName) == -1) { continue; } if ( papsTables[iRecord]->pszSchemaName && osCurrentSchema != papsTables[iRecord]->pszSchemaName ) { osDefnName.Printf("%s.%s", papsTables[iRecord]->pszSchemaName, papsTables[iRecord]->pszTableName ); } else { //no prefix for current_schema in layer name, for backwards compatibility osDefnName = papsTables[iRecord]->pszTableName; } if( osRegisteredLayers.find( osDefnName ) != osRegisteredLayers.end() ) continue; osRegisteredLayers.insert( osDefnName ); OGRPGTableLayer* poLayer; poLayer = OpenTable( osCurrentSchema, papsTables[iRecord]->pszTableName, papsTables[iRecord]->pszSchemaName, NULL, bUpdate, FALSE ); if( psEntry != NULL ) { if( psEntry->nGeomColumnCount > 0 ) { poLayer->SetGeometryInformation(psEntry->pasGeomColumns, psEntry->nGeomColumnCount); } } else { if( papsTables[iRecord]->nGeomColumnCount > 0 ) { poLayer->SetGeometryInformation(papsTables[iRecord]->pasGeomColumns, papsTables[iRecord]->nGeomColumnCount); } } } bRet = TRUE; end: if (hSetTables) CPLHashSetDestroy(hSetTables); CSLDestroy( papszSchemaList ); for(int i=0;i<nTableCount;i++) OGRPGFreeTableEntry(papsTables[i]); CPLFree(papsTables); return bRet; } /************************************************************************/ /* OpenTable() */ /************************************************************************/ OGRPGTableLayer* OGRPGDataSource::OpenTable( CPLString& osCurrentSchema, const char *pszNewName, const char *pszSchemaName, const char * pszGeomColumnForced, int bUpdate, int bTestOpen) { /* -------------------------------------------------------------------- */ /* Create the layer object. */ /* -------------------------------------------------------------------- */ OGRPGTableLayer *poLayer; poLayer = new OGRPGTableLayer( this, osCurrentSchema, pszNewName, pszSchemaName, pszGeomColumnForced, bUpdate ); if( bTestOpen && !(poLayer->ReadTableDefinition()) ) { delete poLayer; return NULL; } /* -------------------------------------------------------------------- */ /* Add layer to data source layer list. */ /* -------------------------------------------------------------------- */ papoLayers = (OGRPGTableLayer **) CPLRealloc( papoLayers, sizeof(OGRPGTableLayer *) * (nLayers+1) ); papoLayers[nLayers++] = poLayer; return poLayer; } /************************************************************************/ /* DeleteLayer() */ /************************************************************************/ int OGRPGDataSource::DeleteLayer( int iLayer ) { if( iLayer < 0 || iLayer >= nLayers ) return OGRERR_FAILURE; /* -------------------------------------------------------------------- */ /* Blow away our OGR structures related to the layer. This is */ /* pretty dangerous if anything has a reference to this layer! */ /* -------------------------------------------------------------------- */ CPLString osLayerName = papoLayers[iLayer]->GetLayerDefn()->GetName(); CPLString osTableName = papoLayers[iLayer]->GetTableName(); CPLString osSchemaName = papoLayers[iLayer]->GetSchemaName(); CPLDebug( "PG", "DeleteLayer(%s)", osLayerName.c_str() ); delete papoLayers[iLayer]; memmove( papoLayers + iLayer, papoLayers + iLayer + 1, sizeof(void *) * (nLayers - iLayer - 1) ); nLayers--; if (osLayerName.size() == 0) return OGRERR_NONE; /* -------------------------------------------------------------------- */ /* Remove from the database. */ /* -------------------------------------------------------------------- */ PGresult *hResult; CPLString osCommand; hResult = OGRPG_PQexec(hPGConn, "BEGIN"); OGRPGClearResult( hResult ); if( bHavePostGIS && sPostGISVersion.nMajor < 2) { /* This is unnecessary if the layer is not a geometry table, or an inherited geometry table */ /* but it shouldn't hurt */ osCommand.Printf( "DELETE FROM geometry_columns WHERE f_table_name='%s' and f_table_schema='%s'", osTableName.c_str(), osSchemaName.c_str() ); hResult = OGRPG_PQexec( hPGConn, osCommand.c_str() ); OGRPGClearResult( hResult ); } osCommand.Printf("DROP TABLE %s.%s CASCADE", OGRPGEscapeColumnName(osSchemaName).c_str(), OGRPGEscapeColumnName(osTableName).c_str() ); hResult = OGRPG_PQexec( hPGConn, osCommand.c_str() ); OGRPGClearResult( hResult ); hResult = OGRPG_PQexec(hPGConn, "COMMIT"); OGRPGClearResult( hResult ); return OGRERR_NONE; } /************************************************************************/ /* CreateLayer() */ /************************************************************************/ OGRLayer * OGRPGDataSource::CreateLayer( const char * pszLayerName, OGRSpatialReference *poSRS, OGRwkbGeometryType eType, char ** papszOptions ) { PGresult *hResult = NULL; CPLString osCommand; const char *pszGeomType = NULL; char *pszTableName = NULL; char *pszSchemaName = NULL; int nDimension = 3; if (pszLayerName == NULL) return NULL; const char* pszFIDColumnName = CSLFetchNameValue(papszOptions, "FID"); CPLString osFIDColumnName; if (pszFIDColumnName == NULL) osFIDColumnName = "OGC_FID"; else { if( CSLFetchBoolean(papszOptions,"LAUNDER", TRUE) ) { char* pszLaunderedFid = LaunderName(pszFIDColumnName); osFIDColumnName += OGRPGEscapeColumnName(pszLaunderedFid); CPLFree(pszLaunderedFid); } else osFIDColumnName += OGRPGEscapeColumnName(pszFIDColumnName); } pszFIDColumnName = osFIDColumnName.c_str(); if (strncmp(pszLayerName, "pg", 2) == 0) { CPLError(CE_Warning, CPLE_AppDefined, "The layer name should not begin by 'pg' as it is a reserved prefix"); } if( wkbFlatten(eType) == eType ) nDimension = 2; int nForcedDimension = -1; if( CSLFetchNameValue( papszOptions, "DIM") != NULL ) { nDimension = atoi(CSLFetchNameValue( papszOptions, "DIM")); nForcedDimension = nDimension; } /* Should we turn layers with None geometry type as Unknown/GEOMETRY */ /* so they are still recorded in geometry_columns table ? (#4012) */ int bNoneAsUnknown = CSLTestBoolean(CSLFetchNameValueDef( papszOptions, "NONE_AS_UNKNOWN", "NO")); if (bNoneAsUnknown && eType == wkbNone) eType = wkbUnknown; int bExtractSchemaFromLayerName = CSLTestBoolean(CSLFetchNameValueDef( papszOptions, "EXTRACT_SCHEMA_FROM_LAYER_NAME", "YES")); /* Postgres Schema handling: Extract schema name from input layer name or passed with -lco SCHEMA. Set layer name to "schema.table" or to "table" if schema == current_schema() Usage without schema name is backwards compatible */ const char* pszDotPos = strstr(pszLayerName,"."); if ( pszDotPos != NULL && bExtractSchemaFromLayerName ) { int length = pszDotPos - pszLayerName; pszSchemaName = (char*)CPLMalloc(length+1); strncpy(pszSchemaName, pszLayerName, length); pszSchemaName[length] = '\0'; if( CSLFetchBoolean(papszOptions,"LAUNDER", TRUE) ) pszTableName = LaunderName( pszDotPos + 1 ); //skip "." else pszTableName = CPLStrdup( pszDotPos + 1 ); //skip "." } else { pszSchemaName = NULL; if( CSLFetchBoolean(papszOptions,"LAUNDER", TRUE) ) pszTableName = LaunderName( pszLayerName ); //skip "." else pszTableName = CPLStrdup( pszLayerName ); //skip "." } /* -------------------------------------------------------------------- */ /* Set the default schema for the layers. */ /* -------------------------------------------------------------------- */ if( CSLFetchNameValue( papszOptions, "SCHEMA" ) != NULL ) { CPLFree(pszSchemaName); pszSchemaName = CPLStrdup(CSLFetchNameValue( papszOptions, "SCHEMA" )); } if ( pszSchemaName == NULL ) { pszSchemaName = CPLStrdup(osCurrentSchema); } /* -------------------------------------------------------------------- */ /* Do we already have this layer? If so, should we blow it */ /* away? */ /* -------------------------------------------------------------------- */ int iLayer; FlushSoftTransaction(); CPLString osSQLLayerName; if (pszSchemaName == NULL || (strlen(osCurrentSchema) > 0 && EQUAL(pszSchemaName, osCurrentSchema.c_str()))) osSQLLayerName = pszTableName; else { osSQLLayerName = pszSchemaName; osSQLLayerName += "."; osSQLLayerName += pszTableName; } /* GetLayerByName() can instanciate layers that would have been */ /* 'hidden' otherwise, for example, non-spatial tables in a */ /* Postgis-enabled database, so this apparently useless command is */ /* not useless... (#4012) */ CPLPushErrorHandler(CPLQuietErrorHandler); GetLayerByName(osSQLLayerName); CPLPopErrorHandler(); CPLErrorReset(); for( iLayer = 0; iLayer < nLayers; iLayer++ ) { if( EQUAL(osSQLLayerName.c_str(),papoLayers[iLayer]->GetName()) ) { if( CSLFetchNameValue( papszOptions, "OVERWRITE" ) != NULL && !EQUAL(CSLFetchNameValue(papszOptions,"OVERWRITE"),"NO") ) { DeleteLayer( iLayer ); } else { CPLError( CE_Failure, CPLE_AppDefined, "Layer %s already exists, CreateLayer failed.\n" "Use the layer creation option OVERWRITE=YES to " "replace it.", osSQLLayerName.c_str() ); CPLFree( pszTableName ); CPLFree( pszSchemaName ); return NULL; } } } /* -------------------------------------------------------------------- */ /* Handle the GEOM_TYPE option. */ /* -------------------------------------------------------------------- */ pszGeomType = CSLFetchNameValue( papszOptions, "GEOM_TYPE" ); if( pszGeomType == NULL ) { if( bHavePostGIS ) pszGeomType = "geometry"; else pszGeomType = "bytea"; } if( eType != wkbNone && EQUAL(pszGeomType, "geography") && !bHaveGeography ) { CPLError( CE_Failure, CPLE_AppDefined, "GEOM_TYPE=geography is only supported in PostGIS >= 1.5.\n" "Creation of layer %s has failed.", pszLayerName ); CPLFree( pszTableName ); CPLFree( pszSchemaName ); return NULL; } if( eType != wkbNone && bHavePostGIS && !EQUAL(pszGeomType,"geometry") && !EQUAL(pszGeomType, "geography") ) { if( bHaveGeography ) CPLError( CE_Failure, CPLE_AppDefined, "GEOM_TYPE in PostGIS enabled databases must be 'geometry' or 'geography'.\n" "Creation of layer %s with GEOM_TYPE %s has failed.", pszLayerName, pszGeomType ); else CPLError( CE_Failure, CPLE_AppDefined, "GEOM_TYPE in PostGIS enabled databases must be 'geometry'.\n" "Creation of layer %s with GEOM_TYPE %s has failed.", pszLayerName, pszGeomType ); CPLFree( pszTableName ); CPLFree( pszSchemaName ); return NULL; } /* -------------------------------------------------------------------- */ /* Try to get the SRS Id of this spatial reference system, */ /* adding tot the srs table if needed. */ /* -------------------------------------------------------------------- */ int nSRSId = nUndefinedSRID; if( poSRS != NULL ) nSRSId = FetchSRSId( poSRS ); const char *pszGeometryType = OGRToOGCGeomType(eType); /* -------------------------------------------------------------------- */ /* Create a basic table with the FID. Also include the */ /* geometry if this is not a PostGIS enabled table. */ /* -------------------------------------------------------------------- */ hResult = OGRPG_PQexec(hPGConn, "BEGIN"); OGRPGClearResult( hResult ); const char *pszGFldName = NULL; CPLString osCreateTable; int bTemporary = CSLFetchNameValue( papszOptions, "TEMPORARY" ) != NULL && CSLTestBoolean(CSLFetchNameValue( papszOptions, "TEMPORARY" )); if (bTemporary) { CPLFree(pszSchemaName); pszSchemaName = CPLStrdup("pg_temp_1"); osCreateTable.Printf("CREATE TEMPORARY TABLE %s", OGRPGEscapeColumnName(pszTableName).c_str()); } else osCreateTable.Printf("CREATE TABLE %s.%s", OGRPGEscapeColumnName(pszSchemaName).c_str(), OGRPGEscapeColumnName(pszTableName).c_str()); if( eType != wkbNone && !bHavePostGIS ) { osCommand.Printf( "%s ( " " %s SERIAL, " " WKB_GEOMETRY %s, " " PRIMARY KEY (%s) )", osCreateTable.c_str(), pszFIDColumnName, pszGeomType, pszFIDColumnName); } else if ( eType != wkbNone && EQUAL(pszGeomType, "geography") ) { if( CSLFetchNameValue( papszOptions, "GEOMETRY_NAME") != NULL ) pszGFldName = CSLFetchNameValue( papszOptions, "GEOMETRY_NAME"); else pszGFldName = "the_geog"; if (nSRSId) osCommand.Printf( "%s ( %s SERIAL, %s geography(%s%s,%d), PRIMARY KEY (%s) )", osCreateTable.c_str(), pszFIDColumnName, OGRPGEscapeColumnName(pszGFldName).c_str(), pszGeometryType, nDimension == 2 ? "" : "Z", nSRSId, pszFIDColumnName); else osCommand.Printf( "%s ( %s SERIAL, %s geography(%s%s), PRIMARY KEY (%s) )", osCreateTable.c_str(), pszFIDColumnName, OGRPGEscapeColumnName(pszGFldName).c_str(), pszGeometryType, nDimension == 2 ? "" : "Z", pszFIDColumnName); } else { osCommand.Printf( "%s ( %s SERIAL, PRIMARY KEY (%s) )", osCreateTable.c_str(), pszFIDColumnName, pszFIDColumnName ); } hResult = OGRPG_PQexec(hPGConn, osCommand.c_str()); if( PQresultStatus(hResult) != PGRES_COMMAND_OK ) { CPLError( CE_Failure, CPLE_AppDefined, "%s\n%s", osCommand.c_str(), PQerrorMessage(hPGConn) ); CPLFree( pszTableName ); CPLFree( pszSchemaName ); OGRPGClearResult( hResult ); hResult = OGRPG_PQexec( hPGConn, "ROLLBACK" ); OGRPGClearResult( hResult ); return NULL; } OGRPGClearResult( hResult ); CPLString osEscapedTableNameSingleQuote = OGRPGEscapeString(hPGConn, pszTableName); const char* pszEscapedTableNameSingleQuote = osEscapedTableNameSingleQuote.c_str(); CPLString osEscapedSchemaNameSingleQuote = OGRPGEscapeString(hPGConn, pszSchemaName); const char* pszEscapedSchemaNameSingleQuote = osEscapedSchemaNameSingleQuote.c_str(); /* -------------------------------------------------------------------- */ /* Eventually we should be adding this table to a table of */ /* "geometric layers", capturing the WKT projection, and */ /* perhaps some other housekeeping. */ /* -------------------------------------------------------------------- */ if( eType != wkbNone && bHavePostGIS && !EQUAL(pszGeomType, "geography")) { if( CSLFetchNameValue( papszOptions, "GEOMETRY_NAME") != NULL ) pszGFldName = CSLFetchNameValue( papszOptions, "GEOMETRY_NAME"); else pszGFldName = "wkb_geometry"; if (sPostGISVersion.nMajor <= 1) { /* Sometimes there is an old cruft entry in the geometry_columns * table if things were not properly cleaned up before. We make * an effort to clean out such cruft. * Note: PostGIS 2.0 defines geometry_columns as a view (no clean up is needed) */ osCommand.Printf( "DELETE FROM geometry_columns WHERE f_table_name = %s AND f_table_schema = %s", pszEscapedTableNameSingleQuote, pszEscapedSchemaNameSingleQuote ); hResult = OGRPG_PQexec(hPGConn, osCommand.c_str()); OGRPGClearResult( hResult ); } osCommand.Printf( "SELECT AddGeometryColumn(%s,%s,%s,%d,'%s',%d)", pszEscapedSchemaNameSingleQuote, pszEscapedTableNameSingleQuote, OGRPGEscapeString(hPGConn, pszGFldName).c_str(), nSRSId, pszGeometryType, nDimension ); hResult = OGRPG_PQexec(hPGConn, osCommand.c_str()); if( !hResult || PQresultStatus(hResult) != PGRES_TUPLES_OK ) { CPLError( CE_Failure, CPLE_AppDefined, "AddGeometryColumn failed for layer %s, layer creation has failed.", pszLayerName ); CPLFree( pszTableName ); CPLFree( pszSchemaName ); OGRPGClearResult( hResult ); hResult = OGRPG_PQexec(hPGConn, "ROLLBACK"); OGRPGClearResult( hResult ); return NULL; } OGRPGClearResult( hResult ); } const char *pszSI = CSLFetchNameValue( papszOptions, "SPATIAL_INDEX" ); int bCreateSpatialIndex = ( pszSI == NULL || CSLTestBoolean(pszSI) ); if( eType != wkbNone && bHavePostGIS && bCreateSpatialIndex ) { /* -------------------------------------------------------------------- */ /* Create the spatial index. */ /* */ /* We're doing this before we add geometry and record to the table */ /* so this may not be exactly the best way to do it. */ /* -------------------------------------------------------------------- */ osCommand.Printf("CREATE INDEX %s ON %s.%s USING GIST (%s)", OGRPGEscapeColumnName( CPLSPrintf("%s_%s_geom_idx", pszTableName, pszGFldName)).c_str(), OGRPGEscapeColumnName(pszSchemaName).c_str(), OGRPGEscapeColumnName(pszTableName).c_str(), OGRPGEscapeColumnName(pszGFldName).c_str()); hResult = OGRPG_PQexec(hPGConn, osCommand.c_str()); if( !hResult || PQresultStatus(hResult) != PGRES_COMMAND_OK ) { CPLError( CE_Failure, CPLE_AppDefined, "'%s' failed for layer %s, index creation has failed.", osCommand.c_str(), pszLayerName ); CPLFree( pszTableName ); CPLFree( pszSchemaName ); OGRPGClearResult( hResult ); hResult = OGRPG_PQexec(hPGConn, "ROLLBACK"); OGRPGClearResult( hResult ); return NULL; } OGRPGClearResult( hResult ); } /* -------------------------------------------------------------------- */ /* Complete, and commit the transaction. */ /* -------------------------------------------------------------------- */ hResult = OGRPG_PQexec(hPGConn, "COMMIT"); OGRPGClearResult( hResult ); /* -------------------------------------------------------------------- */ /* Create the layer object. */ /* -------------------------------------------------------------------- */ OGRPGTableLayer *poLayer; poLayer = new OGRPGTableLayer( this, osCurrentSchema, pszTableName, pszSchemaName, NULL, TRUE ); if( !(poLayer->ReadTableDefinition()) ) { CPLFree( pszTableName ); CPLFree( pszSchemaName ); delete poLayer; return NULL; } poLayer->SetLaunderFlag( CSLFetchBoolean(papszOptions,"LAUNDER",TRUE) ); poLayer->SetPrecisionFlag( CSLFetchBoolean(papszOptions,"PRECISION",TRUE)); //poLayer->SetForcedSRSId(nForcedSRSId); poLayer->SetForcedDimension(nForcedDimension); poLayer->SetCreateSpatialIndexFlag(bCreateSpatialIndex); /* HSTORE_COLUMNS existed at a time during GDAL 1.10dev */ const char* pszHSTOREColumns = CSLFetchNameValue( papszOptions, "HSTORE_COLUMNS" ); if( pszHSTOREColumns != NULL ) CPLError(CE_Warning, CPLE_AppDefined, "HSTORE_COLUMNS not recognized. Use COLUMN_TYPES instead."); const char* pszOverrideColumnTypes = CSLFetchNameValue( papszOptions, "COLUMN_TYPES" ); poLayer->SetOverrideColumnTypes(pszOverrideColumnTypes); /* -------------------------------------------------------------------- */ /* Add layer to data source layer list. */ /* -------------------------------------------------------------------- */ papoLayers = (OGRPGTableLayer **) CPLRealloc( papoLayers, sizeof(OGRPGTableLayer *) * (nLayers+1) ); papoLayers[nLayers++] = poLayer; CPLFree( pszTableName ); CPLFree( pszSchemaName ); return poLayer; } /************************************************************************/ /* TestCapability() */ /************************************************************************/ int OGRPGDataSource::TestCapability( const char * pszCap ) { if( EQUAL(pszCap,ODsCCreateLayer) || EQUAL(pszCap,ODsCDeleteLayer) || EQUAL(pszCap,ODsCCreateGeomFieldAfterCreateLayer) ) return TRUE; else return FALSE; } /************************************************************************/ /* GetLayer() */ /************************************************************************/ OGRLayer *OGRPGDataSource::GetLayer( int iLayer ) { if( iLayer < 0 || iLayer >= nLayers ) return NULL; else return papoLayers[iLayer]; } /************************************************************************/ /* GetLayerByName() */ /************************************************************************/ OGRLayer *OGRPGDataSource::GetLayerByName( const char *pszName ) { char* pszTableName = NULL; char *pszGeomColumnName = NULL; char *pszSchemaName = NULL; if ( ! pszName ) return NULL; int i; int count = GetLayerCount(); /* first a case sensitive check */ for( i = 0; i < count; i++ ) { OGRPGTableLayer *poLayer = papoLayers[i]; if( strcmp( pszName, poLayer->GetName() ) == 0 ) { return poLayer; } } /* then case insensitive */ for( i = 0; i < count; i++ ) { OGRPGTableLayer *poLayer = papoLayers[i]; if( EQUAL( pszName, poLayer->GetName() ) ) { return poLayer; } } char* pszNameWithoutBracket = CPLStrdup(pszName); char *pos = strchr(pszNameWithoutBracket, '('); if (pos != NULL) { *pos = '\0'; pszGeomColumnName = CPLStrdup(pos+1); int len = strlen(pszGeomColumnName); if (len > 0) pszGeomColumnName[len - 1] = '\0'; } pos = strchr(pszNameWithoutBracket, '.'); if (pos != NULL) { *pos = '\0'; pszSchemaName = CPLStrdup(pszNameWithoutBracket); pszTableName = CPLStrdup(pos + 1); } else { pszTableName = CPLStrdup(pszNameWithoutBracket); } CPLFree(pszNameWithoutBracket); pszNameWithoutBracket = NULL; OGRPGTableLayer* poLayer = NULL; if (pszSchemaName != NULL && osCurrentSchema == pszSchemaName && pszGeomColumnName == NULL ) { poLayer = (OGRPGTableLayer*) GetLayerByName(pszTableName); } else { poLayer = OpenTable( osCurrentSchema, pszTableName, pszSchemaName, pszGeomColumnName, bDSUpdate, TRUE ); } CPLFree(pszTableName); CPLFree(pszSchemaName); CPLFree(pszGeomColumnName); return poLayer; } /************************************************************************/ /* OGRPGNoticeProcessor() */ /************************************************************************/ static void OGRPGNoticeProcessor( CPL_UNUSED void *arg, const char * pszMessage ) { CPLDebug( "OGR_PG_NOTICE", "%s", pszMessage ); } /************************************************************************/ /* InitializeMetadataTables() */ /* */ /* Create the metadata tables (SPATIAL_REF_SYS and */ /* GEOMETRY_COLUMNS). */ /************************************************************************/ OGRErr OGRPGDataSource::InitializeMetadataTables() { // implement later. return OGRERR_FAILURE; } /************************************************************************/ /* FetchSRS() */ /* */ /* Return a SRS corresponding to a particular id. Note that */ /* reference counting should be honoured on the returned */ /* OGRSpatialReference, as handles may be cached. */ /************************************************************************/ OGRSpatialReference *OGRPGDataSource::FetchSRS( int nId ) { if( nId < 0 ) return NULL; /* -------------------------------------------------------------------- */ /* First, we look through our SRID cache, is it there? */ /* -------------------------------------------------------------------- */ int i; for( i = 0; i < nKnownSRID; i++ ) { if( panSRID[i] == nId ) return papoSRS[i]; } /* -------------------------------------------------------------------- */ /* Try looking up in spatial_ref_sys table. */ /* -------------------------------------------------------------------- */ PGresult *hResult = NULL; CPLString osCommand; OGRSpatialReference *poSRS = NULL; SoftStartTransaction(); osCommand.Printf( "SELECT srtext FROM spatial_ref_sys " "WHERE srid = %d", nId ); hResult = OGRPG_PQexec(hPGConn, osCommand.c_str() ); if( hResult && PQresultStatus(hResult) == PGRES_TUPLES_OK && PQntuples(hResult) == 1 ) { char *pszWKT; pszWKT = PQgetvalue(hResult,0,0); poSRS = new OGRSpatialReference(); if( poSRS->importFromWkt( &pszWKT ) != OGRERR_NONE ) { delete poSRS; poSRS = NULL; } } else { CPLError( CE_Failure, CPLE_AppDefined, "Could not fetch SRS: %s", PQerrorMessage( hPGConn ) ); } OGRPGClearResult( hResult ); SoftCommit(); /* -------------------------------------------------------------------- */ /* Add to the cache. */ /* -------------------------------------------------------------------- */ panSRID = (int *) CPLRealloc(panSRID,sizeof(int) * (nKnownSRID+1) ); papoSRS = (OGRSpatialReference **) CPLRealloc(papoSRS, sizeof(void*) * (nKnownSRID + 1) ); panSRID[nKnownSRID] = nId; papoSRS[nKnownSRID] = poSRS; nKnownSRID++; return poSRS; } /************************************************************************/ /* FetchSRSId() */ /* */ /* Fetch the id corresponding to an SRS, and if not found, add */ /* it to the table. */ /************************************************************************/ int OGRPGDataSource::FetchSRSId( OGRSpatialReference * poSRS ) { PGresult *hResult = NULL; CPLString osCommand; char *pszWKT = NULL; int nSRSId = nUndefinedSRID; const char* pszAuthorityName; if( poSRS == NULL ) return nUndefinedSRID; OGRSpatialReference oSRS(*poSRS); poSRS = NULL; pszAuthorityName = oSRS.GetAuthorityName(NULL); if( pszAuthorityName == NULL || strlen(pszAuthorityName) == 0 ) { /* -------------------------------------------------------------------- */ /* Try to identify an EPSG code */ /* -------------------------------------------------------------------- */ oSRS.AutoIdentifyEPSG(); pszAuthorityName = oSRS.GetAuthorityName(NULL); if (pszAuthorityName != NULL && EQUAL(pszAuthorityName, "EPSG")) { const char* pszAuthorityCode = oSRS.GetAuthorityCode(NULL); if ( pszAuthorityCode != NULL && strlen(pszAuthorityCode) > 0 ) { /* Import 'clean' SRS */ oSRS.importFromEPSG( atoi(pszAuthorityCode) ); pszAuthorityName = oSRS.GetAuthorityName(NULL); } } } /* -------------------------------------------------------------------- */ /* Check whether the EPSG authority code is already mapped to a */ /* SRS ID. */ /* -------------------------------------------------------------------- */ if( pszAuthorityName != NULL && EQUAL( pszAuthorityName, "EPSG" ) ) { int nAuthorityCode; /* For the root authority name 'EPSG', the authority code * should always be integral */ nAuthorityCode = atoi( oSRS.GetAuthorityCode(NULL) ); osCommand.Printf("SELECT srid FROM spatial_ref_sys WHERE " "auth_name = '%s' AND auth_srid = %d", pszAuthorityName, nAuthorityCode ); hResult = OGRPG_PQexec(hPGConn, osCommand.c_str()); if( hResult && PQresultStatus(hResult) == PGRES_TUPLES_OK && PQntuples(hResult) > 0 ) { nSRSId = atoi(PQgetvalue( hResult, 0, 0 )); OGRPGClearResult( hResult ); return nSRSId; } OGRPGClearResult( hResult ); } /* -------------------------------------------------------------------- */ /* Translate SRS to WKT. */ /* -------------------------------------------------------------------- */ if( oSRS.exportToWkt( &pszWKT ) != OGRERR_NONE ) { CPLFree(pszWKT); return nUndefinedSRID; } /* -------------------------------------------------------------------- */ /* Try to find in the existing table. */ /* -------------------------------------------------------------------- */ hResult = OGRPG_PQexec(hPGConn, "BEGIN"); OGRPGClearResult( hResult ); CPLString osWKT = OGRPGEscapeString(hPGConn, pszWKT, -1, "spatial_ref_sys", "srtext"); osCommand.Printf( "SELECT srid FROM spatial_ref_sys WHERE srtext = %s", osWKT.c_str() ); hResult = OGRPG_PQexec(hPGConn, osCommand.c_str() ); CPLFree( pszWKT ); // CM: Added to prevent mem leaks pszWKT = NULL; // CM: Added /* -------------------------------------------------------------------- */ /* We got it! Return it. */ /* -------------------------------------------------------------------- */ if( hResult && PQresultStatus(hResult) == PGRES_TUPLES_OK && PQntuples(hResult) > 0 ) { nSRSId = atoi(PQgetvalue( hResult, 0, 0 )); OGRPGClearResult( hResult ); hResult = OGRPG_PQexec(hPGConn, "COMMIT"); OGRPGClearResult( hResult ); return nSRSId; } /* -------------------------------------------------------------------- */ /* If the command actually failed, then the metadata table is */ /* likely missing. Try defining it. */ /* -------------------------------------------------------------------- */ int bTableMissing; bTableMissing = hResult == NULL || PQresultStatus(hResult) == PGRES_NONFATAL_ERROR; OGRPGClearResult( hResult ); hResult = OGRPG_PQexec(hPGConn, "COMMIT"); OGRPGClearResult( hResult ); if( bTableMissing ) { if( InitializeMetadataTables() != OGRERR_NONE ) return nUndefinedSRID; } /* -------------------------------------------------------------------- */ /* Get the current maximum srid in the srs table. */ /* -------------------------------------------------------------------- */ hResult = OGRPG_PQexec(hPGConn, "BEGIN"); OGRPGClearResult( hResult ); hResult = OGRPG_PQexec(hPGConn, "SELECT MAX(srid) FROM spatial_ref_sys" ); if( hResult && PQresultStatus(hResult) == PGRES_TUPLES_OK ) { nSRSId = atoi(PQgetvalue(hResult,0,0)) + 1; OGRPGClearResult( hResult ); } else { nSRSId = 1; } /* -------------------------------------------------------------------- */ /* Try adding the SRS to the SRS table. */ /* -------------------------------------------------------------------- */ char *pszProj4 = NULL; if( oSRS.exportToProj4( &pszProj4 ) != OGRERR_NONE ) { CPLFree( pszProj4 ); return nUndefinedSRID; } CPLString osProj4 = OGRPGEscapeString(hPGConn, pszProj4, -1, "spatial_ref_sys", "proj4text"); if( pszAuthorityName != NULL && EQUAL(pszAuthorityName, "EPSG") ) { int nAuthorityCode; nAuthorityCode = atoi( oSRS.GetAuthorityCode(NULL) ); osCommand.Printf( "INSERT INTO spatial_ref_sys (srid,srtext,proj4text,auth_name,auth_srid) " "VALUES (%d, %s, %s, '%s', %d)", nSRSId, osWKT.c_str(), osProj4.c_str(), pszAuthorityName, nAuthorityCode ); } else { osCommand.Printf( "INSERT INTO spatial_ref_sys (srid,srtext,proj4text) VALUES (%d,%s,%s)", nSRSId, osWKT.c_str(), osProj4.c_str() ); } // Free everything that was allocated. CPLFree( pszProj4 ); CPLFree( pszWKT); hResult = OGRPG_PQexec(hPGConn, osCommand.c_str() ); OGRPGClearResult( hResult ); hResult = OGRPG_PQexec(hPGConn, "COMMIT"); OGRPGClearResult( hResult ); return nSRSId; } /************************************************************************/ /* SoftStartTransaction() */ /* */ /* Create a transaction scope. If we already have a */ /* transaction active this isn't a real transaction, but just */ /* an increment to the scope count. */ /************************************************************************/ OGRErr OGRPGDataSource::SoftStartTransaction() { nSoftTransactionLevel++; if( nSoftTransactionLevel == 1 ) { PGresult *hResult = NULL; PGconn *hPGConn = GetPGConn(); //CPLDebug( "PG", "BEGIN Transaction" ); hResult = OGRPG_PQexec(hPGConn, "BEGIN"); if( !hResult || PQresultStatus(hResult) != PGRES_COMMAND_OK ) { OGRPGClearResult( hResult ); CPLDebug( "PG", "BEGIN Transaction failed:\n%s", PQerrorMessage( hPGConn ) ); return OGRERR_FAILURE; } OGRPGClearResult( hResult ); } return OGRERR_NONE; } /************************************************************************/ /* SoftCommit() */ /* */ /* Commit the current transaction if we are at the outer */ /* scope. */ /************************************************************************/ OGRErr OGRPGDataSource::SoftCommit() { EndCopy(); if( nSoftTransactionLevel <= 0 ) { CPLDebug( "PG", "SoftCommit() with no transaction active." ); return OGRERR_FAILURE; } nSoftTransactionLevel--; if( nSoftTransactionLevel == 0 ) { PGresult *hResult = NULL; PGconn *hPGConn = GetPGConn(); //CPLDebug( "PG", "COMMIT Transaction" ); hResult = OGRPG_PQexec(hPGConn, "COMMIT"); if( !hResult || PQresultStatus(hResult) != PGRES_COMMAND_OK ) { OGRPGClearResult( hResult ); CPLDebug( "PG", "COMMIT Transaction failed:\n%s", PQerrorMessage( hPGConn ) ); return OGRERR_FAILURE; } OGRPGClearResult( hResult ); } return OGRERR_NONE; } /************************************************************************/ /* SoftRollback() */ /* */ /* Force a rollback of the current transaction if there is one, */ /* even if we are nested several levels deep. */ /************************************************************************/ OGRErr OGRPGDataSource::SoftRollback() { EndCopy(); if( nSoftTransactionLevel <= 0 ) { CPLDebug( "PG", "SoftRollback() with no transaction active." ); return OGRERR_FAILURE; } nSoftTransactionLevel = 0; PGresult *hResult = NULL; PGconn *hPGConn = GetPGConn(); hResult = OGRPG_PQexec(hPGConn, "ROLLBACK"); if( !hResult || PQresultStatus(hResult) != PGRES_COMMAND_OK ) { OGRPGClearResult( hResult ); return OGRERR_FAILURE; } OGRPGClearResult( hResult ); return OGRERR_NONE; } /************************************************************************/ /* FlushSoftTransaction() */ /* */ /* Force the unwinding of any active transaction, and it's */ /* commit. */ /************************************************************************/ OGRErr OGRPGDataSource::FlushSoftTransaction() { /* This must come first because of ogr2ogr. If you want to use ogr2ogr with COPY support, then you must specify that ogr2ogr does not use transactions. Thus, nSoftTransactionLevel will always be zero, so this has to come first. */ EndCopy(); if( nSoftTransactionLevel <= 0 ) return OGRERR_NONE; nSoftTransactionLevel = 1; return SoftCommit(); } /************************************************************************/ /* OGRPGNoResetResultLayer */ /************************************************************************/ class OGRPGNoResetResultLayer : public OGRPGLayer { public: OGRPGNoResetResultLayer(OGRPGDataSource *poDSIn, PGresult *hResultIn); virtual ~OGRPGNoResetResultLayer(); virtual void ResetReading(); virtual int TestCapability( const char * ) { return FALSE; } virtual OGRFeature *GetNextFeature(); virtual CPLString GetFromClauseForGetExtent() { CPLAssert(FALSE); return ""; } virtual void ResolveSRID(OGRPGGeomFieldDefn* poGFldDefn) { poGFldDefn->nSRSId = -1; } }; /************************************************************************/ /* OGRPGNoResetResultLayer() */ /************************************************************************/ OGRPGNoResetResultLayer::OGRPGNoResetResultLayer( OGRPGDataSource *poDSIn, PGresult *hResultIn ) { poDS = poDSIn; ReadResultDefinition(hResultIn); hCursorResult = hResultIn; CreateMapFromFieldNameToIndex(); } /************************************************************************/ /* ~OGRPGNoResetResultLayer() */ /************************************************************************/ OGRPGNoResetResultLayer::~OGRPGNoResetResultLayer() { OGRPGClearResult( hCursorResult ); hCursorResult = NULL; } /************************************************************************/ /* ResetReading() */ /************************************************************************/ void OGRPGNoResetResultLayer::ResetReading() { iNextShapeId = 0; } /************************************************************************/ /* GetNextFeature() */ /************************************************************************/ OGRFeature *OGRPGNoResetResultLayer::GetNextFeature() { if (iNextShapeId == PQntuples(hCursorResult)) { return NULL; } return RecordToFeature(iNextShapeId ++); } /************************************************************************/ /* OGRPGMemLayerWrapper */ /************************************************************************/ class OGRPGMemLayerWrapper : public OGRLayer { private: OGRDataSource *poMemDS; OGRLayer *poMemLayer; public: OGRPGMemLayerWrapper( OGRDataSource *poMemDSIn ) { poMemDS = poMemDSIn; poMemLayer = poMemDS->GetLayer(0); } ~OGRPGMemLayerWrapper() { delete poMemDS; } virtual void ResetReading() { poMemLayer->ResetReading(); } virtual OGRFeature *GetNextFeature() { return poMemLayer->GetNextFeature(); } virtual OGRFeatureDefn *GetLayerDefn() { return poMemLayer->GetLayerDefn(); } virtual int TestCapability( const char * ) { return FALSE; } }; /************************************************************************/ /* ExecuteSQL() */ /************************************************************************/ OGRLayer * OGRPGDataSource::ExecuteSQL( const char *pszSQLCommand, OGRGeometry *poSpatialFilter, const char *pszDialect ) { /* Skip leading spaces */ while(*pszSQLCommand == ' ') pszSQLCommand ++; /* -------------------------------------------------------------------- */ /* Use generic implementation for recognized dialects */ /* -------------------------------------------------------------------- */ if( IsGenericSQLDialect(pszDialect) ) return OGRDataSource::ExecuteSQL( pszSQLCommand, poSpatialFilter, pszDialect ); /* -------------------------------------------------------------------- */ /* Special case DELLAYER: command. */ /* -------------------------------------------------------------------- */ if( EQUALN(pszSQLCommand,"DELLAYER:",9) ) { const char *pszLayerName = pszSQLCommand + 9; while( *pszLayerName == ' ' ) pszLayerName++; for( int iLayer = 0; iLayer < nLayers; iLayer++ ) { if( EQUAL(papoLayers[iLayer]->GetName(), pszLayerName )) { DeleteLayer( iLayer ); break; } } return NULL; } /* -------------------------------------------------------------------- */ /* Execute the statement. */ /* -------------------------------------------------------------------- */ PGresult *hResult = NULL; FlushSoftTransaction(); if( EQUALN(pszSQLCommand,"VACUUM",6) || SoftStartTransaction() == OGRERR_NONE ) { if (EQUALN(pszSQLCommand, "SELECT", 6) == FALSE || (strstr(pszSQLCommand, "from") == NULL && strstr(pszSQLCommand, "FROM") == NULL)) { hResult = OGRPG_PQexec(hPGConn, pszSQLCommand, TRUE /* multiple allowed */ ); if (hResult && PQresultStatus(hResult) == PGRES_TUPLES_OK) { CPLDebug( "PG", "Command Results Tuples = %d", PQntuples(hResult) ); FlushSoftTransaction(); OGRSFDriver* poMemDriver = OGRSFDriverRegistrar::GetRegistrar()-> GetDriverByName("Memory"); if (poMemDriver) { OGRPGLayer* poResultLayer = new OGRPGNoResetResultLayer( this, hResult ); OGRDataSource* poMemDS = poMemDriver->CreateDataSource(""); poMemDS->CopyLayer(poResultLayer, "sql_statement"); OGRPGMemLayerWrapper* poResLayer = new OGRPGMemLayerWrapper(poMemDS); delete poResultLayer; return poResLayer; } else return NULL; } } else { CPLString osCommand; osCommand.Printf( "DECLARE %s CURSOR for %s", "executeSQLCursor", pszSQLCommand ); hResult = OGRPG_PQexec(hPGConn, osCommand ); /* -------------------------------------------------------------------- */ /* Do we have a tuple result? If so, instantiate a results */ /* layer for it. */ /* -------------------------------------------------------------------- */ if( hResult && PQresultStatus(hResult) == PGRES_COMMAND_OK ) { OGRPGResultLayer *poLayer = NULL; OGRPGClearResult( hResult ); osCommand.Printf( "FETCH 0 in %s", "executeSQLCursor" ); hResult = OGRPG_PQexec(hPGConn, osCommand ); poLayer = new OGRPGResultLayer( this, pszSQLCommand, hResult ); OGRPGClearResult( hResult ); if( poSpatialFilter != NULL ) poLayer->SetSpatialFilter( poSpatialFilter ); return poLayer; } } } /* -------------------------------------------------------------------- */ /* Generate an error report if an error occured. */ /* -------------------------------------------------------------------- */ if( !hResult || (PQresultStatus(hResult) == PGRES_NONFATAL_ERROR || PQresultStatus(hResult) == PGRES_FATAL_ERROR ) ) { CPLError( CE_Failure, CPLE_AppDefined, "%s", PQerrorMessage( hPGConn ) ); } OGRPGClearResult( hResult ); FlushSoftTransaction(); return NULL; } /************************************************************************/ /* ReleaseResultSet() */ /************************************************************************/ void OGRPGDataSource::ReleaseResultSet( OGRLayer * poLayer ) { delete poLayer; } /************************************************************************/ /* LaunderName() */ /************************************************************************/ char *OGRPGDataSource::LaunderName( const char *pszSrcName ) { char *pszSafeName = CPLStrdup( pszSrcName ); for( int i = 0; pszSafeName[i] != '\0'; i++ ) { pszSafeName[i] = (char) tolower( pszSafeName[i] ); if( pszSafeName[i] == '\'' || pszSafeName[i] == '-' || pszSafeName[i] == '#' ) pszSafeName[i] = '_'; } if( strcmp(pszSrcName,pszSafeName) != 0 ) CPLDebug("PG","LaunderName('%s') -> '%s'", pszSrcName, pszSafeName); return pszSafeName; } /************************************************************************/ /* StartCopy() */ /************************************************************************/ void OGRPGDataSource::StartCopy( OGRPGTableLayer *poPGLayer ) { EndCopy(); poLayerInCopyMode = poPGLayer; } /************************************************************************/ /* EndCopy() */ /************************************************************************/ OGRErr OGRPGDataSource::EndCopy( ) { if( poLayerInCopyMode != NULL ) { OGRErr result = poLayerInCopyMode->EndCopy(); poLayerInCopyMode = NULL; return result; } else return OGRERR_NONE; } /************************************************************************/ /* CopyInProgress() */ /************************************************************************/ int OGRPGDataSource::CopyInProgress( ) { return ( poLayerInCopyMode != NULL ); }