优化

internal/data/export.go

@@ -111,21 +111,7 @@ func (r *exportRepo) UpdateExportTemplate(ctx context.Context, v *biz.ExportTemp
 	})
 }
 func (r *exportRepo) DeleteExportTemplates(ctx context.Context, ids []uint) error {
-	return r.data.gormDB.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
-		var tids []string
-		if err := tx.Model(&exportTemplatePO{}).Where("id IN ?", ids).Pluck("template_id", &tids).Error; err != nil {
-			return err
-		}
-		if len(tids) > 0 {
-			if err := tx.Where("template_id IN ?", tids).Delete(&exportConditionPO{}).Error; err != nil {
-				return err
-			}
-			if err := tx.Where("template_id IN ?", tids).Delete(&exportJoinPO{}).Error; err != nil {
-				return err
-			}
-		}
-		return tx.Delete(&exportTemplatePO{}, ids).Error
-	})
+	return r.data.gormDB.WithContext(ctx).Delete(&[]exportTemplatePO{}, "id IN ?", ids).Error
 }
 func (r *exportRepo) FindExportTemplate(ctx context.Context, id uint, tid string) (*biz.ExportTemplate, error) {
 	var po exportTemplatePO
@@ -173,31 +159,9 @@ func (r *exportRepo) ListExportTemplates(ctx context.Context, page, size int, q
 	if err := applyPagination(db, page, size, 0).Find(&pos).Error; err != nil {
 		return nil, 0, err
 	}
-	templateIDs := make([]string, 0, len(pos))
-	for _, po := range pos {
-		templateIDs = append(templateIDs, po.TemplateID)
-	}
-	var conditions []exportConditionPO
-	var joins []exportJoinPO
-	if len(templateIDs) > 0 {
-		if err := r.data.gormDB.WithContext(ctx).Where("template_id IN ?", templateIDs).Find(&conditions).Error; err != nil {
-			return nil, 0, err
-		}
-		if err := r.data.gormDB.WithContext(ctx).Where("template_id IN ?", templateIDs).Find(&joins).Error; err != nil {
-			return nil, 0, err
-		}
-	}
-	conditionsByTemplate := make(map[string][]exportConditionPO, len(pos))
-	for _, condition := range conditions {
-		conditionsByTemplate[condition.TemplateID] = append(conditionsByTemplate[condition.TemplateID], condition)
-	}
-	joinsByTemplate := make(map[string][]exportJoinPO, len(pos))
-	for _, join := range joins {
-		joinsByTemplate[join.TemplateID] = append(joinsByTemplate[join.TemplateID], join)
-	}
 	out := make([]*biz.ExportTemplate, 0, len(pos))
 	for _, po := range pos {
-		out = append(out, exportFromPO(po, conditionsByTemplate[po.TemplateID], joinsByTemplate[po.TemplateID]))
+		out = append(out, exportFromPO(po, nil, nil))
 	}
 	return out, total, nil
 }