6.3 Mongo

6.3.1 第三方库

• go.mongodb.org/mongo-driver@v1.17.4 (opens new window)

6.3.2 配置参数

名称	类型	描述
dsn	string	连接串

6.3.3 示例代码

以下完整示例详见：third-mongo-example (opens new window)

1. 组件封装

文件位置：third-mongo-example/mongo/mongo.go (opens new window)

package mongo

import (
	"context"
	"fmt"
	"log"

	"github.com/dobyte/due/v2/core/pool"
	"github.com/dobyte/due/v2/etc"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
)

var factory = pool.NewFactory(func(name string) (*Client, error) {
	return NewInstance(fmt.Sprintf("etc.mongo.%s", name))
})

type (
	Client = mongo.Client
)

type Config struct {
	DSN string `json:"dsn"` // 连接串
}

// Instance 获取实例
func Instance(name ...string) *Client {
	var (
		err error
		ins *Client
	)

	if len(name) == 0 {
		ins, err = factory.Get("default")
	} else {
		ins, err = factory.Get(name[0])
	}

	if err != nil {
		log.Fatalf("create mongo instance failed: %v", err)
	}

	return ins
}

// NewInstance 新建实例
func NewInstance[T string | Config | *Config](config T) (*Client, error) {
	var (
		conf *Config
		v    any = config
		ctx      = context.Background()
	)

	switch c := v.(type) {
	case string:
		conf = &Config{DSN: etc.Get(fmt.Sprintf("%s.dsn", c)).String()}
	case Config:
		conf = &c
	case *Config:
		conf = c
	}

	opts := options.Client().ApplyURI(conf.DSN)

	client, err := mongo.Connect(ctx, opts)
	if err != nil {
		return nil, err
	}

	return client, nil
}

2. 配置示例

文件位置：third-mongo-example/etc/etc.toml (opens new window)

[mongo.default]
    # 连接串
    dsn = "mongodb://root:12345678@127.0.0.1:27017"

3. 组件调用

文件位置：third-mongo-example/main.go (opens new window)

package main

import (
	"context"
	"third-mongo-example/mongo"

	"github.com/dobyte/due/v2/log"
	"go.mongodb.org/mongo-driver/mongo/readpref"
)

func main() {
	client := mongo.Instance("default")

	if err := client.Ping(context.Background(), readpref.Primary()); err != nil {
		log.Errorf("mongo connection error: %v", err)
	} else {
		log.Infof("mongo connection success")
	}
}