From 0ad83c2dd9b85081752006383c3a2066be6dccbd Mon Sep 17 00:00:00 2001
From: kmetin <kutluhan.metin@hazelcast.com>
Date: Wed, 1 Nov 2023 12:49:51 +0300
Subject: [PATCH] show overall progress

---
 base/commands/migration/migration_stages.go | 50 ++++++++++++++++++++-
 1 file changed, 49 insertions(+), 1 deletion(-)

diff --git a/base/commands/migration/migration_stages.go b/base/commands/migration/migration_stages.go
index 1d24c133..71b1bd26 100644
--- a/base/commands/migration/migration_stages.go
+++ b/base/commands/migration/migration_stages.go
@@ -9,6 +9,7 @@ import (
 	"fmt"
 	"os"
 	"path/filepath"
+	"strconv"
 	"strings"
 	"time"
 
@@ -76,6 +77,13 @@ func createMigrationStages(ctx context.Context, ec plug.ExecContext, ci *hazelca
 					case StatusCanceled, StatusCanceling:
 						execErr = clcerrors.ErrUserCancelled
 						break statusReaderLoop
+					case StatusInProgress:
+						rt, cp, err := fetchOverallProgress(ctx, ci, migrationID)
+						if err != nil {
+							ec.Logger().Error(err)
+						}
+						status.SetProgress(cp)
+						status.SetRemainingDuration(rt)
 					}
 					q := fmt.Sprintf(`SELECT JSON_QUERY(this, '$.migrations[%d]') FROM %s WHERE __key= '%s'`, i, StatusMapName, migrationID)
 					res, err := ci.Client().SQL().Execute(ctx, q)
@@ -104,7 +112,6 @@ func createMigrationStages(ctx context.Context, ec plug.ExecContext, ci *hazelca
 							execErr = err
 							break statusReaderLoop
 						}
-						status.SetProgress(m.CompletionPercentage)
 						switch m.Status {
 						case StatusComplete:
 							return nil, nil
@@ -208,6 +215,7 @@ type OverallMigrationStatus struct {
 	Errors               []string                       `json:"errors"`
 	Report               string                         `json:"report"`
 	CompletionPercentage float32                        `json:"completionPercentage"`
+	RemainingTime        float32                        `json:"remainingTime"`
 	Migrations           []DataStructureMigrationStatus `json:"migrations"`
 }
 
@@ -233,6 +241,46 @@ func fetchMigrationStatus(ctx context.Context, ci *hazelcast.ClientInternal, mig
 	return strings.TrimSuffix(strings.TrimPrefix(string(r.(serialization.JSON)), `"`), `"`), nil
 }
 
+func fetchOverallProgress(ctx context.Context, ci *hazelcast.ClientInternal, migrationID string) (time.Duration, float32, error) {
+	q := fmt.Sprintf(`SELECT JSON_QUERY(this, '$.remainingTime'), JSON_QUERY(this, '$.completionPercentage') FROM %s WHERE __key='%s'`, StatusMapName, migrationID)
+	res, err := ci.Client().SQL().Execute(ctx, q)
+	if err != nil {
+		return 0, 0, err
+	}
+	it, err := res.Iterator()
+	if err != nil {
+		return 0, 0, err
+	}
+	if it.HasNext() {
+		// single iteration is enough that we are reading single result for a single migration
+		row, err := it.Next()
+		if err != nil {
+			return 0, 0, err
+		}
+		remainingTime, err := row.Get(0)
+		if err != nil {
+			return 0, 0, err
+		}
+		completionPercentage, err := row.Get(1)
+		if err != nil {
+			return 0, 0, err
+		}
+		// convert it to parsable duration format
+		rtStr := fmt.Sprintf("%sms", remainingTime.(serialization.JSON).String())
+		rt, err := time.ParseDuration(rtStr)
+		if err != nil {
+			return 0, 0, err
+		}
+		cpStr := completionPercentage.(serialization.JSON).String()
+		cp, err := strconv.ParseFloat(cpStr, 32)
+		if err != nil {
+			return 0, 0, err
+		}
+		return rt, float32(cp), nil
+	}
+	return 0, 0, errors.New("no rows found")
+}
+
 func fetchMigrationReport(ctx context.Context, ci *hazelcast.ClientInternal, migrationID string) (string, error) {
 	q := fmt.Sprintf(`SELECT JSON_QUERY(this, '$.report') FROM %s WHERE __key='%s'`, StatusMapName, migrationID)
 	r, err := querySingleRow(ctx, ci, q)