mirror of
https://github.com/opentofu/opentofu.git
synced 2025-02-25 18:45:20 -06:00
47 lines
1.1 KiB
Go
47 lines
1.1 KiB
Go
// Copyright (c) The OpenTofu Authors
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
// Copyright (c) 2023 HashiCorp, Inc.
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
package statemgr
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/opentofu/opentofu/internal/states"
|
|
)
|
|
|
|
// NewTransientInMemory returns a Transient implementation that retains
|
|
// transient snapshots only in memory, as part of the object.
|
|
//
|
|
// The given initial state, if any, must not be modified concurrently while
|
|
// this function is running, but may be freely modified once this function
|
|
// returns without affecting the stored transient snapshot.
|
|
func NewTransientInMemory(initial *states.State) Transient {
|
|
return &transientInMemory{
|
|
current: initial.DeepCopy(),
|
|
}
|
|
}
|
|
|
|
type transientInMemory struct {
|
|
lock sync.RWMutex
|
|
current *states.State
|
|
}
|
|
|
|
var _ Transient = (*transientInMemory)(nil)
|
|
|
|
func (m *transientInMemory) State() *states.State {
|
|
m.lock.RLock()
|
|
defer m.lock.RUnlock()
|
|
|
|
return m.current.DeepCopy()
|
|
}
|
|
|
|
func (m *transientInMemory) WriteState(new *states.State) error {
|
|
m.lock.Lock()
|
|
defer m.lock.Unlock()
|
|
|
|
m.current = new.DeepCopy()
|
|
return nil
|
|
}
|