package database

import (

"errors"

"github.com/lib/pq"

)

func IsSerializedError(err error) bool {

var pqErr *pq.Error

if errors.As(err, &pqErr) {

return pqErr.Code.Name() == "serialization_failure"

}

return false

}

// IsUniqueViolation checks if the error is due to a unique violation.

// If one or more specific unique constraints are given as arguments,

// the error must be caused by one of them. If no constraints are given,

// this function returns true for any unique violation.

func IsUniqueViolation(err error, uniqueConstraints ...UniqueConstraint) bool {

var pqErr *pq.Error

if errors.As(err, &pqErr) {

if pqErr.Code.Name() == "unique_violation" {

if len(uniqueConstraints) == 0 {

return true

}

for _, uc := range uniqueConstraints {

if pqErr.Constraint == string(uc) {

return true

}

}

}

}

return false

}

// IsQueryCanceledError checks if the error is due to a query being canceled.

func IsQueryCanceledError(err error) bool {

var pqErr *pq.Error

if errors.As(err, &pqErr) {

return pqErr.Code.Name() == "query_canceled"

}

return false

}

func IsStartupLogsLimitError(err error) bool {

var pqErr *pq.Error

if errors.As(err, &pqErr) {

return pqErr.Constraint == "max_startup_logs_length" && pqErr.Table == "workspace_agents"

}

return false

}