yesBad/proton-bridge
1
0
Fork 0
forked from Silverfish/proton-bridge
Code Actions Activity

GODT-1737: Improve logging during import

Browse Source
This commit is contained in:
Jakub
2022-07-15 18:28:15 +02:00
parent 9bb16dec48
commit b3f8866ef7
2 changed files with 13 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
pkg/pmapi/import.go
View File

@ -90,7 +90,14 @@ type ImportMsgRes struct {
// Import imports messages to the user's account.
func (c *client) Import(ctx context.Context, reqs ImportMsgReqs) ([]*ImportMsgRes, error) {
if len(reqs) == 0 {
return nil, errors.New("missing import requests")
}
if len(reqs) > MaxImportMessageRequestLength {
log.
WithField("count", len(reqs)).
Warn("Importing too many messages at once.")
return nil, errors.New("request is too long")
}
@ -98,6 +105,10 @@ func (c *client) Import(ctx context.Context, reqs ImportMsgReqs) ([]*ImportMsgRe
for _, req := range reqs {
remainingSize -= len(req.Message)
if remainingSize < 0 {
log.
WithField("count", len(reqs)).
WithField("size", MaxImportMessageRequestLength-remainingSize).
Warn("Importing too big message(s)")
return nil, errors.New("request size is too big")
}
}