}
}
+func (m *MmeApiHandler) GetModelInfoById(cont *gin.Context) {
+ logging.INFO("Get model info by id ...")
+ id := cont.Param("id")
+ modelInfo, err := m.iDB.GetModelInfoById(id)
+ if err != nil {
+ logging.ERROR("error:", err)
+ cont.JSON(http.StatusInternalServerError, gin.H{
+ "code": http.StatusInternalServerError,
+ "message": err.Error(),
+ })
+ return
+ }
+ if modelInfo.Id == ""{
+ statusCode := http.StatusNotFound
+ errMessage := fmt.Sprintf("Record not found with id: %s", id)
+ logging.ERROR("Record not found, send status code: ", statusCode)
+ cont.JSON(statusCode, gin.H{
+ "code": statusCode,
+ "message": errMessage,
+ })
+ return
+ }
+ cont.JSON(http.StatusOK, modelInfo)
+ return
+}
+
/*
Provides the model details by param model name
*/
GetAll() ([]models.ModelInfo, error)
GetModelInfoByName(modelName string)([]models.ModelInfo, error)
GetModelInfoByNameAndVer(modelName string, modelVersion string)(*models.ModelInfo, error)
+ GetModelInfoById(id string)(*models.ModelInfo, error)
Update(modelInfo models.ModelInfo) error
Delete(id string) error
}
return nil, result.Error
}
return &modelInfo, nil
+}
+func (repo *ModelInfoRepository) GetModelInfoById(id string)(*models.ModelInfo, error){
+ logging.INFO("id is:", id)
+ var modelInfo models.ModelInfo
+ result := repo.db.Where("id = ?", id).Find(&modelInfo)
+ if result.Error != nil {
+ return nil, result.Error
+ }
+ return &modelInfo, nil
}
\ No newline at end of file
r.POST("/registerModel", handler.RegisterModel)
r.GET("/getModelInfo", handler.GetModelInfo)
r.PUT("/modelInfo/:id", handler.UpdateModel)
+ r.GET("/modelInfo/:id", handler.GetModelInfoById)
r.DELETE("/modelInfo/:id", handler.DeleteModel)
r.GET("/getModelInfo/:modelName", handler.GetModelInfoByName)
r.POST("/uploadModel/:modelName", handler.UploadModel)