如何在 Go 中以惯用的方式将数据库访问权限转换为函数

Question

我在 Go 中构建了一个后端 API，它可以工作，但是我想将数据库访问层的代码重构为一个函数——惯用地。// Get the form data entered by client; FirstName, LastName, phone Number,// assign the person a unique i.d// check to see if that user isn't in the database already// if they are send an error message with the a  'bad' response code// if they aren't in db add to db and send a message with successfunc CreateStudentAccountEndpoint(response http.ResponseWriter, request *http.Request){    client, err := mongo.NewClient("mongodb://localhost:27017")    if err != nil {        log.Fatalf("Error connecting to mongoDB client Host: Err-> %v\n ", err)    }    ctx, cancel := context.WithTimeout(context.Background(), 20*time.Second)    defer cancel()    err = client.Connect(ctx)    if err != nil {        log.Fatalf("Error Connecting to MongoDB at context.WtihTimeout: Err-> %v\n ", err)    }    response.Header().Set("Content-Type", "application/json")    studentCollection := client.Database(dbName).Collection("students")    _, err = studentCollection.InsertOne(context.Background(),data)    if err != nil {        response.WriteHeader(501)        response.Write([]byte(`{ "message": "` + err.Error() + `" }`))    }    // encoding json object for returning to the client    jsonStudent, err := json.Marshal(student)    if err != nil {        http.Error(response, err.Error(), http.StatusInternalServerError)    }    response.Write(jsonStudent)}我知道我可以创建一个返回 (*mongoClient, err) 的方法，因为我稍后会在代码中使用客户端局部变量。但是我不知道如何实现该defer cancel()部分，因为它会在方法CreateStudenAccountEndpoint结束时执行。但是我不知道如何defer在一个方法中实现这个部分，该方法将识别我希望延迟发生在调用数据库访问层方法的函数的末尾，而不是CreateStudentAccountEndpoint实际的数据库访问方法本身。

Answer 1

据我了解，连接应该长期存在并设置为构造函数的一部分，即不是请求流的一部分。

这通常看起来像这样：

type BackendAPI struct {

    client *mongo.Client

}

func NewBackendAPI(mongoURI string) (*BackendAPI, error) {

    client, err := mongo.NewClient(mongoURI)

    if err != nil {

        return nil, err

    }

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

    defer cancel()

    err = client.Connect(ctx)

    if err != nil {

        return nil, err

    }

    return &BackendAPI{client}, nil

}

func (api *BackendAPI) func CreateStudentAccountEndpoint(response http.ResponseWriter, request *http.Request) {

    response.Header().Set("Content-Type", "application/json")

    // note the use of the long-lived api.client, which is connected already.

    studentCollection := api.client.Database(dbName).Collection("students")

    _, err = studentCollection.InsertOne(context.Background() ,data)

    if err != nil {

        response.WriteHeader(501)

        response.Write([]byte(`{ "message": "` + err.Error() + `" }`))

        return // at this point, the method should return

    }

    // encoding json object for returning to the client

    jsonStudent, err := json.Marshal(student)

    if err != nil {

        http.Error(response, err.Error(), http.StatusInternalServerError)

    }

    response.Write(jsonStudent)

}

如果您担心失去连接，您可以api.client.Ping在那里实现对 in 的调用，但在我看来，只有当您遇到您认为可以通过重新连接恢复的故障时才应尝试这样做。