Created
November 2, 2021 20:55
-
-
Save xkisu/d4888c368708fb2a17a88e64bd487705 to your computer and use it in GitHub Desktop.
pgx error wrapping
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import ( | |
"errors" | |
"github.com/jackc/pgconn" | |
errors2 "indieport.io/indieport/platform/errors" | |
) | |
// WrapError attempts to wrap a pgx error | |
// in a suitable error type. | |
// | |
// Returns nil if the error is not a fatal error. | |
func WrapError(err error) error { | |
if err != nil { | |
var pgErr *pgconn.PgError | |
if errors.As(err, &pgErr) { | |
e := &errors2.Error{ | |
Message: pgErr.Message, | |
Err: err, | |
} | |
code := ErrorCode(pgErr.Code) | |
// Determine the suitable error code from the | |
switch code.Class() { | |
case CIntegrityConstraintViolation: | |
e.Code = errors2.EConflict | |
default: | |
e.Code = errors2.EInternal | |
} | |
return e | |
} | |
return err | |
} | |
return nil | |
} | |
// Error severities | |
const ( | |
Efatal = "FATAL" | |
Epanic = "PANIC" | |
Ewarning = "WARNING" | |
Enotice = "NOTICE" | |
Edebug = "DEBUG" | |
Einfo = "INFO" | |
Elog = "LOG" | |
) | |
// ErrorCode is a five-character error code. | |
type ErrorCode string | |
// Name returns a more human friendly rendering of the error code, namely the | |
// "condition name". | |
// | |
// See http://www.postgresql.org/docs/9.3/static/errcodes-appendix.html for | |
// details. | |
func (ec ErrorCode) Name() string { | |
return errorCodeNames[ec] | |
} | |
// ErrorClass is only the class part of an error code. | |
type ErrorClass string | |
// Name returns the condition name of an error class. It is equivalent to the | |
// condition name of the "standard" error code (i.e. the one having the last | |
// three characters "000"). | |
func (ec ErrorClass) Name() string { | |
return errorCodeNames[ErrorCode(ec+"000")] | |
} | |
// Class returns the error class, e.g. "28". | |
// | |
// See http://www.postgresql.org/docs/9.3/static/errcodes-appendix.html for | |
// details. | |
func (ec ErrorCode) Class() ErrorClass { | |
return ErrorClass(ec[0:2]) | |
} | |
// errorCodeNames is a mapping between the five-character error codes and the | |
// human readable "condition names". It is derived from the list at | |
// http://www.postgresql.org/docs/9.3/static/errcodes-appendix.html | |
var errorCodeNames = map[ErrorCode]string{ | |
// Class 00 - Successful Completion | |
"00000": "successful_completion", | |
// Class 01 - Warning | |
"01000": "warning", | |
"0100C": "dynamic_result_sets_returned", | |
"01008": "implicit_zero_bit_padding", | |
"01003": "null_value_eliminated_in_set_function", | |
"01007": "privilege_not_granted", | |
"01006": "privilege_not_revoked", | |
"01004": "string_data_right_truncation", | |
"01P01": "deprecated_feature", | |
// Class 02 - No Data (this is also a warning class per the SQL standard) | |
"02000": "no_data", | |
"02001": "no_additional_dynamic_result_sets_returned", | |
// Class 03 - SQL Statement Not Yet Complete | |
"03000": "sql_statement_not_yet_complete", | |
// Class 08 - Connection Exception | |
"08000": "connection_exception", | |
"08003": "connection_does_not_exist", | |
"08006": "connection_failure", | |
"08001": "sqlclient_unable_to_establish_sqlconnection", | |
"08004": "sqlserver_rejected_establishment_of_sqlconnection", | |
"08007": "transaction_resolution_unknown", | |
"08P01": "protocol_violation", | |
// Class 09 - Triggered Action Exception | |
"09000": "triggered_action_exception", | |
// Class 0A - Feature Not Supported | |
"0A000": "feature_not_supported", | |
// Class 0B - Invalid Transaction Initiation | |
"0B000": "invalid_transaction_initiation", | |
// Class 0F - Locator Exception | |
"0F000": "locator_exception", | |
"0F001": "invalid_locator_specification", | |
// Class 0L - Invalid Grantor | |
"0L000": "invalid_grantor", | |
"0LP01": "invalid_grant_operation", | |
// Class 0P - Invalid Role Specification | |
"0P000": "invalid_role_specification", | |
// Class 0Z - Diagnostics Exception | |
"0Z000": "diagnostics_exception", | |
"0Z002": "stacked_diagnostics_accessed_without_active_handler", | |
// Class 20 - Case Not Found | |
"20000": "case_not_found", | |
// Class 21 - Cardinality Violation | |
"21000": "cardinality_violation", | |
// Class 22 - Data Exception | |
"22000": "data_exception", | |
"2202E": "array_subscript_error", | |
"22021": "character_not_in_repertoire", | |
"22008": "datetime_field_overflow", | |
"22012": "division_by_zero", | |
"22005": "error_in_assignment", | |
"2200B": "escape_character_conflict", | |
"22022": "indicator_overflow", | |
"22015": "interval_field_overflow", | |
"2201E": "invalid_argument_for_logarithm", | |
"22014": "invalid_argument_for_ntile_function", | |
"22016": "invalid_argument_for_nth_value_function", | |
"2201F": "invalid_argument_for_power_function", | |
"2201G": "invalid_argument_for_width_bucket_function", | |
"22018": "invalid_character_value_for_cast", | |
"22007": "invalid_datetime_format", | |
"22019": "invalid_escape_character", | |
"2200D": "invalid_escape_octet", | |
"22025": "invalid_escape_sequence", | |
"22P06": "nonstandard_use_of_escape_character", | |
"22010": "invalid_indicator_parameter_value", | |
"22023": "invalid_parameter_value", | |
"2201B": "invalid_regular_expression", | |
"2201W": "invalid_row_count_in_limit_clause", | |
"2201X": "invalid_row_count_in_result_offset_clause", | |
"22009": "invalid_time_zone_displacement_value", | |
"2200C": "invalid_use_of_escape_character", | |
"2200G": "most_specific_type_mismatch", | |
"22004": "null_value_not_allowed", | |
"22002": "null_value_no_indicator_parameter", | |
"22003": "numeric_value_out_of_range", | |
"2200H": "sequence_generator_limit_exceeded", | |
"22026": "string_data_length_mismatch", | |
"22001": "string_data_right_truncation", | |
"22011": "substring_error", | |
"22027": "trim_error", | |
"22024": "unterminated_c_string", | |
"2200F": "zero_length_character_string", | |
"22P01": "floating_point_exception", | |
"22P02": "invalid_text_representation", | |
"22P03": "invalid_binary_representation", | |
"22P04": "bad_copy_file_format", | |
"22P05": "untranslatable_character", | |
"2200L": "not_an_xml_document", | |
"2200M": "invalid_xml_document", | |
"2200N": "invalid_xml_content", | |
"2200S": "invalid_xml_comment", | |
"2200T": "invalid_xml_processing_instruction", | |
// Class 23 - Integrity Constraint Violation | |
"23000": "integrity_constraint_violation", | |
"23001": "restrict_violation", | |
"23502": "not_null_violation", | |
"23503": "foreign_key_violation", | |
"23505": "unique_violation", | |
"23514": "check_violation", | |
"23P01": "exclusion_violation", | |
// Class 24 - Invalid Cursor State | |
"24000": "invalid_cursor_state", | |
// Class 25 - Invalid Transaction State | |
"25000": "invalid_transaction_state", | |
"25001": "active_sql_transaction", | |
"25002": "branch_transaction_already_active", | |
"25008": "held_cursor_requires_same_isolation_level", | |
"25003": "inappropriate_access_mode_for_branch_transaction", | |
"25004": "inappropriate_isolation_level_for_branch_transaction", | |
"25005": "no_active_sql_transaction_for_branch_transaction", | |
"25006": "read_only_sql_transaction", | |
"25007": "schema_and_data_statement_mixing_not_supported", | |
"25P01": "no_active_sql_transaction", | |
"25P02": "in_failed_sql_transaction", | |
// Class 26 - Invalid SQL Statement Name | |
"26000": "invalid_sql_statement_name", | |
// Class 27 - Triggered Data Change Violation | |
"27000": "triggered_data_change_violation", | |
// Class 28 - Invalid Authorization Specification | |
"28000": "invalid_authorization_specification", | |
"28P01": "invalid_password", | |
// Class 2B - Dependent Privilege Descriptors Still Exist | |
"2B000": "dependent_privilege_descriptors_still_exist", | |
"2BP01": "dependent_objects_still_exist", | |
// Class 2D - Invalid Transaction Termination | |
"2D000": "invalid_transaction_termination", | |
// Class 2F - SQL Routine Exception | |
"2F000": "sql_routine_exception", | |
"2F005": "function_executed_no_return_statement", | |
"2F002": "modifying_sql_data_not_permitted", | |
"2F003": "prohibited_sql_statement_attempted", | |
"2F004": "reading_sql_data_not_permitted", | |
// Class 34 - Invalid Cursor Name | |
"34000": "invalid_cursor_name", | |
// Class 38 - External Routine Exception | |
"38000": "external_routine_exception", | |
"38001": "containing_sql_not_permitted", | |
"38002": "modifying_sql_data_not_permitted", | |
"38003": "prohibited_sql_statement_attempted", | |
"38004": "reading_sql_data_not_permitted", | |
// Class 39 - External Routine Invocation Exception | |
"39000": "external_routine_invocation_exception", | |
"39001": "invalid_sqlstate_returned", | |
"39004": "null_value_not_allowed", | |
"39P01": "trigger_protocol_violated", | |
"39P02": "srf_protocol_violated", | |
// Class 3B - Savepoint Exception | |
"3B000": "savepoint_exception", | |
"3B001": "invalid_savepoint_specification", | |
// Class 3D - Invalid Catalog Name | |
"3D000": "invalid_catalog_name", | |
// Class 3F - Invalid Schema Name | |
"3F000": "invalid_schema_name", | |
// Class 40 - Transaction Rollback | |
"40000": "transaction_rollback", | |
"40002": "transaction_integrity_constraint_violation", | |
"40001": "serialization_failure", | |
"40003": "statement_completion_unknown", | |
"40P01": "deadlock_detected", | |
// Class 42 - Syntax Error or Access Rule Violation | |
"42000": "syntax_error_or_access_rule_violation", | |
"42601": "syntax_error", | |
"42501": "insufficient_privilege", | |
"42846": "cannot_coerce", | |
"42803": "grouping_error", | |
"42P20": "windowing_error", | |
"42P19": "invalid_recursion", | |
"42830": "invalid_foreign_key", | |
"42602": "invalid_name", | |
"42622": "name_too_long", | |
"42939": "reserved_name", | |
"42804": "datatype_mismatch", | |
"42P18": "indeterminate_datatype", | |
"42P21": "collation_mismatch", | |
"42P22": "indeterminate_collation", | |
"42809": "wrong_object_type", | |
"42703": "undefined_column", | |
"42883": "undefined_function", | |
"42P01": "undefined_table", | |
"42P02": "undefined_parameter", | |
"42704": "undefined_object", | |
"42701": "duplicate_column", | |
"42P03": "duplicate_cursor", | |
"42P04": "duplicate_database", | |
"42723": "duplicate_function", | |
"42P05": "duplicate_prepared_statement", | |
"42P06": "duplicate_schema", | |
"42P07": "duplicate_table", | |
"42712": "duplicate_alias", | |
"42710": "duplicate_object", | |
"42702": "ambiguous_column", | |
"42725": "ambiguous_function", | |
"42P08": "ambiguous_parameter", | |
"42P09": "ambiguous_alias", | |
"42P10": "invalid_column_reference", | |
"42611": "invalid_column_definition", | |
"42P11": "invalid_cursor_definition", | |
"42P12": "invalid_database_definition", | |
"42P13": "invalid_function_definition", | |
"42P14": "invalid_prepared_statement_definition", | |
"42P15": "invalid_schema_definition", | |
"42P16": "invalid_table_definition", | |
"42P17": "invalid_object_definition", | |
// Class 44 - WITH CHECK OPTION Violation | |
"44000": "with_check_option_violation", | |
// Class 53 - Insufficient Resources | |
"53000": "insufficient_resources", | |
"53100": "disk_full", | |
"53200": "out_of_memory", | |
"53300": "too_many_connections", | |
"53400": "configuration_limit_exceeded", | |
// Class 54 - Program Limit Exceeded | |
"54000": "program_limit_exceeded", | |
"54001": "statement_too_complex", | |
"54011": "too_many_columns", | |
"54023": "too_many_arguments", | |
// Class 55 - Object Not In Prerequisite State | |
"55000": "object_not_in_prerequisite_state", | |
"55006": "object_in_use", | |
"55P02": "cant_change_runtime_param", | |
"55P03": "lock_not_available", | |
// Class 57 - Operator Intervention | |
"57000": "operator_intervention", | |
"57014": "query_canceled", | |
"57P01": "admin_shutdown", | |
"57P02": "crash_shutdown", | |
"57P03": "cannot_connect_now", | |
"57P04": "database_dropped", | |
// Class 58 - System Error (errors external to PostgreSQL itself) | |
"58000": "system_error", | |
"58030": "io_error", | |
"58P01": "undefined_file", | |
"58P02": "duplicate_file", | |
// Class F0 - Configuration File Error | |
"F0000": "config_file_error", | |
"F0001": "lock_file_exists", | |
// Class HV - Foreign Data Wrapper Error (SQL/MED) | |
"HV000": "fdw_error", | |
"HV005": "fdw_column_name_not_found", | |
"HV002": "fdw_dynamic_parameter_value_needed", | |
"HV010": "fdw_function_sequence_error", | |
"HV021": "fdw_inconsistent_descriptor_information", | |
"HV024": "fdw_invalid_attribute_value", | |
"HV007": "fdw_invalid_column_name", | |
"HV008": "fdw_invalid_column_number", | |
"HV004": "fdw_invalid_data_type", | |
"HV006": "fdw_invalid_data_type_descriptors", | |
"HV091": "fdw_invalid_descriptor_field_identifier", | |
"HV00B": "fdw_invalid_handle", | |
"HV00C": "fdw_invalid_option_index", | |
"HV00D": "fdw_invalid_option_name", | |
"HV090": "fdw_invalid_string_length_or_buffer_length", | |
"HV00A": "fdw_invalid_string_format", | |
"HV009": "fdw_invalid_use_of_null_pointer", | |
"HV014": "fdw_too_many_handles", | |
"HV001": "fdw_out_of_memory", | |
"HV00P": "fdw_no_schemas", | |
"HV00J": "fdw_option_name_not_found", | |
"HV00K": "fdw_reply_handle", | |
"HV00Q": "fdw_schema_not_found", | |
"HV00R": "fdw_table_not_found", | |
"HV00L": "fdw_unable_to_create_execution", | |
"HV00M": "fdw_unable_to_create_reply", | |
"HV00N": "fdw_unable_to_establish_connection", | |
// Class P0 - PL/pgSQL Error | |
"P0000": "plpgsql_error", | |
"P0001": "raise_exception", | |
"P0002": "no_data_found", | |
"P0003": "too_many_rows", | |
// Class XX - Internal Error | |
"XX000": "internal_error", | |
"XX001": "data_corrupted", | |
"XX002": "index_corrupted", | |
} | |
// Error classes | |
const ( | |
CSuccessfulCompletion = ErrorClass("00") | |
CWarning = ErrorClass("01") | |
CNoData = ErrorClass("02") | |
CSqlStatementNotYetComplete = ErrorClass("03") | |
CConnectionException = ErrorClass("08") | |
CTriggeredActionException = ErrorClass("09") | |
CFeatureNotSupported = ErrorClass("0A") | |
CInvalidTransactionInitiation = ErrorClass("0B") | |
CLocatorException = ErrorClass("0F") | |
CInvalidGrantor = ErrorClass("0L") | |
CInvalidRoleSpecification = ErrorClass("0P") | |
CDiagnosticsException = ErrorClass("0Z") | |
CCaseNotFound = ErrorClass("20") | |
CCardinalityViolation = ErrorClass("21") | |
CDataException = ErrorClass("22") | |
CIntegrityConstraintViolation = ErrorClass("23") | |
CInvalidCursorState = ErrorClass("24") | |
CInvalidTransactionState = ErrorClass("25") | |
CInvalidSqlStatementName = ErrorClass("26") | |
CTriggeredDataChangeViolation = ErrorClass("27") | |
CInvalidAuthorizationSpecification = ErrorClass("28") | |
CDependentPrivilegeDescriptorsStillExist = ErrorClass("2B") | |
CInvalidTransactionTermination = ErrorClass("2D") | |
CSqlRoutineException = ErrorClass("2F") | |
CInvalidCursorName = ErrorClass("34") | |
CExternalRoutineException = ErrorClass("38") | |
CExternalRoutineInvocationException = ErrorClass("39") | |
CSavepointException = ErrorClass("3B") | |
CInvalidCatalogName = ErrorClass("3D") | |
CInvalidSchemaName = ErrorClass("3F") | |
CTransactionRollback = ErrorClass("40") | |
CSyntaxErrorOrAccessRuleViolation = ErrorClass("42") | |
CWithCheckOptionViolation = ErrorClass("44") | |
CInsufficientResources = ErrorClass("53") | |
CProgramLimitExceeded = ErrorClass("54") | |
CObjectNotInPrerequisiteState = ErrorClass("55") | |
COperatorIntervention = ErrorClass("57") | |
CSystemError = ErrorClass("58") | |
CConfigFileError = ErrorClass("F0") | |
CFdwError = ErrorClass("HV") | |
CPlpgsqlError = ErrorClass("P0") | |
CInternalError = ErrorClass("XX") | |
) | |
/* | |
package main | |
import ( | |
"fmt" | |
"strings" | |
"sort" | |
) | |
var errorClassNames = map[string]string{ | |
"00": "successful_completion", | |
"01": "warning", | |
"02": "no_data", | |
"03": "sql_statement_not_yet_complete", | |
"08": "connection_exception", | |
"09": "triggered_action_exception", | |
"0A": "feature_not_supported", | |
"0B": "invalid_transaction_initiation", | |
"0F": "locator_exception", | |
"0L": "invalid_grantor", | |
"0P": "invalid_role_specification", | |
"0Z": "diagnostics_exception", | |
"20": "case_not_found", | |
"21": "cardinality_violation", | |
"22": "data_exception", | |
"23": "integrity_constraint_violation", | |
"24": "invalid_cursor_state", | |
"25": "invalid_transaction_state", | |
"26": "invalid_sql_statement_name", | |
"27": "triggered_data_change_violation", | |
"28": "invalid_authorization_specification", | |
"2B": "dependent_privilege_descriptors_still_exist", | |
"2D": "invalid_transaction_termination", | |
"2F": "sql_routine_exception", | |
"34": "invalid_cursor_name", | |
"38": "external_routine_exception", | |
"39": "external_routine_invocation_exception", | |
"3B": "savepoint_exception", | |
"3D": "invalid_catalog_name", | |
"3F": "invalid_schema_name", | |
"40": "transaction_rollback", | |
"42": "syntax_error_or_access_rule_violation", | |
"44": "with_check_option_violation", | |
"53": "insufficient_resources", | |
"54": "program_limit_exceeded", | |
"55": "object_not_in_prerequisite_state", | |
"57": "operator_intervention", | |
"58": "system_error", | |
"F0": "config_file_error", | |
"HV": "fdw_error", | |
"P0": "plpgsql_error", | |
"XX": "internal_error", | |
} | |
func snakeCaseToCamelCase(inputUnderScoreStr string) (camelCase string) { | |
//snake_case to camelCase | |
isToUpper := false | |
for k, v := range inputUnderScoreStr { | |
if k == 0 { | |
camelCase = strings.ToUpper(string(inputUnderScoreStr[0])) | |
} else { | |
if isToUpper { | |
camelCase += strings.ToUpper(string(v)) | |
isToUpper = false | |
} else { | |
if v == '_' { | |
isToUpper = true | |
} else { | |
camelCase += string(v) | |
} | |
} | |
} | |
} | |
return | |
} | |
func main() { | |
keys := make([]string, 0) | |
for k, _ := range errorClassNames { | |
keys = append(keys, k) | |
} | |
sort.Strings(keys) | |
txt := "" | |
for _, k := range keys { | |
txt += fmt.Sprintf("C%s = ErrorClass(\"%s\")\n", snakeCaseToCamelCase(errorClassNames[k]), k) | |
} | |
fmt.Println(txt) | |
} | |
*/ | |
var errorClassNames = map[ErrorClass]string{ | |
"00": "successful_completion", | |
"01": "warning", | |
"02": "no_data", | |
"03": "sql_statement_not_yet_complete", | |
"08": "connection_exception", | |
"09": "triggered_action_exception", | |
"0A": "feature_not_supported", | |
"0B": "invalid_transaction_initiation", | |
"0F": "locator_exception", | |
"0L": "invalid_grantor", | |
"0P": "invalid_role_specification", | |
"0Z": "diagnostics_exception", | |
"20": "case_not_found", | |
"21": "cardinality_violation", | |
"22": "data_exception", | |
"23": "integrity_constraint_violation", | |
"24": "invalid_cursor_state", | |
"25": "invalid_transaction_state", | |
"26": "invalid_sql_statement_name", | |
"27": "triggered_data_change_violation", | |
"28": "invalid_authorization_specification", | |
"2B": "dependent_privilege_descriptors_still_exist", | |
"2D": "invalid_transaction_termination", | |
"2F": "sql_routine_exception", | |
"34": "invalid_cursor_name", | |
"38": "external_routine_exception", | |
"39": "external_routine_invocation_exception", | |
"3B": "savepoint_exception", | |
"3D": "invalid_catalog_name", | |
"3F": "invalid_schema_name", | |
"40": "transaction_rollback", | |
"42": "syntax_error_or_access_rule_violation", | |
"44": "with_check_option_violation", | |
"53": "insufficient_resources", | |
"54": "program_limit_exceeded", | |
"55": "object_not_in_prerequisite_state", | |
"57": "operator_intervention", | |
"58": "system_error", | |
"F0": "config_file_error", | |
"HV": "fdw_error", | |
"P0": "plpgsql_error", | |
"XX": "internal_error", | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment