如何使用 Go 中的方法创建单例数据库实例？

带有方法的单例数据库实例

在 Go 中使用方法创建单例可以通过利用未导出的实现类型和导出的接口来实现。

使用接口和包初始化

使用所需的方法定义导出的接口，例如：

<code class="go">package dbprovider

type Manager interface {
    AddArticle(article *article.Article) error
}</code>

创建实现该接口的未导出类型：

<code class="go">type manager struct {
    db *gorm.DB
}</code>

在包初始化函数中初始化单例实例，该函数在任何包引用之前执行一次：

<code class="go">var Mgr Manager

func init() {
    db, err := gorm.Open("sqlite3", "../articles.db")
    if err != nil {
        log.Fatal("Failed to init db:", err)
    }
    Mgr = &manager{db: db}
}</code>

使用单例

通过引用导出的接口变量来使用单例实例，例如:

<code class="go">if err := dbprovider.Mgr.AddArticle(someArticle); err != nil {
    // Handle error
}</code>

gorm.Create(..) 中的异常处理

从 gorm.Create(..) 捕获并返回异常：

<code class="go">func (mgr *manager) AddArticle(article *article.Article) (err error) {
    mgr.db.Create(article)
    if errs := mgr.db.GetErrors(); len(errs) > 0 {
        err = errs[0]
    }
    return
}</code>

替代方法

除了使用包初始化函数之外，您还可以显式初始化单例：

<code class="go">var mgr = newManager()

func newManager() Manager {
    db, err := gorm.Open("sqlite3", "../articles.db")
    if err != nil {
        log.Fatal("Failed to init db:", err)
    }
    return &manager{db: db}
}</code>

这种方法允许用户决定是使用共享实例还是创建一个新实例，例如用于测试目的。

以上是如何使用 Go 中的方法创建单例数据库实例？的详细内容。更多信息请关注PHP中文网其他相关文章！