mirror of
https://github.com/siyuan-note/siyuan.git
synced 2026-07-03 14:09:06 +02:00
Jeffrey Chen
124 lines
3.1 KiB
Go
124 lines
3.1 KiB
Go
// SiYuan - Refactor your thinking
|
|
// Copyright (c) 2020-present, b3log.org
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
package util
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/siyuan-note/httpclient"
|
|
"github.com/siyuan-note/logging"
|
|
"golang.org/x/sync/singleflight"
|
|
)
|
|
|
|
var (
|
|
RhyCacheDuration = int64(3600 * 6)
|
|
|
|
cachedRhyResult = map[string]any{}
|
|
rhyResultCacheTime int64
|
|
rhyResultLock = sync.Mutex{}
|
|
rhyResultFlight singleflight.Group
|
|
|
|
rhyBazaarHash string
|
|
rhyBazaarHashLock sync.RWMutex
|
|
)
|
|
|
|
func RefreshRhyResultJob() {
|
|
_, err := GetRhyResult(context.TODO(), true)
|
|
if nil != err {
|
|
// 系统唤醒后可能还没有网络连接，这里等待后再重试
|
|
go func() {
|
|
time.Sleep(7 * time.Second)
|
|
GetRhyResult(context.TODO(), true)
|
|
}()
|
|
}
|
|
}
|
|
|
|
func GetRhyResult(ctx context.Context, force bool) (map[string]any, error) {
|
|
if ContainerDocker == Container {
|
|
RhyCacheDuration = int64(3600 * 24)
|
|
}
|
|
|
|
if RhyCacheDuration >= time.Now().Unix()-rhyResultCacheTime && !force && 0 < len(cachedRhyResult) {
|
|
return cachedRhyResult, nil
|
|
}
|
|
|
|
// 并发调用只执行一次实际请求
|
|
v, err, _ := rhyResultFlight.Do("rhyResult", func() (any, error) {
|
|
return getRhyResult0(ctx)
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
ret := v.(map[string]any)
|
|
syncRhyBazaarHashFromResult(ret)
|
|
return ret, nil
|
|
}
|
|
|
|
func getRhyResult0(ctx context.Context) (map[string]any, error) {
|
|
rhyResultLock.Lock()
|
|
defer rhyResultLock.Unlock()
|
|
|
|
request := httpclient.NewCloudRequest30s()
|
|
resp, err := request.SetContext(ctx).SetSuccessResult(&cachedRhyResult).Get(GetCloudServer() + "/apis/siyuan/version?ver=" + Ver)
|
|
if err != nil {
|
|
logging.LogErrorf("get version info failed: %s", err)
|
|
return nil, err
|
|
}
|
|
if 200 != resp.StatusCode {
|
|
logging.LogErrorf("get rhy result failed: %d", resp.StatusCode)
|
|
return nil, fmt.Errorf("get rhy result failed: %d", resp.StatusCode)
|
|
}
|
|
rhyResultCacheTime = time.Now().Unix()
|
|
return cachedRhyResult, nil
|
|
}
|
|
|
|
func syncRhyBazaarHashFromResult(m map[string]any) {
|
|
rhyBazaarHashLock.Lock()
|
|
defer rhyBazaarHashLock.Unlock()
|
|
if nil == m {
|
|
rhyBazaarHash = ""
|
|
return
|
|
}
|
|
v, ok := m["bazaar"]
|
|
if !ok || nil == v {
|
|
rhyBazaarHash = ""
|
|
return
|
|
}
|
|
s, ok := v.(string)
|
|
if !ok || "" == s {
|
|
rhyBazaarHash = ""
|
|
return
|
|
}
|
|
rhyBazaarHash = s
|
|
}
|
|
|
|
func GetRhyBazaarHash(ctx context.Context) string {
|
|
rhyBazaarHashLock.RLock()
|
|
h := rhyBazaarHash
|
|
rhyBazaarHashLock.RUnlock()
|
|
if "" != h {
|
|
return h
|
|
}
|
|
_, _ = GetRhyResult(ctx, false)
|
|
rhyBazaarHashLock.RLock()
|
|
defer rhyBazaarHashLock.RUnlock()
|
|
return rhyBazaarHash
|
|
}
|