package model

import (
	"gorm.io/gorm"

	"github.com/pingcap-inc/chaosAPI/pkg/infra/db"
)

func QueryExperimentApprovedRecord(eid uint64) (*db.ExperimentApproveRecord, error) {
	var record db.ExperimentApproveRecord

	err := db.Transaction(func(tx *gorm.DB) error {
		return tx.Model(&db.ExperimentApproveRecord{}).
			Where("experiment_id=?", eid).
			First(&record).Error
	})

	return &record, err
}

func InsertExperimentApprovedRecord(record *db.ExperimentApproveRecord) error {
	return db.Transaction(func(tx *gorm.DB) error {
		return tx.Model(&db.ExperimentApproveRecord{}).Save(record).Error
	})
}