memos/store/db.go

package store

import (
	"database/sql"
	"os"

	_ "github.com/mattn/go-sqlite3"
)

/*
 * Use a global variable to save the db connection: Quick and easy to setup.
 * Reference: https://techinscribed.com/different-approaches-to-pass-database-connection-into-controllers-in-golang/
 */
var DB *sql.DB

func InitDBConn() {
	dbFilePath := "/data/memos.db"

	if _, err := os.Stat(dbFilePath); err != nil {
		dbFilePath = "./resources/memos.db"
		resetDataInDefaultDatabase()
		println("use the default database")
	} else {
		println("use the custom database")
	}

	db, err := sql.Open("sqlite3", dbFilePath)

	if err != nil {
		println("connect failed")
	} else {
		DB = db
		println("connect to sqlite succeed")
	}
}

func FormatDBError(err error) error {
	if err == nil {
		return nil
	}

	switch err.Error() {
	default:
		return err
	}
}

func resetDataInDefaultDatabase() {
	// do nth
}