Skip to content
Snippets Groups Projects
Commit 04ee3119 authored by sxllwx's avatar sxllwx
Browse files

fix latest issue

parent 156bf40f
Branches
Tags
No related merge requests found
...@@ -41,6 +41,8 @@ type RegistryTestSuite struct { ...@@ -41,6 +41,8 @@ type RegistryTestSuite struct {
etcd *embed.Etcd etcd *embed.Etcd
} }
const defaultEtcdV3WorkDir = "/tmp/default-dubbo-go-registry.etcd"
// start etcd server // start etcd server
func (suite *RegistryTestSuite) SetupSuite() { func (suite *RegistryTestSuite) SetupSuite() {
...@@ -48,7 +50,7 @@ func (suite *RegistryTestSuite) SetupSuite() { ...@@ -48,7 +50,7 @@ func (suite *RegistryTestSuite) SetupSuite() {
cfg := embed.NewConfig() cfg := embed.NewConfig()
// avoid conflict with default etcd work-dir // avoid conflict with default etcd work-dir
cfg.Dir = "/tmp/default-dubbo-go-registry.etcd" cfg.Dir = defaultEtcdV3WorkDir
e, err := embed.StartEtcd(cfg) e, err := embed.StartEtcd(cfg)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
...@@ -69,7 +71,7 @@ func (suite *RegistryTestSuite) SetupSuite() { ...@@ -69,7 +71,7 @@ func (suite *RegistryTestSuite) SetupSuite() {
func (suite *RegistryTestSuite) TearDownSuite() { func (suite *RegistryTestSuite) TearDownSuite() {
suite.etcd.Close() suite.etcd.Close()
// clean the etcd workdir // clean the etcd workdir
if err := exec.Command("rm", "-rf", "/tmp/default-dubbo-go-registry.etcd").Run(); err != nil { if err := exec.Command("rm", "-rf", defaultEtcdV3WorkDir).Run(); err != nil {
suite.FailNow(err.Error()) suite.FailNow(err.Error())
} }
} }
......
...@@ -222,10 +222,12 @@ func (r *kubernetesRegistry) registerConsumer(svc common.URL) error { ...@@ -222,10 +222,12 @@ func (r *kubernetesRegistry) registerConsumer(svc common.URL) error {
logger.Errorf("kubernetes client create path %s: %v", consumersNode, err) logger.Errorf("kubernetes client create path %s: %v", consumersNode, err)
return perrors.WithMessage(err, "kubernetes create consumer nodes") return perrors.WithMessage(err, "kubernetes create consumer nodes")
} }
providersNode := fmt.Sprintf("/dubbo/%s/%s", svc.Service(), common.DubboNodes[common.PROVIDER])
if err := r.createDirIfNotExist(providersNode); err != nil { // NOTICE kubernetes && etcdv3 not need create provider metadata dir in consumer logic
return perrors.WithMessage(err, "create provider node") //providersNode := fmt.Sprintf("/dubbo/%s/%s", svc.Service(), common.DubboNodes[common.PROVIDER])
} //if err := r.createDirIfNotExist(providersNode); err != nil {
// return perrors.WithMessage(err, "create provider node")
//}
params := url.Values{} params := url.Values{}
......
...@@ -37,6 +37,8 @@ import ( ...@@ -37,6 +37,8 @@ import (
"google.golang.org/grpc/connectivity" "google.golang.org/grpc/connectivity"
) )
const defaultEtcdV3WorkDir = "/tmp/default-dubbo-go-remote.etcd"
// tests dataset // tests dataset
var tests = []struct { var tests = []struct {
input struct { input struct {
...@@ -92,7 +94,7 @@ func (suite *ClientTestSuite) SetupSuite() { ...@@ -92,7 +94,7 @@ func (suite *ClientTestSuite) SetupSuite() {
cfg := embed.NewConfig() cfg := embed.NewConfig()
cfg.LPUrls = []url.URL{*lpurl} cfg.LPUrls = []url.URL{*lpurl}
cfg.LCUrls = []url.URL{*lcurl} cfg.LCUrls = []url.URL{*lcurl}
cfg.Dir = "/tmp/default-dubbo-go-remote.etcd" cfg.Dir = defaultEtcdV3WorkDir
e, err := embed.StartEtcd(cfg) e, err := embed.StartEtcd(cfg)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
...@@ -112,7 +114,7 @@ func (suite *ClientTestSuite) SetupSuite() { ...@@ -112,7 +114,7 @@ func (suite *ClientTestSuite) SetupSuite() {
// stop etcd server // stop etcd server
func (suite *ClientTestSuite) TearDownSuite() { func (suite *ClientTestSuite) TearDownSuite() {
suite.etcd.Close() suite.etcd.Close()
if err := exec.Command("rm", "-rf", "/tmp/default-dubbo-go-remote.etcd").Run(); err != nil { if err := exec.Command("rm", "-rf", defaultEtcdV3WorkDir).Run(); err != nil {
suite.FailNow(err.Error()) suite.FailNow(err.Error())
} }
} }
......
...@@ -216,8 +216,6 @@ func (c *Client) initStore() error { ...@@ -216,8 +216,6 @@ func (c *Client) initStore() error {
// try to watch kubernetes pods // try to watch kubernetes pods
func (c *Client) maintenanceStatus() error { func (c *Client) maintenanceStatus() error {
c.wg.Add(1)
// try once // try once
watcher, err := c.rawClient.CoreV1().Pods(c.ns).Watch(metav1.ListOptions{ watcher, err := c.rawClient.CoreV1().Pods(c.ns).Watch(metav1.ListOptions{
LabelSelector: fields.OneTermEqualSelector(DubboIOLabelKey, DubboIOLabelValue).String(), LabelSelector: fields.OneTermEqualSelector(DubboIOLabelKey, DubboIOLabelValue).String(),
...@@ -229,6 +227,7 @@ func (c *Client) maintenanceStatus() error { ...@@ -229,6 +227,7 @@ func (c *Client) maintenanceStatus() error {
watcher.Stop() watcher.Stop()
c.wg.Add(1)
// add wg, grace close the client // add wg, grace close the client
go c.maintenanceStatusLoop() go c.maintenanceStatusLoop()
return nil return nil
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment