Function TestConnection

pkg/microservice/aslan/core/system/service/proxy.go:109–147

(args *commonmodels.Proxy, log *zap.SugaredLogger)

}

func TestConnection(args *commonmodels.Proxy, log *zap.SugaredLogger) error {
	if args == nil {
		log.Error("invalid args")
		return e.ErrTestConnection.AddDesc("invalid args")
	}

	request, err := http.NewRequest("GET", "https://www.baidu.com", nil)
	if err != nil {
		log.Errorf("http.NewRequest failed, err:%v", err)
		return e.ErrTestConnection.AddErr(err)
	}

	var ret *http.Response
	client := &http.Client{
		Timeout: 5 * time.Second,
	}

	p, err := url.Parse(args.GetProxyURL())
	if err == nil {
		proxy := http.ProxyURL(p)
		trans := &http.Transport{
			Proxy: proxy,
		}
		client.Transport = trans
	}

	ret, err = client.Do(request)
	if err != nil {
		log.Errorf("client.Do failed, err:%v", err)
		return e.ErrTestConnection.AddErr(err)
	}
	defer func() { _ = ret.Body.Close() }()

	if ret.StatusCode != 200 {
		return e.ErrTestConnection
	}

	return nil
}