Golang备忘录模式对象状态保存与恢复

package main

import "fmt"

// Memento 备忘录结构，保存Originator的状态
type Memento struct {
    state string
}

// Originator 发起人，拥有需要保存的状态
type Originator struct {
    state string
}

// NewOriginator 创建新的发起人
func NewOriginator(state string) *Originator {
    return &Originator{state: state}
}

// SetState 设置状态
func (o *Originator) SetState(state string) {
    o.state = state
}

// SaveToMemento 保存当前状态到备忘录
func (o *Originator) SaveToMemento() *Memento {
    return &Memento{state: o.state}
}

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

// Caretaker 管理者，管理多个备忘录（如历史记录）
type Caretaker struct {
    history []*Memento
}

// NewCaretaker 创建管理者
func NewCaretaker() *Caretaker {
    return &Caretaker{history: make([]*Memento, 0)}
}

// Add 添加备忘录到历史
func (c *Caretaker) Add(m *Memento) {
    c.history = append(c.history, m)
}

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

func main() {
    originator := NewOriginator("初始状态")
    caretaker := NewCaretaker()

    fmt.Println("当前状态:", originator.state)

    // 修改状态并保存
    originator.SetState("第一次修改")
    caretaker.Add(originator.SaveToMemento())
    fmt.Println("保存后状态:", originator.state)

    originator.SetState("第二次修改")
    caretaker.Add(originator.SaveToMemento())
    fmt.Println("再次保存后状态:", originator.state)

    // 恢复到第一个保存点
    memento := caretaker.Get(0)
    if memento != nil {
        originator.RestoreFromMemento(memento)
    }
    fmt.Println("恢复到第一个状态:", originator.state)
}