feat(engine): add the command to generate schema diff to CLI

agneum · agneum · commit 6ceebec2285a · 2022-02-18T10:32:08.000+07:00
diff --git a/engine/cmd/cli/commands/clone/actions.go b/engine/cmd/cli/commands/clone/actions.go
@@ -239,6 +239,25 @@ func destroy(cliCtx *cli.Context) error {
 	return err
 }
 
+// schemaDiff runs a request to generate schema difference between clone and its snapshot.
+func schemaDiff(cliCtx *cli.Context) error {
+	dblabClient, err := commands.ClientByCLIContext(cliCtx)
+	if err != nil {
+		return err
+	}
+
+	cloneID := cliCtx.Args().First()
+
+	response, err := dblabClient.SchemaDiff(cliCtx.Context, cloneID)
+	if err != nil {
+		return err
+	}
+
+	_, err = fmt.Fprint(cliCtx.App.Writer, string(response))
+
+	return err
+}
+
 // startObservation runs a request to startObservation clone.
 func startObservation(cliCtx *cli.Context) error {
 	dblabClient, err := commands.ClientByCLIContext(cliCtx)
diff --git a/engine/cmd/cli/commands/clone/command_list.go b/engine/cmd/cli/commands/clone/command_list.go
@@ -130,6 +130,13 @@ func CommandList() []*cli.Command {
 					},
 				},
 			},
+			{
+				Name:      "schema-diff",
+				Usage:     "generate schema difference between clone and its snapshot",
+				ArgsUsage: "CLONE_ID",
+				Before:    checkCloneIDBefore,
+				Action:    schemaDiff,
+			},
 			{
 				Name:      "start-observation",
 				Usage:     "[EXPERIMENTAL] start clone state monitoring",
diff --git a/engine/pkg/client/dblabapi/clone.go b/engine/pkg/client/dblabapi/clone.go
@@ -323,6 +323,30 @@ func (c *Client) DestroyCloneAsync(ctx context.Context, cloneID string) error {
 	return nil
 }
 
+// SchemaDiff generates diff between Database Lab clone and its snapshot.
+func (c *Client) SchemaDiff(ctx context.Context, cloneID string) ([]byte, error) {
+	u := c.URL(fmt.Sprintf("/clone/diff/%s", cloneID))
+
+	request, err := http.NewRequest(http.MethodGet, u.String(), nil)
+	if err != nil {
+		return nil, errors.Wrap(err, "failed to make a request")
+	}
+
+	response, err := c.Do(ctx, request)
+	if err != nil {
+		return nil, errors.Wrap(err, "failed to get response")
+	}
+
+	defer func() { _ = response.Body.Close() }()
+
+	data, err := io.ReadAll(response.Body)
+	if err != nil {
+		return nil, errors.Wrap(err, "failed to read response")
+	}
+
+	return data, nil
+}
+
 // StartObservation starts a new clone observation.
 func (c *Client) StartObservation(ctx context.Context, startRequest types.StartObservationRequest) (*observer.Session, error) {
 	u := c.URL("/observation/start")
