增加pgx驱动，完善前端的代码，暂定supabase后端数据库

database/database.go

@@ -2,10 +2,12 @@ package db
 
 import (
 	"fmt"
+	//"gorm.io/driver/postgres"
+	"os"
+
 	"gorm.io/driver/postgres"
 	"gorm.io/driver/sqlite"
 	"gorm.io/gorm"
-	"os"
 	"zsxyww.com/scheduler/config"
 	"zsxyww.com/scheduler/model"
 )
@@ -17,12 +19,12 @@ func Connect() {
 	case "SQLite":
 		connectSQLite()
 	case "PostgreSQL":
-		//connectPGSQL()
+		connectPGSQL()
 		PGSQL()
 	default:
 		panic("DBType error")
 	}
-	Main.AutoMigrate(&model.Member{}, &model.Tweak{})
+	//Main.AutoMigrate(&model.Member{}, &model.Tweak{})
 }
 
 func connectSQLite() {
@@ -33,6 +35,7 @@ func connectSQLite() {
 		os.Exit(1)
 
 	}
+	Main.AutoMigrate(&model.Member{}, &model.Tweak{})
 }
 
 func connectPGSQL() {
@@ -40,4 +43,5 @@ func connectPGSQL() {
 	if err != nil {
 		panic(err)
 	}
+	Main.AutoMigrate(&model.Member{}, &model.Tweak{})
 }

database/model.go

@@ -1,7 +1,10 @@
 package db
 
 import (
+	"github.com/jackc/pgx/v5/pgxpool"
 	"gorm.io/gorm"
 )
 
 var Main *gorm.DB //Main database connection
+
+var PGX *pgxpool.Pool

database/pgsql.go

@@ -0,0 +1,26 @@
+package db
+
+import (
+	"context"
+	"fmt"
+	"github.com/jackc/pgx/v5/pgxpool"
+	"os"
+	"zsxyww.com/scheduler/config"
+)
+
+// use pgx to connect
+func PGSQL() {
+	pgx, err := pgxpool.New(context.Background(), config.Default.DB.Path)
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "Unable to create connection pool: %v\n", err)
+		os.Exit(1)
+	}
+	version := ""
+	if err := pgx.QueryRow(context.Background(), "SELECT version()").Scan(&version); err != nil {
+		fmt.Printf("Query failed: %v\n", err)
+		os.Exit(1)
+	}
+
+	fmt.Println("Connected to:", version)
+	PGX = pgx
+}