shop.tonybtw.com

https://git.tonybtw.com/shop.tonybtw.com.git git://git.tonybtw.com/shop.tonybtw.com.git

424 bytes raw

package lib

import (
	"database/sql"
	"os"

	_ "modernc.org/sqlite"
)

func Connect_DB() (*sql.DB, error) {
	db_path := os.Getenv("DATABASE_PATH")
	if db_path == "" {
		db_path = "shop.db"
	}

	db, err := sql.Open("sqlite", db_path)
	if err != nil {
		return nil, err
	}

	if err := db.Ping(); err != nil {
		return nil, err
	}

	db.Exec("PRAGMA foreign_keys = ON")
	db.Exec("PRAGMA journal_mode = WAL")

	return db, nil
}

1	`package lib`
2
3	`import (`
4	`"database/sql"`
5	`"os"`
6
7	`_ "modernc.org/sqlite"`
8	`)`
9
10	`func Connect_DB() (*sql.DB, error) {`
11	`db_path := os.Getenv("DATABASE_PATH")`
12	`if db_path == "" {`
13	`db_path = "shop.db"`
14	`}`
15
16	`db, err := sql.Open("sqlite", db_path)`
17	`if err != nil {`
18	`return nil, err`
19	`}`
20
21	`if err := db.Ping(); err != nil {`
22	`return nil, err`
23	`}`
24
25	`db.Exec("PRAGMA foreign_keys = ON")`
26	`db.Exec("PRAGMA journal_mode = WAL")`
27
28	`return db, nil`
29	`}`