Function run

main.go:251–324

(*cobra.Command, []string)

}

func run(*cobra.Command, []string) {
	if bandwidthLimit > 0 {
		BandwidthLimiter = R.NewBucketWithRate(float64(bandwidthLimit*1024*1024), int64(bandwidthLimit*1024*1024))
		log.Info("Bandwidth limit is set as ", bandwidthLimit, "MB/s")
	}
	if requestLimit > 0 {
		log.Info("Request limit is set as ", requestLimit, " each IP")
		RequestLimiter = NewIPLimiter()
	}
	if denyWebPage {
		log.Info("Denying web page requests")
	}
	if watcher, err := loadDomainList(); err == nil {
		err = watcher.Start()
		if err == nil {
			log.Info("Watching accept domain list")
			defer watcher.Close()
		} else {
			log.Error(E.Cause(err, "Start watch accept domain list"))
			watcher.Close()
		}
	}
	if watcher, err := loadBlackList(); err == nil {
		err = watcher.Start()
		if err == nil {
			log.Info("Watching repository blacklist")
			defer watcher.Close()
		} else {
			log.Error(E.Cause(err, "Start watch repository blacklist"))
			watcher.Close()
		}
	}
	listen := M.ParseSocksaddr(":" + strconv.Itoa(runningPort))
	listener := listenTCP(listen)
	if len(certPath) == 0 || len(keyPath) == 0 {
		log.Info("Listening TCP port ", listen.Port)
	} else if container, err := NewCertContainer(); err != nil {
		log.Warn(E.Cause(err, "Update TCP to TLS"))
		log.Info("Listening TCP port ", listen.Port)
	} else {
		log.Info("Listening TLS port ", listen.Port)
		listener = tls.NewListener(listener, &tls.Config{
			NextProtos: []string{"h2", "http/1.1"},
			GetCertificate: func(info *tls.ClientHelloInfo) (*tls.Certificate, error) {
				return &container.Cert, nil
			},
		})
		defer container.Close()
	}
	chiRouter := chi.NewRouter()
	chiRouter.Group(func(r chi.Router) {
		r.Use(middleware.RealIP)
		r.Use(setContext)
		r.Use(commonLog)
		r.Use(requestLimitHandle)
		r.Get("/", hello)
		r.Mount("/", finalHandle())
	})