Skip to content

Commit 0c684bb

Browse files
committed
Only allow retry functions internally/for testing
Signed-off-by: Salil Chandra <schandra107@bloomberg.net>
1 parent 8acbd4e commit 0c684bb

14 files changed

Lines changed: 29 additions & 10 deletions

File tree

CMakeLists.txt

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,8 @@ endif()
1717

1818
option(COMDB2_TEST "Turn ON to compile with testing features" OFF)
1919
if(COMDB2_TEST)
20-
add_definitions(-DCOMDB2_TEST)
20+
add_definitions(-DCOMDB2_TEST)
21+
add_definitions(-DCDB2API_TEST)
2122
endif()
2223

2324
option(WITH_TINFO "Turn ON on platforms where tinfo library is required for cdb2sql" OFF)

cdb2api/CMakeLists.txt

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -28,10 +28,6 @@ include_directories(
2828
add_definitions(-DDISABLE_HOSTADDR_CACHE)
2929
add_definitions(-DSBUF2_SERVER=0)
3030

31-
if (COMDB2_TEST)
32-
add_definitions(-DCDB2API_TEST)
33-
endif()
34-
3531
if (COMDB2_BBCMAKE)
3632
include(${EXTRA_PLUGINS}/cdb2api/cdb2api.cmake)
3733
add_library(opencdb2api STATIC ${src})

cdb2api/cdb2api.c

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1425,6 +1425,7 @@ struct newsqlheader {
14251425
int length;
14261426
};
14271427

1428+
#ifdef CDB2API_TEST
14281429
void cdb2_set_min_retries(int min_retries)
14291430
{
14301431
if (min_retries > 0) {
@@ -1438,6 +1439,7 @@ void cdb2_set_max_retries(int max_retries)
14381439
MAX_RETRIES = max_retries;
14391440
}
14401441
}
1442+
#endif
14411443

14421444
void cdb2_hndl_set_min_retries(cdb2_hndl_tp *hndl, int min_retries)
14431445
{

cdb2api/cdb2api.h

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -238,10 +238,6 @@ void cdb2_dump_ports(cdb2_hndl_tp *hndl, FILE *out);
238238
void cdb2_cluster_info(cdb2_hndl_tp *hndl, char **cluster, int *ports, int max, int *count);
239239
int cdb2_snapshot_file(cdb2_hndl_tp *hndl, int *file, int *offset);
240240
void cdb2_getinfo(cdb2_hndl_tp *hndl, int *intrans, int *hasql);
241-
void cdb2_set_max_retries(int max_retries);
242-
void cdb2_set_min_retries(int min_retries);
243-
void cdb2_hndl_set_max_retries(cdb2_hndl_tp *hndl, int max_retries);
244-
void cdb2_hndl_set_min_retries(cdb2_hndl_tp *hndl, int min_retries);
245241

246242
int cdb2_bind_param(cdb2_hndl_tp *hndl, const char *name, int type,
247243
const void *varaddr, int length);

cdb2api/cdb2api_int.h

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,9 @@ int cdb2_read_line(char *line, int maxlen, SBUF2 *s, const char *buf, int *chrno
3838

3939
void cdb2_setIdentityBlob(cdb2_hndl_tp *hndl, void *id);
4040

41+
void cdb2_hndl_set_max_retries(cdb2_hndl_tp *hndl, int max_retries);
42+
void cdb2_hndl_set_min_retries(cdb2_hndl_tp *hndl, int min_retries);
43+
4144
#ifndef WITH_DL_LIBS
4245
#define WITH_DL_LIBS 0
4346
#endif

cdb2api/cdb2api_test.h

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,9 @@ struct cdb2_hndl;
8282
const char *get_default_cluster(void);
8383
const char *get_default_cluster_hndl(struct cdb2_hndl *);
8484

85+
void cdb2_set_max_retries(int max_retries);
86+
void cdb2_set_min_retries(int min_retries);
87+
8588
#if defined __cplusplus
8689
}
8790
#endif /* __cplusplus */

db/sqllogfill.c

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
*/
1616
#include <sqllogfill.h>
1717
#include <cdb2api.h>
18+
#include <cdb2api_int.h>
1819
#include <sys_wrap.h>
1920
#include <parse_lsn.h>
2021
#include <unistd.h>

tests/tools/cdb2api_unit.c

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,8 @@
2323
#include <assert.h>
2424
#include <bb_oscompat.h>
2525
#include <cdb2api.c>
26+
#include <cdb2api_test.h>
27+
#include <cdb2api_int.h>
2628

2729
void test_is_sql_read()
2830
{

tests/tools/ins_upd_del.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
#include <string.h>
55
#include <arpa/inet.h>
66
#include <cdb2api.h>
7+
#include <cdb2api_test.h>
78
#include <unistd.h>
89
#include <time.h>
910
#include <sstream>

tests/tools/inscntdel.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
#include <string.h>
55
#include <arpa/inet.h>
66
#include <cdb2api.h>
7+
#include <cdb2api_test.h>
78
#include <unistd.h>
89
#include <time.h>
910
#include <sstream>

0 commit comments

Comments
 (0)