耳机阻抗上报

internal/repository/headphone_impedance.go

@@ -0,0 +1,99 @@
+package repository
+
+import (
+	"context"
+	"database/sql"
+	"fmt"
+
+	"github.com/luxsin/app-api/internal/model"
+)
+
+type HeadphoneImpedanceRepository struct {
+	db *sql.DB
+}
+
+func NewHeadphoneImpedanceRepository(db *sql.DB) *HeadphoneImpedanceRepository {
+	return &HeadphoneImpedanceRepository{db: db}
+}
+
+func (r *HeadphoneImpedanceRepository) FindByMacAndNorm(
+	ctx context.Context,
+	macAddr, brandNorm, modelNorm string,
+) (*model.UserHeadphoneImpedance, error) {
+	const query = `
+		SELECT id, mac_addr, device_model, impedance_ohm,
+			headphone_brand, headphone_model,
+			headphone_brand_norm, headphone_model_norm,
+			ip_addr, create_at, update_at
+		FROM user_headphone_impedance
+		WHERE mac_addr = ? AND headphone_brand_norm = ? AND headphone_model_norm = ?`
+
+	var rec model.UserHeadphoneImpedance
+	err := r.db.QueryRowContext(ctx, query, macAddr, brandNorm, modelNorm).Scan(
+		&rec.ID,
+		&rec.MacAddr,
+		&rec.DeviceModel,
+		&rec.ImpedanceOhm,
+		&rec.HeadphoneBrand,
+		&rec.HeadphoneModel,
+		&rec.HeadphoneBrandNorm,
+		&rec.HeadphoneModelNorm,
+		&rec.IpAddr,
+		&rec.CreateAt,
+		&rec.UpdateAt,
+	)
+	if err == sql.ErrNoRows {
+		return nil, nil
+	}
+	if err != nil {
+		return nil, fmt.Errorf("query user_headphone_impedance: %w", err)
+	}
+	return &rec, nil
+}
+
+func (r *HeadphoneImpedanceRepository) Insert(ctx context.Context, rec model.UserHeadphoneImpedance) error {
+	const query = `
+		INSERT INTO user_headphone_impedance (
+			mac_addr, device_model, impedance_ohm,
+			headphone_brand, headphone_model,
+			headphone_brand_norm, headphone_model_norm,
+			ip_addr
+		) VALUES (?, ?, ?, ?, ?, ?, ?, ?)`
+
+	_, err := r.db.ExecContext(ctx, query,
+		rec.MacAddr,
+		rec.DeviceModel,
+		rec.ImpedanceOhm,
+		rec.HeadphoneBrand,
+		rec.HeadphoneModel,
+		rec.HeadphoneBrandNorm,
+		rec.HeadphoneModelNorm,
+		rec.IpAddr,
+	)
+	if err != nil {
+		return fmt.Errorf("insert user_headphone_impedance: %w", err)
+	}
+	return nil
+}
+
+func (r *HeadphoneImpedanceRepository) Update(ctx context.Context, rec model.UserHeadphoneImpedance) error {
+	const query = `
+		UPDATE user_headphone_impedance
+		SET device_model = ?, impedance_ohm = ?,
+			headphone_brand = ?, headphone_model = ?,
+			ip_addr = ?
+		WHERE id = ?`
+
+	_, err := r.db.ExecContext(ctx, query,
+		rec.DeviceModel,
+		rec.ImpedanceOhm,
+		rec.HeadphoneBrand,
+		rec.HeadphoneModel,
+		rec.IpAddr,
+		rec.ID,
+	)
+	if err != nil {
+		return fmt.Errorf("update user_headphone_impedance: %w", err)
+	}
+	return nil
+}