Added Table Management functions with tests

Author: sharkAndshark

LinqToDBPostGisNetTopologySuite.Tests/TableManagementFunctionsTests.cs

@@ -1,6 +1,7 @@
 using System.Linq;
 
 using LinqToDB;
+using Npgsql;
 using NUnit.Framework;
 
 namespace LinqToDBPostGisNetTopologySuite.Tests
@@ -174,6 +175,113 @@ public void TestFindSrid()
             }
         }
 
+        [Test]
+        public void TestPopulateGeometryColumns()
+        {
+            var expectedConstrainNames = new[] { "enforce_dims_geom", "enforce_geotype_geom", "enforce_srid_geom", };
+            var expectedCheckClause = new[] { "((st_ndims(geom) = 3))", "((geometrytype(geom) = 'LINESTRING'::text))", "((st_srid(geom) = 4326))" };
+
+            //insert geometry data for populating then get table oid
+            int oid = 0;
+            using (var conn = new NpgsqlConnection(TestDatabaseConnectionString))
+            {
+                conn.Open();
+                using (var cmd = conn.CreateCommand())
+                {
+                    cmd.CommandText = "INSERT INTO test_geometry_ddl(id,geom) VALUES(1,ST_GeomFromText('LINESTRING(1 2 1, 3 4 1)',4326) );";
+                    cmd.ExecuteNonQuery();
+
+                    cmd.CommandText = "select oid::int4 from pg_class where oid = 'test_geometry_ddl'::regclass";
+                    var reader = cmd.ExecuteReader();
+                    reader.Read();
+                    oid = reader.GetInt32(0);
+                }
+            }
+
+            using (var db = new PostGisTestDataConnection(TestDatabaseConnectionString))
+            {
+                db.Select(() => TableManagementFunctions.PopulateGeometryColumns(oid, false));
+                using (var cmd = db.CreateCommand())
+                {
+                    cmd.CommandText = @"SELECT
+                            	tc.table_schema,
+                            	tc.TABLE_NAME,
+                            	string_agg ( col.COLUMN_NAME, ', ' ) AS COLUMNS,
+                            	tc.CONSTRAINT_NAME,
+                            	cc.check_clause 
+                            FROM
+                            	information_schema.table_constraints tc
+                            	JOIN information_schema.check_constraints cc ON tc.CONSTRAINT_SCHEMA = cc.CONSTRAINT_SCHEMA 
+                            	AND tc.CONSTRAINT_NAME = cc.
+                            	CONSTRAINT_NAME JOIN pg_namespace nsp ON nsp.nspname = cc.
+                            	CONSTRAINT_SCHEMA JOIN pg_constraint pgc ON pgc.conname = cc.CONSTRAINT_NAME 
+                            	AND pgc.connamespace = nsp.oid 
+                            	AND pgc.contype = 'c'
+                            	JOIN information_schema.COLUMNS col ON col.table_schema = tc.table_schema 
+                            	AND col.TABLE_NAME = tc.TABLE_NAME 
+                            	AND col.ordinal_position = ANY ( pgc.conkey ) 
+                            WHERE
+                            	tc.CONSTRAINT_SCHEMA NOT IN ( 'pg_catalog', 'information_schema' ) and tc.TABLE_NAME = 'test_geometry_ddl' 
+                            GROUP BY
+                            	tc.table_schema,
+                            	tc.TABLE_NAME,
+                            	tc.CONSTRAINT_NAME,
+                            	cc.check_clause 
+                            ORDER BY
+                            	tc.table_schema,
+                            	tc.TABLE_NAME";
+                    var reader = cmd.ExecuteReader();
+                    int constraintCount = 0;
+                    while (reader.Read())
+                    {
+                        var constraintName = reader.GetString(3);
+                        var check = reader.GetString(4);
+
+                        Assert.True(expectedConstrainNames.Contains(constraintName));
+                        Assert.True(expectedCheckClause.Contains(check));
+                        constraintCount++;
+                    }
+                    Assert.AreEqual(3, constraintCount);
+                }
+            }
+        }
+
+        [Test]
+        public void TestUpdateGeometrySRID()
+        {
+            using (var db = new PostGisTestDataConnection(TestDatabaseConnectionString))
+            {
+                var updateGeomText = db.Select(() => TableManagementFunctions.UpdateGeometrySRID("test_geometry_ddl", "geom", 4326));
+
+                var srid = db.Select(() => TableManagementFunctions.FindSrid("public", "test_geometry_ddl", "geom"));
+                Assert.AreEqual(4326, srid.GetValueOrDefault());
+            }
+        }
+
+        [Test]
+        public void TestUpdateGeometrySRIDSchema()
+        {
+            using (var db = new PostGisTestDataConnection(TestDatabaseConnectionString))
+            {
+                var updateGeomText = db.Select(() => TableManagementFunctions.UpdateGeometrySRID("public", "test_geometry_ddl", "geom", 4326));
+                var srid = db.Select(() => TableManagementFunctions.FindSrid("public", "test_geometry_ddl", "geom"));
+
+                Assert.AreEqual(4326, srid.GetValueOrDefault());
+            }
+        }
+
+        [Test]
+        public void TestUpdateGeometrySRIDCatalog()
+        {
+            using (var db = new PostGisTestDataConnection(TestDatabaseConnectionString))
+            {
+                var updateGeomText = db.Select(() => TableManagementFunctions.UpdateGeometrySRID("postgistest", "public", "test_geometry_ddl", "geom", 4326));
+                var srid = db.Select(() => TableManagementFunctions.FindSrid("public", "test_geometry_ddl", "geom"));
+
+                Assert.AreEqual(4326, srid.GetValueOrDefault());
+            }
+        }
+
         [TearDown]
         public void TearDown()
         {

LinqToDBPostGisNetTopologySuite/TableManagementFunctions.cs

@@ -179,5 +179,85 @@ public static string DropGeometryTable(string catalogName, string schemaName, st
         {
             throw new InvalidOperationException();
         }
+
+        /// <summary>
+        /// Ensures geometry columns have appropriate type modifiers or spatial constraints to ensure they are registered correctly in the geometry_columns view. By default will convert all geometry columns with no type modifier to ones with type modifiers
+        /// </summary>
+        /// <remarks>
+        /// See https://postgis.net/docs/manual-3.0/Populate_Geometry_Columns.html
+        /// </remarks>
+        /// <param name="useTypeMod">Use type modifiers</param>
+        /// <returns>Populate result</returns>
+        [Sql.Function("Populate_Geometry_Columns", ServerSideOnly = true)]
+        public static string PopulateGeometryColumns(bool useTypeMod)
+        {
+            throw new InvalidOperationException();
+        }
+
+        /// <summary>
+        /// Ensures geometry columns have appropriate type modifiers or spatial constraints to ensure they are registered correctly in the geometry_columns view. By default will convert all geometry columns with no type modifier to ones with type modifiers
+        /// </summary>
+        /// <remarks>
+        /// See https://postgis.net/docs/manual-3.0/Populate_Geometry_Columns.html
+        /// </remarks>
+        /// <param name="oId">OId</param>
+        /// <param name="useTypeMod">Use type modifiers</param>
+        /// <returns>Populate result</returns>
+        [Sql.Function("Populate_Geometry_Columns", ServerSideOnly = true)]
+        public static int PopulateGeometryColumns(int oId, bool useTypeMod)
+        {
+            throw new InvalidOperationException();
+        }
+
+        /// <summary>
+        /// Updates the SRID of all features in a geometry column, updating constraints and reference in geometry_columns. If the column was enforced by a type definition, the type definition will be changed
+        /// </summary>
+        /// <remarks>
+        /// See https://postgis.net/docs/manual-3.0/UpdateGeometrySRID.html
+        /// </remarks>
+        /// <param name="tableName"></param>
+        /// <param name="columnName"></param>
+        /// <param name="srid"></param>
+        /// <returns>SRID update information</returns>
+        [Sql.Function("UpdateGeometrySRID", ServerSideOnly = true)]
+        public static string UpdateGeometrySRID(string tableName, string columnName, int srid)
+        {
+            throw new InvalidOperationException();
+        }
+
+        /// <summary>
+        /// Updates the SRID of all features in a geometry column, updating constraints and reference in geometry_columns. If the column was enforced by a type definition, the type definition will be changed
+        /// </summary>
+        /// <remarks>
+        /// See https://postgis.net/docs/manual-3.0/UpdateGeometrySRID.html
+        /// </remarks>
+        /// <param name="schemaName">Schema name</param>
+        /// <param name="tableName"></param>
+        /// <param name="columnName"></param>
+        /// <param name="srid"></param>
+        /// <returns>SRID update information</returns>
+        [Sql.Function("UpdateGeometrySRID", ServerSideOnly = true)]
+        public static string UpdateGeometrySRID(string schemaName, string tableName, string columnName, int srid)
+        {
+            throw new InvalidOperationException();
+        }
+
+        /// <summary>
+        /// Updates the SRID of all features in a geometry column, updating constraints and reference in geometry_columns. If the column was enforced by a type definition, the type definition will be changed
+        /// </summary>
+        /// <remarks>
+        /// See https://postgis.net/docs/manual-3.0/UpdateGeometrySRID.html
+        /// </remarks>
+        /// <param name="catalogName">Catalog name</param>
+        /// <param name="schemaName">Schema name</param>
+        /// <param name="tableName">Table name</param>
+        /// <param name="columnName">Column name</param>
+        /// <param name="srid">Spatial Reference System Identifier</param>
+        /// <returns>SRID update information</returns>
+        [Sql.Function("UpdateGeometrySRID", ServerSideOnly = true)]
+        public static string UpdateGeometrySRID(string catalogName, string schemaName, string tableName, string columnName, int srid)
+        {
+            throw new InvalidOperationException();
+        }
     }
 }