First pass at adding in exception for servers which have gone away.
[m6w6/libmemcached] / libtest / fatal.cc
1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
2 *
3 * libtest
4 *
5 * Copyright (C) 2012 Data Differential, http://datadifferential.com/
6 *
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 3 of the License, or (at your option) any later version.
11 *
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this library; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22 #include <config.h>
23 #include <libtest/common.h>
24 #include <cstdarg>
25
26 namespace libtest {
27
28 fatal::fatal(const char *file, int line, const char *func, const char *format, ...) :
29 std::runtime_error(func)
30 {
31 va_list args;
32 va_start(args, format);
33 char last_error[BUFSIZ];
34 (void)vsnprintf(last_error, sizeof(last_error), format, args);
35 va_end(args);
36
37 snprintf(_error_message, sizeof(_error_message), "%s:%d FATAL:%s (%s)", file, int(line), last_error, func);
38 }
39
40 static bool _disabled= false;
41 static uint32_t _counter= 0;
42
43 bool fatal::is_disabled()
44 {
45 return _disabled;
46 }
47
48 void fatal::disable()
49 {
50 _disabled= true;
51 }
52
53 void fatal::enable()
54 {
55 _disabled= false;
56 }
57
58 uint32_t fatal::disabled_counter()
59 {
60 return _counter;
61 }
62
63 void fatal::increment_disabled_counter()
64 {
65 _counter++;
66 }
67
68 disconnected::disconnected(const char *file, int line, const char *func, const char *instance, const in_port_t port, const char *format, ...) :
69 _port(port),
70 std::runtime_error(func)
71 {
72 strncpy(_instance, instance, sizeof(_instance));
73 va_list args;
74 va_start(args, format);
75 char last_error[BUFSIZ];
76 (void)vsnprintf(last_error, sizeof(last_error), format, args);
77 va_end(args);
78
79 snprintf(_error_message, sizeof(_error_message), "%s:%d FATAL:%s (%s)", file, int(line), last_error, func);
80 }
81
82 } // namespace libtest
83