package etcd import ( "context" "fmt" "sort" "strings" etcdv3 "github.com/coreos/etcd/clientv3" "github.com/hashicorp/terraform/backend" "github.com/hashicorp/terraform/states" "github.com/hashicorp/terraform/states/remote" "github.com/hashicorp/terraform/states/statemgr" ) func (b *Backend) Workspaces() ([]string, error) { res, err := b.client.Get(context.TODO(), b.prefix, etcdv3.WithPrefix(), etcdv3.WithKeysOnly()) if err != nil { return nil, err } result := make([]string, 1, len(res.Kvs)+1) result[0] = backend.DefaultStateName for _, kv := range res.Kvs { if strings.TrimPrefix(string(kv.Key), b.prefix) != backend.DefaultStateName { result = append(result, strings.TrimPrefix(string(kv.Key), b.prefix)) } } sort.Strings(result[1:]) return result, nil } func (b *Backend) DeleteWorkspace(name string) error { if name == backend.DefaultStateName || name == "" { return fmt.Errorf("Can't delete default state.") } key := b.determineKey(name) _, err := b.client.Delete(context.TODO(), key) return err } func (b *Backend) StateMgr(name string) (statemgr.Full, error) { var stateMgr statemgr.Full = &remote.State{ Client: &RemoteClient{ Client: b.client, DoLock: b.lock, Key: b.determineKey(name), }, } if !b.lock { stateMgr = &statemgr.LockDisabled{Inner: stateMgr} } lockInfo := statemgr.NewLockInfo() lockInfo.Operation = "init" lockId, err := stateMgr.Lock(lockInfo) if err != nil { return nil, fmt.Errorf("Failed to lock state in etcd: %s.", err) } lockUnlock := func(parent error) error { if err := stateMgr.Unlock(lockId); err != nil { return fmt.Errorf(strings.TrimSpace(errStateUnlock), lockId, err) } return parent } if err := stateMgr.RefreshState(); err != nil { err = lockUnlock(err) return nil, err } if v := stateMgr.State(); v == nil { if err := stateMgr.WriteState(states.NewState()); err != nil { err = lockUnlock(err) return nil, err } if err := stateMgr.PersistState(); err != nil { err = lockUnlock(err) return nil, err } } if err := lockUnlock(nil); err != nil { return nil, err } return stateMgr, nil } func (b *Backend) determineKey(name string) string { return b.prefix + name } const errStateUnlock = ` Error unlocking etcd state. Lock ID: %s Error: %s You may have to force-unlock this state in order to use it again. `