Improved error handling

This commit is contained in:
BLM 2015-07-14 15:07:04 -05:00
parent f07269b2ab
commit adbcd9b878
1 changed files with 28 additions and 6 deletions

View File

@ -21,6 +21,13 @@ void reportOutOfMemory() {
(errcode(ERRCODE_OUT_OF_MEMORY))); (errcode(ERRCODE_OUT_OF_MEMORY)));
} }
void reportParsingError(const char* phone_number, const char* msg = "") {
ereport(ERROR,
(errcode(ERRCODE_INVALID_TEXT_REPRESENTATION),
errmsg("unable to parse '%s' as a phone number", phone_number),
errdetail("%s", msg)));
}
void reportGenericError(std::exception& exception) { void reportGenericError(std::exception& exception) {
ereport(ERROR, ereport(ERROR,
(errcode(ERRCODE_EXTERNAL_ROUTINE_INVOCATION_EXCEPTION), (errcode(ERRCODE_EXTERNAL_ROUTINE_INVOCATION_EXCEPTION),
@ -40,7 +47,7 @@ extern "C" {
Datum Datum
phone_number_in(PG_FUNCTION_ARGS) phone_number_in(PG_FUNCTION_ARGS)
{ {
char *str = PG_GETARG_CSTRING(0); const char *str = PG_GETARG_CSTRING(0);
PhoneNumber *number; PhoneNumber *number;
number = (PhoneNumber*)palloc0(sizeof(PhoneNumber)); number = (PhoneNumber*)palloc0(sizeof(PhoneNumber));
@ -49,13 +56,20 @@ extern "C" {
} }
try { try {
PhoneNumberUtil::ErrorType error; using PNU = i18n::phonenumbers::PhoneNumberUtil;
PNU::ErrorType error;
error = phoneUtil->Parse(str, "US", number); error = phoneUtil->Parse(str, "US", number);
if(error == PhoneNumberUtil::NO_PARSING_ERROR) { switch(error) {
case PNU::NO_PARSING_ERROR:
//The number was parsed correctly; return it. //The number was parsed correctly; return it.
PG_RETURN_POINTER(number); PG_RETURN_POINTER(number);
break;
default:
//We have some generic parsing error.
reportParsingError(str);
} }
//TODO: handle errors. //TODO: handle errors.
//TODO: check number validity.
} catch(std::bad_alloc& e) { } catch(std::bad_alloc& e) {
reportOutOfMemory(); reportOutOfMemory();
} catch (std::exception& e) { } catch (std::exception& e) {
@ -77,16 +91,23 @@ extern "C" {
try { try {
phoneUtil->Format(*number, PhoneNumberUtil::INTERNATIONAL, &formatted); phoneUtil->Format(*number, PhoneNumberUtil::INTERNATIONAL, &formatted);
} catch (std::exception e) { } catch(std::bad_alloc& e) {
//TODO: implement. reportOutOfMemory();
} catch (std::exception& e) {
reportGenericError(e);
} }
//Copy the formatted number to a C-style string.
//We must use the PostgreSQL allocator, not new/malloc.
size_t len = formatted.length(); size_t len = formatted.length();
ereport(INFO, ereport(INFO,
(errcode(ERRCODE_SUCCESSFUL_COMPLETION), (errcode(ERRCODE_SUCCESSFUL_COMPLETION),
errmsg("Length: %zu", len))); errmsg("Length: %zu", len)));
result = (char*)palloc(len + 1); result = (char*)palloc(len + 1);
//TODO: test allocation. if(result == nullptr) {
reportOutOfMemory();
PG_RETURN_NULL();
}
memcpy(result, formatted.data(), len); memcpy(result, formatted.data(), len);
result[len] = '\0'; result[len] = '\0';
@ -102,6 +123,7 @@ extern "C" {
StringInfo buf = (StringInfo)PG_GETARG_POINTER(0); StringInfo buf = (StringInfo)PG_GETARG_POINTER(0);
PhoneNumber *result; PhoneNumber *result;
//TODO: error handling?
result = (PhoneNumber*)palloc0(sizeof(PhoneNumber)); result = (PhoneNumber*)palloc0(sizeof(PhoneNumber));
result->set_country_code(pq_getmsgint(buf, 4)); result->set_country_code(pq_getmsgint(buf, 4));
result->set_national_number(pq_getmsgint64(buf)); result->set_national_number(pq_getmsgint64(buf));