From d092f342934e9c9d2cc8ae48f29275428f1e2a1a Mon Sep 17 00:00:00 2001
From: soulteary <soulteary@gmail.com>
Date: Wed, 15 Jun 2022 10:22:53 +0800
Subject: [PATCH] feat: add custom mirror

---
 linux/common.go       |  4 ++--
 linux/custom.go       | 24 ++++++++++++++++++++++++
 linux/mirrors.go      |  8 ++++----
 linux/mirrors_test.go |  2 +-
 4 files changed, 31 insertions(+), 7 deletions(-)
 create mode 100644 linux/custom.go

diff --git a/linux/common.go b/linux/common.go
index ab9a8da..e315770 100644
--- a/linux/common.go
+++ b/linux/common.go
@@ -32,7 +32,7 @@ const (
 	DEBIAN_BENCHMAKR_URL = "dists/bullseye/main/binary-amd64/Release"
 )
 
-var BUILDIN_DEBIAN_MIRRORS = []string{
+var BUILDIN_OFFICAL_DEBIAN_MIRRORS = []string{
 	"http://ftp.cn.debian.org/debian/",
 	"http://mirror.bjtu.edu.cn/debian/",
 	"http://mirror.lzu.edu.cn/debian/",
@@ -70,7 +70,7 @@ const (
 	UBUNTU_BENCHMAKR_URL  = "dists/jammy/main/binary-amd64/Release"
 )
 
-var BUILDIN_UBUNTU_MIRRORS = []string{
+var BUILDIN_OFFICAL_UBUNTU_MIRRORS = []string{
 	"http://mirrors.aliyun.com/ubuntu/",
 	"http://mirrors.huaweicloud.com/repository/ubuntu/",
 	"http://mirror.dlut.edu.cn/ubuntu/",
diff --git a/linux/custom.go b/linux/custom.go
new file mode 100644
index 0000000..266ec4d
--- /dev/null
+++ b/linux/custom.go
@@ -0,0 +1,24 @@
+package linux
+
+type buildin_custom_mirror struct {
+	url   string
+	alias string
+}
+
+var BUILDIN_CUSTOM_UBUNTU_MIRRORS = []buildin_custom_mirror{
+	{url: "https://mirrors.tuna.tsinghua.edu.cn/ubuntu/", alias: "cn:tsinghua"},
+	{url: "http://mirrors.ustc.edu.cn/ubuntu/", alias: "cn:ustc"},
+	{url: "https://mirrors.163.com/ubuntu/", alias: "cn:163"},
+	{url: "https://mirrors.aliyun.com/ubuntu/", alias: "cn:aliyun"},
+	{url: "https://repo.huaweicloud.com/ubuntu/", alias: "cn:huawei"},
+	{url: "https://mirrors.cloud.tencent.com/ubuntu/", alias: "cn:tencent"},
+}
+
+var BUILDIN_CUSTOM_DEBIAN_MIRRORS = []buildin_custom_mirror{
+	{url: "https://mirrors.tuna.tsinghua.edu.cn/debian/", alias: "cn:tsinghua"},
+	{url: "http://mirrors.ustc.edu.cn/debian/", alias: "cn:ustc"},
+	{url: "https://mirrors.163.com/debian/", alias: "cn:163"},
+	{url: "https://mirrors.aliyun.com/debian/", alias: "cn:aliyun"},
+	{url: "https://repo.huaweicloud.com/debian/", alias: "cn:huawei"},
+	{url: "https://mirrors.cloud.tencent.com/debian/", alias: "cn:tencent"},
+}
diff --git a/linux/mirrors.go b/linux/mirrors.go
index e758da9..8351bc7 100644
--- a/linux/mirrors.go
+++ b/linux/mirrors.go
@@ -10,17 +10,17 @@ func getGeoMirrorUrlsByMode(mode int) (mirrors []string) {
 	if mode == TYPE_LINUX_DISTROS_UBUNTU {
 		ubuntuMirrorsOnline, err := getUbuntuMirrorUrlsByGeo()
 		if err != nil {
-			return BUILDIN_UBUNTU_MIRRORS
+			return BUILDIN_OFFICAL_UBUNTU_MIRRORS
 		}
 		return ubuntuMirrorsOnline
 	}
 
 	if mode == TYPE_LINUX_DISTROS_DEBIAN {
-		return BUILDIN_DEBIAN_MIRRORS
+		return BUILDIN_OFFICAL_DEBIAN_MIRRORS
 	}
 
-	mirrors = append(mirrors, BUILDIN_UBUNTU_MIRRORS...)
-	mirrors = append(mirrors, BUILDIN_DEBIAN_MIRRORS...)
+	mirrors = append(mirrors, BUILDIN_OFFICAL_UBUNTU_MIRRORS...)
+	mirrors = append(mirrors, BUILDIN_OFFICAL_DEBIAN_MIRRORS...)
 	return mirrors
 }
 
diff --git a/linux/mirrors_test.go b/linux/mirrors_test.go
index 60064f9..e02504d 100644
--- a/linux/mirrors_test.go
+++ b/linux/mirrors_test.go
@@ -22,7 +22,7 @@ func TestGetMirrorUrlsByGeo(t *testing.T) {
 	}
 
 	mirrors = getGeoMirrorUrlsByMode(TYPE_LINUX_DISTROS_DEBIAN)
-	if len(mirrors) != len(BUILDIN_DEBIAN_MIRRORS) {
+	if len(mirrors) != len(BUILDIN_OFFICAL_DEBIAN_MIRRORS) {
 		t.Fatal("Get mirrors error")
 	}