57 lines
1.7 KiB
Go
57 lines
1.7 KiB
Go
// Package mhttp extends the standard package with extra functionality which is
|
|
// commonly useful
|
|
package mhttp
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/mediocregopher/mediocre-go-lib/m"
|
|
"github.com/mediocregopher/mediocre-go-lib/mcfg"
|
|
"github.com/mediocregopher/mediocre-go-lib/mlog"
|
|
"github.com/mediocregopher/mediocre-go-lib/mnet"
|
|
)
|
|
|
|
// CfgServer initializes and returns an *http.Server which will initialize on
|
|
// the Start hook. This also sets up config params for configuring the address
|
|
// being listened on.
|
|
func CfgServer(cfg *mcfg.Cfg, h http.Handler) *http.Server {
|
|
cfg = cfg.Child("http")
|
|
addr := cfg.ParamString("addr", ":0", "Address to listen on. Default is any open port")
|
|
|
|
srv := http.Server{Handler: h}
|
|
cfg.Start.Then(func(ctx context.Context) error {
|
|
srv.Addr = *addr
|
|
kv := mlog.KV{"addr": *addr}
|
|
log := m.Log(cfg, kv)
|
|
log.Info("listening")
|
|
go func() {
|
|
err := srv.ListenAndServe()
|
|
// TODO the listening log should happen here, somehow, now that we
|
|
// know the actual address being listened on
|
|
log.Fatal("http server fataled", mlog.ErrKV(err))
|
|
}()
|
|
return nil
|
|
})
|
|
|
|
// TODO shutdown logic
|
|
return &srv
|
|
}
|
|
|
|
// AddXForwardedFor populates the X-Forwarded-For header on the Request to
|
|
// convey that the request is being proxied for IP.
|
|
//
|
|
// If the IP is invalid, loopback, or otherwise part of a reserved range, this
|
|
// does nothing.
|
|
func AddXForwardedFor(r *http.Request, ipStr string) {
|
|
const xff = "X-Forwarded-For"
|
|
ip := net.ParseIP(ipStr)
|
|
if ip == nil || mnet.IsReservedIP(ip) { // IsReservedIP includes loopback
|
|
return
|
|
}
|
|
prev, _ := r.Header[xff]
|
|
r.Header.Set(xff, strings.Join(append(prev, ip.String()), ", "))
|
|
}
|