rust-course/server/internal/user/secret.go

package user

import (
	"database/sql"
	"net/http"
	"time"

	"github.com/imdotdev/im.dev/server/pkg/db"
	"github.com/imdotdev/im.dev/server/pkg/e"
	"github.com/imdotdev/im.dev/server/pkg/models"
	"github.com/imdotdev/im.dev/server/pkg/utils"
)

func GenSecret(userID string) (string, *e.Error) {
	secret := utils.GenID(models.IDTypeSecret)
	tx, err := db.Conn.Begin()
	if err != nil {
		logger.Warn("start transaction error", "error", err)
		return "", e.New(http.StatusInternalServerError, e.Internal)
	}

	_, err = tx.Exec("DELETE FROM user_secret WHERE user_id=?", userID)
	if err != nil {
		logger.Warn("delete secret error", "error", err)
		return "", e.New(http.StatusInternalServerError, e.Internal)
	}

	_, err = tx.Exec("INSERT INTO user_secret (user_id,secret,created) VALUES (?,?,?)", userID, secret, time.Now())
	if err != nil {
		logger.Warn("insert secret error", "error", err)
		tx.Rollback()
		return "", e.New(http.StatusInternalServerError, e.Internal)
	}

	tx.Commit()
	return secret, nil
}

func GetSecret(userID string) (string, *e.Error) {
	var secret string
	err := db.Conn.QueryRow("SELECT secret from user_secret WHERE user_id=?", userID).Scan(&secret)
	if err != nil {
		if err == sql.ErrNoRows {
			return "", e.New(http.StatusNotFound, "找不到对应的secret，请重新生成")
		}

		logger.Warn("select secret error", "error", err)
		return "", e.New(http.StatusInternalServerError, e.Internal)
	}

	return secret, nil
}