From 98cde438271265735cd63f9687f53694b96c06e1 Mon Sep 17 00:00:00 2001
From: ryoha000 <the7uya@gmail.com>
Date: Sat, 7 Dec 2024 16:19:04 +0900
Subject: [PATCH] =?UTF-8?q?post=20/evaluation=20=E3=81=8C=E3=82=BF?=
 =?UTF-8?q?=E3=82=A4=E3=83=A0=E3=82=A2=E3=82=A6=E3=83=88=E3=81=97=E3=81=9F?=
 =?UTF-8?q?=E3=82=89=E5=A4=B1=E6=95=97=E3=81=99=E3=82=8B=E3=82=88=E3=81=86?=
 =?UTF-8?q?=E3=81=AB?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 bench/benchmarker/world/errors.go | 4 ++++
 bench/benchmarker/world/user.go   | 4 ++++
 2 files changed, 8 insertions(+)

diff --git a/bench/benchmarker/world/errors.go b/bench/benchmarker/world/errors.go
index 7335d542..79911086 100644
--- a/bench/benchmarker/world/errors.go
+++ b/bench/benchmarker/world/errors.go
@@ -25,6 +25,8 @@ const (
 	_
 	// ErrorCodeFailedToEvaluate ユーザーが送迎の評価をしようとしたが失敗した
 	ErrorCodeFailedToEvaluate
+	// ErrorCodeEvaluateTimeout ユーザーが送迎の評価をしようとしたがタイムアウトした
+	ErrorCodeEvaluateTimeout
 	// ErrorCodeFailedToCheckRequestHistory ユーザーがリクエスト履歴を確認しようとしたが失敗した
 	ErrorCodeFailedToCheckRequestHistory
 	// ErrorCodeFailedToCreateRequest ユーザーがリクエストを作成しようとしたが失敗した
@@ -88,6 +90,7 @@ var CriticalErrorCodes = map[ErrorCode]bool{
 	ErrorCodeIncorrectAmountOfFareCharged:                   true,
 	ErrorCodeUncontrollableRequestReceived:                  true,
 	ErrorCodeMatchingTimeout:                                true,
+	ErrorCodeEvaluateTimeout:                                true,
 }
 
 var ErrorTexts = map[ErrorCode]string{
@@ -95,6 +98,7 @@ var ErrorTexts = map[ErrorCode]string{
 	ErrorCodeFailedToDepart:                                 "椅子が出発できませんでした",
 	ErrorCodeFailedToAcceptRequest:                          "椅子がライドを受理できませんでした",
 	ErrorCodeFailedToEvaluate:                               "ユーザーのライド評価に失敗しました",
+	ErrorCodeEvaluateTimeout:                                "ユーザーのライド評価がタイムアウトしました",
 	ErrorCodeFailedToCheckRequestHistory:                    "ユーザーがライド履歴の取得に失敗しました",
 	ErrorCodeFailedToCreateRequest:                          "ユーザーが新しくライドを作成できませんでした",
 	ErrorCodeUserNotRequestingButStatusChanged:              "ユーザーが想定していない通知を受け取りました",
diff --git a/bench/benchmarker/world/user.go b/bench/benchmarker/world/user.go
index 25c035aa..b590e4df 100644
--- a/bench/benchmarker/world/user.go
+++ b/bench/benchmarker/world/user.go
@@ -1,6 +1,7 @@
 package world
 
 import (
+	"context"
 	"errors"
 	"fmt"
 	"log/slog"
@@ -175,6 +176,9 @@ func (u *User) Tick(ctx *Context) error {
 				res, err := u.Client.SendEvaluation(ctx, u.Request, score)
 				if err != nil {
 					u.Request.Statuses.Unlock()
+					if errors.Is(err, context.DeadlineExceeded) {
+						return WrapCodeError(ErrorCodeEvaluateTimeout, err)
+					}
 					return WrapCodeError(ErrorCodeFailedToEvaluate, err)
 				}