From e69b42f54b222b9a48f08ebd97afcda4dafe9ae8 Mon Sep 17 00:00:00 2001
From: Zhuoyuan Liu <zhuoyuan.liu@maersk.com>
Date: Thu, 14 Nov 2024 11:26:09 +0100
Subject: [PATCH] update for osctrl-admin

---
 admin/handlers/post.go | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/admin/handlers/post.go b/admin/handlers/post.go
index 3e070f8a..0c153a7e 100644
--- a/admin/handlers/post.go
+++ b/admin/handlers/post.go
@@ -223,6 +223,18 @@ func (h *HandlersAdmin) QueryRunPOSTHandler(w http.ResponseWriter, r *http.Reque
 	}
 	// Remove duplicates from expected
 	expectedClear := removeStringDuplicates(expected)
+
+	// Create new record for query list
+	for _, nodeUUID := range expectedClear {
+		node, err := h.Nodes.GetByUUID(nodeUUID)
+		if err != nil {
+			log.Err(err).Msgf("error getting node %s and failed to create node query for it", nodeUUID)
+			continue
+		}
+		if err := h.Queries.CreateNodeQuery(node.ID, newQuery.ID); err != nil {
+			log.Err(err).Msgf("error creating node query for query %s and node %s", newQuery.Name, nodeUUID)
+		}
+	}
 	// Update value for expected
 	if err := h.Queries.SetExpected(newQuery.Name, len(expectedClear), env.ID); err != nil {
 		adminErrorResponse(w, "error setting expected", http.StatusInternalServerError, err)