Http-сервер Golang возвращает html или json в зависимости от типа контента - PullRequest
Новая
       47

Http-сервер Golang возвращает html или json в зависимости от типа контента

0 голосов
/ 23 июня 2019

Я пытаюсь написать простое приложение RESTful на Голанге, используя гориллу мукс. Я написал несколько обработчиков, которые выглядят следующим образом: 

func getUser(w http.ResponseWriter, r *http.Request) {
    if r.Header.Get("Content-type") == "application/json" {
        w.Header().Set("Content-Type", "application/json")
        u, err := _getUser(r)
        if err != nil {
            http.NotFound(w, r)
            return
        }
        json.NewEncoder(w).Encode(u) //asked for json, return json
    } else {
        w.Header().Set("Content-Type", "text/html")
        u, err := _getUser(r)
        if err != nil {
            http.NotFound(w, r)
            return
        }
        renderTemplate(w, "view", u) // asked for html, return html
    }
}
func _getUser(r *http.Request) (*User, error) {
    params := mux.Vars(r)
    for _, u := range users {
        if u.ID == params["id"] {
            return &u, nil
        }
    }
    return nil, errors.New("")
}

func main() {
    router := mux.NewRouter()
    router.HandleFunc("/v1/users/{id}", getUser).Methods("GET")
}

Проблема, с которой я столкнулся, заключается в том, что у меня много дубликатов. Каждый метод CRUD должен проверять тип содержимого и возвращать либо json, либо html.

Я думал о написании закрытия 

func htmlOrJSON(fn func(http.ResponseWriter, *http.Request) (interface {}, error), templateName string) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        if r.Header.Get("Content-type") == "application/json" {
            w.Header().Set("Content-Type", "application/json")
            result, err := fn(w, r)
            if err != nil {
                http.NotFound(w, r)
                return
            }
            json.NewEncoder(w).Encode(result)
        } else {
            w.Header().Set("Content-Type", "text/html")
            result, err := fn(w, r)
            if err != nil {
                http.NotFound(w, r)
                return
            }
            renderTemplate(w, templateName, result)
        }
    }
}

// and use as:
router.HandleFunc("/v1/users/{id}", htmlOrJSON(getUser, "view")).Methods("GET")

Удалить дублирование, но оно также выглядит не очень хорошо. Может ли кто-нибудь помочь мне сделать этот код чище?

1 Ответ

1 голос
/ 23 июня 2019

Хотя это вопрос проверки кода и должен быть в сообществе CodeReview, я постараюсь ответить на него.

Напишите универсальную функцию, которая обрабатывает рендеринг HTML и JSON.Обработка ошибок IMO должна происходить на каждом обработчике, даже если вы дублируете некоторый код.Это имеет больше смысла и делает код более читабельным и понятным.Вскоре вы увидите, что будут другие ошибки, требующие специальной обработки.

По логике, большинство API принимают параметры запроса http://api.com/user/1?fomtat=json.Это имеет больше смысла, потому что, когда клиент принимает больше, чем типы контента, вы застреваете. 

const JSON = "application/json"

func getUser(w http.ResponseWriter, r *http.Request) {
    u, err := _getUser(r)
    if err != nil {
        http.NotFound(w, r)
        return
    }
    responseBody(u, r.Header.Get("Content-type"), &w)
}

func responseBody(u User, contentType string, w io.writer) {
    switch contentType {
    case JSON:
        w.Header().Set("Content-Type", JSON)
        json.NewEncoder(w).Encode(u) //asked for json, return json
    default:
        w.Header().Set("Content-Type", "text/html")
        renderTemplate(w, "view", u) // asked for html, return html
    }
}
...