changes for the update artifact version

Change-Id: I622c96805d43f1dca4c5c2068fb10527aac169a5
Signed-off-by: rajdeep11 <rajdeep.sin@samsung.com>

diff --git a/apis/mmes_apis.go b/apis/mmes_apis.go
index 914495d..aa5e2b9 100644 (file)
--- 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 (file)
--- 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)