在golang中如何实现main()等待直到init()完成？

学习Golang要努力，但是不要急！今天的这篇文章《在golang中如何实现main()等待直到init()完成？》将会介绍到等等知识点，如果你想深入学习Golang，可以关注我！我会持续更新相关文章的，希望对大家都能有所帮助！

当我使用以下代码运行 main.go 时，它可以正常工作并且客户端连接到 mongo 数据库。

package main

import (
    "context"
    "fmt"
    "log"

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

type trainer struct {
    name string
    age  int
    city string
}

func main() {
    clientoptions := options.client().applyuri("mongodb://localhost:27017")

    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("successful connection")
}

但是，当我尝试分离 init() 和 main() goroutine 之间的代码逻辑时，我收到内存引用错误，这是正常的，因为 main 在 init 实际上与数据库建立 tcp 连接之前执行。我尝试将它们与通道连接，但它没有按预期工作。

package main

import (
    "context"
    "fmt"
    "log"

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

type Trainer struct {
    Name string
    Age  int
    City string
}

var client *mongo.Client
var c = make(chan *mongo.Client)

func init() {
    // configures the connection options
    clientOptions := options.Client().ApplyURI("mongodb://localhost:27017")

    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)
    }

    c <- client
}

func main() {
    fmt.Println(<-c)
}

由于我没有足够的 golang 经验，谁能解释一下为什么我的解决方案不起作用以及如何修复它？如果可能的话，我希望 init() 和 main() 分开。

解决方案

您的代码存在三个问题：

1) 函数 init 和 main 不是 goroutine，它们在程序启动时按顺序运行。因此使用无缓冲通道是没有意义的。

2) 变量 client 在 init 中使用“:=”运算符重新声明。

3) 我不建议在 init 中初始化客户端连接。如有必要，请将代码放入辅助函数中并从 main 中调用它。

覆盖（1）和（2）并排除（3）的修复是：

package main

import (
    "context"
    "fmt"
    "log"

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

type Trainer struct {
    Name string
    Age  int
    City string
}

var client *mongo.Client
// We don't need a channel since init and main are not goroutines. Both
// are executed sequentially in the main thread.
// var c = make(chan *mongo.Client)


// It is not recommended to use init to setup database connections. It
// is definitely part of the program and belongs into main.
func init() {
    // configures the connection options
    clientOptions := options.Client().ApplyURI("mongodb://localhost:27017")

    // Don't redeclare client here. We have however to declare err.
    var err error
    // Note that we use assignment (=) and not declaration (:=).
    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)
    }
}

func main() {
    // Use the global client variable instead reading from the channel.
    fmt.Println(client)
}

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。