iProbe/otpm
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

alpha

Browse source
This commit is contained in:
“xHuPo” 2025-05-22 16:07:55 +08:00
parent 079542e431
commit 25c5f530b8
2 changed files with 6 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

4
database/init/otp.sql
View file

@ -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
);

11
handlers/otp.go
View file

@ -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