Andreas/siyuan
1
0
Fork 0
mirror of https://github.com/siyuan-note/siyuan.git synced 2025-12-17 23:20:13 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

🎨 Improve kernel stability by eliminating some data races https://github.com/siyuan-note/siyuan/issues/9842

Browse source
This commit is contained in:
Daniel 2023-12-08 21:46:46 +08:00
parent f9d476ebf5
commit 1d54de679f
No known key found for this signature in database
GPG key ID: 86211BA83DF03017
7 changed files with 83 additions and 62 deletions
Download patch file Download diff file Expand all files Collapse all files

2
kernel/api/setting.go
View file

@ -452,7 +452,7 @@ func getCloudUser(c *gin.Context) {
token = t.(string) token = t.(string)
} }
model.RefreshUser(token) model.RefreshUser(token)
ret.Data = model.Conf.User ret.Data = model.Conf.GetUser()
} }
func logoutCloudUser(c *gin.Context) { func logoutCloudUser(c *gin.Context) {

2
kernel/api/sync.go
View file

@ -387,7 +387,7 @@ func performSync(c *gin.Context) {
mobileSwitch = mobileSwitchArg.(bool) mobileSwitch = mobileSwitchArg.(bool)
} }
if mobileSwitch { if mobileSwitch {
if nil == model.Conf.User || !model.Conf.Sync.Enabled { if nil == model.Conf.GetUser() || !model.Conf.Sync.Enabled {
return return
} }
} }

56
kernel/model/cloud_service.go
View file

@ -39,7 +39,7 @@ import (
var ErrFailedToConnectCloudServer = errors.New("failed to connect cloud server") var ErrFailedToConnectCloudServer = errors.New("failed to connect cloud server")
func CloudChatGPT(msg string, contextMsgs []string) (ret string, stop bool, err error) { func CloudChatGPT(msg string, contextMsgs []string) (ret string, stop bool, err error) {
if nil == Conf.User { if nil == Conf.GetUser() {
return return
} }
@ -61,7 +61,7 @@ func CloudChatGPT(msg string, contextMsgs []string) (ret string, stop bool, err
request := httpclient.NewCloudRequest30s() request := httpclient.NewCloudRequest30s()
_, err = request. _, err = request.
SetSuccessResult(requestResult). SetSuccessResult(requestResult).
SetCookies(&http.Cookie{Name: "symphony", Value: Conf.User.UserToken}). SetCookies(&http.Cookie{Name: "symphony", Value: Conf.GetUser().UserToken}).
SetBody(payload). SetBody(payload).
Post(util.GetCloudServer() + "/apis/siyuan/ai/chatGPT") Post(util.GetCloudServer() + "/apis/siyuan/ai/chatGPT")
if nil != err { if nil != err {
@ -99,7 +99,7 @@ func CloudChatGPT(msg string, contextMsgs []string) (ret string, stop bool, err
} }
func StartFreeTrial() (err error) { func StartFreeTrial() (err error) {
if nil == Conf.User { if nil == Conf.GetUser() {
return errors.New(Conf.Language(31)) return errors.New(Conf.Language(31))
} }
@ -107,7 +107,7 @@ func StartFreeTrial() (err error) {
request := httpclient.NewCloudRequest30s() request := httpclient.NewCloudRequest30s()
_, err = request. _, err = request.
SetSuccessResult(requestResult). SetSuccessResult(requestResult).
SetCookies(&http.Cookie{Name: "symphony", Value: Conf.User.UserToken}). SetCookies(&http.Cookie{Name: "symphony", Value: Conf.GetUser().UserToken}).
Post(util.GetCloudServer() + "/apis/siyuan/user/startFreeTrial") Post(util.GetCloudServer() + "/apis/siyuan/user/startFreeTrial")
if nil != err { if nil != err {
logging.LogErrorf("start free trial failed: %s", err) logging.LogErrorf("start free trial failed: %s", err)
@ -124,7 +124,7 @@ func DeactivateUser() (err error) {
request := httpclient.NewCloudRequest30s() request := httpclient.NewCloudRequest30s()
resp, err := request. resp, err := request.
SetSuccessResult(requestResult). SetSuccessResult(requestResult).
SetCookies(&http.Cookie{Name: "symphony", Value: Conf.User.UserToken}). SetCookies(&http.Cookie{Name: "symphony", Value: Conf.GetUser().UserToken}).
Post(util.GetCloudServer() + "/apis/siyuan/user/deactivate") Post(util.GetCloudServer() + "/apis/siyuan/user/deactivate")
if nil != err { if nil != err {
logging.LogErrorf("deactivate user failed: %s", err) logging.LogErrorf("deactivate user failed: %s", err)
@ -150,7 +150,7 @@ func SetCloudBlockReminder(id, data string, timed int64) (err error) {
resp, err := request. resp, err := request.
SetSuccessResult(requestResult). SetSuccessResult(requestResult).
SetBody(payload). SetBody(payload).
SetCookies(&http.Cookie{Name: "symphony", Value: Conf.User.UserToken}). SetCookies(&http.Cookie{Name: "symphony", Value: Conf.GetUser().UserToken}).
Post(util.GetCloudServer() + "/apis/siyuan/calendar/setBlockReminder") Post(util.GetCloudServer() + "/apis/siyuan/calendar/setBlockReminder")
if nil != err { if nil != err {
logging.LogErrorf("set block reminder failed: %s", err) logging.LogErrorf("set block reminder failed: %s", err)
@ -182,7 +182,7 @@ func LoadUploadToken() (err error) {
request := httpclient.NewCloudRequest30s() request := httpclient.NewCloudRequest30s()
resp, err := request. resp, err := request.
SetSuccessResult(requestResult). SetSuccessResult(requestResult).
SetCookies(&http.Cookie{Name: "symphony", Value: Conf.User.UserToken}). SetCookies(&http.Cookie{Name: "symphony", Value: Conf.GetUser().UserToken}).
Post(util.GetCloudServer() + "/apis/siyuan/upload/token") Post(util.GetCloudServer() + "/apis/siyuan/upload/token")
if nil != err { if nil != err {
logging.LogErrorf("get upload token failed: %s", err) logging.LogErrorf("get upload token failed: %s", err)
@ -228,8 +228,8 @@ func refreshSubscriptionExpirationRemind() {
defer logging.Recover() defer logging.Recover()
if IsSubscriber() && -1 != Conf.User.UserSiYuanProExpireTime { if IsSubscriber() && -1 != Conf.GetUser().UserSiYuanProExpireTime {
expired := int64(Conf.User.UserSiYuanProExpireTime) expired := int64(Conf.GetUser().UserSiYuanProExpireTime)
now := time.Now().UnixMilli() now := time.Now().UnixMilli()
if now >= expired { // 已经过期 if now >= expired { // 已经过期
if now-expired <= 1000*60*60*24*2 { // 2 天内提醒 https://github.com/siyuan-note/siyuan/issues/7816 if now-expired <= 1000*60*60*24*2 { // 2 天内提醒 https://github.com/siyuan-note/siyuan/issues/7816
@ -240,7 +240,7 @@ func refreshSubscriptionExpirationRemind() {
} }
remains := int((expired - now) / 1000 / 60 / 60 / 24) remains := int((expired - now) / 1000 / 60 / 60 / 24)
expireDay := 15 // 付费订阅提前 15 天提醒 expireDay := 15 // 付费订阅提前 15 天提醒
if 2 == Conf.User.UserSiYuanSubscriptionPlan { if 2 == Conf.GetUser().UserSiYuanSubscriptionPlan {
expireDay = 3 // 试用订阅提前 3 天提醒 expireDay = 3 // 试用订阅提前 3 天提醒
} }
@ -256,10 +256,10 @@ func refreshSubscriptionExpirationRemind() {
func refreshUser() { func refreshUser() {
defer logging.Recover() defer logging.Recover()
if nil != Conf.User { if nil != Conf.GetUser() {
time.Sleep(2 * time.Minute) time.Sleep(2 * time.Minute)
if nil != Conf.User { if nil != Conf.GetUser() {
RefreshUser(Conf.User.UserToken) RefreshUser(Conf.GetUser().UserToken)
} }
subscriptionExpirationReminded = false subscriptionExpirationReminded = false
} }
@ -329,22 +329,22 @@ func RefreshUser(token string) {
threeDaysAfter := util.CurrentTimeMillis() + 1000*60*60*24*3 threeDaysAfter := util.CurrentTimeMillis() + 1000*60*60*24*3
if "" == token { if "" == token {
if "" != Conf.UserData { if "" != Conf.UserData {
Conf.User = loadUserFromConf() Conf.SetUser(loadUserFromConf())
} }
if nil == Conf.User { if nil == Conf.GetUser() {
return return
} }
var tokenExpireTime int64 var tokenExpireTime int64
tokenExpireTime, err := strconv.ParseInt(Conf.User.UserTokenExpireTime+"000", 10, 64) tokenExpireTime, err := strconv.ParseInt(Conf.GetUser().UserTokenExpireTime+"000", 10, 64)
if nil != err { if nil != err {
logging.LogErrorf("convert token expire time [%s] failed: %s", Conf.User.UserTokenExpireTime, err) logging.LogErrorf("convert token expire time [%s] failed: %s", Conf.GetUser().UserTokenExpireTime, err)
util.PushErrMsg(Conf.Language(19), 5000) util.PushErrMsg(Conf.Language(19), 5000)
return return
} }
if threeDaysAfter > tokenExpireTime { if threeDaysAfter > tokenExpireTime {
token = Conf.User.UserToken token = Conf.GetUser().UserToken
goto Net goto Net
} }
return return
@ -354,15 +354,15 @@ Net:
start := time.Now() start := time.Now()
user, err := getUser(token) user, err := getUser(token)
if err != nil { if err != nil {
if nil == Conf.User || errInvalidUser == err { if nil == Conf.GetUser() || errInvalidUser == err {
util.PushErrMsg(Conf.Language(19), 5000) util.PushErrMsg(Conf.Language(19), 5000)
return return
} }
var tokenExpireTime int64 var tokenExpireTime int64
tokenExpireTime, err = strconv.ParseInt(Conf.User.UserTokenExpireTime+"000", 10, 64) tokenExpireTime, err = strconv.ParseInt(Conf.GetUser().UserTokenExpireTime+"000", 10, 64)
if nil != err { if nil != err {
logging.LogErrorf("convert token expire time [%s] failed: %s", Conf.User.UserTokenExpireTime, err) logging.LogErrorf("convert token expire time [%s] failed: %s", Conf.GetUser().UserTokenExpireTime, err)
util.PushErrMsg(Conf.Language(19), 5000) util.PushErrMsg(Conf.Language(19), 5000)
return return
} }
@ -374,7 +374,7 @@ Net:
return return
} }
Conf.User = user Conf.SetUser(user)
data, _ := gulu.JSON.MarshalJSON(user) data, _ := gulu.JSON.MarshalJSON(user)
Conf.UserData = util.AESEncrypt(string(data)) Conf.UserData = util.AESEncrypt(string(data))
Conf.Save() Conf.Save()
@ -407,7 +407,7 @@ func RemoveCloudShorthands(ids []string) (err error) {
} }
resp, err := request. resp, err := request.
SetSuccessResult(&result). SetSuccessResult(&result).
SetCookies(&http.Cookie{Name: "symphony", Value: Conf.User.UserToken}). SetCookies(&http.Cookie{Name: "symphony", Value: Conf.GetUser().UserToken}).
SetBody(body). SetBody(body).
Post(util.GetCloudServer() + "/apis/siyuan/inbox/removeCloudShorthands") Post(util.GetCloudServer() + "/apis/siyuan/inbox/removeCloudShorthands")
if nil != err { if nil != err {
@ -435,7 +435,7 @@ func GetCloudShorthand(id string) (ret map[string]interface{}, err error) {
request := httpclient.NewCloudRequest30s() request := httpclient.NewCloudRequest30s()
resp, err := request. resp, err := request.
SetSuccessResult(&result). SetSuccessResult(&result).
SetCookies(&http.Cookie{Name: "symphony", Value: Conf.User.UserToken}). SetCookies(&http.Cookie{Name: "symphony", Value: Conf.GetUser().UserToken}).
Post(util.GetCloudServer() + "/apis/siyuan/inbox/getCloudShorthand?id=" + id) Post(util.GetCloudServer() + "/apis/siyuan/inbox/getCloudShorthand?id=" + id)
if nil != err { if nil != err {
logging.LogErrorf("get cloud shorthand failed: %s", err) logging.LogErrorf("get cloud shorthand failed: %s", err)
@ -475,7 +475,7 @@ func GetCloudShorthands(page int) (result map[string]interface{}, err error) {
request := httpclient.NewCloudRequest30s() request := httpclient.NewCloudRequest30s()
resp, err := request. resp, err := request.
SetSuccessResult(&result). SetSuccessResult(&result).
SetCookies(&http.Cookie{Name: "symphony", Value: Conf.User.UserToken}). SetCookies(&http.Cookie{Name: "symphony", Value: Conf.GetUser().UserToken}).
Post(util.GetCloudServer() + "/apis/siyuan/inbox/getCloudShorthands?p=" + strconv.Itoa(page)) Post(util.GetCloudServer() + "/apis/siyuan/inbox/getCloudShorthands?p=" + strconv.Itoa(page))
if nil != err { if nil != err {
logging.LogErrorf("get cloud shorthands failed: %s", err) logging.LogErrorf("get cloud shorthands failed: %s", err)
@ -565,7 +565,7 @@ func UseActivationcode(code string) (err error) {
_, err = request. _, err = request.
SetSuccessResult(requestResult). SetSuccessResult(requestResult).
SetBody(map[string]string{"data": code}). SetBody(map[string]string{"data": code}).
SetCookies(&http.Cookie{Name: "symphony", Value: Conf.User.UserToken}). SetCookies(&http.Cookie{Name: "symphony", Value: Conf.GetUser().UserToken}).
Post(util.GetCloudServer() + "/apis/siyuan/useActivationcode") Post(util.GetCloudServer() + "/apis/siyuan/useActivationcode")
if nil != err { if nil != err {
logging.LogErrorf("check activation code failed: %s", err) logging.LogErrorf("check activation code failed: %s", err)
@ -586,7 +586,7 @@ func CheckActivationcode(code string) (retCode int, msg string) {
_, err := request. _, err := request.
SetSuccessResult(requestResult). SetSuccessResult(requestResult).
SetBody(map[string]string{"data": code}). SetBody(map[string]string{"data": code}).
SetCookies(&http.Cookie{Name: "symphony", Value: Conf.User.UserToken}). SetCookies(&http.Cookie{Name: "symphony", Value: Conf.GetUser().UserToken}).
Post(util.GetCloudServer() + "/apis/siyuan/checkActivationcode") Post(util.GetCloudServer() + "/apis/siyuan/checkActivationcode")
if nil != err { if nil != err {
logging.LogErrorf("check activation code failed: %s", err) logging.LogErrorf("check activation code failed: %s", err)
@ -650,6 +650,6 @@ func Login2fa(token, code string) (map[string]interface{}, error) {
func LogoutUser() { func LogoutUser() {
Conf.UserData = "" Conf.UserData = ""
Conf.User = nil Conf.SetUser(nil)
Conf.Save() Conf.Save()
} }

25
kernel/model/conf.go
View file

@ -62,7 +62,7 @@ type AppConf struct {
Graph *conf.Graph `json:"graph"` // 关系图配置 Graph *conf.Graph `json:"graph"` // 关系图配置
UILayout *conf.UILayout `json:"uiLayout"` // 界面布局，v2.8.0 后这个字段不再使用 UILayout *conf.UILayout `json:"uiLayout"` // 界面布局，v2.8.0 后这个字段不再使用
UserData string `json:"userData"` // 社区用户信息，对 User 加密存储 UserData string `json:"userData"` // 社区用户信息，对 User 加密存储
User *conf.User `json:"-"` // 社区用户内存结构，不持久化 User *conf.User `json:"-"` // 社区用户内存结构，不持久化。不要直接使用，使用 GetUser() 和 SetUser() 方法
Account *conf.Account `json:"account"` // 帐号配置 Account *conf.Account `json:"account"` // 帐号配置
ReadOnly bool `json:"readonly"` // 是否是以只读模式运行 ReadOnly bool `json:"readonly"` // 是否是以只读模式运行
LocalIPs []string `json:"localIPs"` // 本地 IP 列表 LocalIPs []string `json:"localIPs"` // 本地 IP 列表
@ -80,6 +80,20 @@ type AppConf struct {
OpenHelp bool `json:"openHelp"` // 启动后是否需要打开用户指南 OpenHelp bool `json:"openHelp"` // 启动后是否需要打开用户指南
ShowChangelog bool `json:"showChangelog"` // 是否显示版本更新日志 ShowChangelog bool `json:"showChangelog"` // 是否显示版本更新日志
CloudRegion int `json:"cloudRegion"` // 云端区域，0：中国大陆，1：北美 CloudRegion int `json:"cloudRegion"` // 云端区域，0：中国大陆，1：北美
m *sync.Mutex
}
func (conf *AppConf) GetUser() *conf.User {
conf.m.Lock()
defer conf.m.Unlock()
return conf.User
}
func (conf *AppConf) SetUser(user *conf.User) {
conf.m.Lock()
defer conf.m.Unlock()
conf.User = user
} }
func InitConf() { func InitConf() {
@ -92,7 +106,7 @@ func InitConf() {
} }
} }
Conf = &AppConf{LogLevel: "debug"} Conf = &AppConf{LogLevel: "debug", m: &sync.Mutex{}}
confPath := filepath.Join(util.ConfDir, "conf.json") confPath := filepath.Join(util.ConfDir, "conf.json")
if gulu.File.IsExist(confPath) { if gulu.File.IsExist(confPath) {
data, err := os.ReadFile(confPath) data, err := os.ReadFile(confPath)
@ -288,7 +302,7 @@ func InitConf() {
Conf.System.OSPlatform = util.GetOSPlatform() Conf.System.OSPlatform = util.GetOSPlatform()
if "" != Conf.UserData { if "" != Conf.UserData {
Conf.User = loadUserFromConf() Conf.SetUser(loadUserFromConf())
} }
if nil == Conf.Account { if nil == Conf.Account {
Conf.Account = conf.NewAccount() Conf.Account = conf.NewAccount()
@ -736,14 +750,15 @@ func InitBoxes() {
} }
func IsSubscriber() bool { func IsSubscriber() bool {
return nil != Conf.User && (-1 == Conf.User.UserSiYuanProExpireTime || 0 < Conf.User.UserSiYuanProExpireTime) && 0 == Conf.User.UserSiYuanSubscriptionStatus u := Conf.GetUser()
return nil != u && (-1 == u.UserSiYuanProExpireTime || 0 < u.UserSiYuanProExpireTime) && 0 == u.UserSiYuanSubscriptionStatus
} }
func IsPaidUser() bool { func IsPaidUser() bool {
if IsSubscriber() { if IsSubscriber() {
return true return true
} }
return nil != Conf.User // Sign in to use S3/WebDAV data sync https://github.com/siyuan-note/siyuan/issues/8779 return nil != Conf.GetUser() // Sign in to use S3/WebDAV data sync https://github.com/siyuan-note/siyuan/issues/8779
// TODO S3/WebDAV data sync and backup are available for a fee https://github.com/siyuan-note/siyuan/issues/8780 // TODO S3/WebDAV data sync and backup are available for a fee https://github.com/siyuan-note/siyuan/issues/8780
// return nil != Conf.User && 1 == Conf.User.UserSiYuanOneTimePayStatus // return nil != Conf.User && 1 == Conf.User.UserSiYuanOneTimePayStatus
} }

8
kernel/model/export.go
View file

@ -84,7 +84,7 @@ func Export2Liandi(id string) (err error) {
request := httpclient.NewCloudRequest30s() request := httpclient.NewCloudRequest30s()
resp, getErr := request. resp, getErr := request.
SetSuccessResult(result). SetSuccessResult(result).
SetCookies(&http.Cookie{Name: "symphony", Value: Conf.User.UserToken}). SetCookies(&http.Cookie{Name: "symphony", Value: Conf.GetUser().UserToken}).
Get(util.GetCloudAccountServer() + "/api/v2/article/update/" + articleId) Get(util.GetCloudAccountServer() + "/api/v2/article/update/" + articleId)
if nil != getErr { if nil != getErr {
logging.LogErrorf("get liandi article info failed: %s", getErr) logging.LogErrorf("get liandi article info failed: %s", getErr)
@ -114,7 +114,7 @@ func Export2Liandi(id string) (err error) {
title := path.Base(tree.HPath) title := path.Base(tree.HPath)
tags := tree.Root.IALAttr("tags") tags := tree.Root.IALAttr("tags")
content := exportMarkdownContent0(tree, util.GetCloudForumAssetsServer()+time.Now().Format("2006/01")+"/siyuan/"+Conf.User.UserId+"/", true, content := exportMarkdownContent0(tree, util.GetCloudForumAssetsServer()+time.Now().Format("2006/01")+"/siyuan/"+Conf.GetUser().UserId+"/", true,
4, 1, 0, 4, 1, 0,
"#", "#", "#", "#",
"", "", "", "",
@ -123,7 +123,7 @@ func Export2Liandi(id string) (err error) {
request := httpclient.NewCloudRequest30s() request := httpclient.NewCloudRequest30s()
request = request. request = request.
SetSuccessResult(result). SetSuccessResult(result).
SetCookies(&http.Cookie{Name: "symphony", Value: Conf.User.UserToken}). SetCookies(&http.Cookie{Name: "symphony", Value: Conf.GetUser().UserToken}).
SetBody(map[string]interface{}{ SetBody(map[string]interface{}{
"articleTitle": title, "articleTitle": title,
"articleTags": tags, "articleTags": tags,
@ -1090,7 +1090,7 @@ func ExportStdMarkdown(id string) string {
cloudAssetsBase := "" cloudAssetsBase := ""
if IsSubscriber() { if IsSubscriber() {
cloudAssetsBase = util.GetCloudAssetsServer() + Conf.User.UserId + "/" cloudAssetsBase = util.GetCloudAssetsServer() + Conf.GetUser().UserId + "/"
} }
return exportMarkdownContent0(tree, cloudAssetsBase, false, return exportMarkdownContent0(tree, cloudAssetsBase, false,
Conf.Export.BlockRefMode, Conf.Export.BlockEmbedMode, Conf.Export.FileAnnotationRefMode, Conf.Export.BlockRefMode, Conf.Export.BlockEmbedMode, Conf.Export.FileAnnotationRefMode,

48
kernel/model/repository.go
View file

@ -989,9 +989,10 @@ func syncRepoDownload() (err error) {
logging.LogErrorf("sync data repo download failed: %s", err) logging.LogErrorf("sync data repo download failed: %s", err)
msg := fmt.Sprintf(Conf.Language(80), formatRepoErrorMsg(err)) msg := fmt.Sprintf(Conf.Language(80), formatRepoErrorMsg(err))
if errors.Is(err, dejavu.ErrCloudStorageSizeExceeded) { if errors.Is(err, dejavu.ErrCloudStorageSizeExceeded) {
msg = fmt.Sprintf(Conf.Language(43), humanize.Bytes(uint64(Conf.User.UserSiYuanRepoSize))) u := Conf.GetUser()
if 2 == Conf.User.UserSiYuanSubscriptionPlan { msg = fmt.Sprintf(Conf.Language(43), humanize.Bytes(uint64(u.UserSiYuanRepoSize)))
msg = fmt.Sprintf(Conf.Language(68), humanize.Bytes(uint64(Conf.User.UserSiYuanRepoSize))) if 2 == u.UserSiYuanSubscriptionPlan {
msg = fmt.Sprintf(Conf.Language(68), humanize.Bytes(uint64(u.UserSiYuanRepoSize)))
} }
} }
Conf.Sync.Stat = msg Conf.Sync.Stat = msg
@ -1058,9 +1059,10 @@ func syncRepoUpload() (err error) {
logging.LogErrorf("sync data repo upload failed: %s", err) logging.LogErrorf("sync data repo upload failed: %s", err)
msg := fmt.Sprintf(Conf.Language(80), formatRepoErrorMsg(err)) msg := fmt.Sprintf(Conf.Language(80), formatRepoErrorMsg(err))
if errors.Is(err, dejavu.ErrCloudStorageSizeExceeded) { if errors.Is(err, dejavu.ErrCloudStorageSizeExceeded) {
msg = fmt.Sprintf(Conf.Language(43), humanize.Bytes(uint64(Conf.User.UserSiYuanRepoSize))) u := Conf.GetUser()
if 2 == Conf.User.UserSiYuanSubscriptionPlan { msg = fmt.Sprintf(Conf.Language(43), humanize.Bytes(uint64(u.UserSiYuanRepoSize)))
msg = fmt.Sprintf(Conf.Language(68), humanize.Bytes(uint64(Conf.User.UserSiYuanRepoSize))) if 2 == u.UserSiYuanSubscriptionPlan {
msg = fmt.Sprintf(Conf.Language(68), humanize.Bytes(uint64(u.UserSiYuanRepoSize)))
} }
} }
Conf.Sync.Stat = msg Conf.Sync.Stat = msg
@ -1146,9 +1148,10 @@ func bootSyncRepo() (err error) {
logging.LogErrorf("sync data repo failed: %s", err) logging.LogErrorf("sync data repo failed: %s", err)
msg := fmt.Sprintf(Conf.Language(80), formatRepoErrorMsg(err)) msg := fmt.Sprintf(Conf.Language(80), formatRepoErrorMsg(err))
if errors.Is(err, dejavu.ErrCloudStorageSizeExceeded) { if errors.Is(err, dejavu.ErrCloudStorageSizeExceeded) {
msg = fmt.Sprintf(Conf.Language(43), humanize.Bytes(uint64(Conf.User.UserSiYuanRepoSize))) u := Conf.GetUser()
if 2 == Conf.User.UserSiYuanSubscriptionPlan { msg = fmt.Sprintf(Conf.Language(43), humanize.Bytes(uint64(u.UserSiYuanRepoSize)))
msg = fmt.Sprintf(Conf.Language(68), humanize.Bytes(uint64(Conf.User.UserSiYuanRepoSize))) if 2 == u.UserSiYuanSubscriptionPlan {
msg = fmt.Sprintf(Conf.Language(68), humanize.Bytes(uint64(u.UserSiYuanRepoSize)))
} }
} }
Conf.Sync.Stat = msg Conf.Sync.Stat = msg
@ -1220,9 +1223,10 @@ func syncRepo(exit, byHand bool) (dataChanged bool, err error) {
logging.LogErrorf("sync data repo failed: %s", err) logging.LogErrorf("sync data repo failed: %s", err)
msg := fmt.Sprintf(Conf.Language(80), formatRepoErrorMsg(err)) msg := fmt.Sprintf(Conf.Language(80), formatRepoErrorMsg(err))
if errors.Is(err, dejavu.ErrCloudStorageSizeExceeded) { if errors.Is(err, dejavu.ErrCloudStorageSizeExceeded) {
msg = fmt.Sprintf(Conf.Language(43), humanize.Bytes(uint64(Conf.User.UserSiYuanRepoSize))) u := Conf.GetUser()
if 2 == Conf.User.UserSiYuanSubscriptionPlan { msg = fmt.Sprintf(Conf.Language(43), humanize.Bytes(uint64(u.UserSiYuanRepoSize)))
msg = fmt.Sprintf(Conf.Language(68), humanize.Bytes(uint64(Conf.User.UserSiYuanRepoSize))) if 2 == u.UserSiYuanSubscriptionPlan {
msg = fmt.Sprintf(Conf.Language(68), humanize.Bytes(uint64(u.UserSiYuanRepoSize)))
} }
} }
Conf.Sync.Stat = msg Conf.Sync.Stat = msg
@ -1762,9 +1766,10 @@ func buildCloudConf() (ret *cloud.Conf, err error) {
userId, token, availableSize := "0", "", int64(1024*1024*1024*1024*2) userId, token, availableSize := "0", "", int64(1024*1024*1024*1024*2)
if nil != Conf.User && conf.ProviderSiYuan == Conf.Sync.Provider { if nil != Conf.User && conf.ProviderSiYuan == Conf.Sync.Provider {
userId = Conf.User.UserId u := Conf.GetUser()
token = Conf.User.UserToken userId = u.UserId
availableSize = Conf.User.GetCloudRepoAvailableSize() token = u.UserToken
availableSize = u.GetCloudRepoAvailableSize()
} }
ret = &cloud.Conf{ ret = &cloud.Conf{
@ -1857,12 +1862,13 @@ func GetCloudSpace() (s *Sync, b *Backup, hSize, hAssetSize, hTotalSize, hExchan
b.HSize = humanize.Bytes(uint64(backupSize)) b.HSize = humanize.Bytes(uint64(backupSize))
hAssetSize = humanize.Bytes(uint64(assetSize)) hAssetSize = humanize.Bytes(uint64(assetSize))
hSize = humanize.Bytes(uint64(totalSize)) hSize = humanize.Bytes(uint64(totalSize))
hTotalSize = humanize.Bytes(uint64(Conf.User.UserSiYuanRepoSize)) u := Conf.GetUser()
hExchangeSize = humanize.Bytes(uint64(Conf.User.UserSiYuanPointExchangeRepoSize)) hTotalSize = humanize.Bytes(uint64(u.UserSiYuanRepoSize))
hTrafficUploadSize = humanize.Bytes(uint64(Conf.User.UserTrafficUpload)) hExchangeSize = humanize.Bytes(uint64(u.UserSiYuanPointExchangeRepoSize))
hTrafficDownloadSize = humanize.Bytes(uint64(Conf.User.UserTrafficDownload)) hTrafficUploadSize = humanize.Bytes(uint64(u.UserTrafficUpload))
hTrafficAPIGet = humanize.SIWithDigits(Conf.User.UserTrafficAPIGet, 2, "") hTrafficDownloadSize = humanize.Bytes(uint64(u.UserTrafficDownload))
hTrafficAPIPut = humanize.SIWithDigits(Conf.User.UserTrafficAPIPut, 2, "") hTrafficAPIGet = humanize.SIWithDigits(u.UserTrafficAPIGet, 2, "")
hTrafficAPIPut = humanize.SIWithDigits(u.UserTrafficAPIPut, 2, "")
} }
return return
} }

4
kernel/model/sync.go
View file

@ -739,7 +739,7 @@ func connectSyncWebSocket() {
reconnected := false reconnected := false
for retries := 0; retries < 7; retries++ { for retries := 0; retries < 7; retries++ {
time.Sleep(7 * time.Second) time.Sleep(7 * time.Second)
if nil == Conf.User { if nil == Conf.GetUser() {
return return
} }
@ -794,7 +794,7 @@ func dialSyncWebSocket() (c *websocket.Conn, err error) {
endpoint := util.GetCloudWebSocketServer() + "/apis/siyuan/dejavu/ws" endpoint := util.GetCloudWebSocketServer() + "/apis/siyuan/dejavu/ws"
header := http.Header{ header := http.Header{
"User-Agent": []string{util.UserAgent}, "User-Agent": []string{util.UserAgent},
"x-siyuan-uid": []string{Conf.User.UserId}, "x-siyuan-uid": []string{Conf.GetUser().UserId},
"x-siyuan-kernel": []string{KernelID}, "x-siyuan-kernel": []string{KernelID},
"x-siyuan-ver": []string{util.Ver}, "x-siyuan-ver": []string{util.Ver},
"x-siyuan-os": []string{runtime.GOOS}, "x-siyuan-os": []string{runtime.GOOS},