tamakyi
0124e392c7
四处都属于「并发/失败路径下状态不一致」，用户侧表现为「点了没反应」「发了没收到」。 1) 删除会话重复回退配额 MarkDeletedByOwner 是 check-then-act：双击时两个请求都读到 is_deleted_by_owner=false， 都执行 UPDATE、都执行一次 IncrementQuota(-1)。计数不会变负，但会白吃掉其他会话的额度， 表现为定量收集提前关闭，箱主重置计数也校准不回来。 → 改条件更新 + RowsAffected 决定是否回退。 2) 双击产生两条重复追问 「上一条追问必须先被回复」原来只在 handler 里判断（事务外）：两个请求都能通过那次检查， 随后被 CreateIfBelowLimit 的问题行锁串行化，各自 count 都 < 3 —— 两条相同追问都建成， 吃掉 2/3 名额，箱主还收两封邮件。 → 判定挪进事务，新增 ErrFollowUpAwaitingReply；handler 的预检查保留作为快速失败。 3) 双击置顶结果与预期相反 SetPinned(id, !question.IsPinned) 是 read-modify-write：两个请求都读到 false 都写 true， 用户点两次期望「置顶后取消」，实际仍是「置顶」。 → 新增 SetPinnedIfCurrent(current, next)，第二次因状态已变而 0 行，语义是「只切换一次」。 4) 私聊首条三步写入无事务 建会话 → 写首条消息 → 计数是三次独立写入，第二步失败会留下 message_count=0 的空会话， 箱主列表里能点进去却什么都没有；而 BeforeCreate 已占掉一个 6 字符 token（唯一索引）， 孤儿会话永久占用它。 → 新增 CreateWithFirstMessage，三步收进一个事务。 5) 验证码失败丢正文 该分支原来走 SetErrorFlash + Redirect，而其它失败分支都是 SetError + Success 回填表单。 验证码失败恰恰是最常见的失败（token 过期、网络抖动），结果用户写好的正文整段丢失。 → 改为同请求渲染并回填；错误文案不再透传 err.Error()（可能含上游原始信息）。 6) 5 处裸类型断言改带 ok（share.go ×3 / context.go ×1 / user/vote.go ×1）， 断言失败时丢弃脏缓存并走重建，而不是 panic 掉整个请求。share.go 的 ttlCache 补 Delete。 验证：内部 DB 测试 3 项（并发删除只回退一次配额 5→4、双击只产生一条追问、 双击只切换一次置顶）；verify_round3.py 6/6（验证码失败 200+正文回填+不建提问、 私聊首条 message_count 与实际消息数一致）。
220 lines
8.8 KiB
Go
220 lines
8.8 KiB
Go
package question
|
||
|
||
import (
|
||
stdctx "context"
|
||
"fmt"
|
||
"unicode/utf8"
|
||
|
||
"github.com/pkg/errors"
|
||
"github.com/sirupsen/logrus"
|
||
|
||
"github.com/tamakyi/TamaBox/internal/conf"
|
||
"github.com/tamakyi/TamaBox/internal/context"
|
||
"github.com/tamakyi/TamaBox/internal/db"
|
||
"github.com/tamakyi/TamaBox/internal/form"
|
||
"github.com/tamakyi/TamaBox/internal/i18n"
|
||
"github.com/tamakyi/TamaBox/internal/mail"
|
||
"github.com/tamakyi/TamaBox/internal/security/moderation"
|
||
"github.com/tamakyi/TamaBox/internal/shared"
|
||
)
|
||
|
||
// isQuestioner 判断当前请求者是否为该提问的提问者。
|
||
// 匿名提问者通过 token（URL 或表单隐藏域）验证；登录提问者通过账号 ID 验证。
|
||
func isQuestioner(ctx context.Context, question *db.Question, token string) bool {
|
||
if question.Token != "" && question.Token == token {
|
||
return true
|
||
}
|
||
if ctx.IsLogged && question.AskerUserID != 0 && ctx.User.ID == question.AskerUserID {
|
||
return true
|
||
}
|
||
return false
|
||
}
|
||
|
||
// PublishFollowUp 提问者在问题被回答后，用 token 继续追问。
|
||
func PublishFollowUp(ctx context.Context, pageUser *db.User, question *db.Question, f form.PublishFollowUpQuestion) {
|
||
// govalid 校验失败（追问内容为空）时回显表单，与 AnswerFollowUp 同路径。
|
||
// （旧实现漏掉该早退：required 失败后仍会继续 censor/入库，追问空内容也会被创建。）
|
||
if ctx.HasError() {
|
||
ctx.Success("question/item")
|
||
return
|
||
}
|
||
|
||
// 追问内容长度上限（conf.Limits.FollowUpContentLen，[limits] 配置段），
|
||
// 与前端追问框 maxlength 同源，rune 计数。
|
||
if utf8.RuneCountInString(f.Content) > conf.Limits.FollowUpContentLen {
|
||
ctx.SetError(errors.New(i18n.T(ctx.Lang(), "form.followup.contentTooLong", conf.Limits.FollowUpContentLen)), f)
|
||
ctx.Success("question/item")
|
||
return
|
||
}
|
||
|
||
token := ctx.Query("t")
|
||
if token == "" {
|
||
token = f.Token
|
||
}
|
||
|
||
if !isQuestioner(ctx, question, token) {
|
||
ctx.SetErrorFlash(i18n.T(ctx.Lang(), "followup.errNoPermission"))
|
||
ctx.Redirect("/")
|
||
return
|
||
}
|
||
|
||
// 只能对已回答的问题追问
|
||
if question.Answer == "" {
|
||
ctx.SetErrorFlash(i18n.T(ctx.Lang(), "followup.errNotAnswered"))
|
||
ctx.Redirect(fmt.Sprintf("/_/%s/%d?t=%s", pageUser.Domain, question.ID, question.Token))
|
||
return
|
||
}
|
||
|
||
// 追问次数限制：先快速预检查（友好提示），CreateIfBelowLimit 内再原子兜底
|
||
// （并发场景下 count-then-create 存在竞态，必须以数据库原子检查为准）。
|
||
count, err := db.FollowUps.CountByQuestionID(ctx.Request().Context(), question.ID)
|
||
if err != nil {
|
||
logrus.WithContext(ctx.Request().Context()).WithError(err).Error("Failed to count follow ups")
|
||
ctx.SetInternalErrorFlash()
|
||
ctx.Redirect(fmt.Sprintf("/_/%s/%d?t=%s", pageUser.Domain, question.ID, question.Token))
|
||
return
|
||
}
|
||
if count >= db.MaxFollowUpCount {
|
||
ctx.SetErrorFlash(i18n.T(ctx.Lang(), "followup.errLimit", db.MaxFollowUpCount))
|
||
ctx.Redirect(fmt.Sprintf("/_/%s/%d?t=%s", pageUser.Domain, question.ID, question.Token))
|
||
return
|
||
}
|
||
|
||
// 服务端校验"等主人回复"：模板仅在前一条被回复后才展示追问表单，
|
||
// 但 POST 可被直接构造绕过，因此必须检查最新一条追问是否已回复。
|
||
//
|
||
// 这一段只是「快速失败」省掉一次事务开销；并发/双击下的正确性由
|
||
// db.FollowUps.CreateIfBelowLimit 在事务内重新判定（见 ErrFollowUpAwaitingReply）。
|
||
latest, err := db.FollowUps.GetLatestByQuestionID(ctx.Request().Context(), question.ID)
|
||
if err != nil {
|
||
logrus.WithContext(ctx.Request().Context()).WithError(err).Error("Failed to get latest follow up")
|
||
ctx.SetInternalErrorFlash()
|
||
ctx.Redirect(fmt.Sprintf("/_/%s/%d?t=%s", pageUser.Domain, question.ID, question.Token))
|
||
return
|
||
}
|
||
if latest != nil && latest.Answer == "" {
|
||
ctx.SetErrorFlash(i18n.T(ctx.Lang(), "followup.errWaitReply"))
|
||
ctx.Redirect(fmt.Sprintf("/_/%s/%d?t=%s", pageUser.Domain, question.ID, question.Token))
|
||
return
|
||
}
|
||
|
||
if ctx.HasError() {
|
||
ctx.Success("question/item")
|
||
return
|
||
}
|
||
|
||
// 内容安全检测：统一守卫，按箱主「审核严格度」裁决是否放行
|
||
// （提问 / 投票 / 私聊共用同一份判定，见 internal/security/moderation）。
|
||
if _, err := moderation.Check(ctx.Request().Context(), moderation.EffectiveStrictness(ctx.Request().Context(), pageUser.ID, pageUser.ModerationStrictness), ctx.Lang(), f.Content); err != nil {
|
||
ctx.SetErrorFlash(err.Error())
|
||
ctx.Redirect(fmt.Sprintf("/_/%s/%d?t=%s", pageUser.Domain, question.ID, question.Token))
|
||
return
|
||
}
|
||
|
||
created, err := db.FollowUps.CreateIfBelowLimit(ctx.Request().Context(), db.CreateFollowUpOptions{
|
||
QuestionID: question.ID,
|
||
Content: f.Content,
|
||
})
|
||
if err != nil {
|
||
if errors.Is(err, db.ErrFollowUpLimitReached) {
|
||
ctx.SetErrorFlash(i18n.T(ctx.Lang(), "followup.errLimit", db.MaxFollowUpCount))
|
||
ctx.Redirect(fmt.Sprintf("/_/%s/%d?t=%s", pageUser.Domain, question.ID, question.Token))
|
||
return
|
||
}
|
||
if errors.Is(err, db.ErrFollowUpAwaitingReply) {
|
||
// 事务层的兜底（并发双击时会走到这里）
|
||
ctx.SetErrorFlash(i18n.T(ctx.Lang(), "followup.errWaitReply"))
|
||
ctx.Redirect(fmt.Sprintf("/_/%s/%d?t=%s", pageUser.Domain, question.ID, question.Token))
|
||
return
|
||
}
|
||
logrus.WithContext(ctx.Request().Context()).WithError(err).Error("Failed to create follow up")
|
||
ctx.SetInternalErrorFlash()
|
||
ctx.Redirect(fmt.Sprintf("/_/%s/%d?t=%s", pageUser.Domain, question.ID, question.Token))
|
||
return
|
||
}
|
||
_ = created
|
||
|
||
// 邮件通知提问箱主人（开启邮件提醒时）
|
||
shared.SafeGo(func() {
|
||
if pageUser.Notify == db.NotifyTypeEmail {
|
||
if err := mail.SendNewFollowUpMail(pageUser.Email, pageUser.Domain, question.ID, question.Content, f.Content, question.Token); err != nil {
|
||
logrus.WithContext(stdctx.Background()).WithError(err).Error("Failed to send new follow up mail to user")
|
||
}
|
||
}
|
||
})
|
||
|
||
ctx.SetSuccessFlash(i18n.T(ctx.Lang(), "followup.flash.sent"))
|
||
ctx.Redirect(fmt.Sprintf("/_/%s/%d?t=%s", pageUser.Domain, question.ID, question.Token))
|
||
}
|
||
|
||
// AnswerFollowUp 提问箱主人回答某条追问。
|
||
func AnswerFollowUp(ctx context.Context, pageUser *db.User, question *db.Question, f form.AnswerFollowUpQuestion) {
|
||
if ctx.User.ID != pageUser.ID {
|
||
ctx.Redirect("/")
|
||
return
|
||
}
|
||
|
||
if ctx.HasError() {
|
||
ctx.Success("question/item")
|
||
return
|
||
}
|
||
|
||
// 追问的回答内容长度上限：与主回答共用 conf.Limits.AnswerContentLen
|
||
// （[limits] 配置段），与前端回复框 maxlength 同源，rune 计数。
|
||
if utf8.RuneCountInString(f.Answer) > conf.Limits.AnswerContentLen {
|
||
ctx.SetError(errors.New(i18n.T(ctx.Lang(), "form.followup.answerTooLong", conf.Limits.AnswerContentLen)), f)
|
||
ctx.Success("question/item")
|
||
return
|
||
}
|
||
|
||
// 站方处置 L2 起拒绝回答追问：追问回答同样是公开内容，且没有「私密」这一档可以降级。
|
||
restrictionLevel, _, err := db.Restrictions.ActiveLevel(ctx.Request().Context(), pageUser.ID)
|
||
if err != nil {
|
||
logrus.WithContext(ctx.Request().Context()).WithError(err).Error("Failed to query user restriction")
|
||
ctx.SetInternalError()
|
||
ctx.Success("question/item")
|
||
return
|
||
}
|
||
if db.RestrictionAtLeast(restrictionLevel, db.RestrictionL2) {
|
||
// 同请求渲染，用 SetError（SetErrorFlash 要等下一次请求才显示）。
|
||
ctx.SetError(errors.New(i18n.T(ctx.Lang(), "question.errRestrictedPublish")), f)
|
||
ctx.Success("question/item")
|
||
return
|
||
}
|
||
|
||
followUpID := uint(ctx.ParamInt("followUpID"))
|
||
|
||
// 内容安全检测：统一守卫，箱主本人回复同样按自己的审核严格度裁决。
|
||
if _, err := moderation.Check(ctx.Request().Context(), moderation.EffectiveStrictnessFor(restrictionLevel, pageUser.ModerationStrictness), ctx.Lang(), f.Answer); err != nil {
|
||
// 同请求渲染，用 SetError；原先的 SetErrorFlash 在这里不会显示出来（既有缺陷）。
|
||
ctx.SetError(err, f)
|
||
ctx.Success("question/item")
|
||
return
|
||
}
|
||
|
||
if err := db.FollowUps.UpdateAnswerByID(ctx.Request().Context(), question.ID, followUpID, f.Answer); err != nil {
|
||
if errors.Is(err, db.ErrFollowUpNotFound) {
|
||
// 同请求渲染需用 SetError（SetErrorFlash 要等下一次请求才显示）。
|
||
ctx.SetError(errors.New(i18n.T(ctx.Lang(), "followup.errNotFound")), f)
|
||
ctx.Success("question/item")
|
||
return
|
||
}
|
||
logrus.WithContext(ctx.Request().Context()).WithError(err).Error("Failed to answer follow up")
|
||
ctx.SetInternalError()
|
||
ctx.Success("question/item")
|
||
return
|
||
}
|
||
|
||
// 邮件通知提问者（留下邮箱时）：复用"您的提问有了回复"模板，带 token 链接
|
||
shared.SafeGo(func() {
|
||
if question.ReceiveReplyEmail != "" {
|
||
if err := mail.SendNewAnswerMail(question.ReceiveReplyEmail, pageUser.Domain, question.ID, question.Content, f.Answer, question.Token); err != nil {
|
||
logrus.WithContext(stdctx.Background()).WithError(err).Error("Failed to send follow up answer mail to questioner")
|
||
}
|
||
}
|
||
})
|
||
|
||
ctx.SetSuccessFlash(i18n.T(ctx.Lang(), "followup.flash.answered"))
|
||
ctx.Redirect(fmt.Sprintf("/_/%s/%d", pageUser.Domain, question.ID))
|
||
}
|