|
|
@ -29,8 +29,8 @@ func TestRegistry(t *testing.T) { |
|
|
|
) |
|
|
|
) |
|
|
|
|
|
|
|
|
|
|
|
ins := ®istry.ServiceInstance{ |
|
|
|
ins := ®istry.ServiceInstance{ |
|
|
|
ID: "test-ut-1", |
|
|
|
ID: "test-ut", |
|
|
|
Name: "test-ut-1", |
|
|
|
Name: "test-ut", |
|
|
|
Version: "v1.0.0", |
|
|
|
Version: "v1.0.0", |
|
|
|
Endpoints: []string{ |
|
|
|
Endpoints: []string{ |
|
|
|
"grpc://127.0.0.1:8080", |
|
|
|
"grpc://127.0.0.1:8080", |
|
|
@ -50,115 +50,9 @@ func TestRegistry(t *testing.T) { |
|
|
|
t.Fatal(err) |
|
|
|
t.Fatal(err) |
|
|
|
} |
|
|
|
} |
|
|
|
time.Sleep(time.Second * 3) |
|
|
|
time.Sleep(time.Second * 3) |
|
|
|
service, err := r.GetService(context.Background(), "test-ut-1") |
|
|
|
service, err := r.GetService(context.Background(), "test-ut") |
|
|
|
if err != nil { |
|
|
|
if err != nil { |
|
|
|
t.Fatal(err) |
|
|
|
t.Fatal(err) |
|
|
|
} |
|
|
|
} |
|
|
|
t.Log(service) |
|
|
|
t.Log(service) |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
func TestWatch(t *testing.T) { |
|
|
|
|
|
|
|
sdk, err := polaris.NewSDKContextByAddress("127.0.0.1:8091") |
|
|
|
|
|
|
|
if err != nil { |
|
|
|
|
|
|
|
t.Fatal(err) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
p := New(sdk) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
r := p.Registry( |
|
|
|
|
|
|
|
WithRegistryTimeout(time.Second), |
|
|
|
|
|
|
|
WithRegistryHealthy(true), |
|
|
|
|
|
|
|
WithRegistryIsolate(false), |
|
|
|
|
|
|
|
WithRegistryRetryCount(3), |
|
|
|
|
|
|
|
WithRegistryWeight(100), |
|
|
|
|
|
|
|
WithRegistryTTL(1000), |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
err = r.Register(context.Background(), ®istry.ServiceInstance{ |
|
|
|
|
|
|
|
ID: "test-ut", |
|
|
|
|
|
|
|
Name: "test-ut", |
|
|
|
|
|
|
|
Version: "v1.0.0", |
|
|
|
|
|
|
|
Endpoints: []string{ |
|
|
|
|
|
|
|
"grpc://127.0.0.1:8080", |
|
|
|
|
|
|
|
"http://127.0.0.1:9090", |
|
|
|
|
|
|
|
}, |
|
|
|
|
|
|
|
}) |
|
|
|
|
|
|
|
if err != nil { |
|
|
|
|
|
|
|
t.Fatal(err) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
time.Sleep(time.Second * 3) |
|
|
|
|
|
|
|
w, err := r.Watch(context.Background(), "test-ut") |
|
|
|
|
|
|
|
if err != nil { |
|
|
|
|
|
|
|
t.Fatal(err) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
service, err := w.Next() |
|
|
|
|
|
|
|
if err != nil { |
|
|
|
|
|
|
|
t.Fatal(err) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if len(service) != 1 { |
|
|
|
|
|
|
|
t.Fatalf("want 1, got %d, service %+v", len(service), service) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
err = r.Register(context.Background(), ®istry.ServiceInstance{ |
|
|
|
|
|
|
|
ID: "test-ut2", |
|
|
|
|
|
|
|
Name: "test-ut", |
|
|
|
|
|
|
|
Version: "v1.0.0", |
|
|
|
|
|
|
|
Endpoints: []string{ |
|
|
|
|
|
|
|
"grpc://127.0.0.2:8080", |
|
|
|
|
|
|
|
"http://127.0.0.2:9090", |
|
|
|
|
|
|
|
}, |
|
|
|
|
|
|
|
}) |
|
|
|
|
|
|
|
if err != nil { |
|
|
|
|
|
|
|
t.Fatal(err) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
time.Sleep(time.Second * 3) |
|
|
|
|
|
|
|
service, err = w.Next() |
|
|
|
|
|
|
|
if err != nil { |
|
|
|
|
|
|
|
t.Fatal(err) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
if len(service) != 2 { |
|
|
|
|
|
|
|
t.Fatalf("want 1, got %d, service %+v", len(service), service) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
err = r.Deregister(context.Background(), ®istry.ServiceInstance{ |
|
|
|
|
|
|
|
ID: "test-ut", |
|
|
|
|
|
|
|
Name: "test-ut", |
|
|
|
|
|
|
|
Version: "v1.0.0", |
|
|
|
|
|
|
|
Endpoints: []string{ |
|
|
|
|
|
|
|
"grpc://127.0.0.1:8080", |
|
|
|
|
|
|
|
"http://127.0.0.1:9090", |
|
|
|
|
|
|
|
}, |
|
|
|
|
|
|
|
}) |
|
|
|
|
|
|
|
if err != nil { |
|
|
|
|
|
|
|
t.Fatal(err) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
time.Sleep(time.Second * 3) |
|
|
|
|
|
|
|
service, err = w.Next() |
|
|
|
|
|
|
|
if err != nil { |
|
|
|
|
|
|
|
t.Fatal(err) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
if len(service) != 1 { |
|
|
|
|
|
|
|
t.Fatalf("want 1, got %d, service %+v", len(service), service) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
err = r.Deregister(context.Background(), ®istry.ServiceInstance{ |
|
|
|
|
|
|
|
ID: "test-ut", |
|
|
|
|
|
|
|
Name: "test-ut", |
|
|
|
|
|
|
|
Version: "v1.0.0", |
|
|
|
|
|
|
|
Endpoints: []string{ |
|
|
|
|
|
|
|
"grpc://127.0.0.2:8080", |
|
|
|
|
|
|
|
"http://127.0.0.2:9090", |
|
|
|
|
|
|
|
}, |
|
|
|
|
|
|
|
}) |
|
|
|
|
|
|
|
if err != nil { |
|
|
|
|
|
|
|
t.Fatal(err) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
time.Sleep(time.Second * 3) |
|
|
|
|
|
|
|
service, err = w.Next() |
|
|
|
|
|
|
|
if err != nil { |
|
|
|
|
|
|
|
t.Fatal(err) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
if len(service) != 0 { |
|
|
|
|
|
|
|
t.Fatalf("want 0, got %d", len(service)) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|