ProfessionalUwU/echoip
1
0
Fork 0
mirror of https://github.com/mpolden/echoip.git synced 2025-07-16 22:13:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

Implement response cache

Browse Source
This commit is contained in:
Martin Polden
2019-12-25 21:04:26 +01:00
parent 66719b9932
commit 78116f69ad
6 changed files with 111 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
cmd/echoip/main.go
View File

@ -22,6 +22,7 @@ func main() {
PortLookup bool `short:"p" long:"port-lookup" description:"Enable port lookup"`
Template string `short:"t" long:"template" description:"Path to template" default:"index.html" value-name:"FILE"`
IPHeaders []string `short:"H" long:"trusted-header" description:"Header to trust for remote IP, if present (e.g. X-Real-IP)" value-name:"NAME"`
CacheCapacity int `short:"C" long:"cache-size" description:"Size of response cache. Set to 0 to disable" value-name:"SIZE"`
}
_, err := flags.ParseArgs(&opts, os.Args)
if err != nil {
@ -33,8 +34,8 @@ func main() {
if err != nil {
log.Fatal(err)
}
server := http.New(r)
cache := http.NewCache(opts.CacheCapacity)
server := http.New(r, cache)
server.IPHeaders = opts.IPHeaders
if _, err := os.Stat(opts.Template); err == nil {
server.Template = opts.Template