mps/cert/mem_provider.go

package cert

import (
	"crypto/tls"
	"fmt"
	"strings"
	"sync"
)

var DefaultMemProvider = NewMemProvider()

// MemProvider A simple in-memory certificate cache
type MemProvider struct {
	cache map[string]*tls.Certificate
	rw sync.RWMutex
}

// Create a MemProvider
func NewMemProvider() *MemProvider {
	return &MemProvider{
		cache: make(map[string]*tls.Certificate),
		rw: sync.RWMutex{},
	}
}

// Get the certificate for the Host from the cache
func (m *MemProvider) Get(host string) (cert *tls.Certificate, err error) {
	var ok bool
	cert, ok = m.cache[strings.TrimSpace(host)]
	if !ok {
		err = fmt.Errorf("cert not exist")
	}
	return
}

// Set the Host certificate to the cache
func (m *MemProvider) Set(host string, cert *tls.Certificate) error {
	host = strings.TrimSpace(host)
	m.rw.Lock()
	m.cache[host] = cert
	m.rw.Unlock()
	return nil
}