fix: register geos codecs in postgis example main.go (#5)

Co-authored-by: sebdotv <sebdotv@gmail.com>
twpayne · Aug 13, 2024 · b84b56f · b84b56f
1 parent 04dbb3e
commit b84b56f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/examples/postgis/main.go b/examples/postgis/main.go
@@ -5,6 +5,7 @@ import (
 	"encoding/json"
 	"flag"
 	"fmt"
+	pgxgeos "github.com/twpayne/pgx-geos"
 	"io"
 	"os"
 
@@ -61,6 +62,11 @@ func (w *Waypoint) UnmarshalJSON(data []byte) error {
 	return nil
 }
 
+// registerGeos registers required codecs
+func registerGeos(ctx context.Context, conn *pgx.Conn) error {
+	return pgxgeos.Register(ctx, conn, geos.NewContext())
+}
+
 // createDB demonstrates create a PostgreSQL/PostGIS database with a table with
 // a geometry column.
 func createDB(ctx context.Context, conn *pgx.Conn) error {
@@ -165,6 +171,9 @@ func run() error {
 		}
 	}
 	if *write {
+		if err := registerGeos(ctx, conn); err != nil {
+			return err
+		}
 		if err := writeGeoJSON(ctx, conn, os.Stdout); err != nil {
 			return err
 		}

diff --git a/examples/postgis/main_test.go b/examples/postgis/main_test.go
@@ -13,8 +13,6 @@ import (
 	"github.com/testcontainers/testcontainers-go"
 	"github.com/testcontainers/testcontainers-go/modules/postgres"
 	"github.com/testcontainers/testcontainers-go/wait"
-	"github.com/twpayne/go-geos"
-	pgxgeos "github.com/twpayne/pgx-geos"
 )
 
 func TestIntegration(t *testing.T) {
@@ -54,7 +52,7 @@ func TestIntegration(t *testing.T) {
 
 	conn, err := pgx.Connect(ctx, connStr)
 	assert.NoError(t, err)
-	assert.NoError(t, pgxgeos.Register(ctx, conn, geos.NewContext()))
+	assert.NoError(t, registerGeos(ctx, conn))
 
 	assert.NoError(t, createDB(ctx, conn))