package model

import (

"context"

_ "fmt"

"go.mongodb.org/mongo-driver/mongo"

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

"log"

"time"

)

type mgo struct {

uri        string //数据库网络地址

database   string //要连接的数据库

//collection string //要连接的集合

}

var (

DB *mongo.Database

)

func Connect() (*mongo.Database, error) {

var m = &mgo{

"mongodb://localhost:27017",

"数据库名",

//"数据库表名",

}

ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)

defer cancel()

client, err := mongo.Connect(ctx, options.Client().ApplyURI(m.uri))

if err != nil {

log.Print(err)

}

DB = client.Database(m.database)

return DB, err

}