本期詳細介紹了通過 Twilio 實現(xiàn) OTP 傳送、使用 goroutine 優(yōu)化 OTP 異步發(fā)送，以及建立強大的基于令牌的身份驗證系統(tǒng)。

使用 Twilio 發(fā)送 OTP

使用 Twilio 消息 API 發(fā)送 OTP 的核心功能如下所示：

func (app *application) sendOTPViaTwilio(otp, phoneNumber string) error {
    client := twilio.NewRestClientWithParams(twilio.ClientParams{
        Username: os.Getenv("TWILIO_SID"),
        Password: os.Getenv("TWILIO_API_KEY"),
    })

    params := &api.CreateMessageParams{}
    params.SetBody(fmt.Sprintf(
        "Thank you for choosing Cheershare! Your one-time password is %v.",
        otp,
    ))
    params.SetFrom(os.Getenv("TWILIO_PHONE_NUMBER"))
    params.SetTo(fmt.Sprintf("+91%v", phoneNumber))

    const maxRetries = 3
    var lastErr error

    for attempt := 1; attempt <= maxRetries; attempt++ {
        resp, err := client.SendSms(params)
        if err == nil {
            app.logger.Printf("Message SID: %s", resp.Sid)
            return nil
        }
        lastErr = err
        time.Sleep(time.Duration(attempt) * 100 * time.Millisecond)
    }

    return fmt.Errorf("failed to send OTP after %d retries: %w", maxRetries, lastErr)
}

該函數(shù)利用 Twilio 的 Go SDK 發(fā)送消息。 from 號碼是預先配置的 Twilio 號碼。 包含重試機制以確?？煽啃浴?/p>

使用 Goroutine 進行異步 OTP 發(fā)送

順序 OTP 發(fā)送會影響服務器性能。 該解決方案涉及利用 goroutine 來同時處理 OTP 傳遞。 application 結構已更新：

type application struct {
    wg     sync.WaitGroup
    config config
    models data.Models
    logger *log.Logger
    cache  *redis.Client
}

輔助函數(shù)有助于后臺任務執(zhí)行：

func (app *application) background(fn func()) {
    app.wg.Add(1)
    go func() {
        defer app.wg.Done()
        defer func() {
            if err := recover(); err != nil {
                app.logger.Printf("Error in background function: %v\n", err)
            }
        }()
        fn()
    }()
}

這使用 sync.WaitGroup 來管理 goroutine，確保在關閉之前完成。

數(shù)據(jù)庫令牌表

創(chuàng)建一個新的數(shù)據(jù)庫表來存儲用戶令牌：

-- 000002_create-token.up.sql
CREATE TABLE IF NOT EXISTS tokens (
    hash bytea PRIMARY KEY,
    user_id bigint NOT NULL REFERENCES users ON DELETE CASCADE,
    expiry timestamp(0) with time zone NOT NULL,
    scope text NOT NULL
);

-- 000002_create-token.down.sql
DROP TABLE IF EXISTS tokens;

該表存儲哈希令牌、用戶 ID、到期時間和令牌范圍。 數(shù)據(jù)庫遷移是使用migrate.

代幣模型和功能

data/models.go 文件包含用于令牌生成、插入和檢索的函數(shù)：

// ... (other imports) ...

package data

// ... (other code) ...

func generateToken(userId int64, ttl time.Duration, scope string) (*Token, error) {
    // ... (token generation logic) ...
}

func (m TokenModel) Insert(token *Token) error {
    // ... (database insertion logic) ...
}

func (m TokenModel) DeleteAllForUser(scope string, userID int64) error {
    // ... (database deletion logic) ...
}

func (m TokenModel) New(userId int64, ttl time.Duration, scope string) (*Token, error) {
    // ... (token creation and insertion logic) ...
}

此代碼處理令牌創(chuàng)建、散列和數(shù)據(jù)庫交互。 New 函數(shù)創(chuàng)建并存儲新令牌。

注冊處理程序更新

cmd/api/user.go 文件的注冊處理程序已修改為在成功 OTP 驗證后頒發(fā)令牌：

// ... (other functions) ...

func (app *application) handleUserSignupAndVerification(w http.ResponseWriter, r *http.Request) {
    // ... (input parsing and validation) ...

    // ... (OTP generation and sending logic) ...

    // ... (OTP verification logic) ...

    // ... (user creation or retrieval) ...

    token, err := app.generateTokenForUser(user.ID)
    if err != nil {
        // ... (error handling) ...
    }

    // ... (success response with token) ...
}

這將令牌生成集成到注冊流程中。

中間件層

三個中間件層增強了安全性和請求處理：recoverPanic、authenticate 和 requireAuthenticatedUser。 這些已實現(xiàn)并應用于路由，如原文所示。 上下文管理函數(shù)（contextSetUser 和 contextGetUser）用于在請求上下文中存儲和檢索用戶數(shù)據(jù)。

服務器配置集成了這些中間件，示例展示了如何使用requireAuthenticatedUser保護路由。 未來的增強功能包括文件上傳、正常關閉和指標集成。 完整的代碼可以在 GitHub 上找到。

以上是使用 Go 構建基于 OTP 的身份驗證服務器：第 3 部分的詳細內容。更多信息請關注PHP中文網(wǎng)其他相關文章！