123 lines
3.9 KiB
Go
123 lines
3.9 KiB
Go
package app
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"fmt"
|
|
"strconv"
|
|
)
|
|
|
|
const defaultFreeDailyActiveChatLimit = 5
|
|
|
|
type rowQuerier interface {
|
|
QueryRowContext(context.Context, string, ...any) *sql.Row
|
|
}
|
|
|
|
type dailyActiveChatLimitError struct {
|
|
Limit int
|
|
}
|
|
|
|
func (e *dailyActiveChatLimitError) Error() string {
|
|
return fmt.Sprintf("今日主动聊天人数已达上限（%d人），回复收到的消息不受此限制", e.Limit)
|
|
}
|
|
|
|
func (a *App) resolveDailyActiveChatLimit(ctx context.Context, queryer rowQuerier, userID int64) int {
|
|
var limit int
|
|
err := queryer.QueryRowContext(ctx, `SELECT p.daily_active_chat_limit
|
|
FROM subscriptions s JOIN membership_plans p ON p.id=s.plan_id
|
|
WHERE s.user_id=? AND s.status=1 AND s.started_at<=NOW(3) AND s.expires_at>NOW(3)
|
|
ORDER BY p.level DESC,s.expires_at DESC LIMIT 1`, userID).Scan(&limit)
|
|
if err == nil {
|
|
return limit
|
|
}
|
|
|
|
var raw string
|
|
if err = queryer.QueryRowContext(ctx, `SELECT config_value FROM system_configs WHERE config_key='membership.free_daily_active_chat_limit'`).Scan(&raw); err == nil {
|
|
if parsed, parseErr := strconv.Atoi(raw); parseErr == nil && parsed >= 0 {
|
|
return parsed
|
|
}
|
|
}
|
|
return defaultFreeDailyActiveChatLimit
|
|
}
|
|
|
|
func dailyActiveChatQuotaView(limit, used int) map[string]any {
|
|
remaining := -1
|
|
unlimited := limit == 0
|
|
if !unlimited {
|
|
remaining = limit - used
|
|
if remaining < 0 {
|
|
remaining = 0
|
|
}
|
|
}
|
|
return map[string]any{
|
|
"limit": limit,
|
|
"remaining": remaining,
|
|
"unlimited": unlimited,
|
|
"used": used,
|
|
}
|
|
}
|
|
|
|
func (a *App) dailyActiveChatQuota(ctx context.Context, userID int64) map[string]any {
|
|
limit := a.resolveDailyActiveChatLimit(ctx, a.db, userID)
|
|
var used int
|
|
_ = a.db.QueryRowContext(ctx, `SELECT used_count FROM im_daily_active_chat_usage WHERE user_id=? AND usage_date=CURRENT_DATE()`, userID).Scan(&used)
|
|
return dailyActiveChatQuotaView(limit, used)
|
|
}
|
|
|
|
func (a *App) reserveDailyActiveChat(ctx context.Context, tx *sql.Tx, conversationID, senderID int64) error {
|
|
var user1ID, user2ID int64
|
|
err := tx.QueryRowContext(ctx, `SELECT user1_id,user2_id FROM im_direct_conversations WHERE conversation_id=?`, conversationID).Scan(&user1ID, &user2ID)
|
|
if err == sql.ErrNoRows {
|
|
return nil
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
targetUserID := user1ID
|
|
if senderID == user1ID {
|
|
targetUserID = user2ID
|
|
} else if senderID != user2ID {
|
|
return fmt.Errorf("不是会话成员")
|
|
}
|
|
|
|
var inboundToday int
|
|
if err = tx.QueryRowContext(ctx, `SELECT EXISTS(
|
|
SELECT 1 FROM im_messages
|
|
WHERE conversation_id=? AND sender_id=?
|
|
AND created_at>=CURRENT_DATE() AND created_at<DATE_ADD(CURRENT_DATE(),INTERVAL 1 DAY)
|
|
)`, conversationID, targetUserID).Scan(&inboundToday); err != nil {
|
|
return err
|
|
}
|
|
if inboundToday == 1 {
|
|
return nil
|
|
}
|
|
|
|
if _, err = tx.ExecContext(ctx, `INSERT IGNORE INTO im_daily_active_chat_usage(user_id,usage_date,used_count) VALUES(?,CURRENT_DATE(),0)`, senderID); err != nil {
|
|
return err
|
|
}
|
|
var used int
|
|
if err = tx.QueryRowContext(ctx, `SELECT used_count FROM im_daily_active_chat_usage WHERE user_id=? AND usage_date=CURRENT_DATE() FOR UPDATE`, senderID).Scan(&used); err != nil {
|
|
return err
|
|
}
|
|
var alreadyCounted int
|
|
if err = tx.QueryRowContext(ctx, `SELECT EXISTS(
|
|
SELECT 1 FROM im_daily_active_chat_targets
|
|
WHERE user_id=? AND target_user_id=? AND usage_date=CURRENT_DATE()
|
|
)`, senderID, targetUserID).Scan(&alreadyCounted); err != nil {
|
|
return err
|
|
}
|
|
if alreadyCounted == 1 {
|
|
return nil
|
|
}
|
|
|
|
limit := a.resolveDailyActiveChatLimit(ctx, tx, senderID)
|
|
if limit > 0 && used >= limit {
|
|
return &dailyActiveChatLimitError{Limit: limit}
|
|
}
|
|
if _, err = tx.ExecContext(ctx, `INSERT INTO im_daily_active_chat_targets(user_id,target_user_id,usage_date,conversation_id) VALUES(?,?,CURRENT_DATE(),?)`, senderID, targetUserID, conversationID); err != nil {
|
|
return err
|
|
}
|
|
_, err = tx.ExecContext(ctx, `UPDATE im_daily_active_chat_usage SET used_count=used_count+1 WHERE user_id=? AND usage_date=CURRENT_DATE()`, senderID)
|
|
return err
|
|
}
|