paopao-ce/internal/conf/db.go

// Copyright 2023 ROC. All rights reserved.
// Use of this source code is governed by a MIT style
// license that can be found in the LICENSE file.

package conf

import (
	"database/sql"
	"sync"

	"github.com/alimy/cfg"
	"github.com/go-redis/redis/v8"
	"github.com/sirupsen/logrus"
)

var (
	_sqldb               *sql.DB
	_redisClient         *redis.Client
	_onceSql, _onceRedis sync.Once
)

const (
	TableAnouncement        = "user"
	TableAnouncementContent = "anouncement_content"
	TableAttachment         = "attachment"
	TableCaptcha            = "captcha"
	TableComment            = "comment"
	TableCommentContent     = "comment_content"
	TableCommentReply       = "comment_reply"
	TableContact            = "contact"
	TableContactGroup       = "contact_group"
	TableMessage            = "message"
	TablePost               = "post"
	TablePostAttachmentBill = "post_attachment_bill"
	TablePostCollection     = "post_collection"
	TablePostContent        = "post_content"
	TablePostStar           = "post_star"
	TableTag                = "tag"
	TableUser               = "user"
	TableWalletRecharge     = "wallet_recharge"
	TableWalletStatement    = "wallet_statement"
)

type TableNameMap map[string]string

func MustSqlDB() *sql.DB {
	_onceSql.Do(func() {
		var err error
		if _, _sqldb, err = newSqlDB(); err != nil {
			logrus.Fatalf("new sql db failed: %s", err)
		}
	})
	return _sqldb
}

func MustRedis() *redis.Client {
	_onceRedis.Do(func() {
		_redisClient = redis.NewClient(&redis.Options{
			Addr:     redisSetting.Host,
			Password: redisSetting.Password,
			DB:       redisSetting.DB,
		})
	})
	return _redisClient
}

func newSqlDB() (driver string, db *sql.DB, err error) {
	if cfg.If("MySQL") {
		driver = "mysql"
		db, err = sql.Open(driver, MysqlSetting.Dsn())
	} else if cfg.If("PostgreSQL") || cfg.If("Postgres") {
		driver = "pgx"
		db, err = sql.Open(driver, PostgresSetting.Dsn())
	} else if cfg.If("Sqlite3") {
		driver, db, err = OpenSqlite3()
	} else {
		driver = "mysql"
		db, err = sql.Open(driver, MysqlSetting.Dsn())
	}
	return
}