-
-
Notifications
You must be signed in to change notification settings - Fork 159
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge branch 'feature/unread' into develop
- Loading branch information
Showing
7 changed files
with
158 additions
and
50 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,103 @@ | ||
| package storage | ||
|
|
||
| import ( | ||
| "sync" | ||
|
|
||
| "github.com/axllent/mailpit/data" | ||
| "github.com/axllent/mailpit/logger" | ||
| "github.com/ostafen/clover/v2" | ||
| ) | ||
|
|
||
| var ( | ||
| mailboxStats = map[string]data.MailboxStats{} | ||
| statsLock = sync.RWMutex{} | ||
| ) | ||
|
|
||
| // StatsGet returns the total/unread statistics for a mailbox | ||
| func StatsGet(mailbox string) data.MailboxStats { | ||
| statsLock.Lock() | ||
| defer statsLock.Unlock() | ||
| s, ok := mailboxStats[mailbox] | ||
| if !ok { | ||
| return data.MailboxStats{ | ||
| Total: 0, | ||
| Unread: 0, | ||
| } | ||
| } | ||
| return s | ||
| } | ||
|
|
||
| // Refresh will completely refresh the existing stats for a given mailbox | ||
| func statsRefresh(mailbox string) error { | ||
| logger.Log().Debugf("[stats] refreshing stats for %s", mailbox) | ||
|
|
||
| total, err := db.Count(clover.NewQuery(mailbox)) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| unread, err := db.Count(clover.NewQuery(mailbox).Where(clover.Field("Read").IsFalse())) | ||
| if err != nil { | ||
| return nil | ||
| } | ||
|
|
||
| statsLock.Lock() | ||
| mailboxStats[mailbox] = data.MailboxStats{ | ||
| Total: total, | ||
| Unread: unread, | ||
| } | ||
| statsLock.Unlock() | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| func statsAddNewMessage(mailbox string) { | ||
| statsLock.Lock() | ||
| s, ok := mailboxStats[mailbox] | ||
| if ok { | ||
| mailboxStats[mailbox] = data.MailboxStats{ | ||
| Total: s.Total + 1, | ||
| Unread: s.Unread + 1, | ||
| } | ||
| } | ||
| statsLock.Unlock() | ||
| } | ||
|
|
||
| // Deleting one will always mean it was read | ||
| func statsDeleteOneMessage(mailbox string) { | ||
| statsLock.Lock() | ||
| s, ok := mailboxStats[mailbox] | ||
| if ok { | ||
| mailboxStats[mailbox] = data.MailboxStats{ | ||
| Total: s.Total - 1, | ||
| Unread: s.Unread, | ||
| } | ||
| } | ||
| statsLock.Unlock() | ||
| } | ||
|
|
||
| // Mark one message as read | ||
| func statsReadOneMessage(mailbox string) { | ||
| statsLock.Lock() | ||
| s, ok := mailboxStats[mailbox] | ||
| if ok { | ||
| mailboxStats[mailbox] = data.MailboxStats{ | ||
| Total: s.Total, | ||
| Unread: s.Unread - 1, | ||
| } | ||
| } | ||
| statsLock.Unlock() | ||
| } | ||
|
|
||
| // Mark one message as unread | ||
| func statsUnreadOneMessage(mailbox string) { | ||
| statsLock.Lock() | ||
| s, ok := mailboxStats[mailbox] | ||
| if ok { | ||
| mailboxStats[mailbox] = data.MailboxStats{ | ||
| Total: s.Total, | ||
| Unread: s.Unread + 1, | ||
| } | ||
| } | ||
| statsLock.Unlock() | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters