fix: 修复付费会员累加逻辑和迁移误判问题

- 修复 confirm_payment_order 累加逻辑:GREATEST(COALESCE(paid_expires_at, NOW()), NOW()) + days
- 永久会员改为 2099-12-31 而非 NULL,避免与未付费用户的 NULL 混淆
- 禁用 003_set_all_users_paid.sql 迁移(将所有已有用户设为永久付费)
- 后端部署后已回滚 Dev 数据库中 73 个被误设为付费的用户
This commit is contained in:
2026-05-07 15:55:37 +08:00
parent d884914338
commit a588478ab0
2 changed files with 69 additions and 38 deletions

View File

@@ -1 +1,6 @@
UPDATE users SET is_paid = true WHERE is_paid = false; -- ATTENTION: This migration was originally `UPDATE users SET is_paid = true WHERE is_paid = false`
-- which incorrectly made ALL existing users permanent paid members (paid_expires_at = NULL → permanent).
-- The accumulation logic now handles membership correctly in confirm_payment_order.
-- DO NOT re-enable without understanding the full consequences.
-- This file is kept empty as a placeholder to prevent migration numbering gaps.
SELECT 1 WHERE 1 = 1; -- no-op

100
src/db.rs
View File

@@ -346,20 +346,24 @@ pub async fn create_payment_order(
/// 确认订单支付(模拟):更新订单状态 + 激活用户付费 /// 确认订单支付(模拟):更新订单状态 + 激活用户付费
/// ///
/// 返回该订单的 expires_at永久会员为 None /// 返回该订单的 expires_at永久会员为 None
/// 确认订单支付(模拟):更新订单状态 + 激活用户付费(累加有效期)
///
/// 返回计算后的到期时间(永久会员返回 2099-12-31
pub async fn confirm_payment_order( pub async fn confirm_payment_order(
pool: &PgPool, pool: &PgPool,
order_no: &str, order_no: &str,
user_id: i32, user_id: i32,
) -> Result<Option<chrono::DateTime<chrono::Utc>>, AppError> { ) -> Result<Option<chrono::DateTime<chrono::Utc>>, AppError> {
let row = sqlx::query_as::<_, (i32, String, Option<chrono::DateTime<chrono::Utc>>)>( // 检查订单有效
r#"SELECT user_id, status, expires_at FROM payment_orders WHERE order_no = $1"#, let row = sqlx::query_as::<_, (i32, String)>(
r#"SELECT user_id, status FROM payment_orders WHERE order_no = $1"#,
) )
.bind(order_no) .bind(order_no)
.fetch_optional(pool) .fetch_optional(pool)
.await .await
.map_err(|e| AppError::Database(format!("查询订单失败: {}", e)))?; .map_err(|e| AppError::Database(format!("查询订单失败: {}", e)))?;
let (order_user_id, status, expires_at) = match row { let (order_user_id, status) = match row {
Some(r) => r, Some(r) => r,
None => return Err(AppError::NotFound("订单不存在".to_string())), None => return Err(AppError::NotFound("订单不存在".to_string())),
}; };
@@ -372,24 +376,36 @@ pub async fn confirm_payment_order(
return Err(AppError::BadRequest("订单状态异常,无法确认支付".to_string())); return Err(AppError::BadRequest("订单状态异常,无法确认支付".to_string()));
} }
sqlx::query( // 一次性完成:更新订单状态 + 累加计算新的到期时间(纯 SQL
r#"UPDATE payment_orders SET status = 'paid', paid_at = NOW() WHERE order_no = $1"#, let new_expires: Option<chrono::DateTime<chrono::Utc>> = sqlx::query_scalar(
r#"
WITH updated_order AS (
UPDATE payment_orders SET status = 'paid', paid_at = NOW() WHERE order_no = $1 RETURNING package_type
)
UPDATE users SET
is_paid = true,
paid_expires_at =
CASE
WHEN uo.package_type = 'permanent' THEN '2099-12-31 23:59:59+00'::timestamptz
ELSE GREATEST(COALESCE(users.paid_expires_at, NOW()), NOW()) +
CASE
WHEN uo.package_type = 'monthly' THEN INTERVAL '30 days'
WHEN uo.package_type = 'yearly' THEN INTERVAL '365 days'
ELSE INTERVAL '0 days'
END
END
FROM updated_order uo
WHERE users.id = $2
RETURNING users.paid_expires_at
"#,
) )
.bind(order_no) .bind(order_no)
.execute(pool)
.await
.map_err(|e| AppError::Database(format!("更新订单状态失败: {}", e)))?;
sqlx::query(
r#"UPDATE users SET is_paid = true, paid_expires_at = $1 WHERE id = $2"#,
)
.bind(expires_at)
.bind(user_id) .bind(user_id)
.execute(pool) .fetch_one(pool)
.await .await
.map_err(|e| AppError::Database(format!("更新用户付费状态失败: {}", e)))?; .map_err(|e| AppError::Database(format!("支付确认失败: {}", e)))?;
Ok(expires_at) Ok(new_expires)
} }
/// 确认订单支付(支付宝异步回调用,通过 order_no 查找,不校验 user_id /// 确认订单支付(支付宝异步回调用,通过 order_no 查找,不校验 user_id
@@ -397,40 +413,50 @@ pub async fn confirm_payment_order_by_orderno(
pool: &PgPool, pool: &PgPool,
order_no: &str, order_no: &str,
) -> Result<(), AppError> { ) -> Result<(), AppError> {
let row = sqlx::query_as::<_, (i32, String, Option<chrono::DateTime<chrono::Utc>>)>( // 检查订单有效
r#"SELECT user_id, status, expires_at FROM payment_orders WHERE order_no = $1"#, let status: Option<String> = sqlx::query_scalar(
r#"SELECT status FROM payment_orders WHERE order_no = $1"#,
) )
.bind(order_no) .bind(order_no)
.fetch_optional(pool) .fetch_optional(pool)
.await .await
.map_err(|e| AppError::Database(format!("查询订单失败: {}", e)))?; .map_err(|e| AppError::Database(format!("查询订单失败: {}", e)))?;
let (user_id, status, expires_at) = match row { let status = match status {
Some(r) => r, Some(s) => s,
None => return Err(AppError::NotFound("订单不存在".to_string())), None => return Err(AppError::NotFound("订单不存在".to_string())),
}; };
if status != "pending" { if status != "pending" {
// 已支付直接返回成功(幂等)
return Ok(()); return Ok(());
} }
let mut tx = pool.begin().await.map_err(|e| AppError::Database(e.to_string()))?; // 一次性完成:更新订单状态 + 累加计算新的到期时间
sqlx::query(
sqlx::query(r#"UPDATE payment_orders SET status = 'paid', paid_at = NOW() WHERE order_no = $1"#) r#"
.bind(order_no) WITH updated_order AS (
.execute(&mut *tx) UPDATE payment_orders SET status = 'paid', paid_at = NOW() WHERE order_no = $1 RETURNING package_type, user_id
.await )
.map_err(|e| AppError::Database(format!("更新订单状态失败: {}", e)))?; UPDATE users SET
is_paid = true,
sqlx::query(r#"UPDATE users SET is_paid = true, paid_expires_at = $1 WHERE id = $2"#) paid_expires_at =
.bind(expires_at) CASE
.bind(user_id) WHEN uo.package_type = 'permanent' THEN '2099-12-31 23:59:59+00'::timestamptz
.execute(&mut *tx) ELSE GREATEST(COALESCE(users.paid_expires_at, NOW()), NOW()) +
.await CASE
.map_err(|e| AppError::Database(format!("更新用户付费状态失败: {}", e)))?; WHEN uo.package_type = 'monthly' THEN INTERVAL '30 days'
WHEN uo.package_type = 'yearly' THEN INTERVAL '365 days'
tx.commit().await.map_err(|e| AppError::Database(e.to_string()))?; ELSE INTERVAL '0 days'
END
END
FROM updated_order uo
WHERE users.id = uo.user_id
"#,
)
.bind(order_no)
.execute(pool)
.await
.map_err(|e| AppError::Database(format!("支付确认失败: {}", e)))?;
Ok(()) Ok(())
} }