init push

internal/repository/mysql/after_flight_inspection_repo.go

@@ -0,0 +1,72 @@
+package mysql
+
+import (
+	"context"
+	"errors"
+
+	"gorm.io/gorm"
+
+	afterflightinspection "wucher/internal/domain/after_flight_inspection"
+	flightinspection "wucher/internal/domain/flight_inspection"
+	"wucher/internal/domain/helicopter"
+)
+
+type AfterFlightInspectionRepository struct {
+	db *gorm.DB
+}
+
+func NewAfterFlightInspectionRepository(db *gorm.DB) *AfterFlightInspectionRepository {
+	return &AfterFlightInspectionRepository{db: db}
+}
+
+func (r *AfterFlightInspectionRepository) Upsert(ctx context.Context, row *afterflightinspection.AfterFlightInspection) error {
+	return r.db.WithContext(ctx).Save(row).Error
+}
+
+func (r *AfterFlightInspectionRepository) GetByFlightInspectionID(ctx context.Context, flightInspectionID []byte) (*afterflightinspection.AfterFlightInspection, error) {
+	var rows []afterflightinspection.AfterFlightInspection
+	err := r.db.WithContext(ctx).Where("flight_inspection_id = ?", flightInspectionID).Limit(1).Find(&rows).Error
+	if err != nil {
+		return nil, err
+	}
+	if len(rows) == 0 {
+		return nil, nil
+	}
+	return &rows[0], nil
+}
+
+func (r *AfterFlightInspectionRepository) DeleteByFlightInspectionID(ctx context.Context, flightInspectionID []byte) error {
+	return r.db.WithContext(ctx).
+		Where("flight_inspection_id = ?", flightInspectionID).
+		Delete(&afterflightinspection.AfterFlightInspection{}).Error
+}
+
+func (r *AfterFlightInspectionRepository) FlightInspectionExists(ctx context.Context, flightInspectionID []byte) (bool, error) {
+	if len(flightInspectionID) != 16 {
+		return false, nil
+	}
+	var total int64
+	err := r.db.WithContext(ctx).
+		Model(&flightinspection.FlightInspection{}).
+		Where("id = ?", flightInspectionID).
+		Count(&total).Error
+	return total > 0, err
+}
+
+func (r *AfterFlightInspectionRepository) GetHelicopterCapabilities(ctx context.Context, flightInspectionID []byte) (*afterflightinspection.HelicopterCapabilities, error) {
+	var caps afterflightinspection.HelicopterCapabilities
+	tx := r.db.WithContext(ctx).
+		Model(&helicopter.Helicopter{}).
+		Joins("JOIN reserve_acs ra ON ra.helicopter_id = helicopters.id").
+		Where("ra.inspection_id = ? AND ra.deleted_at IS NULL", flightInspectionID).
+		Select("helicopters.nr1, helicopters.nr2, helicopters.lh, helicopters.rh, helicopters.mgb, helicopters.igb, helicopters.tgb").
+		Limit(1).
+		Scan(&caps)
+	if tx.Error != nil {
+		return nil, tx.Error
+	}
+	if tx.RowsAffected == 0 {
+		return nil, errors.New("helicopter not found for flight inspection")
+	}
+	return &caps, nil
+}