用go语言写的github加速程序

作者：陈进坚
CSDN：https://blog.csdn.net/c_jian
简书：https://www.jianshu.com/u/8ba9ac5706b6
声明：有错欢迎纠正，转载请注明出处。

github地址

https://github.com/jian1098/github-speedup/releases

注意：需要管理员身份执行，如果访问变慢，再次执行程序即可

原理

通过DNS接口解析github.com的域名得到各个服务器节点的ip地址,然后将这些ip地址写入到hosts文件中，本地请求就可以直接绕过DNS服务器，直接访问网站，达到加速的效果。其中解析域名需要请求多个dns服务器，用go语言协程并发请求加快请求速度。

完整代码

package main

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

func main() {
	fmt.Println("正在解析域名...")
	//需要解析的域名
	var host = "github.com"
	var filename = "C:\\Windows\\System32\\drivers\\etc\\hosts"

	//dns服务器id
	var guid = []string{
		"29d2a14f-accc-43b8-9444-fd6b9e7902bc",
		"0e215450-a287-486e-b758-49b00e432bd4",
		"38806ef7-4638-4808-96e4-85047dfa5853",
		"53e79941-312c-4343-8739-6be3cd105805",
		"e00f2cc8-3f66-452d-b57f-804f84b774ca",
		"97a2da5b-4aa4-4811-aff7-2eb038b6f742",
		"7ddde139-f06e-40aa-b585-25d2ee6fad5f",
		"d9041619-7d90-42ea-9811-2b2fe11cb2b0",
		"02a01d5d-5111-481f-aade-e999a584d8a4",
		"80a828bd-19ed-48c3-a035-e69f6468da03",
		"fc7b8db4-f81d-4432-8d27-fa43dd13df3c",
		"91937e5b-1db0-47b5-b114-c9294694f377",
		"1f4c5976-8cf3-47e7-be10-aa9270461477",
		"3c1c826d-3444-4350-849b-0b9b9755df78",
		"7a2c1fe7-f9de-4fee-b797-bfe343d49f15",
		"eac78784-07a1-4869-be7c-3870a8dcebfc",
		"87c200e0-0059-479f-8103-e9e504f735d0",
		"dc440a55-1148-480f-90a7-9d1e0269b682",
		"08117724-8437-4ebb-88ae-93e50f660867",
		"5fb9012d-b47c-4087-84a2-0b0dfa8c94ab",
		"4250e220-157f-4831-8e6b-ad7cead81ca0",
		"1e375923-e5ee-491e-ba21-621a95ef9de9",
		"39bed414-9402-4266-aa9a-8252e958558f",
		"a0be885d-24ad-487d-bbb0-c94cd02a137d",
		"29d2a14f-accc-43b8-9444-fd6b9e7902bc",
	}
	var ipArr []string
	var wg sync.WaitGroup
	for _, val := range guid {
		data := url.Values{}
		data.Set("guid", val)
		data.Set("host", host)
		wg.Add(1)
		go func() {
			res := httpPost("http://mping.chinaz.com/Handle/AjaxHandler.ashx?action=Ping", data)
			ip := getIp(res)
			if ip != "" {
				ipArr = append(ipArr, ip+" "+host)
			}
			wg.Done()
		}()
	}
	wg.Wait()
	ipArr = arrayUnique(ipArr)

	content, err := readFile(filename, host)
	if err != nil {
		fmt.Println(err)
		return
	}

	for _, val := range ipArr {
		content += val + "\n"
	}
	//将获取到的ip写入hosts文件
	err = writeFile(filename, content)
	if err != nil {
		fmt.Println(err)
		return
	}
	fmt.Println("写入数据成功!")
	time.Sleep(time.Duration(time.Second * 3))
}

//post请求
func httpPost(link string, data url.Values) string {
	resp, err := http.PostForm(link, data)
	if err != nil {
		log.Fatal(err)
	}
	defer resp.Body.Close()
	body, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		log.Fatal(err)
	}

	return string(body)
}

//获取ip地址
func getIp(str string) string {
	gexp := "^(25[0-5]|2[0-4][0-9]|[0-1]{1}[0-9]{2}|[1-9]{1}[0-9]{1}|[1-9])\\.(25[0-5]|2[0-4][0-9]|[0-1]{1}[0-9]{2}|[1-9]{1}[0-9]{1}|[1-9]|0)\\.(25[0-5]|2[0-4][0-9]|[0-1]{1}[0-9]{2}|[1-9]{1}[0-9]{1}|[1-9]|0)\\.(25[0-5]|2[0-4][0-9]|[0-1]{1}[0-9]{2}|[1-9]{1}[0-9]{1}|[0-9])$"
	arr := strings.Split(str, "'")
	var ip string
	for _, str := range arr {
		match, _ := regexp.MatchString(gexp, str)
		if match {
			ip = str
			break
		}
	}
	return ip
}

//数组去重
func arrayUnique(arr []string) []string {
	size := len(arr)
	result := make([]string, 0, size)
	temp := map[string]struct{}{}
	for i := 0; i < size; i++ {
		if _, ok := temp[arr[i]]; ok != true {
			temp[arr[i]] = struct{}{}
			result = append(result, arr[i])
		}
	}
	return result
}

//读取文件内容
func readFile(filename, hosts string) (string, error) {
	file, err := os.Open(filename)
	if err != nil {
		return "", err
	}
	defer file.Close()

	scanner := bufio.NewScanner(file)
	var content string //用来保存文件内容
	for scanner.Scan() {
		// 读取当前行内容
		line := scanner.Text()
		if strings.Contains(line, hosts) {
			continue
		}
		content += line + "\n"
	}
	return content, nil
}

//写入文件
func writeFile(filename, content string) error {
	//覆盖写入
	err := ioutil.WriteFile(filename, []byte(content), 0664)
	if err != nil {
		return err
	}
	return nil
}