Add support for Postgres17 (+ misc linting changes) (stripe#187)

Author: bplunkett-stripe

.github/workflows/run-tests.yaml

@@ -10,11 +10,19 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        pg_package: ["postgresql14", "postgresql15", "postgresql16"]
+        include:
+          - base_image: "golang:1.20.14-alpine3.19"
+            pg_package: "postgresql14"
+          - base_image: "golang:1.20.14-alpine3.19"
+            pg_package: "postgresql15"
+          - base_image: "golang:1.20.14-alpine3.19"
+            pg_package: "postgresql16"
+          - base_image: "golang:1.22.10-alpine3.21"
+            pg_package: "postgresql17"
     steps:
       - name: Checkout code
         uses: actions/checkout@v3
       - name: Build Docker image
-        run: docker build -t pg-schema-diff-test-runner -f ./build/Dockerfile.test --build-arg POSTGRES_PACKAGE=${{ matrix.pg_package }} .
+        run: docker build -t pg-schema-diff-test-runner -f ./build/Dockerfile.test --build-arg BASE_IMAGE=${{ matrix.base_image }} --build-arg POSTGRES_PACKAGE=${{ matrix.pg_package }} .
       - name: Run tests
         run: docker run pg-schema-diff-test-runner

.golangci.yaml

@@ -1,6 +1,7 @@
 linters:
   disable-all: true
   enable:
+  - gofmt
   - goimports
   - ineffassign
   - staticcheck

build/Dockerfile.lint

@@ -11,7 +11,7 @@ RUN wget -O- -nv https://raw.githubusercontent.com/golangci/golangci-lint/master
 # Install sqlfluff
 RUN pip install wheel # It complains if we attempt to install this in the same command as Cython
 RUN pip install "Cython<3.0" pyyaml --no-build-isolation # Fix for https://github.com/yaml/pyyaml/issues/601
-RUN pip install "sqlfluff==2.1.2"
+RUN pip install "sqlfluff==3.3.0"
 
 WORKDIR /pg-schema-diff
 COPY . .

build/Dockerfile.test

@@ -1,7 +1,8 @@
-FROM golang:1.20.14-alpine3.19
+ARG BASE_IMAGE=golang:1.20.14-alpine3.19
 
-ARG POSTGRES_PACKAGE=postgresql14
+FROM $BASE_IMAGE
 
+ARG POSTGRES_PACKAGE=postgresql14
 RUN apk update && \
     apk add --no-cache  \
       build-base \

internal/graph/graph.go

@@ -88,7 +88,7 @@ func (g *Graph[V]) HasVertexWithId(id string) bool {
 // Reverse reverses the edges of the map. The sources become the sinks and vice versa.
 func (g *Graph[V]) Reverse() {
 	reversedEdges := make(AdjacencyMatrix)
-	for vertexId, _ := range g.verticesById {
+	for vertexId := range g.verticesById {
 		reversedEdges[vertexId] = make(map[string]bool)
 	}
 	for source, adjacentEdgesMap := range g.edges {

internal/graph/graph_test.go

@@ -136,7 +136,7 @@ func TestCopy(t *testing.T) {
 		"shared_1", "shared_2", "shared_3", "g_1", "g_2", "g_3",
 	})
 
-	// validate overrides weren't copied over and non-overriden shared nodes are the same
+	// validate overrides weren't copied over and non-overridden shared nodes are the same
 	assert.NotEqual(t, g.GetVertex("shared_1"), gC.GetVertex("shared_1"))
 	assert.Equal(t, gC.GetVertex("shared_1"), copyOverrideShared1)
 	assert.NotEqual(t, g.GetVertex("shared_2"), gC.GetVertex("shared_2"))
@@ -425,7 +425,7 @@ func (v vertex) GetId() string {
 
 func getVertexIds(g *Graph[vertex]) []string {
 	var output []string
-	for id, _ := range g.verticesById {
+	for id := range g.verticesById {
 		output = append(output, id)
 	}
 	return output

internal/migration_acceptance_tests/sequence_cases_test.go

@@ -200,7 +200,7 @@ var sequenceAcceptanceTests = []acceptanceTestCase{
 		},
 	},
 	{
-		name: "And and Drop sequences (conflicing schemas)",
+		name: "And and Drop sequences (conflicting schemas)",
 		oldSchemaDDL: []string{
 			`
             CREATE SCHEMA schema_1;

internal/queries/queries.sql

@@ -342,7 +342,7 @@ WHERE
     AND seq_ns.nspname !~ '^pg_temp'
     -- Exclude sequences owned by identity columns.
     --  These manifest as internal dependency on the column
-    AND (depend.deptype IS NULL OR depend.deptype != 'i')
+    AND (depend.deptype IS null OR depend.deptype != 'i')
     -- Exclude sequences belonging to extensions
     AND NOT EXISTS (
         SELECT ext_depend.objid

internal/queries/queries.sql.go

@@ -865,7 +865,8 @@ func (s *schemaFetcher) buildTable(
 		}
 
 		var identity *ColumnIdentity
-		if len(column.IdentityType) > 0 {
+		if len(column.IdentityType) > 0 && table.ParentTableName == "" {
+			// Exclude identity columns from table partitions because they are owned by the parent.
 			identity = &ColumnIdentity{
 				Type:       ColumnIdentityType(column.IdentityType),
 				StartValue: column.StartValue.Int64,