筛选 MongoDB 在 Go api 中呈现的数据

一分耕耘，一分收获！既然打开了这篇文章《筛选 MongoDB 在 Go api 中呈现的数据》，就坚持看下去吧！文中内容包含等等知识点...希望你能在阅读本文后，能真真实实学到知识或者帮你解决心中的疑惑，也欢迎大佬或者新人朋友们多留言评论，多给建议！谢谢！

我是 go 和 react 新手，我在这个迷你项目中使用了这两种语言。 go 正在使用 mongodb 运行后端 api。

我正在 go 中从 mongo 获取用户列表，然后将其发送到 react，问题是 mongo 为我提供了用户的所有字段（_id、密码和用户名），我只想要用户名。我不明白如何过滤它并防止所有字段从 go 发送到 react。

go api 的 json 输出：

{
    "success": true,
    "data": [
        {
            "_id": "6205ac3d72c15c920a424608",
            "password": {
                "subtype": 0,
                "data": "removed for security"
            },
            "username": "removed for security"
        },
        {
            "_id": "6206b44afb8b044fdba76b8f",
            "password": {
                "subtype": 0,
                "data": "removed for security"
            },
            "username": "removed for security"
        }
    ]
}

这是我针对指定路线的 go 代码：

// Route: Get Users, for getting a list of users
func RouteGetUsers(w http.ResponseWriter, r *http.Request, p httprouter.Params) {
    // Set content-type to JSON
    w.Header().Set("Content-Type", "application/json")

    type Response struct {
        Success bool     `json:"Success"`
        Data    []bson.M `json:"Data"`
    }

    // Load the env file
    err := godotenv.Load("variables.env")
    if err != nil {
        log.Fatal("Error loading .env file")
    }

    // Get Mongo DB environment variable
    uri := os.Getenv("MONGO_URI")
    if uri == "" {
        log.Fatal("You must set your 'MONGO_URI' environmental variable. See\n\t https://docs.mongodb.com/drivers/go/current/usage-examples/#environment-variable")
    }

    // Connect to Mongo Database
    client, err := mongo.Connect(context.TODO(), options.Client().ApplyURI(uri))
    if err != nil {
        panic(err)
    }

    // Close the database connection at the end of the function
    defer func() {
        if err := client.Disconnect(context.TODO()); err != nil {
            panic(err)
        }
    }()

    // Select the database name and collection name
    coll := client.Database("go_project1").Collection("users")

    // Query the database for the user list
    cursor, err := coll.Find(context.TODO(), bson.D{})
    // If no documents were found, send a response and return
    if err == mongo.ErrNoDocuments {
        fmt.Printf("No documents were found")
        return
    }

    // Setup a variable for the database results
    var results []bson.M

    // Send all database results to results variable
    if err = cursor.All(context.TODO(), &results); err != nil {
        panic(err)
    }

    // Setup a variable with the ResponseStandard struct
    response := Response{
        Success: true,
        Data:    results,
    }

    // Marshal into JSON
    responseJson, err := json.Marshal(response)
    if err != nil {
        fmt.Println(err)
    }

    // Send success response to user in JSON
    fmt.Fprintf(w, "%s\n", responseJson)
}

正确答案

使用投影选项：

opts := options.find().setprojection(bson.d{{"username", 1}})
cursor, err := coll.find(context.todo(), bson.d{}, opts)

第二种方法：声明一个包含所需字段的类型，然后获取该类型。

type data struct {
   username string `bson:"username" json:"username"`
}

...

var data []data
if err = cursor.all(context.todo(), &data); err != nil { ...

...

var response = struct {
    success bool     `json:"success"`
    data    []data   `json:"data"`
}{
    true,
    data,
}
responsejson, err := json.marshal(response)
...

第三种方法：过滤问题中的地图：

for _, result := range results {
    for k := range result {
       if k != "username" {
          delete(result, k)
       }
    }
}

今天关于《筛选 MongoDB 在 Go api 中呈现的数据》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！