diff --git a/channeldb/kvdb/etcd/embed.go b/channeldb/kvdb/etcd/embed.go index 3e6b9b168..d99b901ef 100644 --- a/channeldb/kvdb/etcd/embed.go +++ b/channeldb/kvdb/etcd/embed.go @@ -64,8 +64,10 @@ func NewEmbeddedEtcdInstance(path string) (*BackendConfig, func(), error) { fmt.Errorf("etcd failed to start after: %v", readyTimeout) } + ctx, cancel := context.WithCancel(context.Background()) + connConfig := &BackendConfig{ - Ctx: context.Background(), + Ctx: ctx, Host: "http://" + peerURL, User: "user", Pass: "pass", @@ -73,6 +75,7 @@ func NewEmbeddedEtcdInstance(path string) (*BackendConfig, func(), error) { } return connConfig, func() { + cancel() etcd.Close() }, nil }