package audit

import (
	"github.com/thrasher-corp/gocryptotrader/database"
	"github.com/thrasher-corp/gocryptotrader/database/models"
	"github.com/thrasher-corp/gocryptotrader/database/repository/audit"
	log "github.com/thrasher-corp/gocryptotrader/logger"
)

type auditRepo struct{}

// Audit returns a new instance of auditRepo
func Audit() audit.Repository {
	return &auditRepo{}
}

// AddEventTx writes multiple event to database
// writes are done using a transaction with a rollback on error
func (pg *auditRepo) AddEventTx(event []*models.AuditEvent) {
	if pg == nil {
		return
	}

	tx, err := database.Conn.SQL.Begin()
	if err != nil {
		log.Errorf(log.Global, "Failed to create transaction: %v\n", err)
		return
	}

	query := `INSERT INTO audit_event (type, identifier, message) VALUES($1, $2, $3)`

	for x := range event {
		_, err = tx.Exec(query, &event[x].Type, &event[x].Identifier, &event[x].Message)

		if err != nil {
			err = tx.Rollback()
			if err != nil {
				log.Errorf(log.Global, "Tx Rollback has failed: %v", err)
			}
			return
		}
	}

	err = tx.Commit()
	if err != nil {
		err = tx.Rollback()
		if err != nil {
			log.Errorf(log.Global, "Tx Rollback has failed: %v", err)
		}
		return
	}

}