make model entity interface compatible

Signed-off-by: Pranav Singh <pranavsingh02@hotmail.com>
meshery · Sep 28, 2023 · a85f6ae · a85f6ae
1 parent 4577fd6
commit a85f6ae
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/models/meshmodel/core/v1alpha1/models.go b/models/meshmodel/core/v1alpha1/models.go
@@ -7,6 +7,7 @@ import (
 
 	"github.com/google/uuid"
 	"github.com/layer5io/meshkit/database"
+	"github.com/layer5io/meshkit/models/meshmodel/core/types"
 	"gorm.io/gorm"
 )
 
@@ -53,6 +54,14 @@ type ModelDB struct {
 	Metadata    []byte    `json:"modelMetadata" gorm:"modelMetadata"`
 }
 
+func (m Model) Type() types.CapabilityType {
+	return types.Model
+}
+func (m Model) GetID() uuid.UUID {
+	return m.ID
+}
+
+
 func CreateModel(db *database.Handler, cmodel Model) (uuid.UUID, error) {
 	byt, err := json.Marshal(cmodel)
 	if err != nil {