added pool finder
This commit is contained in:
parent
07f48d4369
commit
59ad9d8ec5
3 changed files with 53 additions and 11 deletions
53
pool.go
53
pool.go
|
|
@ -74,11 +74,31 @@ func (c *PoolClient) Delete(ctx context.Context, name string) (err error) {
|
|||
return
|
||||
}
|
||||
|
||||
type pools struct {
|
||||
poolid string `json:"poolid"`
|
||||
}
|
||||
|
||||
func (c *PoolClient) ListPools(ctx context.Context) (list []string, err error) {
|
||||
var ps []pools
|
||||
req, err := c.client.NewRequest(ctx, "GET", "/pools", nil)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
_, err = c.client.Do(req, &ps)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
for _, p := range ps {
|
||||
list = append(list, p.poolid)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
type pool struct {
|
||||
members ServerList `json:"members"`
|
||||
}
|
||||
|
||||
func (c *PoolClient) List(ctx context.Context, name string) (sl ServerList, err error) {
|
||||
func (c *PoolClient) ListMembers(ctx context.Context, name string) (sl ServerList, err error) {
|
||||
if name == "" {
|
||||
err = ErrEmptyPoolName
|
||||
return
|
||||
|
|
@ -94,16 +114,29 @@ func (c *PoolClient) List(ctx context.Context, name string) (sl ServerList, err
|
|||
}
|
||||
|
||||
func (c *PoolClient) FindServer(ctx context.Context, poolname, servername, id string) (s *Server, err error) {
|
||||
sl, err := c.List(ctx, poolname)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
for _, srv := range sl {
|
||||
if id != "" && srv.ID == id {
|
||||
return srv, nil
|
||||
var pools []string
|
||||
if poolname != "" {
|
||||
pools = append(pools, poolname)
|
||||
} else {
|
||||
pools, err = c.ListPools(ctx)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
if servername != "" && srv.Name == servername {
|
||||
return srv, nil
|
||||
}
|
||||
for _, pool := range pools {
|
||||
sl, err := c.ListMembers(ctx, pool)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
for _, srv := range sl {
|
||||
if id != "" && srv.ID == id {
|
||||
srv.Pool = pool
|
||||
return srv, nil
|
||||
}
|
||||
if servername != "" && srv.Name == servername {
|
||||
srv.Pool = pool
|
||||
return srv, nil
|
||||
}
|
||||
}
|
||||
}
|
||||
return nil, ErrServerNotFound
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue