base.go

package sql

import (
	"encoding/json"
	"fmt"
	"reflect"

	"github.com/jinzhu/gorm"

	"sample-app/model"
	"sample-app/service"
	"sample-app/util"
)

type baseRepository struct {
	*gorm.DB
}

func NewBaseRepository(db *gorm.DB) service.BaseRepository {
	return &baseRepository{db}
}

func (br *baseRepository) GetRows(models []string) json.RawMessage {
	data := map[string]interface{}{}

	for _, modelName := range models {
		modelSlice, ok := util.NewInstanceSlice(modelName)
		if !ok {
			panic("Model is not registered")
		}

		rows := reflect.New(modelSlice.Type()).Interface()
		if err := br.Model(rows).Find(rows).Error; err != nil {
			fmt.Printf("Error while getting data for model: %s and the error is: %v", modelName, err)
		}

		data[modelName] = rows
	}

	jsonData, _ := json.Marshal(data)
	return jsonData
}