如何将一个插入到 mongo 数据库

Question

我试图将文档插入 mongo 数据库。我尝试使用下面编写的代码，但无法插入结构数据。我正在使用 mongo-驱动程序。谁能帮我找出为什么我无法在 RegisterNFInstance 处理程序函数中执行插入？

这是我的代码。

const (
    COLLECTION = "nrfcoll"
)

db  *mongo.Database

func  Connect() {
    ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
    defer cancel()
    client, err := mongo.NewClient(options.Client().ApplyURI("mongodb://localhost:27017"))
    if err != nil {
        log.Fatal(err)
    }
    err = client.Connect(ctx)
    if err != nil {
        log.Fatal(err)
    }

    defer func() {
        if err = client.Disconnect(ctx); err != nil {
            panic(err)
        }
    }()
    // Check the connection
    err = client.Ping(ctx, readpref.Primary())
    if err != nil {
        log.Fatal(err)
    }
    // Collection types can be used to access the database
    db = client.Database("profiledb")
    fmt.Println("Successfully connected to MongoDB!")
}


type  NfProfile struct {

    NfInstanceId string `json:"nfInstanceId,omitempty" bson:"nfInstanceId,omitempty"`

    NfInstanceName string `json:"nfInstanceName,omitempty" bson:"nfInstanceName,omitempty"`

    NfType string `json:"nfType" bson:"nfType"`

    NfStatus string `json:"nfStatus,omitempty" bson:"nfStatus"`

}

func Insert(ctx context.Context, nfinstance NfProfile) bool {
    _, err := db.Collection(COLLECTION).InsertOne(ctx, nfinstance)
    if err != nil {
        return false
    }
    return true
}

func respondWithJson(response http.ResponseWriter, code int, payload interface{}) {
    res, err := json.Marshal(payload)
    if err != nil {
        response.WriteHeader(http.StatusInternalServerError)
        response.Write([]byte(err.Error()))
        return
    }
    response.Header().Set("Content-Type", "application/json; charset=UTF-8")
    response.WriteHeader(code)
    response.Write([]byte(res))

}


func RegisterNFInstance(response http.ResponseWriter, request *http.Request) {

    var nfProfile  NfProfile
    
    if request.Header.Get("Content-Type") != "application/json" {
        WriteError(response, ErrStatusUnsupportedMediaType)
        return
    }
        err := json.NewDecoder(request.Body).Decode(&nfProfile)
        if err != nil {
            WriteError(response, ErrBadRequest)
            return
        }
        defer request.Body.Close()

        ctx := context.Background()

        success := Insert(ctx, nfProfile)
        if !success {
            WriteError(response, ErrInternalServer)
            return
        } 

     respondWithJson(response, http.StatusCreated, nfProfile)
}

Answer 1

您的 Connect() 函数包含调用 client.Disconnect() 的延迟调用。这意味着在您 Connect() returns 之前，连接已经关闭。

删除此断开连接代码：

defer func() {
    if err = client.Disconnect(ctx); err != nil {
        panic(err)
    }
}()

如何将一个插入到 mongo 数据库

How to InsertOne to mongo database

mongodb

go

mongo-go