chore(matchmaking): sync with upstream

This commit is contained in:
MingxuanGame
2025-11-22 13:15:24 +00:00
parent 97f8713c71
commit 4b942e3cb3
2 changed files with 73 additions and 9 deletions

View File

@@ -24,9 +24,9 @@ class MatchmakingUserStatsBase(SQLModel, UTCBaseModel):
default=None, default=None,
sa_column=Column(BigInteger, ForeignKey("lazer_users.id"), primary_key=True), sa_column=Column(BigInteger, ForeignKey("lazer_users.id"), primary_key=True),
) )
ruleset_id: int = Field( pool_id: int = Field(
default=None, default=None,
sa_column=Column(SmallInteger, primary_key=True), sa_column=Column(ForeignKey("matchmaking_pools.id"), primary_key=True, nullable=True),
) )
first_placements: int = Field(default=0, ge=0) first_placements: int = Field(default=0, ge=0)
total_points: int = Field(default=0, ge=0) total_points: int = Field(default=0, ge=0)
@@ -43,8 +43,13 @@ class MatchmakingUserStatsBase(SQLModel, UTCBaseModel):
class MatchmakingUserStats(MatchmakingUserStatsBase, table=True): class MatchmakingUserStats(MatchmakingUserStatsBase, table=True):
__tablename__: str = "matchmaking_user_stats" __tablename__: str = "matchmaking_user_stats"
__table_args__ = (
Index("matchmaking_user_stats_pool_first_idx", "pool_id", "first_placements"),
Index("matchmaking_user_stats_pool_points_idx", "pool_id", "total_points"),
)
user: "User" = Relationship(back_populates="matchmaking_stats", sa_relationship_kwargs={"lazy": "joined"}) user: "User" = Relationship(back_populates="matchmaking_stats", sa_relationship_kwargs={"lazy": "joined"})
pool: "MatchmakingPool" = Relationship()
class MatchmakingPoolBase(SQLModel, UTCBaseModel): class MatchmakingPoolBase(SQLModel, UTCBaseModel):
@@ -53,13 +58,11 @@ class MatchmakingPoolBase(SQLModel, UTCBaseModel):
default=0, default=0,
sa_column=Column(SmallInteger, nullable=False), sa_column=Column(SmallInteger, nullable=False),
) )
variant_id: int = Field(
default=0,
ge=0,
sa_column=Column(SmallInteger, nullable=False, server_default="0"),
)
name: str = Field(max_length=255) name: str = Field(max_length=255)
active: bool = Field(default=True) active: bool = Field(default=True)
lobby_size: int = Field(default=8)
rating_search_radius: int = Field(default=20)
rating_search_radius_exp: int = Field(default=15)
created_at: datetime | None = Field( created_at: datetime | None = Field(
default=None, default=None,
sa_column=Column(DateTime(timezone=True), server_default=func.now()), sa_column=Column(DateTime(timezone=True), server_default=func.now()),
@@ -72,7 +75,7 @@ class MatchmakingPoolBase(SQLModel, UTCBaseModel):
class MatchmakingPool(MatchmakingPoolBase, table=True): class MatchmakingPool(MatchmakingPoolBase, table=True):
__tablename__: str = "matchmaking_pools" __tablename__: str = "matchmaking_pools"
__table_args__ = (Index("matchmaking_pools_ruleset_variant_active_idx", "ruleset_id", "variant_id", "active"),) __table_args__ = (Index("matchmaking_pools_ruleset_active_idx", "ruleset_id", "active"),)
beatmaps: list["MatchmakingPoolBeatmap"] = Relationship( beatmaps: list["MatchmakingPoolBeatmap"] = Relationship(
back_populates="pool", back_populates="pool",
@@ -93,7 +96,7 @@ class MatchmakingPoolBeatmapBase(SQLModel, UTCBaseModel):
sa_column=Column(ForeignKey("beatmaps.id"), nullable=False), sa_column=Column(ForeignKey("beatmaps.id"), nullable=False),
) )
mods: list[APIMod] | None = Field(default=None, sa_column=Column(JSON)) mods: list[APIMod] | None = Field(default=None, sa_column=Column(JSON))
rating: int = Field(default=1500) rating: int | None = Field(default=1500)
selection_count: int = Field(default=0) selection_count: int = Field(default=0)

View File

@@ -0,0 +1,61 @@
"""matchmaking: sync with osu-web
Revision ID: 3f0f22f38c3d
Revises: a7646e082906
Create Date: 2025-11-22 12:45:36.767097
"""
from collections.abc import Sequence
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import mysql
# revision identifiers, used by Alembic.
revision: str = "3f0f22f38c3d"
down_revision: str | Sequence[str] | None = "a7646e082906"
branch_labels: str | Sequence[str] | None = None
depends_on: str | Sequence[str] | None = None
def upgrade() -> None:
"""Upgrade schema."""
# ### commands auto generated by Alembic - please adjust! ###
op.drop_index(op.f("matchmaking_pools_ruleset_variant_active_idx"), table_name="matchmaking_pools")
op.create_index("matchmaking_pools_ruleset_active_idx", "matchmaking_pools", ["ruleset_id", "active"], unique=False)
op.drop_column("matchmaking_pools", "variant_id")
op.add_column("matchmaking_user_stats", sa.Column("pool_id", sa.Integer(), nullable=True))
op.create_index(
"matchmaking_user_stats_pool_first_idx", "matchmaking_user_stats", ["pool_id", "first_placements"], unique=False
)
op.create_index(
"matchmaking_user_stats_pool_points_idx", "matchmaking_user_stats", ["pool_id", "total_points"], unique=False
)
op.create_foreign_key(None, "matchmaking_user_stats", "matchmaking_pools", ["pool_id"], ["id"])
op.drop_column("matchmaking_user_stats", "ruleset_id")
# ### end Alembic commands ###
def downgrade() -> None:
"""Downgrade schema."""
# ### commands auto generated by Alembic - please adjust! ###
op.add_column(
"matchmaking_user_stats",
sa.Column("ruleset_id", mysql.SMALLINT(), nullable=False, server_default="0"),
)
op.drop_index("matchmaking_user_stats_pool_points_idx", table_name="matchmaking_user_stats")
op.drop_index("matchmaking_user_stats_pool_first_idx", table_name="matchmaking_user_stats")
op.drop_column("matchmaking_user_stats", "pool_id")
op.add_column(
"matchmaking_pools",
sa.Column("variant_id", mysql.SMALLINT(), nullable=False, server_default="0"),
)
op.drop_index("matchmaking_pools_ruleset_active_idx", table_name="matchmaking_pools")
op.create_index(
"matchmaking_pools_ruleset_variant_active_idx",
"matchmaking_pools",
["ruleset_id", "variant_id", "active"],
unique=False,
)
# ### end Alembic commands ###