如何在go-gin框架中验证API密钥？

package main

import (
    "net/http"

    "github.com/gin-gonic/gin"
)

func main() {
    r := gin.Default()

    // 定义一个中间件来验证API密钥
    // 这里假设API密钥存储在一个名为"API_KEY"的请求头中
    apiKeyMiddleware := func(c *gin.Context) {
        apiKey := c.GetHeader("API_KEY")

        // 在这里进行API密钥的验证逻辑
        // 这里只是一个示例，你可以根据实际需求进行修改
        if apiKey != "YOUR_API_KEY" {
            c.JSON(http.StatusUnauthorized, gin.H{"error": "Invalid API key"})
            c.Abort()
            return
        }

        c.Next()
    }

    // 在需要验证API密钥的路由组中使用中间件
    authorized := r.Group("/")
    authorized.Use(apiKeyMiddleware)
    {
        authorized.GET("/protected", func(c *gin.Context) {
            c.JSON(http.StatusOK, gin.H{"message": "API key is valid"})
        })
    }

    r.Run(":8080")
}