jinql
/
goproxy
mirror of https://github.com/goproxyio/goproxy
1
0
Fork 0
Code Issues Releases Wiki Activity

improve server with graceful shutdown

pull/25/head
nafisfaysal 2018-12-26 07:36:02 +06:00 committed by kun
parent 1ff5252114
commit afb11b3683
1 changed files with 21 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
main.go
View File

@ -7,7 +7,9 @@ import (
"log" "log"
"net/http" "net/http"
"os" "os"
"os/signal"
"path/filepath" "path/filepath"
"syscall"
"time" "time"
"github.com/goproxyio/goproxy/pkg/proxy" "github.com/goproxyio/goproxy/pkg/proxy"
@ -24,6 +26,8 @@ func init() {
} }
func main() { func main() {
errCh := make(chan error)
log.Printf("goproxy: %s inited. listen on %s\n", time.Now().Format("2006-01-02 15:04:05"), listen) log.Printf("goproxy: %s inited. listen on %s\n", time.Now().Format("2006-01-02 15:04:05"), listen)
if cacheDir == "" { if cacheDir == "" {
@ -43,9 +47,22 @@ func main() {
} }
http.Handle("/", proxy.NewProxy(cacheDir)) http.Handle("/", proxy.NewProxy(cacheDir))
// TODO: TLS, graceful shutdown
err := http.ListenAndServe(listen, nil) go func() {
if nil != err { err := http.ListenAndServe(listen, nil)
panic(err) if err != nil {
errCh <- err
}
}()
signCh := make(chan os.Signal)
signal.Notify(signCh, os.Interrupt, syscall.SIGTERM)
select {
case err := <-errCh:
log.Fatal(err)
case sign := <-signCh:
log.Printf("Server gracefully %s", sign)
} }
} }