Mongo-go-driver 从插入结果中获取 objectID

使用Mongo-go-driver从插入结果中获取ObjectID

go get go.mongodb.org/mongo-driver

go
package main
import (
	"context"
	"fmt"
	"log"
	"time"
	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
)
func main() {
	// 设置MongoDB连接选项
	clientOptions := options.Client().ApplyURI("mongodb://localhost:27017")
	// 连接到MongoDB
	client, err := mongo.Connect(context.TODO(), clientOptions)
	if err != nil {
		log.Fatal(err)
	}
	// 检查连接
	err = client.Ping(context.TODO(), nil)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println("Connected to MongoDB!")
	// 获取要插入的数据
	data := bson.D{
		{Key: "name", Value: "John"},
		{Key: "age", Value: 30},
		{Key: "city", Value: "New York"},
	}
	// 选择数据库和集合
	collection := client.Database("mydatabase").Collection("mycollection")
	// 插入数据
	insertResult, err := collection.InsertOne(context.TODO(), data)
	if err != nil {
		log.Fatal(err)
	}
	// 获取插入数据的ObjectID
	objectID := insertResult.InsertedID.(primitive.ObjectID)
	fmt.Println("Inserted document with ID:", objectID.Hex())
	// 断开与MongoDB的连接
	err = client.Disconnect(context.TODO())
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println("Connection to MongoDB closed.")
}