1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
/*
* pg_errors.c - Definition and lookup of error classes.
*
*/
#include "pg.h"
VALUE rb_hErrors;
VALUE rb_ePGerror;
VALUE rb_eServerError;
VALUE rb_eUnableToSend;
VALUE rb_eConnectionBad;
VALUE rb_eInvalidResultStatus;
VALUE rb_eNoResultError;
VALUE rb_eInvalidChangeOfResultFields;
static VALUE
define_error_class(const char *name, const char *baseclass_code)
{
VALUE baseclass = rb_eServerError;
if(baseclass_code)
{
baseclass = rb_hash_aref( rb_hErrors, rb_str_new2(baseclass_code) );
}
return rb_define_class_under( rb_mPG, name, baseclass );
}
static void
register_error_class(const char *code, VALUE klass)
{
rb_hash_aset( rb_hErrors, rb_str_new2(code), klass );
}
/* Find a proper error class for the given SQLSTATE string
*/
VALUE
lookup_error_class(const char *sqlstate)
{
VALUE klass;
if(sqlstate)
{
/* Find the proper error class by the 5-characters SQLSTATE. */
klass = rb_hash_aref( rb_hErrors, rb_str_new2(sqlstate) );
if(NIL_P(klass))
{
/* The given SQLSTATE couldn't be found. This might happen, if
* the server side uses a newer version than the client.
* Try to find a error class by using the 2-characters SQLSTATE.
*/
klass = rb_hash_aref( rb_hErrors, rb_str_new(sqlstate, 2) );
if(NIL_P(klass))
{
/* Also the 2-characters SQLSTATE is unknown.
* Use the generic server error instead.
*/
klass = rb_eServerError;
}
}
}
else
{
/* Unable to retrieve the PG_DIAG_SQLSTATE.
* Use the generic error instead.
*/
klass = rb_eUnableToSend;
}
return klass;
}
void
init_pg_errors(void)
{
rb_hErrors = rb_hash_new();
rb_define_const( rb_mPG, "ERROR_CLASSES", rb_hErrors );
rb_ePGerror = rb_define_class_under( rb_mPG, "Error", rb_eStandardError );
/*************************
* PG::Error
*************************/
rb_define_alias( rb_ePGerror, "error", "message" );
rb_define_attr( rb_ePGerror, "connection", 1, 0 );
rb_define_attr( rb_ePGerror, "result", 1, 0 );
rb_eServerError = rb_define_class_under( rb_mPG, "ServerError", rb_ePGerror );
rb_eUnableToSend = rb_define_class_under( rb_mPG, "UnableToSend", rb_ePGerror );
rb_eConnectionBad = rb_define_class_under( rb_mPG, "ConnectionBad", rb_ePGerror );
rb_eInvalidResultStatus = rb_define_class_under( rb_mPG, "InvalidResultStatus", rb_ePGerror );
rb_eNoResultError = rb_define_class_under( rb_mPG, "NoResultError", rb_ePGerror );
rb_eInvalidChangeOfResultFields = rb_define_class_under( rb_mPG, "InvalidChangeOfResultFields", rb_ePGerror );
#include "errorcodes.def"
}
|
