list.go

package client

import (
	"fmt"
	"strings"

	"github.com/coreos/etcd/clientv3"
	"github.com/coreos/etcd/mvcc/mvccpb"
)

// list a directory
func (clt *EtcdHRCHYClient) List(key string) ([]*Node, error) {
	key, _, err := clt.ensureKey(key)
	if err != nil {
		return nil, err
	}

	// directory start with /
	var dir = key
	if key != "/" {
		dir = key + "/"
	}
	fmt.Println(key, dir)

	txn := clt.client.Txn(clt.ctx)
	// make sure the list key is a directory
	cmp := []clientv3.Cmp{}
	if dir != "/" {
		cmp = append(cmp, clientv3.Compare(
			clientv3.Value(key),
			"=",
			clt.dirValue,
		))
	}
	txn.If(cmp...).Then(
		clientv3.OpGet(dir, clientv3.WithPrefix()),
	)

	txnResp, err := txn.Commit()
	if err != nil {
		return nil, err
	}

	//fmt.Println("resp===", txnResp.Responses)
	if !txnResp.Succeeded {
		return nil, ErrorListKey
	} else {
		if len(txnResp.Responses) > 0 {
			rangeResp := txnResp.Responses[0].GetResponseRange()
			return clt.list(dir, rangeResp.Kvs)
		} else {
			// empty directory
			return []*Node{}, nil
		}
	}
}

// pick key/value under the dir
func (clt *EtcdHRCHYClient) list(dir string, kvs []*mvccpb.KeyValue) ([]*Node, error) {
	nodes := []*Node{}
	for _, kv := range kvs {
		name := strings.TrimPrefix(string(kv.Key), dir)
		if strings.Contains(name, "/") {
			// secondary directory
			continue
		}
		nodes = append(nodes, clt.createNode(kv))
	}
	return nodes, nil
}