mutex around signing method registration. shouldn't matter, but couldn't hurt

This commit is contained in:
Dave Grijalva 2016-04-12 17:06:56 -07:00
parent b9283128ba
commit b6d201ffa0
1 changed files with 11 additions and 0 deletions

View File

@ -1,6 +1,11 @@
package jwt package jwt
import (
"sync"
)
var signingMethods = map[string]func() SigningMethod{} var signingMethods = map[string]func() SigningMethod{}
var signingMethodLock = new(sync.RWMutex)
// Implement SigningMethod to add new methods for signing or verifying tokens. // Implement SigningMethod to add new methods for signing or verifying tokens.
type SigningMethod interface { type SigningMethod interface {
@ -12,11 +17,17 @@ type SigningMethod interface {
// Register the "alg" name and a factory function for signing method. // Register the "alg" name and a factory function for signing method.
// This is typically done during init() in the method's implementation // This is typically done during init() in the method's implementation
func RegisterSigningMethod(alg string, f func() SigningMethod) { func RegisterSigningMethod(alg string, f func() SigningMethod) {
signingMethodLock.Lock()
defer signingMethodLock.Unlock()
signingMethods[alg] = f signingMethods[alg] = f
} }
// Get a signing method from an "alg" string // Get a signing method from an "alg" string
func GetSigningMethod(alg string) (method SigningMethod) { func GetSigningMethod(alg string) (method SigningMethod) {
signingMethodLock.RLock()
defer signingMethodLock.RUnlock()
if methodF, ok := signingMethods[alg]; ok { if methodF, ok := signingMethods[alg]; ok {
method = methodF() method = methodF()
} }