Golang备忘录模式快照实现实例

package main

import (
	"fmt"
	"time"
)

// DataState 表示要保存的状态快照
type DataState struct {
	Value     string
	Timestamp time.Time
}

// Originator 发起人，持有当前状态
type Originator struct {
	currentState DataState
}

// SaveToMemento 创建一个备忘录，保存当前状态
func (o *Originator) SaveToMemento() *Memento {
	return &Memento{
		state: o.currentState,
	}
}

// RestoreFromMemento 从备忘录恢复状态
func (o *Originator) RestoreFromMemento(m *Memento) {
	o.currentState = m.GetState()
}

// Memento 备忘录，封装状态
type Memento struct {
	state DataState
}

// GetState 提供对状态的只读访问（仅Originator应调用）
func (m *Memento) GetState() DataState {
	return m.state
}

// Caretaker 管理者，保存多个快照
type Caretaker struct {
	history []*Memento
}

// Add 保存一个备忘录
func (c *Caretaker) Add(m *Memento) {
	c.history = append(c.history, m)
}

// Get 获取指定索引的备忘录
func (c *Caretaker) Get(index int) *Memento {
	if index = len(c.history) {
		return nil
	}
	return c.history[index]
}

// Size 返回快照数量
func (c *Caretaker) Size() int {
	return len(c.history)
}

func main() {
	originator := &Originator{}
	caretaker := &Caretaker{}

	// 修改状态并保存快照1
	originator.currentState = DataState{Value: "第一次修改", Timestamp: time.Now()}
	caretaker.Add(originator.SaveToMemento())
	fmt.Println("保存快照1:", originator.currentState.Value)

	// 修改状态并保存快照2
	originator.currentState = DataState{Value: "第二次修改", Timestamp: time.Now()}
	caretaker.Add(originator.SaveToMemento())
	fmt.Println("保存快照2:", originator.currentState.Value)

	// 再次修改状态（不保存）
	originator.currentState = DataState{Value: "未保存的更改", Timestamp: time.Now()}
	fmt.Println("当前状态:", originator.currentState.Value)

	// 恢复到第一个快照
	firstMemento := caretaker.Get(0)
	if firstMemento != nil {
		originator.RestoreFromMemento(firstMemento)
		fmt.Println("恢复到第一个快照:", originator.currentState.Value)
	} else {
		fmt.Println("无法获取指定快照")
	}
}