HIMBOT RETURNS
This commit is contained in:
29
lib/db.go
29
lib/db.go
@ -17,35 +17,16 @@ var DBClient *sql.DB
|
||||
var DBConnector *libsql.Connector
|
||||
|
||||
func InitDB() error {
|
||||
dbUrl := os.Getenv("DATABASE_URL")
|
||||
dbToken := os.Getenv("DATABASE_AUTH_TOKEN")
|
||||
|
||||
if dbUrl == "" || dbToken == "" {
|
||||
return fmt.Errorf("database configuration missing")
|
||||
}
|
||||
|
||||
// Determine DB path based on /data directory existence
|
||||
dbPath := "himbot.db" // default to local
|
||||
if _, err := os.Stat("/data"); !os.IsNotExist(err) {
|
||||
dbPath = "/data/himbot.db"
|
||||
}
|
||||
dbName := "file:./himbot.db"
|
||||
|
||||
connector, connectorError := libsql.NewEmbeddedReplicaConnector(
|
||||
dbPath,
|
||||
dbUrl,
|
||||
libsql.WithAuthToken(dbToken),
|
||||
)
|
||||
|
||||
if connectorError != nil {
|
||||
fmt.Fprintf(os.Stderr, "failed to open db %s: %s", dbUrl, connectorError)
|
||||
db, err := sql.Open("libsql", dbName)
|
||||
if err != nil {
|
||||
fmt.Fprintf(os.Stderr, "failed to open db %s", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
// finalDBUrl := fmt.Sprintf("%s?authToken=%s", dbUrl, dbToken)
|
||||
|
||||
client := sql.OpenDB(connector)
|
||||
|
||||
DBClient = client
|
||||
DBConnector = connector
|
||||
DBClient = db
|
||||
|
||||
return runMigrations()
|
||||
}
|
||||
|
Reference in New Issue
Block a user