71 lines
1.4 KiB
Go
71 lines
1.4 KiB
Go
package share
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/yaoapp/kun/log"
|
|
"github.com/yaoapp/xun/capsule"
|
|
"github.com/yaoapp/yao/config"
|
|
)
|
|
|
|
// DBConnect 建立数据库连接
|
|
func DBConnect(dbconfig config.DBConfig) (err error) {
|
|
|
|
if dbconfig.Primary == nil {
|
|
return fmt.Errorf("YAO_DB_PRIMARY was not set")
|
|
}
|
|
|
|
manager := capsule.New()
|
|
for i, dsn := range dbconfig.Primary {
|
|
_, err = manager.Add(fmt.Sprintf("primary-%d", i), dbconfig.Driver, dsn, false)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
if dbconfig.Secondary != nil {
|
|
for i, dsn := range dbconfig.Secondary {
|
|
_, err = manager.Add(fmt.Sprintf("secondary-%d", i), dbconfig.Driver, dsn, true)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
manager.SetAsGlobal()
|
|
go func() {
|
|
for _, c := range manager.Pool.Primary {
|
|
err = c.Ping(5 * time.Second)
|
|
if err != nil {
|
|
log.Error("%s error %v", c.Config.Name, err.Error())
|
|
}
|
|
}
|
|
}()
|
|
|
|
return err
|
|
}
|
|
|
|
// DBClose close the database connections
|
|
func DBClose() error {
|
|
messages := []string{}
|
|
capsule.Global.Connections.Range(func(key, value any) bool {
|
|
log.Trace("[DBClose] %s", key)
|
|
if conn, ok := value.(*capsule.Connection); ok {
|
|
err := conn.Close()
|
|
if err != nil {
|
|
messages = append(messages, err.Error())
|
|
}
|
|
}
|
|
return true
|
|
})
|
|
|
|
if len(messages) > 0 {
|
|
msg := fmt.Sprintf("[DBClose] %s ", strings.Join(messages, ";"))
|
|
log.Error(msg)
|
|
return fmt.Errorf(msg)
|
|
}
|
|
|
|
return nil
|
|
}
|