-
Notifications
You must be signed in to change notification settings - Fork 10
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
optimsitic and new comment/admire push notifications #1159
Changes from 1 commit
File filter
Filter by extension
Conversations
Jump to
Diff view
Diff view
There are no files selected for viewing
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -48,6 +48,7 @@ type pushLimiter struct { | |
comments *limiters.KeyRateLimiter | ||
admires *limiters.KeyRateLimiter | ||
follows *limiters.KeyRateLimiter | ||
tokens *limiters.KeyRateLimiter | ||
} | ||
|
||
func newPushLimiter() *pushLimiter { | ||
|
@@ -57,6 +58,7 @@ func newPushLimiter() *pushLimiter { | |
comments: limiters.NewKeyRateLimiter(ctx, cache, "comments", 5, time.Minute), | ||
admires: limiters.NewKeyRateLimiter(ctx, cache, "admires", 1, time.Minute*10), | ||
follows: limiters.NewKeyRateLimiter(ctx, cache, "follows", 1, time.Minute*10), | ||
tokens: limiters.NewKeyRateLimiter(ctx, cache, "tokens", 1, time.Minute*10), | ||
} | ||
} | ||
|
||
|
@@ -101,6 +103,18 @@ func (p *pushLimiter) tryFollow(ctx context.Context, sendingUserID persist.DBID, | |
} | ||
} | ||
|
||
func (p *pushLimiter) tryTokens(ctx context.Context, sendingUserID persist.DBID, tokenID persist.DBID) error { | ||
key := fmt.Sprintf("%s:%s", sendingUserID, tokenID) | ||
if p.isActionAllowed(ctx, p.tokens, key) { | ||
return nil | ||
} | ||
|
||
return errRateLimited{ | ||
limiterName: p.follows.Name(), | ||
senderID: sendingUserID, | ||
} | ||
} | ||
|
||
func (p *pushLimiter) isActionAllowed(ctx context.Context, limiter *limiters.KeyRateLimiter, key string) bool { | ||
canContinue, _, err := limiter.ForKey(ctx, key) | ||
if err != nil { | ||
|
@@ -493,7 +507,7 @@ func createPushMessage(ctx context.Context, notif db.Notification, queries *db.Q | |
}, | ||
} | ||
|
||
if notif.Action == persist.ActionAdmiredFeedEvent { | ||
if notif.Action == persist.ActionAdmiredFeedEvent || notif.Action == persist.ActionAdmiredPost { | ||
admirer, err := queries.GetUserById(ctx, notif.Data.AdmirerIDs[0]) | ||
if err != nil { | ||
return task.PushNotificationMessage{}, err | ||
|
@@ -507,11 +521,16 @@ func createPushMessage(ctx context.Context, notif db.Notification, queries *db.Q | |
return task.PushNotificationMessage{}, fmt.Errorf("user with ID=%s has no username", admirer.ID) | ||
} | ||
|
||
message.Body = fmt.Sprintf("%s admired your activity", admirer.Username.String) | ||
ac := "activity" | ||
if notif.Action == persist.ActionAdmiredPost { | ||
ac = "post" | ||
} | ||
|
||
message.Body = fmt.Sprintf("%s admired your %s", admirer.Username.String, ac) | ||
return message, nil | ||
} | ||
|
||
if notif.Action == persist.ActionCommentedOnFeedEvent { | ||
if notif.Action == persist.ActionCommentedOnFeedEvent || notif.Action == persist.ActionCommentedOnPost { | ||
comment, err := queries.GetCommentByCommentID(ctx, notif.CommentID) | ||
if err != nil { | ||
return task.PushNotificationMessage{}, err | ||
|
@@ -530,7 +549,12 @@ func createPushMessage(ctx context.Context, notif db.Notification, queries *db.Q | |
return task.PushNotificationMessage{}, fmt.Errorf("user with ID=%s has no username", commenter.ID) | ||
} | ||
|
||
message.Body = fmt.Sprintf("%s commented on your activity", commenter.Username.String) | ||
ac := "activity" | ||
if notif.Action == persist.ActionCommentedOnPost { | ||
ac = "post" | ||
} | ||
|
||
message.Body = fmt.Sprintf("%s commented on your %s", commenter.Username.String, ac) | ||
return message, nil | ||
} | ||
|
||
|
@@ -558,6 +582,32 @@ func createPushMessage(ctx context.Context, notif db.Notification, queries *db.Q | |
message.Body = body | ||
return message, nil | ||
} | ||
if notif.Action == persist.ActionNewTokensReceived { | ||
|
||
tm, err := queries.GetTokenMediaByTokenId(ctx, notif.Data.NewTokenID) | ||
if err != nil { | ||
return task.PushNotificationMessage{}, err | ||
} | ||
name := tm.Name | ||
if name == "" { | ||
to, err := queries.GetTokenById(ctx, notif.Data.NewTokenID) | ||
if err != nil { | ||
return task.PushNotificationMessage{}, err | ||
} | ||
name = to.Name.String | ||
} | ||
Comment on lines
+587
to
+598
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Not exactly happy with this. Just like I know we have also been talking about joining There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Hmm, yeah, the duplication here is kind of annoying. Agreed that you could coalesce them and write a more specific query that hits both tables and just returns a name instead of all token media columns. Filling in some of There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. I think I will make a follow-up, maybe fix it week, for creating token_medias immediately and filling them out partially. |
||
|
||
if err := limiter.tryTokens(ctx, notif.OwnerID, notif.Data.NewTokenID); err != nil { | ||
return task.PushNotificationMessage{}, err | ||
} | ||
amount := notif.Data.NewTokenQuantity | ||
i := amount.BigInt().Uint64() | ||
if i > 1 { | ||
message.Body = fmt.Sprintf("You received %d new %s tokens", i, name) | ||
} else { | ||
message.Body = fmt.Sprintf("You received a new %s token", name) | ||
} | ||
} | ||
|
||
return task.PushNotificationMessage{}, fmt.Errorf("unsupported notification action: %s", notif.Action) | ||
} | ||
|
@@ -570,6 +620,12 @@ func actionSupportsPushNotifications(action persist.Action) bool { | |
return true | ||
case persist.ActionUserFollowedUsers: | ||
return true | ||
case persist.ActionNewTokensReceived: | ||
return true | ||
case persist.ActionCommentedOnPost: | ||
return true | ||
case persist.ActionAdmiredPost: | ||
return true | ||
default: | ||
return false | ||
} | ||
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Pretty sure
tm.token_id
is the on-chain "token ID," not the token's actual DBID, right? I think you wanttokens.token_media_id = tm.id
.