Skip to content

Commit

Permalink
feat: add custom mirror
Browse files Browse the repository at this point in the history
  • Loading branch information
soulteary committed Jun 15, 2022
1 parent a5dc244 commit d092f34
Show file tree
Hide file tree
Showing 4 changed files with 31 additions and 7 deletions.
4 changes: 2 additions & 2 deletions linux/common.go
Original file line number Diff line number Diff line change
Expand Up @@ -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/",
Expand Down Expand Up @@ -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/",
Expand Down
24 changes: 24 additions & 0 deletions linux/custom.go
Original file line number Diff line number Diff line change
@@ -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"},
}
8 changes: 4 additions & 4 deletions linux/mirrors.go
Original file line number Diff line number Diff line change
Expand Up @@ -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
}

Expand Down
2 changes: 1 addition & 1 deletion linux/mirrors_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -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")
}

Expand Down

0 comments on commit d092f34

Please sign in to comment.