synapsecns/sanguine

package base

import (
    "context"
    "errors"
    "fmt"

    "github.com/ethereum/go-ethereum/common/hexutil"
    "github.com/synapsecns/sanguine/services/rfq/guard/guarddb"
    "gorm.io/gorm"
    "gorm.io/gorm/clause"
)

// StoreBridgeRequest stores a quote request.
func (s Store) StoreBridgeRequest(ctx context.Context, request guarddb.BridgeRequest) error {
    model := FromBridgeRequest(request)
    dbTx := s.DB().WithContext(ctx).Clauses(clause.OnConflict{
        Columns:   []clause.Column{{Name: transactionIDFieldName}},
        DoUpdates: clause.AssignmentColumns([]string{transactionIDFieldName}),
    }).Create(&model)
    if dbTx.Error != nil {
        return fmt.Errorf("could not store request: %w", dbTx.Error)
    }
    return nil
}

// GetBridgeRequestByID gets a quote request by id. Should return ErrNoBridgeRequestForID if not found.
func (s Store) GetBridgeRequestByID(ctx context.Context, id [32]byte) (*guarddb.BridgeRequest, error) {
    var modelResult BridgeRequestModel
    tx := s.DB().WithContext(ctx).Where(fmt.Sprintf("%s = ?", transactionIDFieldName), hexutil.Encode(id[:])).First(&modelResult)
    if errors.Is(tx.Error, gorm.ErrRecordNotFound) {
        return nil, guarddb.ErrNoBridgeRequestForID
    }

    if tx.Error != nil {
        return nil, fmt.Errorf("could not get request")
    }

    qr, err := modelResult.ToBridgeRequest()
    if err != nil {
        return nil, err
    }
    return qr, nil
}