Proof of concept: River operating with pgx simple protocol

internal/riverinternaltest/riverinternaltest.go

@@ -73,6 +73,7 @@ func DatabaseConfig(databaseName string) *pgxpool.Config {
 	// are unlikely to succeed even with more time:
 	config.ConnConfig.ConnectTimeout = 2 * time.Second
 	config.ConnConfig.RuntimeParams["timezone"] = "UTC"
+	config.ConnConfig.DefaultQueryExecMode = pgx.QueryExecModeSimpleProtocol
 	return config
 }
 

riverdriver/riverdatabasesql/internal/dbsqlc/sqlc.yaml

@@ -26,6 +26,15 @@ sql:
           ttl: "TTL"
 
         overrides:
+          - db_type: "jsonb"
+            go_type: "string"
+
+          - db_type: "jsonb"
+            go_type:
+              type: "string"
+              pointer: true
+            nullable: true
+
           - db_type: "pg_catalog.interval"
             go_type: "time.Duration"
 
@@ -40,14 +49,6 @@ sql:
 
           # specific columns
 
-          # This one is necessary because `args` is nullable (this seems to have
-          # been an oversight, but one we're determined isn't worth correcting
-          # for now), and the `database/sql` variant of sqlc will give it a
-          # crazy type by default, so here we give it something more reasonable.
-          - column: "river_job.args"
-            go_type:
-              type: "[]byte"
-
           - column: "river_job.errors"
             go_type:
               type: "[]AttemptError"

riverdriver/riverpgxv5/internal/dbsqlc/river_job.sql

@@ -174,7 +174,7 @@ INSERT INTO river_job(
     @finalized_at,
     @kind::text,
     @max_attempts::smallint,
-    coalesce(@metadata::jsonb, '{}'),
+    coalesce(sqlc.narg('metadata')::jsonb, '{}'),
     @priority::smallint,
     @queue::text,
     coalesce(sqlc.narg('scheduled_at')::timestamptz, now()),
@@ -199,7 +199,7 @@ INSERT INTO river_job(
     state,
     tags
 ) VALUES (
-    @args::jsonb,
+    @args,
     coalesce(@attempt::smallint, 0),
     @attempted_at,
     coalesce(sqlc.narg('created_at')::timestamptz, now()),

riverdriver/riverpgxv5/internal/dbsqlc/sqlc.yaml

@@ -29,6 +29,15 @@ sql:
           - db_type: "pg_catalog.interval"
             go_type: "time.Duration"
 
+          - db_type: "jsonb"
+            go_type: "string"
+
+          - db_type: "jsonb"
+            go_type:
+              type: "string"
+              pointer: true
+            nullable: true
+
           - db_type: "timestamptz"
             go_type: "time.Time"