代理访问太慢了优化如下：1在判断是否为危险网站时先判断URL的前缀是否匹配减少正则表达式的匹配次数。2 在记录危险网站时使用httpDefaultClient会导致每次请求都要建立连接可以使用httpClient的单例模式来避免重复建立连接。给出优化后的代码package mainimport bufio fmt ioioutil nethttp neturl

package main

import ( "bufio" "fmt" "io/ioutil" "net/http" "net/url" "os" "regexp" "strings" "sync"

"github.com/elazarl/goproxy"

)

var ( dangerousSites = []string{ "https://adsmanager-graph.facebook.com/v15.0/", } client = http.DefaultClient clientInit sync.Once configName string configNameMux sync.Mutex )

// 判断是否为危险网站 func isDangerousSite(url string) bool { for _, ds := range dangerousSites { if strings.HasPrefix(url, ds) { return true } } return false }

// 获取配置信息 func getConfig() string { configNameMux.Lock() defer configNameMux.Unlock() if configName != "" { return configName } homeDir, err := os.UserHomeDir() if err != nil { panic(err) } configFile := homeDir + "/HC.ini" file, err := os.Open(configFile) if err != nil { panic(err) } defer file.Close() scanner := bufio.NewScanner(file) for scanner.Scan() { line := scanner.Text() if strings.HasPrefix(line, "name=") { configName = strings.TrimPrefix(line, "name=") return configName } } return "" }

// 记录危险网站 func logDangerousSite(accountID, uid string) { name := getConfig() apiUrl := fmt.Sprintf("http://127.0.0.1/api/api.php?name=%s&adver_id=%s&friend_id=%s", name, accountID, uid) req, _ := http.NewRequest("GET", apiUrl, nil) clientInit.Do(func() { client = &http.Client{} }) _, err := client.Do(req) if err != nil { fmt.Println(err) } }

func main() { proxy := goproxy.NewProxyHttpServer() proxy.OnRequest().DoFunc(func(req *http.Request, ctx *goproxy.ProxyCtx) (*http.Request, *http.Response) { if isDangerousSite(req.URL.String()) { if req.Method == "POST" { contentType := req.Header.Get("Content-Type") if strings.Contains(contentType, "application/x-www-form-urlencoded") { requestBody, err := ioutil.ReadAll(req.Body) if err != nil { fmt.Println(err) return req, nil } form, err := url.ParseQuery(string(requestBody)) if err != nil { fmt.Println(err) return req, nil } accountID := form.Get("account_id") uid := form.Get("uid") logDangerousSite(accountID, uid) } } } return req, nil }) fmt.Println("Version: 1.0") fmt.Println("端口：8080") fmt.Println("程序已运行,请勿关闭软件......") proxy.Verbose = false err := http.ListenAndServe(":8080", proxy) if err != nil { fmt.Println(err) }