From 25c5f530b833568d4124d5fad15930dba8850dcc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E2=80=9CxHuPo=E2=80=9D?=
 <7513325+vrocwang@users.noreply.github.com>
Date: Thu, 22 May 2025 16:07:55 +0800
Subject: [PATCH] alpha

---
 database/init/otp.sql |  4 +---
 handlers/otp.go       | 11 +++++------
 2 files changed, 6 insertions(+), 9 deletions(-)

diff --git a/database/init/otp.sql b/database/init/otp.sql
index db2e481..e1c176c 100644
--- a/database/init/otp.sql
+++ b/database/init/otp.sql
@@ -1,7 +1,5 @@
 CREATE TABLE IF NOT EXISTS otp (
-    id SERIAL PRIMARY KEY,
-    openid VARCHAR(255),
-    num INTEGER,
+    openid VARCHAR(255) PRIMARY KEY,
     token VARCHAR(255),
     createtime TIMESTAMP DEFAULT CURRENT_TIMESTAMP
 );
\ No newline at end of file
diff --git a/handlers/otp.go b/handlers/otp.go
index 9895e76..eb5789e 100644
--- a/handlers/otp.go
+++ b/handlers/otp.go
@@ -38,13 +38,12 @@ func (h *Handler) UpdateOrCreateOtp(w http.ResponseWriter, r *http.Request) {
 
 	// 插入或更新 OTP 记录
 	query := `
-		INSERT INTO otp (openid, num, token)
-		VALUES ($1, $2, $3)
-		ON CONFLICT (openid) DO UPDATE SET num = EXCLUDED.num, token = EXCLUDED.token
-	}
+		INSERT INTO otp (openid, token)
+		VALUES ($1, $2)
+		ON CONFLICT (openid) DO UPDATE SET token = EXCLUDED.token
 	`
 
-	_, err := h.DB.Exec(query, req.OpenID, len(*req.Token), req.Token)
+	_, err := h.DB.Exec(query, req.OpenID, req.Token)
 	if err != nil {
 		WriteError(w, "Failed to update or create OTP", http.StatusInternalServerError)
 		return
@@ -62,7 +61,7 @@ func (h *Handler) GetOtp(w http.ResponseWriter, r *http.Request) {
 
 	var otp OtpRequest
 
-	err := h.DB.Get(&otp, "SELECT openid, token, num FROM otp WHERE openid=$1", openid)
+	err := h.DB.Get(&otp, "SELECT openid, token FROM otp WHERE openid=$1", openid)
 	if err != nil {
 		WriteError(w, "Failed to get OTP", http.StatusInternalServerError)
 		return