feat: add manager client list scheduler interface (#694)

* feat: add manager client list scheduler interface

Signed-off-by: Gaius <[email protected]>

pkg/rpc/manager/client/client.go

@@ -39,11 +39,23 @@ const (
 )
 
 type Client interface {
-	Close() error
+	// Get Scheduler and Scheduler cluster configuration
 	GetScheduler(*manager.GetSchedulerRequest) (*manager.Scheduler, error)
+
+	// Update scheduler configuration
 	UpdateScheduler(*manager.UpdateSchedulerRequest) (*manager.Scheduler, error)
+
+	// Update CDN configuration
 	UpdateCDN(*manager.UpdateCDNRequest) (*manager.CDN, error)
+
+	// List acitve schedulers configuration
+	ListSchedulers(*manager.ListSchedulersRequest) (*manager.ListSchedulersResponse, error)
+
+	// KeepAlive with manager
 	KeepAlive(time.Duration, *manager.KeepAliveRequest)
+
+	// Close client connect
+	Close() error
 }
 
 type client struct {
@@ -79,25 +91,32 @@ func New(target string) (Client, error) {
 	}, nil
 }
 
-func (c *client) GetScheduler(scheduler *manager.GetSchedulerRequest) (*manager.Scheduler, error) {
+func (c *client) GetScheduler(req *manager.GetSchedulerRequest) (*manager.Scheduler, error) {
+	ctx, cancel := context.WithTimeout(context.Background(), contextTimeout)
+	defer cancel()
+
+	return c.ManagerClient.GetScheduler(ctx, req)
+}
+
+func (c *client) UpdateScheduler(req *manager.UpdateSchedulerRequest) (*manager.Scheduler, error) {
 	ctx, cancel := context.WithTimeout(context.Background(), contextTimeout)
 	defer cancel()
 
-	return c.ManagerClient.GetScheduler(ctx, scheduler)
+	return c.ManagerClient.UpdateScheduler(ctx, req)
 }
 
-func (c *client) UpdateScheduler(scheduler *manager.UpdateSchedulerRequest) (*manager.Scheduler, error) {
+func (c *client) UpdateCDN(req *manager.UpdateCDNRequest) (*manager.CDN, error) {
 	ctx, cancel := context.WithTimeout(context.Background(), contextTimeout)
 	defer cancel()
 
-	return c.ManagerClient.UpdateScheduler(ctx, scheduler)
+	return c.ManagerClient.UpdateCDN(ctx, req)
 }
 
-func (c *client) UpdateCDN(cdn *manager.UpdateCDNRequest) (*manager.CDN, error) {
+func (c *client) ListSchedulers(req *manager.ListSchedulersRequest) (*manager.ListSchedulersResponse, error) {
 	ctx, cancel := context.WithTimeout(context.Background(), contextTimeout)
 	defer cancel()
 
-	return c.ManagerClient.UpdateCDN(ctx, cdn)
+	return c.ManagerClient.ListSchedulers(ctx, req)
 }
 
 func (c *client) KeepAlive(interval time.Duration, keepalive *manager.KeepAliveRequest) {