添加mongodb使用,添加用户前端配置的上传及获取接口
This commit is contained in:
parent
a2e0dae65a
commit
97dfd18eb4
20
dao/db.go
20
dao/db.go
|
|
@ -148,21 +148,25 @@ var mongoClient *mongo.Client
|
||||||
|
|
||||||
//var collection *mongo.Collection
|
//var collection *mongo.Collection
|
||||||
|
|
||||||
func InitMongoDB() {
|
func InitMongoDB() error {
|
||||||
// 设置 MongoDB 客户端选项
|
// 设置 MongoDB 客户端选项
|
||||||
clientOptions := options.Client().ApplyURI(proto.Config.MONGO_URI)
|
clientOptions := options.Client().ApplyURI(proto.Config.MONGO_URI)
|
||||||
// 连接到 MongoDB
|
// 连接到 MongoDB
|
||||||
var err error
|
var err error
|
||||||
mongoClient, err = mongo.Connect(context.TODO(), clientOptions)
|
mongoClient, err = mongo.Connect(context.TODO(), clientOptions)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal(err)
|
log.Println("Error connecting to MongoDB:", err)
|
||||||
|
} else {
|
||||||
|
// 检查连接
|
||||||
|
err = mongoClient.Ping(context.TODO(), nil)
|
||||||
|
if err != nil {
|
||||||
|
log.Println("Error pinging MongoDB:", err)
|
||||||
|
} else {
|
||||||
|
log.Println("Connected to MongoDB!")
|
||||||
|
}
|
||||||
}
|
}
|
||||||
// 检查连接
|
|
||||||
err = mongoClient.Ping(context.TODO(), nil)
|
return err
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
log.Println("Connected to MongoDB!")
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func CloseMongoDB() {
|
func CloseMongoDB() {
|
||||||
|
|
|
||||||
4
main.go
4
main.go
|
|
@ -27,6 +27,10 @@ func main() {
|
||||||
if err != nil {
|
if err != nil {
|
||||||
panic("failed to connect database:" + err.Error())
|
panic("failed to connect database:" + err.Error())
|
||||||
}
|
}
|
||||||
|
err = dao.InitMongoDB()
|
||||||
|
if err != nil {
|
||||||
|
panic("failed to connect mongodb:" + err.Error())
|
||||||
|
}
|
||||||
err = worker.InitRedis()
|
err = worker.InitRedis()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
panic("failed to connect redis:" + err.Error())
|
panic("failed to connect redis:" + err.Error())
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue