chore: adding preload for capture relation data

rulanugrh · Mar 20, 2024 · 208a275 · 208a275
1 parent 7461852
commit 208a275
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 4 deletions.
diff --git a/internal/repository/kunjungan.go b/internal/repository/kunjungan.go
@@ -27,6 +27,11 @@ func(k *kunjungan) Create(req domain.Kunjungan) (*domain.Kunjungan, error) {
 		return nil, util.Errors(err)
 	}
 
+	err = k.client.DB.Preload("Penyakit").Preload("User").Find(&req).Error
+	if err != nil {
+		return nil, util.Errors(err)
+	}
+
 	return &req, nil
 }
 
@@ -41,5 +46,11 @@ func(k *kunjungan) List(id uint) (*[]domain.Kunjungan, error) {
 		return nil, util.Errors(find.Error)
 	}
 
+	err := k.client.DB.Preload("Penyakit").Preload("User").Find(&model).Error
+	if err != nil {
+		return nil, util.Errors(err)
+	}
+
+
 	return &model, nil
 }
diff --git a/internal/repository/obat.go b/internal/repository/obat.go
@@ -35,12 +35,17 @@ func(o *obat) Create(req domain.Obat) (*domain.Obat, error) {
 		return nil, util.Errors(err)
 	}
 
+	err = o.client.DB.Preload("Penyakit").Find(&req).Error
+	if err != nil {
+		return nil, util.Errors(err)
+	}
+
 	return &req, nil
 }
 
 func(o *obat) FindID(id uint) (*domain.Obat, error) {
 	var model domain.Obat
-	err := o.client.DB.Where("id = ?", id).Find(&model).Error
+	err := o.client.DB.Where("id = ?", id).Preload("Penyakit").Find(&model).Error
 
 	if err != nil {
 		return nil, util.Errors(err)
@@ -52,7 +57,7 @@ func(o *obat) FindID(id uint) (*domain.Obat, error) {
 func(o *obat) FindAll() (*[]domain.Obat, error) {
 	var finds []domain.Obat
 
-	err := o.client.DB.Find(&finds).Error
+	err := o.client.DB.Preload("Penyakit").Find(&finds).Error
 	if err != nil {
 		return nil, util.Errors(err)
 	}

diff --git a/internal/repository/penyakit.go b/internal/repository/penyakit.go
@@ -44,7 +44,7 @@ func(p *penyakit) Create(req domain.Penyakit) (*domain.Penyakit, error) {
 
 func(p *penyakit) ListAll() (*[]domain.Penyakit, error) {
 	var model []domain.Penyakit
-	err := p.client.DB.Find(model).Error
+	err := p.client.DB.Preload("Obat").Find(model).Error
 
 	if err != nil {
 		return nil, util.Errors(err)
@@ -55,7 +55,7 @@ func(p *penyakit) ListAll() (*[]domain.Penyakit, error) {
 
 func(p *penyakit) FindID(id uint) (*domain.Penyakit, error) {
 	var model domain.Penyakit
-	find := p.client.DB.Where("id = ?", id).Find(&model)
+	find := p.client.DB.Preload("Obat").Where("id = ?", id).Find(&model)
 
 	if find.RowsAffected == 0 {
 		return nil, util.NotFound()
@@ -65,6 +65,7 @@ func(p *penyakit) FindID(id uint) (*domain.Penyakit, error) {
 		return nil, util.Errors(find.Error)
 	}
 
+
 	return &model, nil
 }