From 7813db6cf248d0c17222ff0d3030bb78950b070b Mon Sep 17 00:00:00 2001
From: Runxi Yu <me@runxiyu.org>
Date: Wed, 12 Feb 2025 00:03:17 +0800
Subject: config.go, etc.: Add PostgreSQL

---
 config.go | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'config.go')

diff --git a/config.go b/config.go
index 6261eb2..883b41f 100644
--- a/config.go
+++ b/config.go
@@ -2,11 +2,18 @@ package main
 
 import (
 	"bufio"
+	"context"
+	"errors"
 	"os"
 
+	"github.com/jackc/pgx/v5/pgxpool"
 	"go.lindenii.runxiyu.org/lindenii-common/scfg"
 )
 
+var database *pgxpool.Pool
+
+var err_unsupported_database_type = errors.New("Unsupported database type")
+
 var config struct {
 	HTTP struct {
 		Net  string `scfg:"net"`
@@ -33,5 +40,13 @@ func load_config(path string) (err error) {
 		return err
 	}
 
+	if config.DB.Type != "postgres" {
+		return err_unsupported_database_type
+	}
+	database, err = pgxpool.New(context.Background(), config.DB.Conn)
+	if err != nil {
+		return err
+	}
+
 	return nil
 }
-- 
cgit v1.2.3