mirror of
https://github.com/dutchcoders/transfer.sh.git
synced 2024-12-03 01:20:17 +01:00
61 lines
1.4 KiB
Go
61 lines
1.4 KiB
Go
|
// Copyright 2017 The Go Authors. All rights reserved.
|
||
|
|
//
|
||
|
|
// Use of this source code is governed by a BSD-style
|
||
|
|
// license that can be found in the LICENSE file or at
|
||
|
|
// https://developers.google.com/open-source/licenses/bsd.
|
||
|
|
|
||
|
|
// This file implements an http.Client with request timeouts set by command
|
||
|
|
// line flags.
|
||
|
|
|
||
|
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"net"
|
||
|
|
"net/http"
|
||
|
|
"os"
|
||
|
|
|
||
|
|
"github.com/gregjones/httpcache"
|
||
|
|
"github.com/gregjones/httpcache/memcache"
|
||
|
|
"github.com/spf13/viper"
|
||
|
|
|
||
|
|
"github.com/golang/gddo/httputil"
|
||
|
|
)
|
||
|
|
|
||
|
|
func newHTTPClient() *http.Client {
|
||
|
|
t := newCacheTransport()
|
||
|
|
|
||
|
|
requestTimeout := viper.GetDuration(ConfigRequestTimeout)
|
||
|
|
t.Transport = &http.Transport{
|
||
|
|
Proxy: http.ProxyFromEnvironment,
|
||
|
|
Dial: (&net.Dialer{
|
||
|
|
Timeout: viper.GetDuration(ConfigDialTimeout),
|
||
|
|
KeepAlive: requestTimeout / 2,
|
||
|
|
}).Dial,
|
||
|
|
ResponseHeaderTimeout: requestTimeout / 2,
|
||
|
|
TLSHandshakeTimeout: requestTimeout / 2,
|
||
|
|
}
|
||
|
|
return &http.Client{
|
||
|
|
// Wrap the cached transport with GitHub authentication.
|
||
|
|
Transport: httputil.NewAuthTransport(t),
|
||
|
|
Timeout: requestTimeout,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func newCacheTransport() *httpcache.Transport {
|
||
|
|
// host and port are set by GAE Flex runtime, can be left blank locally.
|
||
|
|
host := os.Getenv("MEMCACHE_PORT_11211_TCP_ADDR")
|
||
|
|
if host == "" {
|
||
|
|
host = "localhost"
|
||
|
|
}
|
||
|
|
port := os.Getenv("MEMCACHE_PORT_11211_TCP_PORT")
|
||
|
|
if port == "" {
|
||
|
|
port = "11211"
|
||
|
|
}
|
||
|
|
addr := fmt.Sprintf("%s:%s", host, port)
|
||
|
|
|
||
|
|
return httpcache.NewTransport(
|
||
|
|
memcache.New(addr),
|
||
|
|
)
|
||
|
|
}
|