From 6abd11e6d4c35b942896d058949bc60e303e7ff7 Mon Sep 17 00:00:00 2001
From: rajdeep11 <rajdeep.sin@samsung.com>
Date: Thu, 19 Dec 2024 06:11:03 +0530
Subject: [PATCH] changes for the update artifact version

Change-Id: I622c96805d43f1dca4c5c2068fb10527aac169a5
Signed-off-by: rajdeep11 <rajdeep.sin@samsung.com>
---
 apis/mmes_apis.go | 30 +++++++++++++++++++++++++++++-
 routers/router.go |  1 +
 2 files changed, 30 insertions(+), 1 deletion(-)

diff --git a/apis/mmes_apis.go b/apis/mmes_apis.go
index 914495d..aa5e2b9 100644
--- a/apis/mmes_apis.go
+++ b/apis/mmes_apis.go
@@ -71,7 +71,9 @@ func (m *MmeApiHandler) RegisterModel(cont *gin.Context) {
 		})
 		return
 	}
-
+    // by default when a model is registered its artifact version is set to 0.0.0
+    modelInfo.ModelId.ArtifactVersion = "0.0.0"
+    
 	if err := m.iDB.Create(modelInfo); err != nil {
 		logging.ERROR("error", err)
 		cont.JSON(http.StatusBadRequest, gin.H{
@@ -338,3 +340,29 @@ func (m *MmeApiHandler) DeleteModel(cont *gin.Context) {
 	}
 	cont.JSON(http.StatusOK, gin.H{"message": message})
 }
+
+func (m *MmeApiHandler) UpdateArtifact(cont *gin.Context){
+	logging.INFO("Update artifact version of model")
+	// var modelInfo *models.ModelRelatedInformation
+	modelname := cont.Param("modelname")
+	modelversion := cont.Param("modelversion")
+	artifactversion := cont.Param("artifactversion")
+	modelInfo, err := m.iDB.GetModelInfoByNameAndVer(modelname, modelversion)
+	if err != nil {
+		logging.ERROR("error:", err)
+		cont.JSON(http.StatusInternalServerError, gin.H{
+			"code":    http.StatusInternalServerError,
+			"message": err.Error(),
+		})
+		return
+	}
+	modelInfo.ModelId.ArtifactVersion = artifactversion
+	if err := m.iDB.Update(*modelInfo); err != nil {
+		logging.ERROR(err)
+		return
+	}
+	logging.INFO("model updated")
+	cont.JSON(http.StatusOK, gin.H{
+		"modelinfo": modelInfo,
+	})
+}
\ No newline at end of file
diff --git a/routers/router.go b/routers/router.go
index 6173967..f90e5bf 100644
--- a/routers/router.go
+++ b/routers/router.go
@@ -28,6 +28,7 @@ func InitRouter(handler *apis.MmeApiHandler) *gin.Engine {
 	r.Use(gin.Recovery())
 	// As per R1-AP v6
 	r.POST("/model-registrations", handler.RegisterModel)
+	r.POST("/model-registrations/updateArtifact/:modelname/:modelversion/:artifactversion", handler.UpdateArtifact)
 	r.GET("/model-registrations/:modelRegistrationId", handler.GetModelInfoById)
 	r.PUT("/model-registrations/:modelRegistrationId", handler.UpdateModel)
 	r.DELETE("/model-registrations/:modelRegistrationId", handler.DeleteModel)
-- 
2.16.6