From 8d9b8708201efeac698f78c8a347c14b0be78b56 Mon Sep 17 00:00:00 2001
From: Sebastian Tobie <sebastian@sebastian-tobie.de>
Date: Fri, 18 Nov 2022 14:10:59 +0100
Subject: [PATCH] added a small helper function

---
 middleware/db/db.go | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/middleware/db/db.go b/middleware/db/db.go
index 387e804..863f550 100644
--- a/middleware/db/db.go
+++ b/middleware/db/db.go
@@ -3,6 +3,7 @@ package db
 
 import (
 	"context"
+	"errors"
 	"fmt"
 	"io/fs"
 	"sync"
@@ -25,6 +26,14 @@ var _ ConnGet = NewMiddleware().GetConn
 var _ middleware.Middleware = &Middleware{}
 var _ middleware.PostSetupMiddleware = &Middleware{}
 
+// GetConnection is an simple helper function that returns an connection to the db
+func GetConnection(c *gin.Context, db string) (*pgxpool.Conn, error) {
+	if co, ok := c.Get(ContextKey); ok {
+		return co.(ConnGet)(db), nil
+	}
+	return nil, errors.New("No db.Middleware set up. ")
+}
+
 // Middleware return a handler that sets the db into the context of every request.
 // uri is an url in the form dbtype:connectargs
 type Middleware struct {