feat: init repo

templates/project/providers/postgres/config.go.tpl

@@ -0,0 +1,79 @@
+package postgres
+
+import (
+	"fmt"
+
+	"git.ipao.vip/rogeecn/atom/container"
+	"git.ipao.vip/rogeecn/atom/utils/opt"
+)
+
+const DefaultPrefix = "Database"
+
+func DefaultProvider() container.ProviderContainer {
+	return container.ProviderContainer{
+		Provider: Provide,
+		Options: []opt.Option{
+			opt.Prefix(DefaultPrefix),
+		},
+	}
+}
+
+type Config struct {
+	Username     string
+	Password     string
+	Database     string
+	Schema       string
+	Host         string
+	Port         uint
+	SslMode      string
+	TimeZone     string
+	Prefix       string // 表前缀
+	Singular     bool   // 是否开启全局禁用复数，true表示开启
+	MaxIdleConns int    // 空闲中的最大连接数
+	MaxOpenConns int    // 打开到数据库的最大连接数
+}
+
+func (m *Config) checkDefault() {
+	if m.MaxIdleConns == 0 {
+		m.MaxIdleConns = 10
+	}
+
+	if m.MaxOpenConns == 0 {
+		m.MaxOpenConns = 100
+	}
+
+	if m.Username == "" {
+		m.Username = "postgres"
+	}
+
+	if m.SslMode == "" {
+		m.SslMode = "disable"
+	}
+
+	if m.TimeZone == "" {
+		m.TimeZone = "Asia/Shanghai"
+	}
+
+	if m.Port == 0 {
+		m.Port = 5432
+	}
+
+	if m.Schema == "" {
+		m.Schema = "public"
+	}
+}
+
+func (m *Config) EmptyDsn() string {
+	dsnTpl := "host=%s user=%s password=%s port=%d dbname=%s sslmode=%s TimeZone=%s"
+	m.checkDefault()
+
+	return fmt.Sprintf(dsnTpl, m.Host, m.Username, m.Password, m.Port, m.Database, m.SslMode, m.TimeZone)
+}
+
+// DSN connection dsn
+func (m *Config) DSN() string {
+	dsnTpl := "host=%s user=%s password=%s dbname=%s port=%d sslmode=%s TimeZone=%s"
+	m.checkDefault()
+
+	return fmt.Sprintf(dsnTpl, m.Host, m.Username, m.Password, m.Database, m.Port, m.SslMode, m.TimeZone)
+}

templates/project/providers/postgres/postgres.go.tpl

@@ -0,0 +1,34 @@
+package postgres
+
+import (
+	"database/sql"
+
+	"git.ipao.vip/rogeecn/atom/container"
+	"git.ipao.vip/rogeecn/atom/utils/opt"
+	_ "github.com/lib/pq"
+	"github.com/pkg/errors"
+	log "github.com/sirupsen/logrus"
+)
+
+func Provide(opts ...opt.Option) error {
+	o := opt.New(opts...)
+	var conf Config
+	if err := o.UnmarshalConfig(&conf); err != nil {
+		return err
+	}
+
+	return container.Container.Provide(func() (*sql.DB, *Config, error) {
+		log.Debugf("connect postgres with dsn: '%s'", conf.DSN())
+		db, err := sql.Open("postgres", conf.DSN())
+		if err != nil {
+			return nil, nil, errors.Wrap(err, "connect database")
+		}
+
+		if err := db.Ping(); err != nil {
+			db.Close()
+			return nil, nil, errors.Wrap(err, "ping database")
+		}
+
+		return db, &conf, err
+	}, o.DiOptions()...)
+}