Add counts to model, relationship and component

Signed-off-by: abdullah1308 <abdullahrafi.1308@gmail.com>

models/meshmodel/core/v1alpha1/component.go

@@ -80,7 +80,7 @@ func CreateComponent(db *database.Handler, c ComponentDefinition) (uuid.UUID, er
 	err = db.Create(&cdb).Error
 	return c.ID, err
 }
-func GetMeshModelComponents(db *database.Handler, f ComponentFilter) (c []ComponentDefinition) {
+func GetMeshModelComponents(db *database.Handler, f ComponentFilter) (c []ComponentDefinition, count int64) {
 	type componentDefinitionWithModel struct {
 		ComponentDefinitionDB
 		ModelDB
@@ -127,6 +127,9 @@ func GetMeshModelComponents(db *database.Handler, f ComponentFilter) (c []Compon
 			finder = finder.Order(f.OrderOn)
 		}
 	}
+
+	finder.Count(&count)
+
 	finder = finder.Offset(f.Offset)
 	if f.Limit != 0 {
 		finder = finder.Limit(f.Limit)
@@ -143,7 +146,7 @@ func GetMeshModelComponents(db *database.Handler, f ComponentFilter) (c []Compon
 		c = append(c, cm.ComponentDefinitionDB.GetComponentDefinition(cm.ModelDB.GetModel(cm.CategoryDB.GetCategory(db))))
 	}
 
-	return c
+	return c, count
 }
 
 type ComponentFilter struct {

models/meshmodel/core/v1alpha1/relationship.go

@@ -58,7 +58,7 @@ func (rf *RelationshipFilter) Create(m map[string]interface{}) {
 		return
 	}
 }
-func GetMeshModelRelationship(db *database.Handler, f RelationshipFilter) (r []RelationshipDefinition) {
+func GetMeshModelRelationship(db *database.Handler, f RelationshipFilter) (r []RelationshipDefinition, count int64) {
 	type componentDefinitionWithModel struct {
 		RelationshipDefinitionDB
 		ModelDB
@@ -92,6 +92,9 @@ func GetMeshModelRelationship(db *database.Handler, f RelationshipFilter) (r []R
 			finder = finder.Order(f.OrderOn)
 		}
 	}
+
+	finder.Count(&count)
+
 	finder = finder.Offset(f.Offset)
 	if f.Limit != 0 {
 		finder = finder.Limit(f.Limit)
@@ -104,7 +107,7 @@ func GetMeshModelRelationship(db *database.Handler, f RelationshipFilter) (r []R
 	for _, cm := range componentDefinitionsWithModel {
 		r = append(r, cm.RelationshipDefinitionDB.GetRelationshipDefinition(cm.ModelDB.GetModel(cm.CategoryDB.GetCategory(db))))
 	}
-	return r
+	return r, count
 }
 
 func (rdb *RelationshipDefinitionDB) GetRelationshipDefinition(m Model) (r RelationshipDefinition) {

models/meshmodel/registry.go

@@ -157,34 +157,34 @@ func (rm *RegistryManager) RegisterEntity(h Host, en Entity) error {
 	}
 }
 
-func (rm *RegistryManager) GetEntities(f types.Filter) []Entity {
+func (rm *RegistryManager) GetEntities(f types.Filter) ([]Entity, *int64) {
 	switch filter := f.(type) {
 	case *v1alpha1.ComponentFilter:
 		en := make([]Entity, 0)
-		comps := v1alpha1.GetMeshModelComponents(rm.db, *filter)
+		comps, count := v1alpha1.GetMeshModelComponents(rm.db, *filter)
 		for _, comp := range comps {
 			en = append(en, comp)
 		}
-		return en
+		return en, &count
 	case *v1alpha1.RelationshipFilter:
 		en := make([]Entity, 0)
-		relationships := v1alpha1.GetMeshModelRelationship(rm.db, *filter)
+		relationships, count := v1alpha1.GetMeshModelRelationship(rm.db, *filter)
 		for _, rel := range relationships {
 			en = append(en, rel)
 		}
-		return en
+		return en, &count
 	case *v1alpha1.PolicyFilter:
 		en := make([]Entity, 0)
 		policies := v1alpha1.GetMeshModelPolicy(rm.db, *filter)
 		for _, pol := range policies {
 			en = append(en, pol)
 		}
-		return en
+		return en, nil
 	default:
-		return nil
+		return nil, nil
 	}
 }
-func (rm *RegistryManager) GetModels(db *database.Handler, f types.Filter) []v1alpha1.Model {
+func (rm *RegistryManager) GetModels(db *database.Handler, f types.Filter) ([]v1alpha1.Model, int64) {
 	var m []v1alpha1.Model
 	type modelWithCategories struct {
 		v1alpha1.ModelDB
@@ -195,6 +195,10 @@ func (rm *RegistryManager) GetModels(db *database.Handler, f types.Filter) []v1a
 	finder := db.Model(&v1alpha1.ModelDB{}).
 		Select("model_dbs.*, category_dbs.*").
 		Joins("JOIN category_dbs ON model_dbs.category_id = category_dbs.id") //
+
+	// total count before pagination
+	var count int64
+
 	if mf, ok := f.(*v1alpha1.ModelFilter); ok {
 		if mf.Greedy {
 			if mf.Name != "" && mf.DisplayName != "" {
@@ -225,6 +229,9 @@ func (rm *RegistryManager) GetModels(db *database.Handler, f types.Filter) []v1a
 				finder = finder.Order(mf.OrderOn)
 			}
 		}
+
+		finder.Count(&count)
+
 		if mf.Limit != 0 {
 			finder = finder.Limit(mf.Limit)
 		}
@@ -241,7 +248,7 @@ func (rm *RegistryManager) GetModels(db *database.Handler, f types.Filter) []v1a
 	for _, modelDB := range modelWithCategoriess {
 		m = append(m, modelDB.ModelDB.GetModel(modelDB.GetCategory(db)))
 	}
-	return m
+	return m, count
 }
 func (rm *RegistryManager) GetCategories(db *database.Handler, f types.Filter) []v1alpha1.Category {
 	var catdb []v1alpha1.CategoryDB