支持MySQL并移除不必要的lock (#43)

* Support MySQL

* 添加依赖

Author: JohnRichard4096

pyproject.toml

@@ -1,13 +1,15 @@
 [project]
 name = "nonebot-plugin-value"
-version = "0.0.8"
+version = "0.0.9"
 description = "Value API for NoneBot2"
 readme = "README.md"
 requires-python = ">=3.10"
 dependencies = [
+    "aiomysql>=0.2.0",
     "aiosqlite>=0.21.0",
     "nonebot-plugin-orm>=0.8.1",
     "nonebot2>=2.4.2",
+    "psycopg2-binary>=2.9.10",
     "pydantic>=2.11.7",
     "sqlalchemy>=2.0.41",
 ]

src/nonebot_plugin_value/models/balance.py

@@ -22,7 +22,7 @@ class UserAccount(Model):
     __tablename__ = "user_accounts"
 
     # 每种货币账户的唯一ID(id currency_id-UUID.hex)
-    uni_id: MappedColumn[str] = mapped_column(String(64), primary_key=True)
+    uni_id: MappedColumn[str] = mapped_column(String(255), primary_key=True)
 
     # 用户ID
     id: MappedColumn[str] = mapped_column(String(64))
@@ -67,16 +67,16 @@ class Transaction(Model):
     __tablename__ = "transactions"
 
     # UUID作为主键
-    id: MappedColumn[str] = mapped_column(String(64), primary_key=True)
+    id: MappedColumn[str] = mapped_column(String(255), primary_key=True)
 
     # 账户外键
     account_id: MappedColumn[str] = mapped_column(
-        String, ForeignKey("user_accounts.id", ondelete="RESTRICT"), nullable=False
+        String(255), ForeignKey("user_accounts.id", ondelete="RESTRICT"), nullable=False
     )
 
     # 货币外键
     currency_id: MappedColumn[str] = mapped_column(
-        String(32), ForeignKey("currency_meta.id", ondelete="RESTRICT"), nullable=False
+        String(255), ForeignKey("currency_meta.id", ondelete="RESTRICT"), nullable=False
     )
 
     # 交易金额

src/nonebot_plugin_value/models/currency.py

@@ -12,7 +12,7 @@ class CurrencyMeta(Model):
     __tablename__ = "currency_meta"
 
     # 货币ID作为主键
-    id: Mapped[str] = mapped_column(String(64), primary_key=True, default=uuid4)
+    id: Mapped[str] = mapped_column(String(255), primary_key=True, default=uuid4)
 
     # 货币显示名称
     display_name: Mapped[str] = mapped_column(String(64), nullable=False)

src/nonebot_plugin_value/repository.py

@@ -27,11 +27,7 @@ async def createcurrency(self, currency_data: CurrencyData) -> CurrencyMeta:
             stmt = insert(CurrencyMeta).values(**dict(currency_data))
             await session.execute(stmt)
             await session.commit()
-            stmt = (
-                select(CurrencyMeta)
-                .where(CurrencyMeta.id == currency_data.id)
-                .with_for_update()
-            )
+            stmt = select(CurrencyMeta).where(CurrencyMeta.id == currency_data.id)
             result = await session.execute(stmt)
             currency_meta = result.scalar_one()
             session.add(currency_meta)
@@ -112,11 +108,7 @@ async def get_or_create_account(
         async with self.session as session:
             """获取或创建用户账户"""
             # 获取货币配置
-            stmt = (
-                select(CurrencyMeta)
-                .where(CurrencyMeta.id == currency_id)
-                .with_for_update()
-            )
+            stmt = select(CurrencyMeta).where(CurrencyMeta.id == currency_id)
             result = await session.execute(stmt)
             currency = result.scalar_one_or_none()
             if currency is None:
@@ -149,13 +141,9 @@ async def get_or_create_account(
             session.add(account)
             await session.commit()
 
-            stmt = (
-                select(UserAccount)
-                .where(
-                    UserAccount.id == user_id,
-                    UserAccount.currency_id == currency_id,
-                )
-                .with_for_update()
+            stmt = select(UserAccount).where(
+                UserAccount.id == user_id,
+                UserAccount.currency_id == currency_id,
             )
             result = await session.execute(stmt)
             account = result.scalar_one()
@@ -295,7 +283,6 @@ async def get_transaction_history(self, account_id: str, limit: int = 100):
             .where(Transaction.account_id == account_id)
             .order_by(Transaction.timestamp.desc())
             .limit(limit)
-            .with_for_update()
         )
         data = result.scalars().all()
         self.session.add_all(data)