MongoDB集成 #

go get go.mongodb.org/mongo-driver/mongo
go get go.mongodb.org/mongo-driver/mongo/options

package database

import (
    "context"
    "go.mongodb.org/mongo-driver/mongo"
    "go.mongodb.org/mongo-driver/mongo/options"
)

func ConnectMongo(uri string) (*mongo.Client, error) {
    client, err := mongo.Connect(context.Background(), options.Client().ApplyURI(uri))
    if err != nil {
        return nil, err
    }
    
    err = client.Ping(context.Background(), nil)
    return client, err
}

func GetCollection(client *mongo.Client, dbName, collectionName string) *mongo.Collection {
    return client.Database(dbName).Collection(collectionName)
}

package models

import "go.mongodb.org/mongo-driver/bson/primitive"

type User struct {
    ID       primitive.ObjectID `bson:"_id,omitempty" json:"id"`
    Name     string             `bson:"name" json:"name"`
    Email    string             `bson:"email" json:"email"`
    Age      int                `bson:"age" json:"age"`
    Active   bool               `bson:"active" json:"active"`
    Tags     []string           `bson:"tags" json:"tags"`
}

func CreateUser(ctx context.Context, col *mongo.Collection, user *models.User) error {
    result, err := col.InsertOne(ctx, user)
    if err != nil {
        return err
    }
    
    user.ID = result.InsertedID.(primitive.ObjectID)
    return nil
}

func GetUserByID(ctx context.Context, col *mongo.Collection, id primitive.ObjectID) (*models.User, error) {
    var user models.User
    err := col.FindOne(ctx, bson.M{"_id": id}).Decode(&user)
    return &user, err
}

func GetUsers(ctx context.Context, col *mongo.Collection, filter bson.M) ([]models.User, error) {
    cursor, err := col.Find(ctx, filter)
    if err != nil {
        return nil, err
    }
    defer cursor.Close(ctx)
    
    var users []models.User
    if err := cursor.All(ctx, &users); err != nil {
        return nil, err
    }
    
    return users, nil
}

func UpdateUser(ctx context.Context, col *mongo.Collection, id primitive.ObjectID, update bson.M) error {
    _, err := col.UpdateOne(ctx, bson.M{"_id": id}, bson.M{"$set": update})
    return err
}

func DeleteUser(ctx context.Context, col *mongo.Collection, id primitive.ObjectID) error {
    _, err := col.DeleteOne(ctx, bson.M{"_id": id})
    return err
}