GinでJWTを使ったAPI認証

import (
	"github.com/gin-gonic/gin"
	"github.com/dgrijalva/jwt-go"
	"time"
	"net/http"
)

func generateToken(userID string) (string, error) {
	// トークンの生成
	token := jwt.New(jwt.SigningMethodHS256)

	// トークンにクレームを追加する
	claims := token.Claims.(jwt.MapClaims)
	claims["user_id"] = userID
	claims["exp"] = time.Now().Add(time.Hour * 72).Unix()

	// トークンを署名する
	secret := []byte("secret")
	tokenString, err := token.SignedString(secret)
	if err != nil {
		return "", err
	}

	return tokenString, nil
}

func authenticate() gin.HandlerFunc {
  return func(c *gin.Context) {
    // トークンを取得する
    tokenString := c.GetHeader("Authorization")
    if tokenString == "" {
      c.AbortWithStatusJSON(http.StatusUnauthorized, gin.H{
        "error": "Authorization header is missing",
      })

      return
    }

    // トークンを検証する
    token, err := jwt.Parse(tokenString, func(token *jwt.Token) (interface{}, error) {
      if _, ok := token.Method.(*jwt.SigningMethodHMAC); !ok {
        return nil, fmt.Errorf("unexpected signing method: %v", token.Header["alg"])
      }

      return []byte("secret"), nil
    })
    if err != nil {
      c.AbortWithStatusJSON(http.StatusUnauthorized, gin.H{
        "error": "Failed to validate token",
      })

      return
    }

    // トークンが正当である場合、APIリクエストを実行する
    if claims, ok := token.Claims.(jwt.MapClaims); ok && token.Valid {
      c.Set("user_id", claims["user_id"])
      c.Next()
    } else {
      c.AbortWithStatusJSON(http.StatusUnauthorized, gin.H{
        "error": "Token is not valid",
      })
    }
  }
}

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

	// JWT認証ミドルウェアを追加する
	r.Use(authenticate())

	// APIエンドポイントを追加する
	r.GET("/api", func(c *gin.Context) {
		userID := c.MustGet("user_id").(string)
		c.JSON(200, gin.H{
			"message": "Hello, " + userID,
		})
	})

	r.Run()
}