top_channel_args.go

时间:2024-04-15 20:03:04
package http_api

import (
    "fmt"
    "log"
    "net"
    "net/http"
    "strings"

    "github.com/nsqio/nsq/internal/app"
)

type logWriter struct {
    app.Logger
}

func (l logWriter) Write(p []byte) (int, error) {
    l.Logger.Output(2, string(p))
    return len(p), nil
}

func Serve(listener net.Listener, handler http.Handler, proto string, l app.Logger) {
    l.Output(2, fmt.Sprintf("%s: listening on %s", proto, listener.Addr()))

    server := &http.Server{
        Handler:  handler,
        ErrorLog: log.New(logWriter{l}, "", 0),
    }
    err := server.Serve(listener)
    // theres no direct way to detect this error because it is not exposed
    if err != nil && !strings.Contains(err.Error(), "use of closed network connection") {
        l.Output(2, fmt.Sprintf("ERROR: http.Serve() - %s", err))
    }

    l.Output(2, fmt.Sprintf("%s: closing %s", proto, listener.Addr()))
}