From 1d4d42f6bd98fd39eb6022a576ce2d5188672d41 Mon Sep 17 00:00:00 2001
From: vikrantgupta25 <vikrant.thomso@gmail.com>
Date: Mon, 18 Nov 2024 17:06:33 +0530
Subject: [PATCH] feat: use the license v2 key to fill licenses v3 on startup

---
 ee/query-service/license/manager.go | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/ee/query-service/license/manager.go b/ee/query-service/license/manager.go
index 6dcc704e3a..d5bab652bb 100644
--- a/ee/query-service/license/manager.go
+++ b/ee/query-service/license/manager.go
@@ -67,6 +67,26 @@ func StartManager(dbType string, db *sqlx.DB, useLicensesV3 bool, features ...ba
 		repo: &repo,
 	}
 
+	if useLicensesV3 {
+		// get active license from the db
+		active, err := m.repo.GetActiveLicense(context.Background())
+		if err != nil {
+			return m, err
+		}
+
+		// fetch the new license structure from control plane
+		licenseV3, apiError := validate.ValidateLicenseV3(active.Key)
+		if apiError != nil {
+			return m, apiError
+		}
+
+		// insert the licenseV3 in sqlite db
+		apiError = m.repo.InsertLicenseV3(context.Background(), licenseV3)
+		if apiError != nil {
+			return m, apiError
+		}
+	}
+
 	if err := m.start(useLicensesV3, features...); err != nil {
 		return m, err
 	}