tests/integration/clientv3/examples/example_kv_test.go - main - githubmirror/etcd

tests/integration/clientv3/examples/example_kv_test.go
439 строк10 KB
Jefftree
*: add GetStream and RangeStream client support
30 апр 2026, 16:57
30 апр 2026, 16:57a792e7f
// Copyright 2016 The etcd Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package clientv3_test

import (
	"context"
	"errors"
	"fmt"
	"log"

	"go.etcd.io/etcd/api/v3/v3rpc/rpctypes"
	clientv3 "go.etcd.io/etcd/client/v3"
)

func mockKVPut() {}

func ExampleKV_put() {
	forUnitTestsRunInMockedContext(mockKVPut, func() {
		cli, err := clientv3.New(clientv3.Config{
			Endpoints:   exampleEndpoints(),
			DialTimeout: dialTimeout,
		})
		if err != nil {
			log.Fatal(err)
		}
		defer cli.Close()

		ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
		_, err = cli.Put(ctx, "sample_key", "sample_value")
		cancel()
		if err != nil {
			log.Fatal(err)
		}
	})
	// Output:
}

func mockKVPutErrorHandling() {
	fmt.Println("client-side error: etcdserver: key is not provided")
}

func ExampleKV_putErrorHandling() {
	forUnitTestsRunInMockedContext(mockKVPutErrorHandling, func() {
		cli, err := clientv3.New(clientv3.Config{
			Endpoints:   exampleEndpoints(),
			DialTimeout: dialTimeout,
		})
		if err != nil {
			log.Fatal(err)
		}
		defer cli.Close()

		ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
		_, err = cli.Put(ctx, "", "sample_value")
		cancel()
		if err != nil {
			if errors.Is(err, context.Canceled) {
				fmt.Printf("ctx is canceled by another routine: %v\n", err)
			} else if errors.Is(err, context.DeadlineExceeded) {
				fmt.Printf("ctx is attached with a deadline is exceeded: %v\n", err)
			} else if errors.Is(err, rpctypes.ErrEmptyKey) {
				fmt.Printf("client-side error: %v\n", err)
			} else {
				fmt.Printf("bad cluster endpoints, which are not etcd servers: %v\n", err)
			}
		}
	})
	// Output: client-side error: etcdserver: key is not provided
}

func mockKVGet() {
	fmt.Println("foo : bar")
}

func ExampleKV_get() {
	forUnitTestsRunInMockedContext(mockKVGet, func() {
		cli, err := clientv3.New(clientv3.Config{
			Endpoints:   exampleEndpoints(),
			DialTimeout: dialTimeout,
		})
		if err != nil {
			log.Fatal(err)
		}
		defer cli.Close()

		_, err = cli.Put(context.TODO(), "foo", "bar")
		if err != nil {
			log.Fatal(err)
		}

		ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
		resp, err := cli.Get(ctx, "foo")
		cancel()
		if err != nil {
			log.Fatal(err)
		}
		for _, ev := range resp.Kvs {
			fmt.Printf("%s : %s\n", ev.Key, ev.Value)
		}
	})
	// Output: foo : bar
}

func mockKVGetWithRev() {
	fmt.Println("foo : bar1")
}

func ExampleKV_getWithRev() {
	forUnitTestsRunInMockedContext(mockKVGetWithRev, func() {
		cli, err := clientv3.New(clientv3.Config{
			Endpoints:   exampleEndpoints(),
			DialTimeout: dialTimeout,
		})
		if err != nil {
			log.Fatal(err)
		}
		defer cli.Close()

		presp, err := cli.Put(context.TODO(), "foo", "bar1")
		if err != nil {
			log.Fatal(err)
		}
		_, err = cli.Put(context.TODO(), "foo", "bar2")
		if err != nil {
			log.Fatal(err)
		}

		ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
		resp, err := cli.Get(ctx, "foo", clientv3.WithRev(presp.Header.Revision))
		cancel()
		if err != nil {
			log.Fatal(err)
		}
		for _, ev := range resp.Kvs {
			fmt.Printf("%s : %s\n", ev.Key, ev.Value)
		}
	})
	// Output: foo : bar1
}

func mockKVGetSortedPrefix() {
	fmt.Println(`key_2 : value`)
	fmt.Println(`key_1 : value`)
	fmt.Println(`key_0 : value`)
}

func ExampleKV_getSortedPrefix() {
	forUnitTestsRunInMockedContext(mockKVGetSortedPrefix, func() {
		cli, err := clientv3.New(clientv3.Config{
			Endpoints:   exampleEndpoints(),
			DialTimeout: dialTimeout,
		})
		if err != nil {
			log.Fatal(err)
		}
		defer cli.Close()

		for i := range make([]int, 3) {
			ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
			_, err = cli.Put(ctx, fmt.Sprintf("key_%d", i), "value")
			cancel()
			if err != nil {
				log.Fatal(err)
			}
		}

		ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
		resp, err := cli.Get(ctx, "key", clientv3.WithPrefix(), clientv3.WithSort(clientv3.SortByKey, clientv3.SortDescend))
		cancel()
		if err != nil {
			log.Fatal(err)
		}
		for _, ev := range resp.Kvs {
			fmt.Printf("%s : %s\n", ev.Key, ev.Value)
		}
	})
	// Output:
	// key_2 : value
	// key_1 : value
	// key_0 : value
}

func mockKVGetStream() {
	fmt.Println("key_0 : value")
	fmt.Println("key_1 : value")
	fmt.Println("key_2 : value")
	fmt.Println("count: 3, more: false")
}

func ExampleKV_getStream() {
	forUnitTestsRunInMockedContext(mockKVGetStream, func() {
		cli, err := clientv3.New(clientv3.Config{
			Endpoints:   exampleEndpoints(),
			DialTimeout: dialTimeout,
		})
		if err != nil {
			log.Fatal(err)
		}
		defer cli.Close()

		for i := 0; i < 3; i++ {
			ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
			_, err = cli.Put(ctx, fmt.Sprintf("key_%d", i), "value")
			cancel()
			if err != nil {
				log.Fatal(err)
			}
		}

		ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
		defer cancel()
		stream, err := cli.GetStream(ctx, "key", clientv3.WithPrefix())
		if err != nil {
			log.Fatal(err)
		}
		// Header, More, and Count are populated only on the final chunk, and
		// only when the stream completes without error. Track the latest
		// chunk to read them, the same way they appear on a unary Get response.
		var last clientv3.RangeStreamResponse
		for chunk := range stream {
			if err := chunk.Err(); err != nil {
				log.Fatal(err)
			}
			for _, ev := range chunk.Kvs {
				fmt.Printf("%s : %s\n", ev.Key, ev.Value)
			}
			last = chunk
		}
		fmt.Printf("count: %d, more: %v\n", last.Count, last.More)
	})
	// Output:
	// key_0 : value
	// key_1 : value
	// key_2 : value
	// count: 3, more: false
}

func mockKVGetStreamToGetResponse() {
	fmt.Println("count: 3")
	fmt.Println("key_0 : value")
	fmt.Println("key_1 : value")
	fmt.Println("key_2 : value")
}

func ExampleKV_getStreamToGetResponse() {
	forUnitTestsRunInMockedContext(mockKVGetStreamToGetResponse, func() {
		cli, err := clientv3.New(clientv3.Config{
			Endpoints:   exampleEndpoints(),
			DialTimeout: dialTimeout,
		})
		if err != nil {
			log.Fatal(err)
		}
		defer cli.Close()

		for i := 0; i < 3; i++ {
			ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
			_, err = cli.Put(ctx, fmt.Sprintf("key_%d", i), "value")
			cancel()
			if err != nil {
				log.Fatal(err)
			}
		}

		ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
		defer cancel()
		stream, err := cli.GetStream(ctx, "key", clientv3.WithPrefix())
		if err != nil {
			log.Fatal(err)
		}
		resp, err := clientv3.GetStreamToGetResponse(stream)
		if err != nil {
			log.Fatal(err)
		}
		fmt.Printf("count: %d\n", resp.Count)
		for _, ev := range resp.Kvs {
			fmt.Printf("%s : %s\n", ev.Key, ev.Value)
		}
	})
	// Output:
	// count: 3
	// key_0 : value
	// key_1 : value
	// key_2 : value
}

func mockKVDelete() {
	fmt.Println("Deleted all keys: true")
}

func ExampleKV_delete() {
	forUnitTestsRunInMockedContext(mockKVDelete, func() {
		cli, err := clientv3.New(clientv3.Config{
			Endpoints:   exampleEndpoints(),
			DialTimeout: dialTimeout,
		})
		if err != nil {
			log.Fatal(err)
		}
		defer cli.Close()

		ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
		defer cancel()

		// count keys about to be deleted
		gresp, err := cli.Get(ctx, "key", clientv3.WithPrefix())
		if err != nil {
			log.Fatal(err)
		}

		// delete the keys
		dresp, err := cli.Delete(ctx, "key", clientv3.WithPrefix())
		if err != nil {
			log.Fatal(err)
		}

		fmt.Println("Deleted all keys:", int64(len(gresp.Kvs)) == dresp.Deleted)
	})
	// Output:
	// Deleted all keys: true
}

func mockKVCompact() {}

func ExampleKV_compact() {
	forUnitTestsRunInMockedContext(mockKVCompact, func() {
		cli, err := clientv3.New(clientv3.Config{
			Endpoints:   exampleEndpoints(),
			DialTimeout: dialTimeout,
		})
		if err != nil {
			log.Fatal(err)
		}
		defer cli.Close()

		ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
		resp, err := cli.Get(ctx, "foo")
		cancel()
		if err != nil {
			log.Fatal(err)
		}
		compRev := resp.Header.Revision // specify compact revision of your choice

		ctx, cancel = context.WithTimeout(context.Background(), requestTimeout)
		_, err = cli.Compact(ctx, compRev)
		cancel()
		if err != nil {
			log.Fatal(err)
		}
	})
	// Output:
}

func mockKVTxn() {
	fmt.Println("key : XYZ")
}

func ExampleKV_txn() {
	forUnitTestsRunInMockedContext(mockKVTxn, func() {
		cli, err := clientv3.New(clientv3.Config{
			Endpoints:   exampleEndpoints(),
			DialTimeout: dialTimeout,
		})
		if err != nil {
			log.Fatal(err)
		}
		defer cli.Close()

		kvc := clientv3.NewKV(cli)

		_, err = kvc.Put(context.TODO(), "key", "xyz")
		if err != nil {
			log.Fatal(err)
		}

		ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
		_, err = kvc.Txn(ctx).
			// txn value comparisons are lexical
			If(clientv3.Compare(clientv3.Value("key"), ">", "abc")).
			// the "Then" runs, since "xyz" > "abc"
			Then(clientv3.OpPut("key", "XYZ")).
			// the "Else" does not run
			Else(clientv3.OpPut("key", "ABC")).
			Commit()
		cancel()
		if err != nil {
			log.Fatal(err)
		}

		gresp, err := kvc.Get(context.TODO(), "key")
		if err != nil {
			log.Fatal(err)
		}
		for _, ev := range gresp.Kvs {
			fmt.Printf("%s : %s\n", ev.Key, ev.Value)
		}
	})
	// Output: key : XYZ
}

func mockKVDo() {}

func ExampleKV_do() {
	forUnitTestsRunInMockedContext(mockKVDo, func() {
		cli, err := clientv3.New(clientv3.Config{
			Endpoints:   exampleEndpoints(),
			DialTimeout: dialTimeout,
		})
		if err != nil {
			log.Fatal(err)
		}
		defer cli.Close()

		ops := []clientv3.Op{
			clientv3.OpPut("put-key", "123"),
			clientv3.OpGet("put-key"),
			clientv3.OpPut("put-key", "456"),
		}

		for _, op := range ops {
			if _, err := cli.Do(context.TODO(), op); err != nil {
				log.Fatal(err)
			}
		}
	})
	// Output:
}