1 | /* |
2 | * smd.h |
3 | * |
4 | * Copyright (C) 2018 Aerospike, Inc. |
5 | * |
6 | * Portions may be licensed to Aerospike, Inc. under one or more contributor |
7 | * license agreements. |
8 | * |
9 | * This program is free software: you can redistribute it and/or modify it under |
10 | * the terms of the GNU Affero General Public License as published by the Free |
11 | * Software Foundation, either version 3 of the License, or (at your option) any |
12 | * later version. |
13 | * |
14 | * This program is distributed in the hope that it will be useful, but WITHOUT |
15 | * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS |
16 | * FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for more |
17 | * details. |
18 | * |
19 | * You should have received a copy of the GNU Affero General Public License |
20 | * along with this program. If not, see http://www.gnu.org/licenses/ |
21 | */ |
22 | |
23 | #pragma once |
24 | |
25 | //========================================================== |
26 | // Includes. |
27 | // |
28 | |
29 | #include <stdbool.h> |
30 | #include <stdint.h> |
31 | |
32 | #include "citrusleaf/cf_vector.h" |
33 | |
34 | #include "dynbuf.h" |
35 | |
36 | |
37 | //========================================================== |
38 | // Typedefs & constants. |
39 | // |
40 | |
41 | // These values are used on the wire - don't change them. |
42 | typedef enum { |
43 | AS_SMD_MODULE_EVICT, |
44 | AS_SMD_MODULE_ROSTER, |
45 | AS_SMD_MODULE_SECURITY, |
46 | AS_SMD_MODULE_SINDEX, |
47 | AS_SMD_MODULE_TRUNCATE, |
48 | AS_SMD_MODULE_UDF, |
49 | |
50 | AS_SMD_NUM_MODULES |
51 | } as_smd_id; |
52 | |
53 | typedef struct as_smd_item_s { |
54 | char* key; |
55 | char* value; // NULL means delete |
56 | uint64_t timestamp; |
57 | uint32_t generation; |
58 | } as_smd_item; |
59 | |
60 | typedef enum { |
61 | AS_SMD_ACCEPT_OPT_START, |
62 | AS_SMD_ACCEPT_OPT_SET |
63 | } as_smd_accept_type; |
64 | |
65 | typedef void (*as_smd_accept_fn)(const cf_vector* items, as_smd_accept_type accept_type); |
66 | // Return false to choose item0, true to choose item1. |
67 | typedef bool (*as_smd_conflict_fn)(const as_smd_item* item0, const as_smd_item* item1); |
68 | typedef void (*as_smd_get_all_fn)(const cf_vector* items, void* udata); |
69 | typedef void (*as_smd_set_fn)(bool result, void* udata); |
70 | |
71 | |
72 | //========================================================== |
73 | // Public API. |
74 | // |
75 | |
76 | void as_smd_module_load(as_smd_id id, as_smd_accept_fn accept_cb, as_smd_conflict_fn conflict_cb, const cf_vector* default_items); |
77 | void as_smd_start(void); |
78 | // timeout 0 is default timeout in msec. |
79 | void as_smd_set(as_smd_id id, const char* key, const char* value, as_smd_set_fn set_cb, void* udata, uint64_t timeout); |
80 | bool as_smd_set_blocking(as_smd_id id, const char* key, const char* value, uint64_t timeout); |
81 | void as_smd_get_all(as_smd_id id, as_smd_get_all_fn cb, void* udata); |
82 | void as_smd_get_info(cf_dyn_buf* db); |
83 | |
84 | static inline void |
85 | as_smd_set_and_forget(as_smd_id id, const char* key, const char* value) |
86 | { |
87 | as_smd_set(id, key, value, NULL, NULL, 0); |
88 | } |
89 | |
90 | static inline void |
91 | as_smd_delete(as_smd_id id, const char* key, as_smd_set_fn set_cb, void* udata, uint64_t timeout) |
92 | { |
93 | as_smd_set(id, key, NULL, set_cb, udata, timeout); |
94 | } |
95 | |
96 | static inline bool |
97 | as_smd_delete_blocking(as_smd_id id, const char* key, uint64_t timeout) |
98 | { |
99 | return as_smd_set_blocking(id, key, NULL, timeout); |
100 | } |
101 | |
102 | static inline void |
103 | as_smd_delete_and_forget(as_smd_id id, const char* key) |
104 | { |
105 | as_smd_set(id, key, NULL, NULL, NULL, 0); |
106 | } |
107 | |