mirror of https://github.com/goproxyio/goproxy
71 lines
1.4 KiB
Go
71 lines
1.4 KiB
Go
//go:generate bash build/generate.sh
|
|
|
|
package main
|
|
|
|
import (
|
|
"flag"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"path/filepath"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/goproxyio/goproxy/pkg/proxy"
|
|
)
|
|
|
|
var listen string
|
|
var cacheDir string
|
|
|
|
func init() {
|
|
log.SetOutput(os.Stdout)
|
|
flag.StringVar(&cacheDir, "cacheDir", "", "go modules cache dir")
|
|
flag.StringVar(&listen, "listen", "0.0.0.0:8081", "service listen address")
|
|
flag.Parse()
|
|
}
|
|
|
|
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)
|
|
|
|
if cacheDir == "" {
|
|
cacheDir = "/go"
|
|
gpEnv := os.Getenv("GOPATH")
|
|
if gpEnv != "" {
|
|
gp := filepath.SplitList(gpEnv)
|
|
if gp[0] != "" {
|
|
cacheDir = gp[0]
|
|
}
|
|
}
|
|
}
|
|
fullCacheDir := filepath.Join(cacheDir, "pkg", "mod", "cache", "download")
|
|
if _, err := os.Stat(fullCacheDir); os.IsNotExist(err) {
|
|
log.Printf("goproxy: cache dir %s is not exist. To create it.\n", fullCacheDir)
|
|
if err := os.MkdirAll(fullCacheDir, 0755); err != nil {
|
|
log.Fatalf("make cache dir failed: %s", err)
|
|
}
|
|
}
|
|
|
|
http.Handle("/", proxy.NewProxy(cacheDir))
|
|
|
|
go func() {
|
|
err := http.ListenAndServe(listen, nil)
|
|
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)
|
|
}
|
|
}
|