Question

我正在学习Go，正在尝试创建一个具有'fields'参数的api端点。当我尝试将sqlx结果行扫描到结构中时，用户省略的字段将作为空字符串返回。有没有一种方法可以更改结构以仅反映用户传递的字段？我不希望使用omitempty，例如，如果user_name是一个空字符串。

type User struct {
    Id int `db:"id"`
    UserName string `db:"user_name"`
}


func GetUsers(w http.ResponseWriter,r *http.Request,db *sqlx.DB) {
    acceptedFields := map[string]bool {
        "id":true,
        "user_name":true,
    }
    var requestedFields string = "id"
    if r.URL.Query().Get("fields") != ""{
        requestedFields = r.URL.Query().Get("fields");
    }

    for _, requestedField := range strings.Split(requestedFields,",") {
        if !acceptedFields[requestedField] {
            http.Error(w, fmt.Sprintf("Unknown Field '%s'",requestedField), http.StatusBadRequest)
        }
    }

    users := []User{}

    err := db.Select(&users,fmt.Sprintf("SELECT %s FROM users",requestedFields)); 
    if err != nil {
        log.Fatal(err)
    }
    response, _ := json.Marshal(users)
    fmt.Fprintf(w,string(response))
}

最终的端点输出

/users?fields=id => [{"Id":12,"UserName":""}]

所需的端点输出

/users?fields=id => [{"Id":12}]

还使用sql.NullString会导致以下结果：

[{"Id":12,"UserName":{"String":"","Valid":false}}]

Answer 1

感谢mkorpriva这是一个解决方案

type User struct {
    Id int `db:"id"`
    UserName *string `db:"user_name" json:",omitempty"`
}

使用SQLX进入可选字段

1 个答案: