Chapter6. HTTP Server develop - Object Sharing(custom handler), Server Middleware(handler Function)

[robert-min]

핸들러 함수 간에 객체(데이터 공유)

• 데이터 로깅 객체를 공유하기 위해 커스텀 핸들러 핸들러 구축
• app은 appConfig 객체 타입의 필드와 func(http ResponseWriter, *http.Request) 시그니처를 갖는 구조체

[robert-min]

핸들러 함수 간에 객체 공유 코드

type appConfig struct {
    logger *log.Logger
}

type app struct {
    config  appConfig
    handler func(w http.ResponseWriter, r *http.Request, confing appConfig)
}

// 공통된 객체를 공유하는 핸들러 생성
func (a app) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    a.handler(w, r, a.config)
}

func apiHandler(w http.ResponseWriter, r *http.Request, config appConfig) {
    config.logger.Println("Handling API request")
    fmt.Fprintln(w, "Hello, world!")
}

func healthCheckHandler(w http.ResponseWriter, r *http.Request, config appConfig) {
    if r.Method != "GET" {
        http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
        return
    }
    config.logger.Println("Handling healthcheck request")
    fmt.Fprintf(w, "ok")
}

func setupHandler(mux *http.ServeMux, config appConfig) {
    mux.Handle("/healthz", &app{config: config, handler: healthCheckHandler})
    mux.Handle("/api", &app{config: config, handler: apiHandler})
}

func main() {
    listenAddr := os.Getenv("LISTENADDR")
    if len(listenAddr) == 0 {
        listenAddr = ":8080"
    }

    config := appConfig{
        logger: log.New(os.Stdout, "", log.Ldate|log.Ltime|log.Lshortfile),
    }

    mux := http.NewServeMux()
    setupHandler(mux, config)

    log.Fatal(http.ListenAndServe(listenAddr, mux))
}

[robert-min]

Server Middleware

• 요청을 처리할 때 사용되는 일반적인 동작을 자동으로 수행할 수 있도록 처리
• 위와 동일하게 Custom http.Handler 타입을 구현하는 방법과 HandlerFunc 방법 두가지 존재
• 위에 방식은 Custom http.Handler에 등록되지 않은 요청시 핸들러가 호출되지 않지만 HandelerFunc는 핸들러 함수 처리전 middleware에서 처리

[robert-min]

Server Middleware 코드

type appConfig struct {
    logger *log.Logger
}

type app struct {
    config  appConfig
    handler func(w http.ResponseWriter, r *http.Request, config appConfig)
}

func (a app) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    a.handler(w, r, a.config)
}

func healthCheckHandler(w http.ResponseWriter, r *http.Request, config appConfig) {
    if r.Method != "GET" {
        config.logger.Printf("error=\"Invalid request\" path=%s method=%s", r.URL.Path, r.Method)
        http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
        return
    }
    fmt.Fprintf(w, "ok")
}

func apiHandler(w http.ResponseWriter, r *http.Request, config appConfig) {
    fmt.Fprintf(w, "Hello, World")
}

func panicHandler(w http.ResponseWriter, r *http.Request, config appConfig) {
    panic("I panicked")
}

func setupHandlers(mux *http.ServeMux, config appConfig) {
    mux.Handle("/healthz", &app{config: config, handler: healthCheckHandler})
    mux.Handle("/api", &app{config: config, handler: apiHandler})
    mux.Handle("/panic", &app{config: config, handler: panicHandler})
}

func loggingMiddleware(h http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        startTime := time.Now()
        h.ServeHTTP(w, r)
        log.Printf("protocol=%s path=%s method=%s duration=%f", r.Proto, r.URL.Path, r.Method, time.Now().Sub(startTime).Seconds())
    })
}

func panicMiddleware(h http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        defer func() {
            if rValue := recover(); rValue != nil {
                log.Println("panic detected when handling request", rValue)
                w.WriteHeader(http.StatusInternalServerError)
                fmt.Fprintf(w, "Unexpected server error occured")
            }
        }()
        h.ServeHTTP(w, r)
    })
}

func main() {
    listenAddr := os.Getenv("LISTEN_ADDR")
    if len(listenAddr) == 0 {
        listenAddr = ":8080"
    }

    config := appConfig{
        logger: log.New(os.Stdout, "", log.Ldate|log.Ltime|log.Lshortfile),
    }
    mux := http.NewServeMux()
    setupHandlers(mux, config)

    m := loggingMiddleware(panicMiddleware(mux))

    log.Fatal(http.ListenAndServe(listenAddr, m))
}