✏️ 修复 vtb panic 判断失误

This commit is contained in:
fumiama 2021-11-22 22:33:29 +08:00
parent fa063a05f7
commit 12f96bb95c

View File

@ -7,31 +7,36 @@ import (
"os" "os"
"github.com/FloatTech/ZeroBot-Plugin/utils/file" "github.com/FloatTech/ZeroBot-Plugin/utils/file"
"github.com/FloatTech/ZeroBot-Plugin/utils/process"
) )
const pburl = "https://codechina.csdn.net/u011570312/ZeroBot-Plugin/-/raw/master/" + dbfile const pburl = "https://codechina.csdn.net/u011570312/ZeroBot-Plugin/-/raw/master/" + dbfile
// 加载数据库 // 加载数据库
func init() { func init() {
_ = os.MkdirAll(dbpath, 0755) go func() {
if !file.IsExist(dbfile) { // 如果没有数据库,则从 url 下载 process.SleepAbout1sTo2s()
f, err := os.Create(dbfile) _ = os.MkdirAll(dbpath, 0755)
if err != nil { if !file.IsExist(dbfile) { // 如果没有数据库,则从 url 下载
panic(err) f, err := os.Create(dbfile)
} if err != nil {
defer f.Close()
resp, err := http.Get(pburl)
if err == nil {
defer resp.Body.Close()
if resp.ContentLength > 0 {
log.Printf("[vtb]从镜像下载数据库%d字节...", resp.ContentLength)
data, err := io.ReadAll(resp.Body)
if err == nil && len(data) > 0 {
_, _ = f.Write(data)
}
panic(err) panic(err)
} }
defer f.Close()
resp, err := http.Get(pburl)
if err == nil {
defer resp.Body.Close()
if resp.ContentLength > 0 {
log.Printf("[vtb]从镜像下载数据库%d字节...", resp.ContentLength)
data, err := io.ReadAll(resp.Body)
if err == nil && len(data) > 0 {
_, _ = f.Write(data)
return
}
panic(err)
}
}
panic(err)
} }
panic(err) }()
}
} }