YourSelfHosted/memos
1
0
Fork 0
mirror of https://github.com/usememos/memos.git synced 2025-02-16 03:12:13 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix: version checker

Browse Source
This commit is contained in:
Steven 2023-11-06 22:33:12 +08:00
parent 363c107359
commit 342d1aeefb
3 changed files with 17 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
server/server.go
View File

@ -110,7 +110,7 @@ func NewServer(ctx context.Context, profile *profile.Profile, store *store.Store
}
func (s *Server) Start(ctx context.Context) error {
go versionchecker.NewVersionChecker(s.Store).Start(ctx)
go versionchecker.NewVersionChecker(s.Store, s.Profile).Start(ctx)
go s.telegramBot.Start(ctx)
go s.backupRunner.Run(ctx)

35
server/service/version_checker/version_checker.go
View File

@ -11,18 +11,21 @@ import (
"github.com/pkg/errors"
storepb "github.com/usememos/memos/proto/gen/store"
"github.com/usememos/memos/server/profile"
"github.com/usememos/memos/server/version"
"github.com/usememos/memos/store"
)
// nolint
type VersionChecker struct {
Store *store.Store
Store *store.Store
Profile *profile.Profile
}
func NewVersionChecker(store *store.Store) *VersionChecker {
func NewVersionChecker(store *store.Store, profile *profile.Profile) *VersionChecker {
return &VersionChecker{
Store: store,
Store: store,
Profile: profile,
}
}
@ -47,20 +50,11 @@ func (*VersionChecker) GetLatestVersion() (string, error) {
}
func (c *VersionChecker) Check(ctx context.Context) {
migrationHistories, err := c.Store.FindMigrationHistoryList(ctx, &store.FindMigrationHistory{})
if err != nil {
return
}
if len(migrationHistories) == 0 {
return
}
lastVersion := migrationHistories[0].Version
latestVersion, err := c.GetLatestVersion()
if err != nil {
return
}
if !version.IsVersionGreaterThan(latestVersion, lastVersion) {
if !version.IsVersionGreaterThan(latestVersion, version.GetCurrentVersion(c.Profile.Mode)) {
return
}
@ -71,15 +65,16 @@ func (c *VersionChecker) Check(ctx context.Context) {
if err != nil {
return
}
if len(list) == 0 {
return
shouldNotify := true
if len(list) > 0 {
latestVersionUpdateActivity := list[0]
if latestVersionUpdateActivity.Payload != nil && version.IsVersionGreaterOrEqualThan(latestVersionUpdateActivity.Payload.VersionUpdate.Version, latestVersion) {
shouldNotify = false
}
}
latestVersionUpdateActivity := list[0]
if latestVersionUpdateActivity.Payload == nil {
return
}
if version.IsVersionGreaterOrEqualThan(latestVersionUpdateActivity.Payload.VersionUpdate.Version, latestVersion) {
if !shouldNotify {
return
}

2
server/service/version_checker/version_checker_test.go
View File

@ -7,6 +7,6 @@ import (
)
func TestGetLatestVersion(t *testing.T) {
_, err := NewVersionChecker(nil).GetLatestVersion()
_, err := NewVersionChecker(nil, nil).GetLatestVersion()
require.NoError(t, err)
}