Line data    Source code

       1             : /******************************************************************************
       2             :  *
       3             :  * Project:  OpenGIS Simple Features Reference Implementation
       4             :  * Purpose:  Private definitions for OGR/PostgreSQL dump driver.
       5             :  * Author:   Even Rouault, <even dot rouault at spatialys.com>
       6             :  *
       7             :  ******************************************************************************
       8             :  * Copyright (c) 2010-2013, Even Rouault <even dot rouault at spatialys.com>
       9             :  *
      10             :  * SPDX-License-Identifier: MIT
      11             :  ****************************************************************************/
      12             : 
      13             : #ifndef OGR_PGDUMP_H_INCLUDED
      14             : #define OGR_PGDUMP_H_INCLUDED
      15             : 
      16             : #include "ogrsf_frmts.h"
      17             : #include "cpl_string.h"
      18             : 
      19             : #include <vector>
      20             : 
      21             : // Cf https://www.postgresql.org/docs/current/sql-syntax-lexical.html#SQL-SYNTAX-IDENTIFIERS
      22             : constexpr int OGR_PG_NAMEDATALEN = 64;
      23             : 
      24             : CPLString OGRPGDumpEscapeColumnName(const char *pszColumnName);
      25             : CPLString OGRPGDumpEscapeString(const char *pszStrValue, int nMaxLength = -1,
      26             :                                 const char *pszFieldName = "");
      27             : char CPL_DLL *OGRPGCommonGByteArrayToBYTEA(const GByte *pabyData, size_t nLen);
      28             : CPLString CPL_DLL OGRPGCommonLayerGetType(const OGRFieldDefn &oField,
      29             :                                           bool bPreservePrecision,
      30             :                                           bool bApproxOK);
      31             : bool CPL_DLL OGRPGCommonLayerSetType(OGRFieldDefn &oField, const char *pszType,
      32             :                                      const char *pszFormatType, int nWidth);
      33             : void CPL_DLL OGRPGCommonLayerNormalizeDefault(OGRFieldDefn *poFieldDefn,
      34             :                                               const char *pszDefault);
      35             : CPLString CPL_DLL OGRPGCommonLayerGetPGDefault(OGRFieldDefn *poFieldDefn);
      36             : 
      37             : void CPL_DLL OGRPGCommonAppendCopyFID(CPLString &osCommand,
      38             :                                       OGRFeature *poFeature);
      39             : 
      40             : typedef CPLString (*OGRPGCommonEscapeStringCbk)(void *userdata,
      41             :                                                 const char *pszValue,
      42             :                                                 int nWidth,
      43             :                                                 const char *pszLayerName,
      44             :                                                 const char *pszFieldRef);
      45             : void CPL_DLL OGRPGCommonAppendCopyRegularFields(
      46             :     CPLString &osCommand, OGRFeature *poFeature, const char *pszFIDColumn,
      47             :     const std::vector<bool> &abFieldsToInclude,
      48             :     OGRPGCommonEscapeStringCbk pfnEscapeString, void *userdata);
      49             : 
      50             : void CPL_DLL OGRPGCommonAppendFieldValue(
      51             :     CPLString &osCommand, OGRFeature *poFeature, int i,
      52             :     OGRPGCommonEscapeStringCbk pfnEscapeString, void *userdata);
      53             : 
      54             : char CPL_DLL *OGRPGCommonLaunderName(const char *pszSrcName,
      55             :                                      const char *pszDebugPrefix,
      56             :                                      bool bUTF8ToASCII);
      57             : 
      58             : std::string CPL_DLL
      59             : OGRPGCommonGenerateShortEnoughIdentifier(const char *pszIdentifier);
      60             : 
      61             : std::string CPL_DLL OGRPGCommonGenerateSpatialIndexName(
      62             :     const char *pszTableName, const char *pszGeomFieldName, int nGeomFieldIdx);
      63             : 
      64             : /************************************************************************/
      65             : /*                        OGRPGDumpGeomFieldDefn                        */
      66             : /************************************************************************/
      67             : 
      68         216 : class OGRPGDumpGeomFieldDefn final : public OGRGeomFieldDefn
      69             : {
      70             :     OGRPGDumpGeomFieldDefn(const OGRPGDumpGeomFieldDefn &) = delete;
      71             :     OGRPGDumpGeomFieldDefn &operator=(const OGRPGDumpGeomFieldDefn &) = delete;
      72             : 
      73             :   public:
      74         108 :     explicit OGRPGDumpGeomFieldDefn(OGRGeomFieldDefn *poGeomField)
      75         108 :         : OGRGeomFieldDefn(poGeomField), m_nSRSId(-1), m_nGeometryTypeFlags(0)
      76             :     {
      77         108 :     }
      78             : 
      79             :     ~OGRPGDumpGeomFieldDefn() override;
      80             : 
      81             :     int m_nSRSId;
      82             :     int m_nGeometryTypeFlags;
      83             : };
      84             : 
      85             : /************************************************************************/
      86             : /*                            OGRPGDumpLayer                            */
      87             : /************************************************************************/
      88             : 
      89             : class OGRPGDumpDataSource;
      90             : 
      91             : class OGRPGDumpLayer final : public OGRLayer
      92             : {
      93             :     OGRPGDumpLayer(const OGRPGDumpLayer &) = delete;
      94             :     OGRPGDumpLayer &operator=(const OGRPGDumpLayer &) = delete;
      95             : 
      96             :     static constexpr int USE_COPY_UNSET = -1;
      97             : 
      98             :     char *m_pszSchemaName = nullptr;
      99             :     char *m_pszSqlTableName = nullptr;
     100             :     CPLString m_osForcedDescription{};
     101             :     char *m_pszFIDColumn = nullptr;
     102             :     OGRFeatureDefn *m_poFeatureDefn = nullptr;
     103             :     OGRPGDumpDataSource *m_poDS = nullptr;
     104             :     bool m_bLaunderColumnNames = true;
     105             :     bool m_bUTF8ToASCII = false;
     106             :     bool m_bPreservePrecision = true;
     107             :     int m_bUseCopy = USE_COPY_UNSET;
     108             :     bool m_bWriteAsHex = false;
     109             :     bool m_bCopyActive = false;
     110             :     bool m_bFIDColumnInCopyFields = false;
     111             :     int m_bCreateTable = false;
     112             :     bool m_bSkipConflicts = false;
     113             :     int m_nUnknownSRSId = -1;
     114             :     int m_nForcedSRSId = -1;
     115             :     int m_nForcedGeometryTypeFlags = -2;
     116             :     bool m_bCreateSpatialIndexFlag = false;
     117             :     CPLString m_osSpatialIndexType{};
     118             :     int m_nPostGISMajor = 0;
     119             :     int m_nPostGISMinor = 0;
     120             : 
     121             :     GIntBig m_iNextShapeId = 0;
     122             :     int m_iFIDAsRegularColumnIndex = -1;
     123             :     bool m_bAutoFIDOnCreateViaCopy = true;
     124             :     bool m_bCopyStatementWithFID = true;
     125             :     bool m_bNeedToUpdateSequence = false;
     126             :     bool m_bGeomColumnPositionImmediate = true;
     127             :     std::vector<std::string> m_aosDeferredGeomFieldCreationCommands{};
     128             :     std::vector<std::string> m_aosDeferrentNonGeomFieldCreationCommands{};
     129             :     std::vector<std::string> m_aosSpatialIndexCreationCommands{};
     130             : 
     131             :     CPLStringList m_apszOverrideColumnTypes{};
     132             : 
     133             :     CPLString m_osFirstGeometryFieldName{};
     134             : 
     135             :     OGRErr StartCopy(int bSetFID);
     136             :     CPLString BuildCopyFields(int bSetFID);
     137             : 
     138             :     void UpdateSequenceIfNeeded();
     139             : 
     140             :     void LogDeferredFieldCreationIfNeeded();
     141             : 
     142             :   public:
     143             :     OGRPGDumpLayer(OGRPGDumpDataSource *poDS, const char *pszSchemaName,
     144             :                    const char *pszLayerName, const char *pszFIDColumn,
     145             :                    int bWriteAsHexIn, int bCreateTable, bool bSkipConflicts);
     146             :     ~OGRPGDumpLayer() override;
     147             : 
     148             :     using OGRLayer::GetLayerDefn;
     149             : 
     150         583 :     const OGRFeatureDefn *GetLayerDefn() const override
     151             :     {
     152         583 :         return m_poFeatureDefn;
     153             :     }
     154             : 
     155           7 :     const char *GetFIDColumn() const override
     156             :     {
     157           7 :         return m_pszFIDColumn ? m_pszFIDColumn : "";
     158             :     }
     159             : 
     160          16 :     void ResetReading() override
     161             :     {
     162          16 :     }
     163             : 
     164             :     int TestCapability(const char *) const override;
     165             : 
     166             :     OGRErr ICreateFeature(OGRFeature *poFeature) override;
     167             :     OGRErr CreateFeatureViaInsert(OGRFeature *poFeature);
     168             :     OGRErr CreateFeatureViaCopy(OGRFeature *poFeature);
     169             : 
     170             :     virtual OGRErr CreateField(const OGRFieldDefn *poField,
     171             :                                int bApproxOK = TRUE) override;
     172             :     virtual OGRErr CreateGeomField(const OGRGeomFieldDefn *poGeomField,
     173             :                                    int bApproxOK = TRUE) override;
     174             : 
     175             :     OGRFeature *GetNextFeature() override;
     176             : 
     177             :     CPLErr SetMetadata(CSLConstList papszMD,
     178             :                        const char *pszDomain = "") override;
     179             :     CPLErr SetMetadataItem(const char *pszName, const char *pszValue,
     180             :                            const char *pszDomain = "") override;
     181             : 
     182             :     GDALDataset *GetDataset() override;
     183             : 
     184             :     // follow methods are not base class overrides
     185         118 :     void SetLaunderFlag(bool bFlag)
     186             :     {
     187         118 :         m_bLaunderColumnNames = bFlag;
     188         118 :     }
     189             : 
     190         118 :     void SetUTF8ToASCIIFlag(bool bFlag)
     191             :     {
     192         118 :         m_bUTF8ToASCII = bFlag;
     193         118 :     }
     194             : 
     195         118 :     void SetPrecisionFlag(bool bFlag)
     196             :     {
     197         118 :         m_bPreservePrecision = bFlag;
     198         118 :     }
     199             : 
     200             :     void SetOverrideColumnTypes(const char *pszOverrideColumnTypes);
     201             : 
     202         118 :     void SetUnknownSRSId(int nUnknownSRSIdIn)
     203             :     {
     204         118 :         m_nUnknownSRSId = nUnknownSRSIdIn;
     205         118 :     }
     206             : 
     207         118 :     void SetForcedSRSId(int nForcedSRSIdIn)
     208             :     {
     209         118 :         m_nForcedSRSId = nForcedSRSIdIn;
     210         118 :     }
     211             : 
     212         118 :     void SetForcedGeometryTypeFlags(int GeometryTypeFlagsIn)
     213             :     {
     214         118 :         m_nForcedGeometryTypeFlags = GeometryTypeFlagsIn;
     215         118 :     }
     216             : 
     217         118 :     void SetCreateSpatialIndex(bool bFlag, const char *pszSpatialIndexType)
     218             :     {
     219         118 :         m_bCreateSpatialIndexFlag = bFlag;
     220         118 :         m_osSpatialIndexType = pszSpatialIndexType;
     221         118 :     }
     222             : 
     223         118 :     void SetPostGISVersion(int nPostGISMajorIn, int nPostGISMinorIn)
     224             :     {
     225         118 :         m_nPostGISMajor = nPostGISMajorIn;
     226         118 :         m_nPostGISMinor = nPostGISMinorIn;
     227         118 :     }
     228             : 
     229           6 :     void SetGeometryFieldName(const char *pszGeomFieldName)
     230             :     {
     231           6 :         m_osFirstGeometryFieldName = pszGeomFieldName;
     232           6 :     }
     233             : 
     234             :     void SetForcedDescription(const char *pszDescriptionIn);
     235             : 
     236         118 :     void SetGeomColumnPositionImmediate(bool bGeomColumnPositionImmediate)
     237             :     {
     238         118 :         m_bGeomColumnPositionImmediate = bGeomColumnPositionImmediate;
     239         118 :     }
     240             : 
     241           6 :     void SetDeferredGeomFieldCreationCommands(
     242             :         const std::vector<std::string> &aosDeferredGeomFieldCreationCommands)
     243             :     {
     244             :         m_aosDeferredGeomFieldCreationCommands =
     245           6 :             aosDeferredGeomFieldCreationCommands;
     246           6 :     }
     247             : 
     248         118 :     void SetSpatialIndexCreationCommands(
     249             :         const std::vector<std::string> &aosSpatialIndexCreationCommands)
     250             :     {
     251         118 :         m_aosSpatialIndexCreationCommands = aosSpatialIndexCreationCommands;
     252         118 :     }
     253             : 
     254             :     OGRErr EndCopy();
     255             : };
     256             : 
     257             : /************************************************************************/
     258             : /*                         OGRPGDumpDataSource                          */
     259             : /************************************************************************/
     260             : class OGRPGDumpDataSource final : public GDALDataset
     261             : {
     262             :     OGRPGDumpDataSource(const OGRPGDumpDataSource &) = delete;
     263             :     OGRPGDumpDataSource &operator=(const OGRPGDumpDataSource &) = delete;
     264             : 
     265             :     std::vector<std::unique_ptr<OGRPGDumpLayer>> m_apoLayers{};
     266             :     VSILFILE *m_fp = nullptr;
     267             :     bool m_bInTransaction = false;
     268             :     OGRPGDumpLayer *m_poLayerInCopyMode = nullptr;
     269             :     const char *m_pszEOL = "\n";
     270             : 
     271             :   public:
     272             :     OGRPGDumpDataSource(const char *pszName, CSLConstList papszOptions);
     273             :     ~OGRPGDumpDataSource() override;
     274             : 
     275             :     bool Log(const char *pszStr, bool bAddSemiColumn = true);
     276             : 
     277          21 :     int GetLayerCount() const override
     278             :     {
     279          21 :         return static_cast<int>(m_apoLayers.size());
     280             :     }
     281             : 
     282             :     const OGRLayer *GetLayer(int) const override;
     283             : 
     284             :     virtual OGRLayer *ICreateLayer(const char *pszName,
     285             :                                    const OGRGeomFieldDefn *poGeomFieldDefn,
     286             :                                    CSLConstList papszOptions) override;
     287             : 
     288             :     int TestCapability(const char *) const override;
     289             : 
     290             :     void LogStartTransaction();
     291             :     void LogCommit();
     292             : 
     293             :     void StartCopy(OGRPGDumpLayer *poPGLayer);
     294             :     OGRErr EndCopy();
     295             : };
     296             : 
     297             : #endif /* ndef OGR_PGDUMP_H_INCLUDED */