Added conn to pgqueue

Author: djthorpe

pkg/pgqueue/config/task.go

@@ -29,6 +29,7 @@ type task struct {
 }
 
 var _ server.Task = (*task)(nil)
+var _ server.PGQueue = (*task)(nil)
 
 ////////////////////////////////////////////////////////////////////////////////
 // LIFECYCLE
@@ -154,6 +155,11 @@ FOR_LOOP:
 ////////////////////////////////////////////////////////////////////////////////
 // PUBLIC METHODS
 
+// Conn returns the underlying connection pool object.
+func (t *task) Conn() pg.PoolConn {
+	return t.manager.Conn()
+}
+
 // RegisterTicker registers a periodic task (ticker) with a callback function.
 // It returns the metadata of the registered ticker.
 func (t *task) RegisterTicker(ctx context.Context, meta schema.TickerMeta, fn server.PGCallback) (*schema.Ticker, error) {

pkg/pgqueue/manager.go

@@ -90,6 +90,10 @@ func (manager *Manager) Worker() string {
 	return manager.worker
 }
 
+func (manager *Manager) Conn() pg.PoolConn {
+	return manager.conn
+}
+
 ////////////////////////////////////////////////////////////////////////////////
 // PUBLIC METHODS - TICKER
 

plugin.go

@@ -116,6 +116,9 @@ type PGCallback func(context.Context, any) error
 
 // PGQueue defines methods for interacting with a PostgreSQL-backed task queue.
 type PGQueue interface {
+	// Conn returns the underlying connection pool object.
+	Conn() pg.PoolConn
+
 	// RegisterTicker registers a periodic task (ticker) with a callback function.
 	// It returns the metadata of the registered ticker.
 	RegisterTicker(context.Context, pgschema.TickerMeta, PGCallback) (*pgschema.Ticker, error)