mirror of
https://github.com/opentofu/opentofu.git
synced 2025-01-27 17:06:27 -06:00
83 lines
2.2 KiB
Go
83 lines
2.2 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
package views
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/hashicorp/terraform/internal/command/arguments"
|
|
)
|
|
|
|
// The StateLocker view is used to display locking/unlocking status messages
|
|
// if the state lock process takes longer than expected.
|
|
type StateLocker interface {
|
|
Locking()
|
|
Unlocking()
|
|
}
|
|
|
|
// NewStateLocker returns an initialized StateLocker implementation for the given ViewType.
|
|
func NewStateLocker(vt arguments.ViewType, view *View) StateLocker {
|
|
switch vt {
|
|
case arguments.ViewHuman:
|
|
return &StateLockerHuman{view: view}
|
|
case arguments.ViewJSON:
|
|
return &StateLockerJSON{view: view}
|
|
default:
|
|
panic(fmt.Sprintf("unknown view type %v", vt))
|
|
}
|
|
}
|
|
|
|
// StateLockerHuman is an implementation of StateLocker which prints status to
|
|
// a terminal.
|
|
type StateLockerHuman struct {
|
|
view *View
|
|
}
|
|
|
|
var _ StateLocker = (*StateLockerHuman)(nil)
|
|
var _ StateLocker = (*StateLockerJSON)(nil)
|
|
|
|
func (v *StateLockerHuman) Locking() {
|
|
v.view.streams.Println("Acquiring state lock. This may take a few moments...")
|
|
}
|
|
|
|
func (v *StateLockerHuman) Unlocking() {
|
|
v.view.streams.Println("Releasing state lock. This may take a few moments...")
|
|
}
|
|
|
|
// StateLockerJSON is an implementation of StateLocker which prints the state lock status
|
|
// to a terminal in machine-readable JSON form.
|
|
type StateLockerJSON struct {
|
|
view *View
|
|
}
|
|
|
|
func (v *StateLockerJSON) Locking() {
|
|
current_timestamp := time.Now().Format(time.RFC3339)
|
|
|
|
json_data := map[string]string{
|
|
"@level": "info",
|
|
"@message": "Acquiring state lock. This may take a few moments...",
|
|
"@module": "terraform.ui",
|
|
"@timestamp": current_timestamp,
|
|
"type": "state_lock_acquire"}
|
|
|
|
lock_info_message, _ := json.Marshal(json_data)
|
|
v.view.streams.Println(string(lock_info_message))
|
|
}
|
|
|
|
func (v *StateLockerJSON) Unlocking() {
|
|
current_timestamp := time.Now().Format(time.RFC3339)
|
|
|
|
json_data := map[string]string{
|
|
"@level": "info",
|
|
"@message": "Releasing state lock. This may take a few moments...",
|
|
"@module": "terraform.ui",
|
|
"@timestamp": current_timestamp,
|
|
"type": "state_lock_release"}
|
|
|
|
lock_info_message, _ := json.Marshal(json_data)
|
|
v.view.streams.Println(string(lock_info_message))
|
|
}
|