予約枠数を環境変数から読み込めるように

isucon · Nov 1, 2023 · 2488481 · 2488481
1 parent bcb0c15
commit 2488481
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/webapp/go/livestream_handler.go b/webapp/go/livestream_handler.go
@@ -15,9 +15,6 @@ import (
 	"github.com/labstack/echo/v4"
 )
 
-// 同時配信可能予約枠数
-const NumReservationSlot = 2
-
 type ReserveLivestreamRequest struct {
 	Tags        []int64 `json:"tags"`
 	Title       string  `json:"title"`
@@ -122,7 +119,7 @@ func reserveLivestreamHandler(c echo.Context) error {
 		if err := tx.GetContext(ctx, &founds, "SELECT COUNT(*) FROM livestreams WHERE user_id = ? AND  ? >= start_at AND ? <= end_at", user, reserveStartAt, reserveEndAt); err != nil {
 			return echo.NewHTTPError(http.StatusInternalServerError, err)
 		}
-		if founds >= NumReservationSlot {
+		if founds >= numReservationSlot {
 			return echo.NewHTTPError(http.StatusBadRequest, fmt.Sprintf("ユーザ%dが予約できません", user))
 		}
 	}

diff --git a/webapp/go/main.go b/webapp/go/main.go
@@ -31,13 +31,20 @@ var (
 	powerDNSSubdomainAddress string
 	dbConn                   *sqlx.DB
 	secret                   = []byte("isucon13_session_cookiestore_defaultsecret")
+	numReservationSlot       = 2
 )
 
 func init() {
 	log.SetFlags(log.Ldate | log.Ltime | log.Lshortfile)
 	if secretKey, ok := os.LookupEnv("ISUCON13_SESSION_SECRETKEY"); ok {
 		secret = []byte(secretKey)
 	}
+	if slotStr, ok := os.LookupEnv("ISUCON13_NUM_RESERVATION_SLOT"); ok {
+		slot, err := strconv.Atoi(slotStr)
+		if err == nil {
+			numReservationSlot = slot
+		}
+	}
 }
 
 // FIXME: ポータルと足並み揃えて修正