Add route shapees saving with a idempotent upsert

Co-Authored-By: David Costa <david@zarel.net>
phoops · Nov 29, 2021 · 0aa9b28 · 0aa9b28
1 parent 5bae170
commit 0aa9b28
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/internal/infrastructure/postgres/client.go b/internal/infrastructure/postgres/client.go
@@ -59,7 +59,15 @@ func (c *Client) SetRouteShapes(
 			"generated",
 			"geometry",
 			"centroid",
-		)
+		).
+		Suffix(`
+			on conflict (route_id, direction_id) do update set
+			feed_version_id = excluded.feed_version_id,
+			shape_id = excluded.shape_id,
+			"generated" = excluded.generated,
+			geometry = excluded.geometry,
+			centroid = excluded.centroid
+		`)
 
 	rs := chunkRouteShapes(routeShapes, routeShapesInsertChunkSize)