實(shí)戰(zhàn)Go：HashMap緩存的緩存數(shù)據(jù)訪問緩存加密與解密

在Go語言中，使用sync.Map可以實(shí)現(xiàn)一個(gè)簡單的HashMap緩存。為了實(shí)現(xiàn)緩存數(shù)據(jù)的加密與解密，我們可以使用對(duì)稱加密算法，例如AES。下面是一個(gè)簡單的示例，展示了如何使用Go語言實(shí)現(xiàn)一個(gè)帶有加密和解密功能的HashMap緩存。

package main

import (
	"crypto/aes"
	"crypto/cipher"
	"crypto/rand"
	"encoding/base64"
	"fmt"
	"io"
	"sync"
)

type EncryptedCache struct {
	cache    map[string][]byte
	key      []byte
	mu       sync.RWMutex
}

func NewEncryptedCache(key []byte) *EncryptedCache {
	return &EncryptedCache{
		cache: make(map[string][]byte),
		key:   key,
	}
}

func (ec *EncryptedCache) Get(key string) ([]byte, bool) {
	ec.mu.RLock()
	defer ec.mu.RUnlock()

	data, ok := ec.cache[key]
	if !ok {
		return nil, false
	}

	// 解密數(shù)據(jù)
	decryptedData, err := decrypt(data, ec.key)
	if err != nil {
		return nil, false
	}

	return decryptedData, true
}

func (ec *EncryptedCache) Set(key string, value []byte) error {
	// 加密數(shù)據(jù)
	encryptedData, err := encrypt(value, ec.key)
	if err != nil {
		return err
	}

	ec.mu.Lock()
	defer ec.mu.Unlock()

	ec.cache[key] = encryptedData
	return nil
}

func encrypt(data, key []byte) ([]byte, error) {
	block, err := aes.NewCipher(key)
	if err != nil {
		return nil, err
	}

	nonce := make([]byte, 12)
	if _, err := io.ReadFull(rand.Reader, nonce); err != nil {
		return nil, err
	}

	aesgcm, err := cipher.NewGCM(block)
	if err != nil {
		return nil, err
	}

	ciphertext := aesgcm.Seal(nil, nonce, data, nil)
	return append(nonce, ciphertext...), nil
}

func decrypt(data, key []byte) ([]byte, error) {
	block, err := aes.NewCipher(key)
	if err != nil {
		return nil, err
	}

	if len(data) < 12 {
		return nil, fmt.Errorf("ciphertext too short")
	}

	nonce, ciphertext := data[:12], data[12:]

	aesgcm, err := cipher.NewGCM(block)
	if err != nil {
		return nil, err
	}

	plaintext, err := aesgcm.Open(nil, nonce, ciphertext, nil)
	if err != nil {
		return nil, err
	}

	return plaintext, nil
}

func main() {
	key := []byte("0123456789abcdef") // 16字節(jié)的密鑰
	cache := NewEncryptedCache(key)

	// 設(shè)置緩存數(shù)據(jù)
	cache.Set("key1", []byte("value1"))
	cache.Set("key2", []byte("value2"))

	// 獲取緩存數(shù)據(jù)
	value1, ok := cache.Get("key1")
	if ok {
		fmt.Println("key1:", string(value1))
	} else {
		fmt.Println("key1 not found")
	}

	value2, ok := cache.Get("key2")
	if ok {
		fmt.Println("key2:", string(value2))
	} else {
		fmt.Println("key2 not found")
	}
}

在這個(gè)示例中，我們創(chuàng)建了一個(gè)名為EncryptedCache的結(jié)構(gòu)體，它包含一個(gè)加密的緩存數(shù)據(jù)映射、一個(gè)密鑰和一個(gè)讀寫鎖。我們實(shí)現(xiàn)了Get和Set方法，分別用于獲取和設(shè)置緩存數(shù)據(jù)。在獲取數(shù)據(jù)時(shí)，我們會(huì)解密數(shù)據(jù)；在設(shè)置數(shù)據(jù)時(shí)，我們會(huì)加密數(shù)據(jù)。

我們使用了AES加密算法和GCM模式來實(shí)現(xiàn)數(shù)據(jù)的加密和解密。請(qǐng)注意，這個(gè)示例僅用于演示目的，實(shí)際應(yīng)用中可能需要考慮更多的安全性和性能問題。