✏️ 统一懒加载验证

This commit is contained in:
fumiama
2021-12-13 20:49:00 +08:00
parent f7a3c0e6c6
commit 8969c44b3f
9 changed files with 117 additions and 186 deletions

View File

@@ -1,8 +1,6 @@
package omikuji
import (
"io"
"net/http"
"os"
log "github.com/sirupsen/logrus"
@@ -15,7 +13,6 @@ import (
const (
dbpath = "data/Omikuji/"
dbfile = dbpath + "kuji.db"
dburl = "https://codechina.csdn.net/u011570312/ZeroBot-Plugin/-/raw/master/" + dbfile
)
var db = &sql.Sqlite{DBPath: dbfile}
@@ -29,28 +26,7 @@ func init() {
}()
process.SleepAbout1sTo2s()
_ = os.MkdirAll(dbpath, 0755)
if !file.IsExist(dbfile) { // 如果没有数据库,则从 url 下载
f, err := os.Create(dbfile)
if err != nil {
panic(err)
}
defer f.Close()
resp, err := http.Get(dburl)
if err != nil {
panic(err)
}
defer resp.Body.Close()
if resp.ContentLength > 0 {
log.Printf("[omikuji]从镜像下载数据库%d字节...", resp.ContentLength)
data, err := io.ReadAll(resp.Body)
if err == nil && len(data) > 0 {
_, _ = f.Write(data)
} else {
panic(err)
}
}
}
_, _ = file.GetLazyData(dbfile, false, true)
err := db.Create("kuji", &kuji{})
if err != nil {
panic(err)