go任务调度10（操作mongodb）

Posted 2021-02-17

后台启动mongodb服务器端：

nohup ./bin/mongod --dbpath=./data &

package main

//导入
import (
    "context"
    "fmt"
    "go.mongodb.org/mongo-driver/bson"
    "go.mongodb.org/mongo-driver/bson/primitive"
    "go.mongodb.org/mongo-driver/mongo"
    "go.mongodb.org/mongo-driver/mongo/options"
    "go.mongodb.org/mongo-driver/mongo/readpref"
    "os"
    "time"
)

type Howie struct {
    //struct里面获取ObjectID
    HowieId    primitive.ObjectID `bson:"_id"`
    Name       string
    Pwd        string
    Age        int64
    CreateTime int64
}

func main() {
    TestMongo("mongodb://0.0.0.0:27017")
}

func TestMongo(url string) {
    var (
        err             error
        client          *mongo.Client
        collection      *mongo.Collection
        insertOneRes    *mongo.InsertOneResult
        insertManyRes   *mongo.InsertManyResult
        delRes          *mongo.DeleteResult
        updateRes       *mongo.UpdateResult
        cursor          *mongo.Cursor
        howieArray      = GetHowieArray()
        howie           Howie
        howieArrayEmpty []Howie
        size            int64
    )
    //链接mongo服务
    if client, err = mongo.Connect(getContext(), options.Client().ApplyURI(url)); err != nil {
        checkErr(err)
    }
    //判断服务是否可用
    if err = client.Ping(getContext(), readpref.Primary()); err != nil {
        checkErr(err)
    }
    //选择数据库和集合
    collection = client.Database("testing_base").Collection("howie")

    //删除这个集合
    collection.Drop(getContext())

    //插入一条数据
    if insertOneRes, err = collection.InsertOne(getContext(), howieArray[0]); err != nil {
        checkErr(err)
    }

    fmt.Printf("InsertOne插入的消息ID:%v
", insertOneRes.InsertedID)
    //批量插入数据
    if insertManyRes, err = collection.InsertMany(getContext(), howieArray[1:]); err != nil {
        checkErr(err)
    }
    fmt.Printf("InsertMany插入的消息ID:%v
", insertManyRes.InsertedIDs)
    var Dinfo = make(map[string]interface{})
    err = collection.FindOne(getContext(), bson.D{{"name", "howie_2"}, {"age", 11}}).Decode(&Dinfo)
    fmt.Println(Dinfo)
    fmt.Println("_id", Dinfo["_id"])

    //查询单条数据
    if err = collection.FindOne(getContext(), bson.D{{"name", "howie_2"}, {"age", 11}}).Decode(&howie); err != nil {
        checkErr(err)
    }
    fmt.Printf("FindOne查询到的数据:%v
", howie)

    //查询单条数据后删除该数据
    if err = collection.FindOneAndDelete(getContext(), bson.D{{"name", "howie_3"}}).Decode(&howie); err != nil {
        checkErr(err)
    }
    fmt.Printf("FindOneAndDelete查询到的数据:%v
", howie)

    //查询单条数据后修改该数据
    if err = collection.FindOneAndUpdate(getContext(), bson.D{{"name", "howie_4"}}, bson.M{"$set": bson.M{"name": "这条数据我需要修改了"}}).Decode(&howie); err != nil {
        checkErr(err)
    }
    fmt.Printf("FindOneAndUpdate查询到的数据:%v
", howie)

    //查询单条数据后替换该数据(以前的数据全部清空)
    if err = collection.FindOneAndReplace(getContext(), bson.D{{"name", "howie_5"}}, bson.M{"hero": "这条数据我替换了"}).Decode(&howie); err != nil {
        checkErr(err)
    }

    fmt.Printf("FindOneAndReplace查询到的数据:%v
", howie)
    //一次查询多条数据
    // 查询createtime>=3
    // 限制取2条
    // createtime从大到小排序的数据
    if cursor, err = collection.Find(getContext(), bson.M{"createtime": bson.M{"$gte": 2}}, options.Find().SetLimit(2), options.Find().SetSort(bson.M{"createtime": -1})); err != nil {
        checkErr(err)
    }
    if err = cursor.Err(); err != nil {
        checkErr(err)
    }
    defer cursor.Close(context.Background())
    for cursor.Next(context.Background()) {
        if err = cursor.Decode(&howie); err != nil {
            checkErr(err)
        }
        howieArrayEmpty = append(howieArrayEmpty, howie)
    }
    for _, v := range howieArrayEmpty {
        fmt.Printf("Find查询到的数据ObejectId值%s 值:%v
", v.HowieId.Hex(), v)
    }
    //查询集合里面有多少数据
    if size, err = collection.CountDocuments(getContext(),bson.D{}); err != nil {
        checkErr(err)
    }
    fmt.Printf("Count里面有多少条数据:%d
", size)

    //查询集合里面有多少数据(查询createtime>=3的数据)
    if size, err = collection.CountDocuments(getContext(), bson.M{"createtime": bson.M{"$gte": 3}}); err != nil {
        checkErr(err)
    }
    fmt.Printf("Count里面有多少条数据:%d
", size)

    //修改一条数据
    if updateRes, err = collection.UpdateOne(getContext(), bson.M{"name": "howie_2"}, bson.M{"$set": bson.M{"name": "我要改了他的名字"}}); err != nil {
        checkErr(err)
    }
    fmt.Printf("UpdateOne的数据:%d
", updateRes)

    //修改多条数据
    if updateRes, err = collection.UpdateMany(getContext(), bson.M{"createtime": bson.M{"$gte": 3}}, bson.M{"$set": bson.M{"name": "我要批量改了他的名字"}}); err != nil {
        checkErr(err)
    }

    fmt.Printf("UpdateMany的数据:%d
", updateRes)
    //删除一条数据
    if delRes, err = collection.DeleteOne(getContext(), bson.M{"name": "howie_1"}); err != nil {
        checkErr(err)
    }
    fmt.Printf("DeleteOne删除了多少条数据:%d
", delRes.DeletedCount)

    //删除多条数据
    if delRes, err = collection.DeleteMany(getContext(), bson.M{"createtime": bson.M{"$gte": 7}}); err != nil {
        checkErr(err)
    }
    fmt.Printf("DeleteMany删除了多少条数据:%d
", delRes.DeletedCount)

}

func checkErr(err error) {
    if err != nil {
        if err == mongo.ErrNoDocuments {
            fmt.Println("没有查到数据")
            os.Exit(0)
        } else {
            fmt.Println(err)
            os.Exit(0)
        }

    }
}

func getContext() (ctx context.Context) {
    ctx, _ = context.WithTimeout(context.Background(), 10*time.Second)
    return
}

func GetHowieArray() (data []interface{}) {
    var i int64
    for i = 0; i <= 10; i++ {
        data = append(data, Howie{
            HowieId:    primitive.NewObjectID(),
            Name:       fmt.Sprintf("howie_%d", i+1),
            Pwd:        fmt.Sprintf("pwd_%d", i+1),
            Age:        i + 10,
            CreateTime: i + 1,
        })
    }
    return
}

注意：从github.com/mongodb/mongo-go-driver下载下来要放到go.mongodb.org目录对应的位置，不然会报错