signed_keys.go

// MIT License

// Copyright (c) 2020 Tree Xie

// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:

// The above copyright notice and this permission notice shall be included in all
// copies or substantial portions of the Software.

// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
// SOFTWARE.

package elton

import (
	"sync"
	"sync/atomic"
)

type (
	// SignedKeysGenerator signed keys generator
	SignedKeysGenerator interface {
		GetKeys() []string
		SetKeys([]string)
	}
	// SimpleSignedKeys simple sigined key
	SimpleSignedKeys struct {
		keys []string
	}
	// RWMutexSignedKeys read/write mutex signed key
	RWMutexSignedKeys struct {
		sync.RWMutex
		keys []string
	}
	// AtomicSignedKeys atomic toggle signed keys
	AtomicSignedKeys struct {
		value atomic.Value
	}
)

// GetKeys returns the key list of simple signed keys
func (sk *SimpleSignedKeys) GetKeys() []string {
	return sk.keys
}

// SetKeys sets the key list of simple signed keys
func (sk *SimpleSignedKeys) SetKeys(values []string) {
	keys := make([]string, len(values))
	copy(keys, values)
	sk.keys = keys
}

// GetKeys returns the key list of rwmutex signed keys
func (rwSk *RWMutexSignedKeys) GetKeys() []string {
	rwSk.RLock()
	defer rwSk.RUnlock()
	return rwSk.keys
}

// SetKeys sets the key list of rwmutex signed keys
func (rwSk *RWMutexSignedKeys) SetKeys(values []string) {
	keys := make([]string, len(values))
	copy(keys, values)
	rwSk.Lock()
	defer rwSk.Unlock()
	rwSk.keys = keys
}

// GetKeys returns the key list of atomic signed keys
func (atSk *AtomicSignedKeys) GetKeys() []string {
	if value := atSk.value.Load(); value != nil {
		// atomic value只能相同的类型，因此只要值存在，转换时直接转换
		return *value.(*[]string)
	}
	return nil
}

// SetKeys sets the key list of atomic signed keys
func (atSk *AtomicSignedKeys) SetKeys(values []string) {
	keys := make([]string, len(values))
	copy(keys, values)
	atSk.value.Store(&keys)
}