client.cpp 90 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270
  1. #include "smartbotic/database/client.hpp"
  2. #include <database.grpc.pb.h>
  3. #include <grpcpp/grpcpp.h>
  4. #include <grpcpp/security/credentials.h>
  5. #include <grpcpp/security/tls_certificate_verifier.h>
  6. #include <grpcpp/security/tls_credentials_options.h>
  7. #include <spdlog/spdlog.h>
  8. #include <atomic>
  9. #include <chrono>
  10. #include <cstdlib>
  11. #include <fstream>
  12. #include <mutex>
  13. #include <sstream>
  14. #include <thread>
  15. namespace smartbotic::database {
  16. namespace {
  17. // v1.7.0 T11 — transient transport errors that are safe to retry for idempotent
  18. // writes (insert with explicit ID, updateIfVersion, upsert, remove, patch).
  19. // All other codes (INVALID_ARGUMENT, FAILED_PRECONDITION, NOT_FOUND, ...) are
  20. // final — no retry.
  21. bool retryableStatus(const grpc::Status& s) {
  22. switch (s.error_code()) {
  23. case grpc::StatusCode::DEADLINE_EXCEEDED: // server queued behind eviction/WAL
  24. case grpc::StatusCode::RESOURCE_EXHAUSTED: // admission control or gRPC concurrency cap
  25. case grpc::StatusCode::UNAVAILABLE: // transient connection issue
  26. return true;
  27. default:
  28. return false;
  29. }
  30. }
  31. // Exponential backoff with jitter. attempt=0 => baseMs; attempt=1 => baseMs*2; ...
  32. uint32_t computeBackoffMs(uint32_t baseMs, uint32_t attempt, uint32_t capMs, double jitterPct) {
  33. uint64_t exp = static_cast<uint64_t>(baseMs) << attempt;
  34. if (exp > capMs) exp = capMs;
  35. // ±jitterPct jitter; std::rand() is fine here — not a security sensitive RNG.
  36. double jitter = 1.0 + ((double(std::rand()) / RAND_MAX) * 2.0 - 1.0) * jitterPct;
  37. if (jitter < 0.1) jitter = 0.1;
  38. return static_cast<uint32_t>(exp * jitter);
  39. }
  40. // v2.4 — slurp a PEM-encoded CA cert file into a string. Returns empty
  41. // string on read failure; caller logs the surrounding context.
  42. std::string readCaCertFile(const std::string& path) {
  43. std::ifstream in(path, std::ios::binary);
  44. if (!in.is_open()) return {};
  45. std::ostringstream buf;
  46. buf << in.rdbuf();
  47. return buf.str();
  48. }
  49. } // anonymous namespace
  50. // ===== PIMPL Implementation =====
  51. class Client::Impl {
  52. public:
  53. explicit Impl(Config config) : config_(std::move(config)) {
  54. if (config_.project.empty()) config_.project = "default";
  55. }
  56. // v2.3 — qualify a collection name with Config::project unless the
  57. // caller already supplied a "<project>:<collection>" qualified form.
  58. // The qualified form always wins. Server-side parseProjectCollection
  59. // is strict on edge cases (leading colon, double colon, etc.) so
  60. // we just delegate the validation to the server.
  61. std::string qualify(std::string_view collection) const {
  62. if (collection.find(':') != std::string_view::npos) {
  63. return std::string(collection);
  64. }
  65. // v2.7.0 — system collections (`_views`, `_collection_meta`,
  66. // `_policies`, `_files`) are GLOBAL, not per-project: their rows are
  67. // already keyed by a project-qualified id. Qualifying the collection
  68. // name too produced `default:_policies`, which does not exist, so
  69. // reads silently found nothing while writes succeeded (the server
  70. // tolerates both spellings on the policy write path). That asymmetry
  71. // is exactly how a management command can look like it worked and
  72. // then show stale state.
  73. if (!collection.empty() && collection.front() == '_') {
  74. return std::string(collection);
  75. }
  76. std::string out;
  77. out.reserve(config_.project.size() + 1 + collection.size());
  78. out.append(config_.project);
  79. out.push_back(':');
  80. out.append(collection.data(), collection.size());
  81. return out;
  82. }
  83. // Inverse of qualify() for values coming back off the wire. Strips only
  84. // OUR project's prefix, so a name from another namespace stays visibly
  85. // qualified rather than being silently flattened into ours.
  86. std::string unqualify(const std::string& qualified) const {
  87. const std::string prefix = config_.project + ":";
  88. if (qualified.rfind(prefix, 0) == 0) {
  89. return qualified.substr(prefix.size());
  90. }
  91. return qualified;
  92. }
  93. ~Impl() {
  94. disconnect();
  95. }
  96. bool connect() {
  97. try {
  98. auto channelArgs = grpc::ChannelArguments();
  99. // Use longer keepalive intervals to avoid "too_many_pings" errors from server
  100. channelArgs.SetInt(GRPC_ARG_KEEPALIVE_TIME_MS, 60000); // 60 seconds
  101. channelArgs.SetInt(GRPC_ARG_KEEPALIVE_TIMEOUT_MS, 20000); // 20 seconds
  102. channelArgs.SetInt(GRPC_ARG_KEEPALIVE_PERMIT_WITHOUT_CALLS, 0); // Only ping when there are active calls
  103. // Set max message size to match server (100MB for file uploads)
  104. channelArgs.SetMaxReceiveMessageSize(100 * 1024 * 1024);
  105. channelArgs.SetMaxSendMessageSize(100 * 1024 * 1024);
  106. std::shared_ptr<grpc::ChannelCredentials> creds = buildChannelCredentials();
  107. if (!creds) {
  108. spdlog::error("Database client connection failed: could not build channel credentials");
  109. return false;
  110. }
  111. channel_ = grpc::CreateCustomChannel(
  112. config_.address,
  113. creds,
  114. channelArgs
  115. );
  116. stub_ = smartbotic::databasepb::DatabaseService::NewStub(channel_);
  117. connected_ = true;
  118. const char* scheme = config_.tls_enabled ? "tls" : "plaintext";
  119. const char* authed = config_.auth_token.empty() ? "no-auth" : "bearer-auth";
  120. spdlog::info("Database client connected to {} ({}, {})",
  121. config_.address, scheme, authed);
  122. return true;
  123. } catch (const std::exception& e) {
  124. spdlog::error("Database client connection failed: {}", e.what());
  125. return false;
  126. }
  127. }
  128. // v2.4 — build the ChannelCredentials matching Config's TLS settings.
  129. //
  130. // Three cases:
  131. // 1. !tls_enabled → InsecureChannelCredentials
  132. // (v2.3 back-compat default).
  133. // 2. tls_enabled && !insecure_skip_verify → SslCredentials with
  134. // pem_root_certs read from tls_ca_cert_path, or empty (= system
  135. // trust roots) when the path is unset.
  136. // 3. tls_enabled && insecure_skip_verify → experimental
  137. // TlsChannelCredentialsOptions with NoOpCertificateVerifier and
  138. // set_verify_server_certs(false). Loud WARN on every connect.
  139. std::shared_ptr<grpc::ChannelCredentials> buildChannelCredentials() {
  140. if (!config_.tls_enabled) {
  141. return grpc::InsecureChannelCredentials();
  142. }
  143. if (config_.tls_insecure_skip_verify) {
  144. spdlog::warn(
  145. "Database client: connecting with tls_insecure_skip_verify — "
  146. "DEV ONLY. Server certificate will NOT be validated against any "
  147. "trust root. Drop the server's self-signed cert at "
  148. "tls_ca_cert_path for any production-shaped use.");
  149. // Route through gRPC's experimental TLS API. This is the path
  150. // gRPC documents for "accept any cert" — combine
  151. // NoOpCertificateVerifier with set_verify_server_certs(false).
  152. grpc::experimental::TlsChannelCredentialsOptions tlsOpts;
  153. tlsOpts.set_certificate_verifier(
  154. std::make_shared<grpc::experimental::NoOpCertificateVerifier>());
  155. tlsOpts.set_verify_server_certs(false);
  156. tlsOpts.set_check_call_host(false);
  157. auto creds = grpc::experimental::TlsCredentials(tlsOpts);
  158. if (!creds) {
  159. // No fallback that "trusts everything" without
  160. // experimental TLS — if the build/runtime can't produce
  161. // experimental TlsCredentials, refuse to silently degrade
  162. // to system-trust mode (which would reject the
  163. // self-signed cert anyway). The operator should bake the
  164. // server's self-signed cert into tls_ca_cert_path.
  165. spdlog::error(
  166. "Database client: tls_insecure_skip_verify requested but "
  167. "experimental TlsCredentials returned null. Set "
  168. "tls_ca_cert_path to the server's self-signed PEM "
  169. "instead.");
  170. return nullptr;
  171. }
  172. return creds;
  173. }
  174. grpc::SslCredentialsOptions sslOpts;
  175. if (!config_.tls_ca_cert_path.empty()) {
  176. sslOpts.pem_root_certs = readCaCertFile(config_.tls_ca_cert_path);
  177. if (sslOpts.pem_root_certs.empty()) {
  178. spdlog::error(
  179. "Database client: could not read tls_ca_cert_path '{}'",
  180. config_.tls_ca_cert_path);
  181. return nullptr;
  182. }
  183. }
  184. // Else leave pem_root_certs empty → gRPC uses system trust roots
  185. // (or GRPC_DEFAULT_SSL_ROOTS_FILE_PATH env var, per gRPC docs).
  186. return grpc::SslCredentials(sslOpts);
  187. }
  188. void disconnect() {
  189. connected_ = false;
  190. stub_.reset();
  191. channel_.reset();
  192. }
  193. bool isConnected() const {
  194. if (!connected_ || !channel_) {
  195. return false;
  196. }
  197. // Check if channel is in a usable state (not failed or shutdown)
  198. auto state = channel_->GetState(false);
  199. return state == GRPC_CHANNEL_READY ||
  200. state == GRPC_CHANNEL_IDLE ||
  201. state == GRPC_CHANNEL_CONNECTING;
  202. }
  203. // ===== Document Operations =====
  204. std::string insert(const std::string& collection, const nlohmann::json& data,
  205. const std::string& id, uint32_t ttlSeconds,
  206. const std::string& actor) {
  207. smartbotic::databasepb::InsertRequest request;
  208. request.set_collection(qualify(collection));
  209. request.set_data(data.dump());
  210. if (!id.empty()) {
  211. request.set_id(id);
  212. }
  213. if (ttlSeconds > 0) {
  214. request.set_ttl_seconds(ttlSeconds);
  215. }
  216. if (!actor.empty()) {
  217. request.set_actor(actor);
  218. }
  219. smartbotic::databasepb::InsertResponse response;
  220. grpc::Status status;
  221. for (uint32_t attempt = 0; attempt <= config_.writeRetries; ++attempt) {
  222. grpc::ClientContext context;
  223. setDeadline(context);
  224. response.Clear();
  225. status = stub_->Insert(&context, request, &response);
  226. if (status.ok() || !retryableStatus(status)) {
  227. break;
  228. }
  229. if (attempt < config_.writeRetries) {
  230. uint32_t backoffMs = computeBackoffMs(
  231. config_.writeRetryBackoffMs, attempt,
  232. config_.writeRetryMaxBackoffMs, config_.writeRetryJitter);
  233. spdlog::warn("Client::insert {}; retrying in {}ms (attempt {}/{})",
  234. status.error_message(), backoffMs,
  235. attempt + 1, config_.writeRetries);
  236. std::this_thread::sleep_for(std::chrono::milliseconds(backoffMs));
  237. }
  238. }
  239. if (!status.ok()) {
  240. spdlog::error("Client::insert failed after retries: {}", status.error_message());
  241. throw std::runtime_error(status.error_message());
  242. }
  243. return response.id();
  244. }
  245. std::optional<nlohmann::json> get(const std::string& collection, const std::string& id) {
  246. smartbotic::databasepb::GetRequest request;
  247. request.set_collection(qualify(collection));
  248. request.set_id(id);
  249. smartbotic::databasepb::GetResponse response;
  250. grpc::ClientContext context;
  251. setDeadline(context);
  252. auto status = stub_->Get(&context, request, &response);
  253. if (!status.ok()) {
  254. // v2.8.0 — a permission denial must NOT look like "no data".
  255. // Returning empty here would leave an application unable to tell
  256. // "you may not see this" from "there is nothing to see", so it
  257. // would silently take the wrong branch. Same reasoning as the
  258. // v2.4.2 change that stopped Find returning empty for a missing
  259. // collection.
  260. if (status.error_code() == grpc::StatusCode::PERMISSION_DENIED) {
  261. throw std::runtime_error("access denied: get");
  262. }
  263. spdlog::error("Client::get failed: {}", status.error_message());
  264. return std::nullopt;
  265. }
  266. if (!response.found()) {
  267. return std::nullopt;
  268. }
  269. auto json = nlohmann::json::parse(response.document().data());
  270. json["_id"] = response.document().id();
  271. json["_version"] = response.document().version();
  272. json["_created_at"] = response.document().created_at();
  273. json["_updated_at"] = response.document().updated_at();
  274. json["_created_by"] = response.document().created_by();
  275. json["_updated_by"] = response.document().updated_by();
  276. return json;
  277. }
  278. bool update(const std::string& collection, const std::string& id, const nlohmann::json& data,
  279. const std::string& actor) {
  280. // Strip metadata fields from data — callers pass the JSON from get() which
  281. // includes _version, _id, etc. We use _version for optimistic locking.
  282. auto cleanData = data;
  283. cleanData.erase("_id");
  284. cleanData.erase("_version");
  285. cleanData.erase("_created_at");
  286. cleanData.erase("_updated_at");
  287. cleanData.erase("_created_by");
  288. cleanData.erase("_updated_by");
  289. // Optimistic locking with automatic retry:
  290. // 1. Read current version
  291. // 2. Call updateIfVersion
  292. // 3. On version conflict (another writer), re-read and retry
  293. for (uint32_t attempt = 0; attempt <= config_.maxRetries; ++attempt) {
  294. // Get current version
  295. auto current = get(collection, id);
  296. if (!current) {
  297. return false; // document doesn't exist
  298. }
  299. uint64_t version = (*current)["_version"].get<uint64_t>();
  300. // Attempt version-checked update
  301. if (updateIfVersion(collection, id, cleanData, version, actor)) {
  302. return true;
  303. }
  304. // Version conflict — another client wrote between our get() and update
  305. if (attempt < config_.maxRetries) {
  306. spdlog::debug("Client::update version conflict on {}/{}, retry {}/{}",
  307. collection, id, attempt + 1, config_.maxRetries);
  308. }
  309. }
  310. spdlog::warn("Client::update failed after {} retries due to version conflicts on {}/{}",
  311. config_.maxRetries, collection, id);
  312. return false;
  313. }
  314. bool updateIfVersion(const std::string& collection, const std::string& id,
  315. const nlohmann::json& data, uint64_t expectedVersion,
  316. const std::string& actor) {
  317. smartbotic::databasepb::UpdateRequest request;
  318. request.set_collection(qualify(collection));
  319. request.set_id(id);
  320. request.set_data(data.dump());
  321. request.set_expected_version(expectedVersion);
  322. if (!actor.empty()) {
  323. request.set_actor(actor);
  324. }
  325. smartbotic::databasepb::UpdateResponse response;
  326. grpc::Status status;
  327. for (uint32_t attempt = 0; attempt <= config_.writeRetries; ++attempt) {
  328. grpc::ClientContext context;
  329. setDeadline(context);
  330. response.Clear();
  331. status = stub_->Update(&context, request, &response);
  332. if (status.ok() || !retryableStatus(status)) {
  333. break;
  334. }
  335. if (attempt < config_.writeRetries) {
  336. uint32_t backoffMs = computeBackoffMs(
  337. config_.writeRetryBackoffMs, attempt,
  338. config_.writeRetryMaxBackoffMs, config_.writeRetryJitter);
  339. spdlog::warn("Client::updateIfVersion {}; retrying in {}ms (attempt {}/{})",
  340. status.error_message(), backoffMs,
  341. attempt + 1, config_.writeRetries);
  342. std::this_thread::sleep_for(std::chrono::milliseconds(backoffMs));
  343. }
  344. }
  345. if (!status.ok()) {
  346. spdlog::error("Client::updateIfVersion failed after retries: {}", status.error_message());
  347. return false;
  348. }
  349. return response.success();
  350. }
  351. uint64_t patch(const std::string& collection, const std::string& id,
  352. const nlohmann::json& fields, const std::string& actor,
  353. std::optional<uint32_t> ttlSeconds = std::nullopt) {
  354. smartbotic::databasepb::PatchDocumentRequest request;
  355. request.set_collection(qualify(collection));
  356. // v2.8.0 — presence distinguishes "leave the expiry alone" (absent) from
  357. // "clear it" (0) from "set it" (N).
  358. if (ttlSeconds.has_value()) request.set_ttl_seconds(*ttlSeconds);
  359. request.set_id(id);
  360. request.set_patch_json(fields.dump());
  361. if (!actor.empty()) {
  362. request.set_actor(actor);
  363. }
  364. smartbotic::databasepb::PatchDocumentResponse response;
  365. grpc::Status status;
  366. for (uint32_t attempt = 0; attempt <= config_.writeRetries; ++attempt) {
  367. grpc::ClientContext context;
  368. setDeadline(context);
  369. response.Clear();
  370. status = stub_->PatchDocument(&context, request, &response);
  371. if (status.ok() || !retryableStatus(status)) {
  372. break;
  373. }
  374. if (attempt < config_.writeRetries) {
  375. uint32_t backoffMs = computeBackoffMs(
  376. config_.writeRetryBackoffMs, attempt,
  377. config_.writeRetryMaxBackoffMs, config_.writeRetryJitter);
  378. spdlog::warn("Client::patch {}; retrying in {}ms (attempt {}/{})",
  379. status.error_message(), backoffMs,
  380. attempt + 1, config_.writeRetries);
  381. std::this_thread::sleep_for(std::chrono::milliseconds(backoffMs));
  382. }
  383. }
  384. if (!status.ok()) {
  385. spdlog::error("Client::patch failed after retries: {}", status.error_message());
  386. return 0;
  387. }
  388. if (!response.success()) {
  389. spdlog::error("Client::patch failed: {}", response.error());
  390. return 0;
  391. }
  392. return response.new_version();
  393. }
  394. std::pair<std::string, bool> upsert(const std::string& collection, const nlohmann::json& data,
  395. const std::string& id, uint32_t ttlSeconds,
  396. const std::string& actor) {
  397. smartbotic::databasepb::UpsertRequest request;
  398. request.set_collection(qualify(collection));
  399. request.set_data(data.dump());
  400. if (!id.empty()) {
  401. request.set_id(id);
  402. }
  403. if (ttlSeconds > 0) {
  404. request.set_ttl_seconds(ttlSeconds);
  405. }
  406. if (!actor.empty()) {
  407. request.set_actor(actor);
  408. }
  409. smartbotic::databasepb::UpsertResponse response;
  410. grpc::Status status;
  411. for (uint32_t attempt = 0; attempt <= config_.writeRetries; ++attempt) {
  412. grpc::ClientContext context;
  413. setDeadline(context);
  414. response.Clear();
  415. status = stub_->Upsert(&context, request, &response);
  416. if (status.ok() || !retryableStatus(status)) {
  417. break;
  418. }
  419. if (attempt < config_.writeRetries) {
  420. uint32_t backoffMs = computeBackoffMs(
  421. config_.writeRetryBackoffMs, attempt,
  422. config_.writeRetryMaxBackoffMs, config_.writeRetryJitter);
  423. spdlog::warn("Client::upsert {}; retrying in {}ms (attempt {}/{})",
  424. status.error_message(), backoffMs,
  425. attempt + 1, config_.writeRetries);
  426. std::this_thread::sleep_for(std::chrono::milliseconds(backoffMs));
  427. }
  428. }
  429. if (!status.ok()) {
  430. spdlog::error("Client::upsert failed after retries: {}", status.error_message());
  431. throw std::runtime_error(status.error_message());
  432. }
  433. return {response.id(), response.inserted()};
  434. }
  435. bool remove(const std::string& collection, const std::string& id) {
  436. smartbotic::databasepb::DeleteRequest request;
  437. request.set_collection(qualify(collection));
  438. request.set_id(id);
  439. smartbotic::databasepb::DeleteResponse response;
  440. grpc::Status status;
  441. for (uint32_t attempt = 0; attempt <= config_.writeRetries; ++attempt) {
  442. grpc::ClientContext context;
  443. setDeadline(context);
  444. response.Clear();
  445. status = stub_->Delete(&context, request, &response);
  446. if (status.ok() || !retryableStatus(status)) {
  447. break;
  448. }
  449. if (attempt < config_.writeRetries) {
  450. uint32_t backoffMs = computeBackoffMs(
  451. config_.writeRetryBackoffMs, attempt,
  452. config_.writeRetryMaxBackoffMs, config_.writeRetryJitter);
  453. spdlog::warn("Client::remove {}; retrying in {}ms (attempt {}/{})",
  454. status.error_message(), backoffMs,
  455. attempt + 1, config_.writeRetries);
  456. std::this_thread::sleep_for(std::chrono::milliseconds(backoffMs));
  457. }
  458. }
  459. if (!status.ok()) {
  460. spdlog::error("Client::remove failed after retries: {}", status.error_message());
  461. return false;
  462. }
  463. return response.deleted();
  464. }
  465. bool exists(const std::string& collection, const std::string& id) {
  466. smartbotic::databasepb::ExistsRequest request;
  467. request.set_collection(qualify(collection));
  468. request.set_id(id);
  469. smartbotic::databasepb::ExistsResponse response;
  470. grpc::ClientContext context;
  471. setDeadline(context);
  472. auto status = stub_->Exists(&context, request, &response);
  473. if (!status.ok()) {
  474. // v2.8.0 — a permission denial must NOT look like "no data".
  475. // Returning empty here would leave an application unable to tell
  476. // "you may not see this" from "there is nothing to see", so it
  477. // would silently take the wrong branch. Same reasoning as the
  478. // v2.4.2 change that stopped Find returning empty for a missing
  479. // collection.
  480. if (status.error_code() == grpc::StatusCode::PERMISSION_DENIED) {
  481. throw std::runtime_error("access denied: exists");
  482. }
  483. spdlog::error("Client::exists failed: {}", status.error_message());
  484. return false;
  485. }
  486. return response.exists();
  487. }
  488. // ===== Query Operations =====
  489. std::vector<nlohmann::json> find(const std::string& collection,
  490. const Client::QueryOptions& options,
  491. const std::vector<std::string>& projection = {}) {
  492. smartbotic::databasepb::FindRequest request;
  493. request.set_collection(qualify(collection));
  494. // Set filters
  495. for (const auto& filter : options.filters) {
  496. auto* pb = request.add_filters();
  497. pb->set_field(filter.field);
  498. pb->set_value(filter.value.dump());
  499. pb->set_op(static_cast<smartbotic::databasepb::FilterOp>(filter.op));
  500. }
  501. // Set sorting
  502. if (!options.sortField.empty()) {
  503. auto* sort = request.mutable_sort();
  504. sort->set_field(options.sortField);
  505. sort->set_descending(options.sortDescending);
  506. }
  507. // Set pagination
  508. request.set_limit(options.limit);
  509. request.set_offset(options.offset);
  510. // v2.7.1 — field projection, passed alongside QueryOptions rather than
  511. // inside it: adding a member would change the struct size, and with an
  512. // unchanged soname an installed consumer crashes. See client.hpp.
  513. for (const auto& f : projection) {
  514. request.add_projection(f);
  515. }
  516. smartbotic::databasepb::FindResponse response;
  517. grpc::ClientContext context;
  518. setDeadline(context);
  519. auto status = stub_->Find(&context, request, &response);
  520. if (status.error_code() == grpc::StatusCode::NOT_FOUND) {
  521. // Surfaced rather than logged: an unknown collection or view is a
  522. // caller bug, and returning {} here is indistinguishable from a
  523. // legitimately empty result. Other failures keep the old
  524. // log-and-return-empty behaviour.
  525. throw std::runtime_error(status.error_message());
  526. }
  527. if (!status.ok()) {
  528. // v2.8.0 — a permission denial must NOT look like "no data".
  529. // Returning empty here would leave an application unable to tell
  530. // "you may not see this" from "there is nothing to see", so it
  531. // would silently take the wrong branch. Same reasoning as the
  532. // v2.4.2 change that stopped Find returning empty for a missing
  533. // collection.
  534. if (status.error_code() == grpc::StatusCode::PERMISSION_DENIED) {
  535. throw std::runtime_error("access denied: find");
  536. }
  537. spdlog::error("Client::find failed: {}", status.error_message());
  538. return {};
  539. }
  540. std::vector<nlohmann::json> results;
  541. results.reserve(response.documents_size());
  542. for (const auto& doc : response.documents()) {
  543. auto json = nlohmann::json::parse(doc.data());
  544. json["_id"] = doc.id();
  545. json["_version"] = doc.version();
  546. json["_created_at"] = doc.created_at();
  547. json["_updated_at"] = doc.updated_at();
  548. json["_created_by"] = doc.created_by();
  549. json["_updated_by"] = doc.updated_by();
  550. results.push_back(json);
  551. }
  552. return results;
  553. }
  554. Client::FindResult findWithMetrics(const std::string& collection,
  555. const Client::QueryOptions& options,
  556. const std::vector<std::string>& projection = {}) {
  557. smartbotic::databasepb::FindRequest request;
  558. request.set_collection(qualify(collection));
  559. // Set filters
  560. for (const auto& filter : options.filters) {
  561. auto* pb = request.add_filters();
  562. pb->set_field(filter.field);
  563. pb->set_value(filter.value.dump());
  564. pb->set_op(static_cast<smartbotic::databasepb::FilterOp>(filter.op));
  565. }
  566. // Set sorting
  567. if (!options.sortField.empty()) {
  568. auto* sort = request.mutable_sort();
  569. sort->set_field(options.sortField);
  570. sort->set_descending(options.sortDescending);
  571. }
  572. // Set pagination
  573. request.set_limit(options.limit);
  574. request.set_offset(options.offset);
  575. // v2.7.1 — field projection, passed alongside QueryOptions rather than
  576. // inside it: adding a member would change the struct size, and with an
  577. // unchanged soname an installed consumer crashes. See client.hpp.
  578. for (const auto& f : projection) {
  579. request.add_projection(f);
  580. }
  581. smartbotic::databasepb::FindResponse response;
  582. grpc::ClientContext context;
  583. setDeadline(context);
  584. auto status = stub_->Find(&context, request, &response);
  585. if (status.error_code() == grpc::StatusCode::NOT_FOUND) {
  586. throw std::runtime_error(status.error_message()); // see find()
  587. }
  588. if (!status.ok()) {
  589. // v2.8.0 — a permission denial must NOT look like "no data".
  590. // Returning empty here would leave an application unable to tell
  591. // "you may not see this" from "there is nothing to see", so it
  592. // would silently take the wrong branch. Same reasoning as the
  593. // v2.4.2 change that stopped Find returning empty for a missing
  594. // collection.
  595. if (status.error_code() == grpc::StatusCode::PERMISSION_DENIED) {
  596. throw std::runtime_error("access denied: findWithMetrics");
  597. }
  598. spdlog::error("Client::findWithMetrics failed: {}", status.error_message());
  599. return {};
  600. }
  601. Client::FindResult result;
  602. result.totalCount = response.total_count();
  603. result.hasMore = response.has_more();
  604. result.usedWalFallback = response.used_wal_fallback();
  605. result.memoryMatchCount = response.memory_match_count();
  606. result.walMatchCount = response.wal_match_count();
  607. result.memorySearchMicros = response.memory_search_micros();
  608. result.walSearchMicros = response.wal_search_micros();
  609. result.documents.reserve(response.documents_size());
  610. for (const auto& doc : response.documents()) {
  611. auto json = nlohmann::json::parse(doc.data());
  612. json["_id"] = doc.id();
  613. json["_version"] = doc.version();
  614. json["_created_at"] = doc.created_at();
  615. json["_updated_at"] = doc.updated_at();
  616. json["_created_by"] = doc.created_by();
  617. json["_updated_by"] = doc.updated_by();
  618. result.documents.push_back(json);
  619. }
  620. return result;
  621. }
  622. uint64_t count(const std::string& collection,
  623. const std::vector<Client::Filter>& filters) {
  624. smartbotic::databasepb::CountRequest request;
  625. request.set_collection(qualify(collection));
  626. for (const auto& filter : filters) {
  627. auto* pb = request.add_filters();
  628. pb->set_field(filter.field);
  629. pb->set_value(filter.value.dump());
  630. pb->set_op(static_cast<smartbotic::databasepb::FilterOp>(filter.op));
  631. }
  632. smartbotic::databasepb::CountResponse response;
  633. grpc::ClientContext context;
  634. setDeadline(context);
  635. auto status = stub_->Count(&context, request, &response);
  636. if (!status.ok()) {
  637. // v2.8.0 — a permission denial must NOT look like "no data".
  638. // Returning empty here would leave an application unable to tell
  639. // "you may not see this" from "there is nothing to see", so it
  640. // would silently take the wrong branch. Same reasoning as the
  641. // v2.4.2 change that stopped Find returning empty for a missing
  642. // collection.
  643. if (status.error_code() == grpc::StatusCode::PERMISSION_DENIED) {
  644. throw std::runtime_error("access denied: count");
  645. }
  646. spdlog::error("Client::count failed: {}", status.error_message());
  647. return 0;
  648. }
  649. return response.count();
  650. }
  651. // ===== Set Operations =====
  652. bool setAdd(const std::string& collection, const std::string& setId, const std::string& member) {
  653. smartbotic::databasepb::SetAddRequest request;
  654. request.set_collection(qualify(collection));
  655. request.set_set_id(setId);
  656. request.set_member(member);
  657. smartbotic::databasepb::SetAddResponse response;
  658. grpc::ClientContext context;
  659. setDeadline(context);
  660. auto status = stub_->SetAdd(&context, request, &response);
  661. if (!status.ok()) {
  662. spdlog::error("Client::setAdd failed: {}", status.error_message());
  663. return false;
  664. }
  665. return response.added();
  666. }
  667. bool setRemove(const std::string& collection, const std::string& setId, const std::string& member) {
  668. smartbotic::databasepb::SetRemoveRequest request;
  669. request.set_collection(qualify(collection));
  670. request.set_set_id(setId);
  671. request.set_member(member);
  672. smartbotic::databasepb::SetRemoveResponse response;
  673. grpc::ClientContext context;
  674. setDeadline(context);
  675. auto status = stub_->SetRemove(&context, request, &response);
  676. if (!status.ok()) {
  677. spdlog::error("Client::setRemove failed: {}", status.error_message());
  678. return false;
  679. }
  680. return response.removed();
  681. }
  682. std::vector<std::string> setMembers(const std::string& collection, const std::string& setId) {
  683. smartbotic::databasepb::SetMembersRequest request;
  684. request.set_collection(qualify(collection));
  685. request.set_set_id(setId);
  686. smartbotic::databasepb::SetMembersResponse response;
  687. grpc::ClientContext context;
  688. setDeadline(context);
  689. auto status = stub_->SetMembers(&context, request, &response);
  690. if (!status.ok()) {
  691. spdlog::error("Client::setMembers failed: {}", status.error_message());
  692. return {};
  693. }
  694. return {response.members().begin(), response.members().end()};
  695. }
  696. bool setIsMember(const std::string& collection, const std::string& setId, const std::string& member) {
  697. smartbotic::databasepb::SetIsMemberRequest request;
  698. request.set_collection(qualify(collection));
  699. request.set_set_id(setId);
  700. request.set_member(member);
  701. smartbotic::databasepb::SetIsMemberResponse response;
  702. grpc::ClientContext context;
  703. setDeadline(context);
  704. auto status = stub_->SetIsMember(&context, request, &response);
  705. if (!status.ok()) {
  706. spdlog::error("Client::setIsMember failed: {}", status.error_message());
  707. return false;
  708. }
  709. return response.is_member();
  710. }
  711. // ===== Collection Management =====
  712. std::vector<Client::SimilarityResult> similaritySearch(
  713. const std::string& collection, const std::vector<float>& queryVector,
  714. uint32_t topK, float minScore) {
  715. smartbotic::databasepb::SimilaritySearchRequest request;
  716. request.set_collection(qualify(collection));
  717. for (float v : queryVector) {
  718. request.add_query_vector(v);
  719. }
  720. request.set_top_k(topK);
  721. request.set_min_score(minScore);
  722. smartbotic::databasepb::SimilaritySearchResponse response;
  723. grpc::ClientContext context;
  724. setDeadline(context);
  725. auto status = stub_->SimilaritySearch(&context, request, &response);
  726. if (!status.ok()) {
  727. spdlog::error("Client::similaritySearch failed: {}", status.error_message());
  728. throw std::runtime_error(status.error_message());
  729. }
  730. std::vector<Client::SimilarityResult> results;
  731. results.reserve(response.results_size());
  732. for (const auto& r : response.results()) {
  733. Client::SimilarityResult entry;
  734. entry.id = r.id();
  735. entry.score = r.score();
  736. if (!r.data().empty()) {
  737. entry.data = nlohmann::json::parse(r.data());
  738. }
  739. results.push_back(std::move(entry));
  740. }
  741. return results;
  742. }
  743. bool createCollection(const std::string& name, uint32_t defaultTtlSeconds, bool encrypted,
  744. uint32_t maxVersions, uint32_t vectorDimension) {
  745. smartbotic::databasepb::CreateCollectionRequest request;
  746. // v2.4.5 — MUST qualify. Sending a bare name here while insert/get/
  747. // find all qualify created TWO collections from one call site: the
  748. // bare one received the options (encrypted, maxVersions,
  749. // vectorDimension, TTL) and stayed empty, while the qualified one
  750. // was created implicitly by the first insert with DEFAULTS. Silent,
  751. // and unrecoverable for vectorDimension, which is immutable after
  752. // creation. Same class of bug as the v2.4.2 createView break.
  753. request.set_name(qualify(name));
  754. auto* options = request.mutable_options();
  755. if (defaultTtlSeconds > 0) {
  756. options->set_default_ttl_seconds(defaultTtlSeconds);
  757. }
  758. options->set_encrypted(encrypted);
  759. if (maxVersions > 0) {
  760. options->set_max_versions(maxVersions);
  761. }
  762. if (vectorDimension > 0) {
  763. options->set_vector_dimension(vectorDimension);
  764. }
  765. smartbotic::databasepb::CreateCollectionResponse response;
  766. grpc::ClientContext context;
  767. setDeadline(context);
  768. auto status = stub_->CreateCollection(&context, request, &response);
  769. if (!status.ok()) {
  770. spdlog::error("Client::createCollection failed: {}", status.error_message());
  771. return false;
  772. }
  773. return response.created();
  774. }
  775. bool dropCollection(const std::string& name) {
  776. smartbotic::databasepb::DropCollectionRequest request;
  777. // v2.4.5 — MUST qualify. Unqualified, this dropped the empty phantom
  778. // created by createCollection and returned TRUE while every document
  779. // in <project>:<name> remained readable — a deletion request that
  780. // reports success and deletes nothing.
  781. request.set_name(qualify(name));
  782. smartbotic::databasepb::DropCollectionResponse response;
  783. grpc::ClientContext context;
  784. setDeadline(context);
  785. auto status = stub_->DropCollection(&context, request, &response);
  786. if (!status.ok()) {
  787. spdlog::error("Client::dropCollection failed: {}", status.error_message());
  788. return false;
  789. }
  790. return response.dropped();
  791. }
  792. std::vector<std::string> listCollections() {
  793. smartbotic::databasepb::ListCollectionsRequest request;
  794. // v2.8.0 — scope the listing to this client's project, so a workspace
  795. // never sees another workspace's collection names.
  796. request.set_project(config_.project);
  797. smartbotic::databasepb::ListCollectionsResponse response;
  798. grpc::ClientContext context;
  799. setDeadline(context);
  800. auto status = stub_->ListCollections(&context, request, &response);
  801. if (!status.ok()) {
  802. spdlog::error("Client::listCollections failed: {}", status.error_message());
  803. return {};
  804. }
  805. // v2.4.5 — round-trip the naming: callers pass bare names in, so they
  806. // get bare names back for their OWN project. Names belonging to other
  807. // projects stay visibly qualified rather than being flattened into
  808. // ours (same rule as listViews). Note the server still returns every
  809. // project's collections; ListCollectionsRequest has no project filter
  810. // yet, unlike ListViewsRequest.
  811. std::vector<std::string> out;
  812. out.reserve(static_cast<size_t>(response.names_size()));
  813. for (const auto& n : response.names()) {
  814. out.push_back(unqualify(n));
  815. }
  816. return out;
  817. }
  818. std::optional<Client::CollectionInfo> getCollectionInfo(const std::string& name) {
  819. smartbotic::databasepb::GetCollectionInfoRequest request;
  820. // v2.4.5 — MUST qualify, else this reports on the phantom (0 docs)
  821. // rather than the collection the caller reads and writes.
  822. request.set_name(qualify(name));
  823. smartbotic::databasepb::GetCollectionInfoResponse response;
  824. grpc::ClientContext context;
  825. setDeadline(context);
  826. auto status = stub_->GetCollectionInfo(&context, request, &response);
  827. if (!status.ok()) {
  828. spdlog::error("Client::getCollectionInfo failed: {}", status.error_message());
  829. return std::nullopt;
  830. }
  831. if (!response.found()) {
  832. return std::nullopt;
  833. }
  834. const auto& info = response.info();
  835. Client::CollectionInfo result;
  836. result.name = info.name();
  837. result.documentCount = info.document_count();
  838. result.sizeBytes = info.size_bytes();
  839. result.defaultTtlSeconds = info.options().default_ttl_seconds();
  840. result.encrypted = info.options().encrypted();
  841. result.maxVersions = info.options().max_versions();
  842. result.createdAt = info.created_at();
  843. result.updatedAt = info.updated_at();
  844. return result;
  845. }
  846. // ===== Project Management (v2.3 Stage F) =====
  847. std::vector<std::string> listProjects() {
  848. smartbotic::databasepb::ListProjectsRequest request;
  849. smartbotic::databasepb::ListProjectsResponse response;
  850. grpc::ClientContext context;
  851. setDeadline(context);
  852. auto status = stub_->ListProjects(&context, request, &response);
  853. if (!status.ok()) {
  854. spdlog::error("Client::listProjects failed: {}", status.error_message());
  855. return {};
  856. }
  857. return {response.projects().begin(), response.projects().end()};
  858. }
  859. bool createProject(const std::string& name) {
  860. smartbotic::databasepb::CreateProjectRequest request;
  861. request.set_name(name);
  862. smartbotic::databasepb::CreateProjectResponse response;
  863. grpc::ClientContext context;
  864. setDeadline(context);
  865. auto status = stub_->CreateProject(&context, request, &response);
  866. if (!status.ok()) {
  867. spdlog::error("Client::createProject failed: {}", status.error_message());
  868. throw std::runtime_error("createProject transport failure: " +
  869. status.error_message());
  870. }
  871. if (!response.error().empty()) {
  872. // Validation error (invalid name etc.) — surfaced via the
  873. // response field, not the gRPC status.
  874. throw std::runtime_error("createProject('" + name + "') rejected: " +
  875. response.error());
  876. }
  877. return response.created();
  878. }
  879. void dropProject(const std::string& name) {
  880. smartbotic::databasepb::DropProjectRequest request;
  881. request.set_name(name);
  882. smartbotic::databasepb::DropProjectResponse response;
  883. grpc::ClientContext context;
  884. setDeadline(context);
  885. auto status = stub_->DropProject(&context, request, &response);
  886. if (!status.ok()) {
  887. spdlog::error("Client::dropProject failed: {}", status.error_message());
  888. throw std::runtime_error("dropProject transport failure: " +
  889. status.error_message());
  890. }
  891. if (!response.error().empty()) {
  892. throw std::runtime_error("dropProject('" + name + "') refused: " +
  893. response.error());
  894. }
  895. // dropped() may still be false here defensively — but the
  896. // server-side contract is dropped==true iff error is empty.
  897. }
  898. // ===== Collection Configuration =====
  899. bool configureCollection(const std::string& collection, const Client::CollectionConfig& cfg) {
  900. smartbotic::databasepb::ConfigureCollectionRequest request;
  901. request.set_collection(qualify(collection));
  902. request.mutable_config()->set_timestamp_precision(cfg.timestampPrecision);
  903. // v2.4.5 — only send versioning when the caller actually set it, so
  904. // the server's partial update leaves it alone otherwise.
  905. if (cfg.versioningEnabled.has_value()) {
  906. request.mutable_config()->set_versioning_enabled(*cfg.versioningEnabled);
  907. }
  908. smartbotic::databasepb::ConfigureCollectionResponse response;
  909. grpc::ClientContext context;
  910. setDeadline(context);
  911. auto status = stub_->ConfigureCollection(&context, request, &response);
  912. if (!status.ok()) {
  913. spdlog::error("Client::configureCollection failed: {}", status.error_message());
  914. return false;
  915. }
  916. if (!response.success()) {
  917. spdlog::error("Client::configureCollection rejected: {}", response.error());
  918. return false;
  919. }
  920. return true;
  921. }
  922. Client::CollectionConfig getCollectionConfig(const std::string& collection) {
  923. smartbotic::databasepb::GetCollectionConfigRequest request;
  924. request.set_collection(qualify(collection));
  925. smartbotic::databasepb::GetCollectionConfigResponse response;
  926. grpc::ClientContext context;
  927. setDeadline(context);
  928. Client::CollectionConfig out;
  929. auto status = stub_->GetCollectionConfig(&context, request, &response);
  930. if (!status.ok()) {
  931. spdlog::error("Client::getCollectionConfig failed: {}", status.error_message());
  932. return out;
  933. }
  934. out.timestampPrecision = response.config().timestamp_precision();
  935. if (out.timestampPrecision.empty()) out.timestampPrecision = "ms";
  936. // v2.4.5 — always populated on read, so callers get the effective
  937. // value rather than the "leave unchanged" sentinel. A server older
  938. // than 2.4.5 omits the field; treat that as versioning on, which is
  939. // what those builds always did.
  940. out.versioningEnabled = response.config().has_versioning_enabled()
  941. ? response.config().versioning_enabled()
  942. : true;
  943. return out;
  944. }
  945. bool hasCollectionConfig(const std::string& collection) {
  946. smartbotic::databasepb::GetCollectionConfigRequest request;
  947. request.set_collection(qualify(collection));
  948. smartbotic::databasepb::GetCollectionConfigResponse response;
  949. grpc::ClientContext context;
  950. setDeadline(context);
  951. auto status = stub_->GetCollectionConfig(&context, request, &response);
  952. if (!status.ok()) return false;
  953. return response.found();
  954. }
  955. Client::TimestampMigrationResult migrateCollectionTimestamps(
  956. const std::string& collection,
  957. const std::string& fromPrecision,
  958. const std::string& toPrecision
  959. ) {
  960. smartbotic::databasepb::MigrateCollectionTimestampsRequest request;
  961. request.set_collection(qualify(collection));
  962. request.set_from_precision(fromPrecision);
  963. request.set_to_precision(toPrecision);
  964. smartbotic::databasepb::MigrateCollectionTimestampsResponse response;
  965. grpc::ClientContext context;
  966. // Longer deadline — can iterate many rows
  967. auto deadline = std::chrono::system_clock::now() + std::chrono::minutes(10);
  968. context.set_deadline(deadline);
  969. attachAuth(context);
  970. Client::TimestampMigrationResult out;
  971. auto status = stub_->MigrateCollectionTimestamps(&context, request, &response);
  972. if (!status.ok()) {
  973. out.success = false;
  974. out.error = status.error_message();
  975. return out;
  976. }
  977. out.success = response.success();
  978. out.error = response.error();
  979. out.rowsMigrated = response.rows_migrated();
  980. out.rowsSkipped = response.rows_skipped();
  981. return out;
  982. }
  983. // ===== View Management =====
  984. bool createView(const std::string& name,
  985. const std::string& collection,
  986. const std::vector<std::string>& include,
  987. const std::vector<std::string>& exclude,
  988. const std::vector<Client::Filter>& where,
  989. const std::optional<Client::Sort>& defaultSort) {
  990. smartbotic::databasepb::CreateViewRequest request;
  991. // Both fields are namespaced. Qualifying `collection` but not `name`
  992. // is what broke view lookups in v2.3: the registry keyed on the bare
  993. // name while every read path sent the qualified one, so the keys
  994. // could never meet.
  995. request.set_name(qualify(name));
  996. request.set_collection(qualify(collection));
  997. for (const auto& p : include) request.add_include(p);
  998. for (const auto& p : exclude) request.add_exclude(p);
  999. for (const auto& f : where) {
  1000. auto* pb = request.add_where();
  1001. pb->set_field(f.field);
  1002. pb->set_op(static_cast<smartbotic::databasepb::FilterOp>(f.op));
  1003. pb->set_value(f.value.dump());
  1004. }
  1005. if (defaultSort) {
  1006. auto* s = request.mutable_default_sort();
  1007. s->set_field(defaultSort->field);
  1008. s->set_descending(defaultSort->descending);
  1009. }
  1010. smartbotic::databasepb::CreateViewResponse response;
  1011. grpc::ClientContext context;
  1012. setDeadline(context);
  1013. auto status = stub_->CreateView(&context, request, &response);
  1014. if (!status.ok()) {
  1015. spdlog::error("Client::createView failed: {}", status.error_message());
  1016. return false;
  1017. }
  1018. if (!response.success()) {
  1019. spdlog::error("Client::createView rejected: {}", response.error());
  1020. return false;
  1021. }
  1022. return true;
  1023. }
  1024. // v2.9.0 — secondary indexes.
  1025. bool createIndex(const std::string& collection, const std::string& field,
  1026. uint64_t* rowsIndexed) {
  1027. smartbotic::databasepb::CreateIndexRequest request;
  1028. request.set_collection(qualify(collection));
  1029. request.set_field(field);
  1030. smartbotic::databasepb::CreateIndexResponse response;
  1031. grpc::ClientContext context;
  1032. setDeadline(context);
  1033. auto status = stub_->CreateIndex(&context, request, &response);
  1034. if (!status.ok()) {
  1035. spdlog::error("Client::createIndex failed: {}", status.error_message());
  1036. return false;
  1037. }
  1038. if (!response.success()) {
  1039. spdlog::error("Client::createIndex rejected: {}", response.error());
  1040. return false;
  1041. }
  1042. if (rowsIndexed != nullptr) *rowsIndexed = response.rows_indexed();
  1043. return true;
  1044. }
  1045. bool dropIndex(const std::string& collection, const std::string& field) {
  1046. smartbotic::databasepb::DropIndexRequest request;
  1047. request.set_collection(qualify(collection));
  1048. request.set_field(field);
  1049. smartbotic::databasepb::DropIndexResponse response;
  1050. grpc::ClientContext context;
  1051. setDeadline(context);
  1052. auto status = stub_->DropIndex(&context, request, &response);
  1053. if (!status.ok()) {
  1054. spdlog::error("Client::dropIndex failed: {}", status.error_message());
  1055. return false;
  1056. }
  1057. return response.success();
  1058. }
  1059. std::vector<Client::IndexDefinition> listIndexes(const std::string& collection) {
  1060. smartbotic::databasepb::ListIndexesRequest request;
  1061. request.set_collection(qualify(collection));
  1062. smartbotic::databasepb::ListIndexesResponse response;
  1063. grpc::ClientContext context;
  1064. setDeadline(context);
  1065. std::vector<Client::IndexDefinition> out;
  1066. auto status = stub_->ListIndexes(&context, request, &response);
  1067. if (!status.ok()) {
  1068. spdlog::error("Client::listIndexes failed: {}", status.error_message());
  1069. return out;
  1070. }
  1071. out.reserve(response.indexes_size());
  1072. for (const auto& i : response.indexes()) {
  1073. Client::IndexDefinition d;
  1074. d.field = i.field();
  1075. d.distinctValues = i.distinct_values();
  1076. d.entries = i.entries();
  1077. out.push_back(std::move(d));
  1078. }
  1079. return out;
  1080. }
  1081. std::vector<Client::IndexValue> indexValues(const std::string& collection,
  1082. const std::string& field,
  1083. uint32_t limit, bool ascending) {
  1084. smartbotic::databasepb::GetIndexValuesRequest request;
  1085. request.set_collection(qualify(collection));
  1086. request.set_field(field);
  1087. request.set_limit(limit);
  1088. request.set_ascending(ascending);
  1089. smartbotic::databasepb::GetIndexValuesResponse response;
  1090. grpc::ClientContext context;
  1091. setDeadline(context);
  1092. std::vector<Client::IndexValue> out;
  1093. auto status = stub_->GetIndexValues(&context, request, &response);
  1094. if (!status.ok()) {
  1095. spdlog::error("Client::indexValues failed: {}", status.error_message());
  1096. return out;
  1097. }
  1098. out.reserve(response.values_size());
  1099. for (const auto& v : response.values()) {
  1100. Client::IndexValue iv;
  1101. try {
  1102. iv.value = nlohmann::json::parse(v.value());
  1103. } catch (const nlohmann::json::exception&) {
  1104. continue;
  1105. }
  1106. iv.count = v.count();
  1107. out.push_back(std::move(iv));
  1108. }
  1109. return out;
  1110. }
  1111. bool dropView(const std::string& name) {
  1112. smartbotic::databasepb::DropViewRequest request;
  1113. request.set_name(qualify(name));
  1114. smartbotic::databasepb::DropViewResponse response;
  1115. grpc::ClientContext context;
  1116. setDeadline(context);
  1117. auto status = stub_->DropView(&context, request, &response);
  1118. if (!status.ok()) {
  1119. spdlog::error("Client::dropView failed: {}", status.error_message());
  1120. return false;
  1121. }
  1122. return response.success();
  1123. }
  1124. std::vector<Client::ViewDefinition> listViews() {
  1125. smartbotic::databasepb::ListViewsRequest request;
  1126. // Scope the listing to this client's workspace.
  1127. request.set_project(config_.project);
  1128. smartbotic::databasepb::ListViewsResponse response;
  1129. grpc::ClientContext context;
  1130. setDeadline(context);
  1131. std::vector<Client::ViewDefinition> out;
  1132. auto status = stub_->ListViews(&context, request, &response);
  1133. if (!status.ok()) {
  1134. spdlog::error("Client::listViews failed: {}", status.error_message());
  1135. return out;
  1136. }
  1137. for (const auto& pb : response.views()) {
  1138. Client::ViewDefinition v;
  1139. // Round-trip: the caller created "adults", so list it as "adults".
  1140. v.name = unqualify(pb.name());
  1141. v.collection = unqualify(pb.collection());
  1142. for (const auto& p : pb.include()) v.include.push_back(p);
  1143. for (const auto& p : pb.exclude()) v.exclude.push_back(p);
  1144. for (const auto& pbf : pb.where()) {
  1145. Client::Filter f;
  1146. f.field = pbf.field();
  1147. f.op = static_cast<Client::FilterOp>(pbf.op());
  1148. try { f.value = nlohmann::json::parse(pbf.value()); }
  1149. catch (...) { f.value = pbf.value(); }
  1150. v.where.push_back(f);
  1151. }
  1152. if (pb.has_default_sort()) {
  1153. Client::Sort s;
  1154. s.field = pb.default_sort().field();
  1155. s.descending = pb.default_sort().descending();
  1156. v.defaultSort = s;
  1157. }
  1158. v.createdAt = pb.created_at();
  1159. v.updatedAt = pb.updated_at();
  1160. out.push_back(std::move(v));
  1161. }
  1162. return out;
  1163. }
  1164. std::optional<Client::ViewDefinition> getViewInfo(const std::string& name) {
  1165. smartbotic::databasepb::GetViewInfoRequest request;
  1166. request.set_name(qualify(name));
  1167. smartbotic::databasepb::GetViewInfoResponse response;
  1168. grpc::ClientContext context;
  1169. setDeadline(context);
  1170. auto status = stub_->GetViewInfo(&context, request, &response);
  1171. if (!status.ok() || !response.found()) {
  1172. return std::nullopt;
  1173. }
  1174. Client::ViewDefinition v;
  1175. v.name = unqualify(response.view().name());
  1176. v.collection = unqualify(response.view().collection());
  1177. for (const auto& p : response.view().include()) v.include.push_back(p);
  1178. for (const auto& p : response.view().exclude()) v.exclude.push_back(p);
  1179. for (const auto& pbf : response.view().where()) {
  1180. Client::Filter f;
  1181. f.field = pbf.field();
  1182. f.op = static_cast<Client::FilterOp>(pbf.op());
  1183. try { f.value = nlohmann::json::parse(pbf.value()); }
  1184. catch (...) { f.value = pbf.value(); }
  1185. v.where.push_back(f);
  1186. }
  1187. if (response.view().has_default_sort()) {
  1188. Client::Sort s;
  1189. s.field = response.view().default_sort().field();
  1190. s.descending = response.view().default_sort().descending();
  1191. v.defaultSort = s;
  1192. }
  1193. v.createdAt = response.view().created_at();
  1194. v.updatedAt = response.view().updated_at();
  1195. return v;
  1196. }
  1197. // ===== Event Subscription =====
  1198. class SubscriptionHandle {
  1199. public:
  1200. SubscriptionHandle(std::shared_ptr<grpc::ClientContext> ctx,
  1201. std::unique_ptr<grpc::ClientReader<smartbotic::databasepb::DatabaseEvent>> reader,
  1202. std::thread readerThread)
  1203. : context_(std::move(ctx))
  1204. , reader_(std::move(reader))
  1205. , readerThread_(std::move(readerThread))
  1206. , active_(true) {}
  1207. ~SubscriptionHandle() {
  1208. cancel();
  1209. }
  1210. void cancel() {
  1211. if (active_.exchange(false)) {
  1212. context_->TryCancel();
  1213. if (readerThread_.joinable()) {
  1214. readerThread_.join();
  1215. }
  1216. }
  1217. }
  1218. private:
  1219. std::shared_ptr<grpc::ClientContext> context_;
  1220. std::unique_ptr<grpc::ClientReader<smartbotic::databasepb::DatabaseEvent>> reader_;
  1221. std::thread readerThread_;
  1222. std::atomic<bool> active_;
  1223. };
  1224. std::shared_ptr<void> subscribe(const std::vector<std::string>& collections,
  1225. Client::EventCallback callback) {
  1226. auto context = std::make_shared<grpc::ClientContext>();
  1227. // Subscribe is a long-running stream — no deadline — but the auth
  1228. // metadata still has to land on the initial request headers.
  1229. attachAuth(*context);
  1230. smartbotic::databasepb::SubscribeRequest request;
  1231. // v2.8.0 — subscribe was the one call left unqualified while every
  1232. // document call qualified. Two consequences, both cross-project leaks:
  1233. //
  1234. // * a bare name matched nothing, because events carry the qualified
  1235. // collection - so subscriptions silently never fired;
  1236. // * an EMPTY list means "every collection" server-side, which meant
  1237. // every collection in every PROJECT.
  1238. //
  1239. // Named collections are qualified like any other. For the "everything"
  1240. // case we send a `<project>:*` pattern instead of an empty request, so
  1241. // "all" means all of MY project. That needs no proto change - the
  1242. // pattern field already existed.
  1243. for (const auto& coll : collections) {
  1244. request.add_collections(qualify(coll));
  1245. }
  1246. if (collections.empty()) {
  1247. request.add_patterns(config_.project + ":*");
  1248. }
  1249. request.set_include_data(true);
  1250. auto reader = stub_->Subscribe(context.get(), request);
  1251. // Create reader thread
  1252. auto readerThread = std::thread([reader = reader.get(), callback = std::move(callback)]() {
  1253. smartbotic::databasepb::DatabaseEvent event;
  1254. while (reader->Read(&event)) {
  1255. std::optional<nlohmann::json> data;
  1256. if (!event.data().empty()) {
  1257. data = nlohmann::json::parse(event.data());
  1258. }
  1259. // Convert proto event type to string
  1260. std::string eventType;
  1261. switch (event.type()) {
  1262. case smartbotic::databasepb::EVENT_INSERT: eventType = "insert"; break;
  1263. case smartbotic::databasepb::EVENT_UPDATE: eventType = "update"; break;
  1264. case smartbotic::databasepb::EVENT_DELETE: eventType = "delete"; break;
  1265. case smartbotic::databasepb::EVENT_EXPIRE: eventType = "expire"; break;
  1266. case smartbotic::databasepb::EVENT_INVALIDATE: eventType = "invalidate"; break;
  1267. default: eventType = "unknown"; break;
  1268. }
  1269. callback(
  1270. event.collection(),
  1271. event.document_id(),
  1272. eventType,
  1273. data
  1274. );
  1275. }
  1276. });
  1277. return std::make_shared<SubscriptionHandle>(
  1278. std::move(context),
  1279. std::move(reader),
  1280. std::move(readerThread)
  1281. );
  1282. }
  1283. // ===== Version History =====
  1284. Client::VersionHistoryResult getVersionHistory(const std::string& collection,
  1285. const std::string& id, uint32_t limit, uint32_t offset) {
  1286. smartbotic::databasepb::GetVersionHistoryRequest request;
  1287. request.set_collection(qualify(collection));
  1288. request.set_id(id);
  1289. if (limit > 0) request.set_limit(limit);
  1290. if (offset > 0) request.set_offset(offset);
  1291. smartbotic::databasepb::GetVersionHistoryResponse response;
  1292. grpc::ClientContext context;
  1293. setDeadline(context);
  1294. auto status = stub_->GetVersionHistory(&context, request, &response);
  1295. if (!status.ok()) {
  1296. spdlog::error("Client::getVersionHistory failed: {}", status.error_message());
  1297. return {};
  1298. }
  1299. Client::VersionHistoryResult result;
  1300. result.currentVersion = response.current_version();
  1301. result.totalCount = response.total_count();
  1302. result.documentDeleted = response.document_deleted();
  1303. result.versions.reserve(response.versions_size());
  1304. for (const auto& ver : response.versions()) {
  1305. Client::VersionEntry entry;
  1306. entry.version = ver.version();
  1307. entry.timestamp = ver.timestamp();
  1308. entry.updatedBy = ver.updated_by();
  1309. if (!ver.data().empty()) {
  1310. entry.data = nlohmann::json::parse(ver.data());
  1311. }
  1312. result.versions.push_back(std::move(entry));
  1313. }
  1314. return result;
  1315. }
  1316. std::optional<Client::VersionEntry> getDocumentVersion(const std::string& collection,
  1317. const std::string& id, uint64_t version) {
  1318. smartbotic::databasepb::GetDocumentVersionRequest request;
  1319. request.set_collection(qualify(collection));
  1320. request.set_id(id);
  1321. request.set_version(version);
  1322. smartbotic::databasepb::GetDocumentVersionResponse response;
  1323. grpc::ClientContext context;
  1324. setDeadline(context);
  1325. auto status = stub_->GetDocumentVersion(&context, request, &response);
  1326. if (!status.ok()) {
  1327. spdlog::error("Client::getDocumentVersion failed: {}", status.error_message());
  1328. return std::nullopt;
  1329. }
  1330. if (!response.found()) {
  1331. return std::nullopt;
  1332. }
  1333. Client::VersionEntry entry;
  1334. entry.version = response.version_entry().version();
  1335. entry.timestamp = response.version_entry().timestamp();
  1336. entry.updatedBy = response.version_entry().updated_by();
  1337. if (!response.version_entry().data().empty()) {
  1338. entry.data = nlohmann::json::parse(response.version_entry().data());
  1339. }
  1340. return entry;
  1341. }
  1342. uint64_t restoreVersion(const std::string& collection, const std::string& id,
  1343. uint64_t version, const std::string& actor) {
  1344. smartbotic::databasepb::RestoreVersionRequest request;
  1345. request.set_collection(qualify(collection));
  1346. request.set_id(id);
  1347. request.set_version(version);
  1348. if (!actor.empty()) request.set_actor(actor);
  1349. smartbotic::databasepb::RestoreVersionResponse response;
  1350. grpc::ClientContext context;
  1351. setDeadline(context);
  1352. auto status = stub_->RestoreVersion(&context, request, &response);
  1353. if (!status.ok()) {
  1354. spdlog::error("Client::restoreVersion failed: {}", status.error_message());
  1355. return 0;
  1356. }
  1357. if (!response.success()) {
  1358. spdlog::error("Client::restoreVersion: {}", response.error());
  1359. return 0;
  1360. }
  1361. return response.new_version();
  1362. }
  1363. std::pair<uint64_t, uint64_t> restoreToDate(const std::string& collection,
  1364. const std::string& id, uint64_t timestamp, const std::string& actor) {
  1365. smartbotic::databasepb::RestoreToDateRequest request;
  1366. request.set_collection(qualify(collection));
  1367. request.set_id(id);
  1368. request.set_timestamp(timestamp);
  1369. if (!actor.empty()) request.set_actor(actor);
  1370. smartbotic::databasepb::RestoreToDateResponse response;
  1371. grpc::ClientContext context;
  1372. setDeadline(context);
  1373. auto status = stub_->RestoreToDate(&context, request, &response);
  1374. if (!status.ok()) {
  1375. spdlog::error("Client::restoreToDate failed: {}", status.error_message());
  1376. return {0, 0};
  1377. }
  1378. if (!response.success()) {
  1379. spdlog::error("Client::restoreToDate: {}", response.error());
  1380. return {0, 0};
  1381. }
  1382. return {response.restored_version(), response.new_version()};
  1383. }
  1384. // ===== Health =====
  1385. bool healthCheck() {
  1386. smartbotic::databasepb::HealthCheckRequest request;
  1387. smartbotic::databasepb::HealthCheckResponse response;
  1388. grpc::ClientContext context;
  1389. setDeadline(context);
  1390. auto status = stub_->HealthCheck(&context, request, &response);
  1391. if (!status.ok()) {
  1392. return false;
  1393. }
  1394. return response.healthy();
  1395. }
  1396. std::optional<Client::HealthInfo> getHealthInfo() {
  1397. smartbotic::databasepb::HealthCheckRequest request;
  1398. smartbotic::databasepb::HealthCheckResponse response;
  1399. grpc::ClientContext context;
  1400. setDeadline(context);
  1401. auto status = stub_->HealthCheck(&context, request, &response);
  1402. if (!status.ok()) {
  1403. return std::nullopt;
  1404. }
  1405. Client::HealthInfo info;
  1406. info.healthy = response.healthy();
  1407. info.uptimeMs = response.uptime_seconds() * 1000;
  1408. info.documentCount = response.document_count();
  1409. info.memoryUsedBytes = response.memory_used_bytes();
  1410. info.walSizeBytes = response.wal_size_bytes();
  1411. return info;
  1412. }
  1413. std::optional<Client::StatsInfo> getStats() {
  1414. smartbotic::databasepb::GetStatsRequest request;
  1415. smartbotic::databasepb::GetStatsResponse response;
  1416. grpc::ClientContext context;
  1417. setDeadline(context);
  1418. auto status = stub_->GetStats(&context, request, &response);
  1419. if (!status.ok()) {
  1420. return std::nullopt;
  1421. }
  1422. Client::StatsInfo info;
  1423. info.totalDocuments = response.total_documents();
  1424. info.totalCollections = response.total_collections();
  1425. info.memoryUsedBytes = response.memory_used_bytes();
  1426. info.walSequence = response.wal_sequence();
  1427. info.walSizeBytes = response.wal_size_bytes();
  1428. info.snapshotCount = response.snapshot_count();
  1429. info.lastSnapshotSequence = response.last_snapshot_sequence();
  1430. info.insertCount = response.insert_count();
  1431. info.updateCount = response.update_count();
  1432. info.deleteCount = response.delete_count();
  1433. info.queryCount = response.query_count();
  1434. // Memory eviction stats
  1435. info.evictedDocuments = response.evicted_documents();
  1436. info.totalEvictions = response.total_evictions();
  1437. info.recoveryCount = response.recovery_count();
  1438. // Memory configuration
  1439. info.maxMemoryBytes = response.max_memory_bytes();
  1440. info.evictionThresholdPercent = response.eviction_threshold_percent();
  1441. info.evictionTargetPercent = response.eviction_target_percent();
  1442. // Operation timing (microseconds) - for performance monitoring
  1443. info.getCount = response.get_count();
  1444. info.getTotalMicros = response.get_total_micros();
  1445. info.getMaxMicros = response.get_max_micros();
  1446. info.insertTotalMicros = response.insert_total_micros();
  1447. info.insertMaxMicros = response.insert_max_micros();
  1448. info.updateTotalMicros = response.update_total_micros();
  1449. info.updateMaxMicros = response.update_max_micros();
  1450. info.queryTotalMicros = response.query_total_micros();
  1451. info.queryMaxMicros = response.query_max_micros();
  1452. return info;
  1453. }
  1454. Client::MemoryStats getMemoryStats() {
  1455. smartbotic::databasepb::GetMemoryStatsRequest request;
  1456. smartbotic::databasepb::GetMemoryStatsResponse response;
  1457. grpc::ClientContext context;
  1458. setDeadline(context);
  1459. Client::MemoryStats out;
  1460. auto status = stub_->GetMemoryStats(&context, request, &response);
  1461. if (!status.ok()) {
  1462. spdlog::error("Client::getMemoryStats failed: {}", status.error_message());
  1463. return out;
  1464. }
  1465. out.totalMemoryBytes = response.total_memory_bytes();
  1466. out.maxMemoryBytes = response.max_memory_bytes();
  1467. out.pressurePercent = response.pressure_percent();
  1468. out.pressureLevel = response.pressure_level();
  1469. out.lastEvictionTimestamp = response.last_eviction_timestamp();
  1470. out.lastEvictionDocs = response.last_eviction_docs();
  1471. out.lastEvictionBytesFreed = response.last_eviction_bytes_freed();
  1472. out.collections.reserve(response.collections_size());
  1473. for (const auto& pbc : response.collections()) {
  1474. Client::MemoryCollectionStats c;
  1475. c.collection = pbc.collection();
  1476. c.documentCount = pbc.document_count();
  1477. c.estimatedBytes = pbc.estimated_bytes();
  1478. c.evictedStubCount = pbc.evicted_stub_count();
  1479. switch (pbc.priority()) {
  1480. case smartbotic::databasepb::MEMORY_PRIORITY_LOW: c.priority = "low"; break;
  1481. case smartbotic::databasepb::MEMORY_PRIORITY_NORMAL: c.priority = "normal"; break;
  1482. case smartbotic::databasepb::MEMORY_PRIORITY_HIGH: c.priority = "high"; break;
  1483. default: c.priority = "normal"; break;
  1484. }
  1485. out.collections.push_back(std::move(c));
  1486. }
  1487. return out;
  1488. }
  1489. // ===== Read-Only Control =====
  1490. bool setReadOnly(bool readOnly) {
  1491. smartbotic::databasepb::SetReadOnlyRequest request;
  1492. request.set_read_only(readOnly);
  1493. smartbotic::databasepb::SetReadOnlyResponse response;
  1494. grpc::ClientContext context;
  1495. setDeadline(context);
  1496. auto status = stub_->SetReadOnly(&context, request, &response);
  1497. if (!status.ok()) {
  1498. spdlog::error("Client::setReadOnly failed: {}", status.error_message());
  1499. return false;
  1500. }
  1501. return response.success();
  1502. }
  1503. Client::ReadOnlyStatus getReadOnlyStatus() {
  1504. smartbotic::databasepb::GetReadOnlyStatusRequest request;
  1505. smartbotic::databasepb::GetReadOnlyStatusResponse response;
  1506. grpc::ClientContext context;
  1507. setDeadline(context);
  1508. Client::ReadOnlyStatus out;
  1509. auto status = stub_->GetReadOnlyStatus(&context, request, &response);
  1510. if (!status.ok()) {
  1511. spdlog::error("Client::getReadOnlyStatus failed: {}", status.error_message());
  1512. return out;
  1513. }
  1514. out.readOnly = response.read_only();
  1515. out.reason = response.reason();
  1516. out.recoveryOutcome = response.recovery_outcome();
  1517. out.expectedSnapshot = response.expected_snapshot();
  1518. out.snapshotUsed = response.snapshot_used();
  1519. out.failureReason = response.failure_reason();
  1520. out.walEntriesReplayed = response.wal_entries_replayed();
  1521. out.snapshotsAttempted = response.snapshots_attempted();
  1522. return out;
  1523. }
  1524. // ===== File Operations =====
  1525. Client::FileUploadResult uploadFile(const std::vector<uint8_t>& data,
  1526. const Client::FileUploadMeta& meta,
  1527. uint32_t ttlSeconds = 0,
  1528. bool explicitTtl = false) {
  1529. grpc::ClientContext context;
  1530. auto timeout_ms = config_.timeoutMs + (data.size() / (1024 * 1024)) * 1000;
  1531. context.set_deadline(
  1532. std::chrono::system_clock::now() + std::chrono::milliseconds(timeout_ms));
  1533. attachAuth(context);
  1534. smartbotic::databasepb::UploadFileResponse response;
  1535. auto writer = stub_->UploadFile(&context, &response);
  1536. // First chunk: metadata
  1537. smartbotic::databasepb::FileChunk metaChunk;
  1538. auto* m = metaChunk.mutable_metadata();
  1539. m->set_name(meta.name);
  1540. m->set_mime_type(meta.mime_type);
  1541. m->set_file_type(meta.file_type);
  1542. m->set_related_id(meta.related_id);
  1543. m->set_is_public(meta.is_public);
  1544. // v2.6.0 — files are namespaced like collections. Filled from
  1545. // Config::project so callers keep passing bare ids and names.
  1546. m->set_project(config_.project);
  1547. // v2.8.0 — relative TTL; the server converts it to an absolute expiry.
  1548. // Set unconditionally on this overload: calling it IS the explicit
  1549. // choice, and 0 therefore means "never expire, ignore any configured
  1550. // default for this file type". The two-argument uploadFile() leaves the
  1551. // field absent so the default applies.
  1552. if (explicitTtl) m->set_ttl_seconds(ttlSeconds);
  1553. for (const auto& [key, value] : meta.metadata) {
  1554. (*m->mutable_metadata())[key] = value;
  1555. }
  1556. writer->Write(metaChunk);
  1557. // Data chunks (64KB each)
  1558. constexpr size_t CHUNK_SIZE = 64 * 1024;
  1559. for (size_t offset = 0; offset < data.size(); offset += CHUNK_SIZE) {
  1560. smartbotic::databasepb::FileChunk dataChunk;
  1561. size_t chunkSize = std::min(CHUNK_SIZE, data.size() - offset);
  1562. dataChunk.set_data(data.data() + offset, chunkSize);
  1563. if (!writer->Write(dataChunk)) break;
  1564. }
  1565. writer->WritesDone();
  1566. auto status = writer->Finish();
  1567. if (!status.ok()) {
  1568. spdlog::error("Client::uploadFile failed: {}", status.error_message());
  1569. throw std::runtime_error(status.error_message());
  1570. }
  1571. return {response.id(), response.size(), response.checksum(), response.deduplicated()};
  1572. }
  1573. std::vector<uint8_t> downloadFile(const std::string& id) {
  1574. // v2.6.0 — see note in uploadFile; project scopes every file call.
  1575. smartbotic::databasepb::DownloadFileRequest request;
  1576. request.set_id(id);
  1577. request.set_project(config_.project);
  1578. grpc::ClientContext context;
  1579. context.set_deadline(
  1580. std::chrono::system_clock::now() + std::chrono::milliseconds(config_.timeoutMs * 10));
  1581. attachAuth(context);
  1582. auto reader = stub_->DownloadFile(&context, request);
  1583. std::vector<uint8_t> fileData;
  1584. smartbotic::databasepb::FileChunk chunk;
  1585. while (reader->Read(&chunk)) {
  1586. if (chunk.has_data()) {
  1587. const auto& d = chunk.data();
  1588. fileData.insert(fileData.end(), d.begin(), d.end());
  1589. }
  1590. }
  1591. auto status = reader->Finish();
  1592. if (!status.ok()) {
  1593. spdlog::error("Client::downloadFile failed: {}", status.error_message());
  1594. throw std::runtime_error(status.error_message());
  1595. }
  1596. return fileData;
  1597. }
  1598. std::optional<Client::FileRecord> getFileInfo(const std::string& id) {
  1599. smartbotic::databasepb::GetFileInfoRequest request;
  1600. request.set_id(id);
  1601. request.set_project(config_.project);
  1602. smartbotic::databasepb::FileInfo response;
  1603. grpc::ClientContext context;
  1604. setDeadline(context);
  1605. auto status = stub_->GetFileInfo(&context, request, &response);
  1606. if (!status.ok()) {
  1607. if (status.error_code() == grpc::StatusCode::NOT_FOUND) return std::nullopt;
  1608. spdlog::error("Client::getFileInfo failed: {}", status.error_message());
  1609. throw std::runtime_error(status.error_message());
  1610. }
  1611. Client::FileRecord record;
  1612. record.id = response.id();
  1613. record.name = response.name();
  1614. record.mime_type = response.mime_type();
  1615. record.size = response.size();
  1616. record.file_type = response.file_type();
  1617. record.related_id = response.related_id();
  1618. record.checksum = response.checksum();
  1619. record.is_public = response.is_public();
  1620. record.ref_count = response.ref_count();
  1621. record.created_at = response.created_at();
  1622. record.project = response.project();
  1623. for (const auto& [key, value] : response.metadata()) {
  1624. record.metadata[key] = value;
  1625. }
  1626. return record;
  1627. }
  1628. bool deleteFile(const std::string& id) {
  1629. smartbotic::databasepb::DeleteFileRequest request;
  1630. request.set_id(id);
  1631. request.set_project(config_.project);
  1632. smartbotic::databasepb::DeleteFileResponse response;
  1633. grpc::ClientContext context;
  1634. setDeadline(context);
  1635. auto status = stub_->DeleteFile(&context, request, &response);
  1636. if (!status.ok()) {
  1637. spdlog::error("Client::deleteFile failed: {}", status.error_message());
  1638. return false;
  1639. }
  1640. return response.deleted();
  1641. }
  1642. bool setFileTtl(const std::string& id, uint32_t ttlSeconds) {
  1643. smartbotic::databasepb::SetFileTtlRequest request;
  1644. request.set_id(id);
  1645. request.set_project(config_.project);
  1646. request.set_ttl_seconds(ttlSeconds);
  1647. smartbotic::databasepb::SetFileTtlResponse response;
  1648. grpc::ClientContext context;
  1649. setDeadline(context);
  1650. auto status = stub_->SetFileTtl(&context, request, &response);
  1651. if (!status.ok()) {
  1652. spdlog::error("Client::setFileTtl failed: {}", status.error_message());
  1653. return false;
  1654. }
  1655. return response.updated();
  1656. }
  1657. Client::FileListResult listFiles(const std::string& file_type,
  1658. const std::string& related_id,
  1659. uint32_t limit, uint32_t offset,
  1660. const std::string& checksum,
  1661. const std::string& name) {
  1662. smartbotic::databasepb::ListFilesRequest request;
  1663. if (!file_type.empty()) request.set_file_type(file_type);
  1664. if (!related_id.empty()) request.set_related_id(related_id);
  1665. request.set_limit(limit);
  1666. request.set_offset(offset);
  1667. if (!checksum.empty()) request.set_checksum(checksum);
  1668. if (!name.empty()) request.set_name(name);
  1669. request.set_project(config_.project);
  1670. smartbotic::databasepb::ListFilesResponse response;
  1671. grpc::ClientContext context;
  1672. setDeadline(context);
  1673. auto status = stub_->ListFiles(&context, request, &response);
  1674. if (!status.ok()) {
  1675. spdlog::error("Client::listFiles failed: {}", status.error_message());
  1676. throw std::runtime_error(status.error_message());
  1677. }
  1678. Client::FileListResult result;
  1679. result.total_count = response.total_count();
  1680. result.has_more = response.has_more();
  1681. for (const auto& f : response.files()) {
  1682. Client::FileRecord record;
  1683. record.id = f.id();
  1684. record.name = f.name();
  1685. record.mime_type = f.mime_type();
  1686. record.size = f.size();
  1687. record.file_type = f.file_type();
  1688. record.related_id = f.related_id();
  1689. record.checksum = f.checksum();
  1690. record.is_public = f.is_public();
  1691. record.created_at = f.created_at();
  1692. record.project = f.project();
  1693. for (const auto& [key, value] : f.metadata()) {
  1694. record.metadata[key] = value;
  1695. }
  1696. result.files.push_back(std::move(record));
  1697. }
  1698. return result;
  1699. }
  1700. private:
  1701. // Per-context setup: deadline + (v2.4) bearer-token auth metadata.
  1702. //
  1703. // Most RPC call sites in this library funnel through this helper —
  1704. // unary (insert, get, update, ...) plus the unary file ops that need
  1705. // the default deadline. Long-running streams (Subscribe) and the
  1706. // file-streaming RPCs (UploadFile/DownloadFile) compute their own
  1707. // deadlines and call `attachAuth()` separately so the auth metadata
  1708. // still lands on every outbound RPC. Centralising both concerns here
  1709. // means we don't need a gRPC ClientInterceptorFactoryInterface — the
  1710. // interceptor route is also offered by gRPC++ but it sits in
  1711. // grpc::experimental:: and would need to be re-validated on every
  1712. // grpc minor; the per-context style keeps Stage E independent of
  1713. // gRPC's TLS/interceptor experimental flux.
  1714. void setDeadline(grpc::ClientContext& context) {
  1715. context.set_deadline(
  1716. std::chrono::system_clock::now() + std::chrono::milliseconds(config_.timeoutMs)
  1717. );
  1718. attachAuth(context);
  1719. }
  1720. // v2.4 — attach the bearer-token authorization metadata if configured.
  1721. // Use this on contexts that set their own deadline (streams,
  1722. // long-running ops) so the auth metadata still lands.
  1723. void attachAuth(grpc::ClientContext& context) {
  1724. if (!config_.auth_token.empty()) {
  1725. context.AddMetadata("authorization", "Bearer " + config_.auth_token);
  1726. }
  1727. }
  1728. Config config_;
  1729. std::shared_ptr<grpc::Channel> channel_;
  1730. std::unique_ptr<smartbotic::databasepb::DatabaseService::Stub> stub_;
  1731. std::atomic<bool> connected_{false};
  1732. };
  1733. // ===== Client Public Interface Implementation =====
  1734. Client::Client(Config config)
  1735. : impl_(std::make_unique<Impl>(std::move(config))) {}
  1736. Client::~Client() = default;
  1737. Client::Client(Client&&) noexcept = default;
  1738. Client& Client::operator=(Client&&) noexcept = default;
  1739. bool Client::connect() {
  1740. return impl_->connect();
  1741. }
  1742. bool Client::isConnected() const {
  1743. return impl_->isConnected();
  1744. }
  1745. std::string Client::insert(const std::string& collection, const nlohmann::json& data,
  1746. const std::string& id, uint32_t ttlSeconds,
  1747. const std::string& actor) {
  1748. return impl_->insert(collection, data, id, ttlSeconds, actor);
  1749. }
  1750. std::optional<nlohmann::json> Client::get(const std::string& collection, const std::string& id) {
  1751. return impl_->get(collection, id);
  1752. }
  1753. bool Client::update(const std::string& collection, const std::string& id, const nlohmann::json& data,
  1754. const std::string& actor) {
  1755. return impl_->update(collection, id, data, actor);
  1756. }
  1757. bool Client::updateIfVersion(const std::string& collection, const std::string& id,
  1758. const nlohmann::json& data, uint64_t expectedVersion,
  1759. const std::string& actor) {
  1760. return impl_->updateIfVersion(collection, id, data, expectedVersion, actor);
  1761. }
  1762. uint64_t Client::patch(const std::string& collection, const std::string& id,
  1763. const nlohmann::json& fields, const std::string& actor) {
  1764. return impl_->patch(collection, id, fields, actor);
  1765. }
  1766. uint64_t Client::patchWithTtl(const std::string& collection, const std::string& id,
  1767. const nlohmann::json& fields, uint32_t ttlSeconds,
  1768. const std::string& actor) {
  1769. return impl_->patch(collection, id, fields, actor, ttlSeconds);
  1770. }
  1771. std::pair<std::string, bool> Client::upsert(const std::string& collection, const nlohmann::json& data,
  1772. const std::string& id, uint32_t ttlSeconds,
  1773. const std::string& actor) {
  1774. return impl_->upsert(collection, data, id, ttlSeconds, actor);
  1775. }
  1776. bool Client::remove(const std::string& collection, const std::string& id) {
  1777. return impl_->remove(collection, id);
  1778. }
  1779. bool Client::exists(const std::string& collection, const std::string& id) {
  1780. return impl_->exists(collection, id);
  1781. }
  1782. Client::VersionHistoryResult Client::getVersionHistory(const std::string& collection,
  1783. const std::string& id, uint32_t limit, uint32_t offset) {
  1784. return impl_->getVersionHistory(collection, id, limit, offset);
  1785. }
  1786. std::optional<Client::VersionEntry> Client::getDocumentVersion(const std::string& collection,
  1787. const std::string& id, uint64_t version) {
  1788. return impl_->getDocumentVersion(collection, id, version);
  1789. }
  1790. uint64_t Client::restoreVersion(const std::string& collection, const std::string& id,
  1791. uint64_t version, const std::string& actor) {
  1792. return impl_->restoreVersion(collection, id, version, actor);
  1793. }
  1794. std::pair<uint64_t, uint64_t> Client::restoreToDate(const std::string& collection,
  1795. const std::string& id, uint64_t timestamp, const std::string& actor) {
  1796. return impl_->restoreToDate(collection, id, timestamp, actor);
  1797. }
  1798. std::vector<nlohmann::json> Client::find(const std::string& collection,
  1799. const QueryOptions& options) {
  1800. return impl_->find(collection, options);
  1801. }
  1802. std::vector<nlohmann::json> Client::find(const std::string& collection,
  1803. const QueryOptions& options,
  1804. const std::vector<std::string>& projection) {
  1805. return impl_->find(collection, options, projection);
  1806. }
  1807. Client::FindResult Client::findWithMetrics(const std::string& collection,
  1808. const QueryOptions& options) {
  1809. return impl_->findWithMetrics(collection, options);
  1810. }
  1811. uint64_t Client::count(const std::string& collection,
  1812. const std::vector<Filter>& filters) {
  1813. return impl_->count(collection, filters);
  1814. }
  1815. uint64_t Client::count(const std::string& collection) {
  1816. return impl_->count(collection, {});
  1817. }
  1818. bool Client::setAdd(const std::string& collection, const std::string& setId, const std::string& member) {
  1819. return impl_->setAdd(collection, setId, member);
  1820. }
  1821. bool Client::setRemove(const std::string& collection, const std::string& setId, const std::string& member) {
  1822. return impl_->setRemove(collection, setId, member);
  1823. }
  1824. std::vector<std::string> Client::setMembers(const std::string& collection, const std::string& setId) {
  1825. return impl_->setMembers(collection, setId);
  1826. }
  1827. bool Client::setIsMember(const std::string& collection, const std::string& setId, const std::string& member) {
  1828. return impl_->setIsMember(collection, setId, member);
  1829. }
  1830. std::vector<Client::SimilarityResult> Client::similaritySearch(
  1831. const std::string& collection, const std::vector<float>& queryVector,
  1832. uint32_t topK, float minScore) {
  1833. return impl_->similaritySearch(collection, queryVector, topK, minScore);
  1834. }
  1835. bool Client::createCollection(const std::string& name, uint32_t defaultTtlSeconds,
  1836. bool encrypted, uint32_t maxVersions, uint32_t vectorDimension) {
  1837. return impl_->createCollection(name, defaultTtlSeconds, encrypted, maxVersions, vectorDimension);
  1838. }
  1839. bool Client::dropCollection(const std::string& name) {
  1840. return impl_->dropCollection(name);
  1841. }
  1842. std::vector<std::string> Client::listCollections() {
  1843. return impl_->listCollections();
  1844. }
  1845. std::optional<Client::CollectionInfo> Client::getCollectionInfo(const std::string& name) {
  1846. return impl_->getCollectionInfo(name);
  1847. }
  1848. // ===== Project Management (v2.3 Stage F) =====
  1849. std::vector<std::string> Client::listProjects() {
  1850. return impl_->listProjects();
  1851. }
  1852. bool Client::createProject(const std::string& name) {
  1853. return impl_->createProject(name);
  1854. }
  1855. void Client::dropProject(const std::string& name) {
  1856. impl_->dropProject(name);
  1857. }
  1858. bool Client::configureCollection(const std::string& collection, const CollectionConfig& cfg) {
  1859. return impl_->configureCollection(collection, cfg);
  1860. }
  1861. Client::CollectionConfig Client::getCollectionConfig(const std::string& collection) {
  1862. return impl_->getCollectionConfig(collection);
  1863. }
  1864. bool Client::hasCollectionConfig(const std::string& collection) {
  1865. return impl_->hasCollectionConfig(collection);
  1866. }
  1867. Client::TimestampMigrationResult Client::migrateCollectionTimestamps(
  1868. const std::string& collection,
  1869. const std::string& fromPrecision,
  1870. const std::string& toPrecision
  1871. ) {
  1872. return impl_->migrateCollectionTimestamps(collection, fromPrecision, toPrecision);
  1873. }
  1874. bool Client::createView(const std::string& name,
  1875. const std::string& collection,
  1876. const std::vector<std::string>& include,
  1877. const std::vector<std::string>& exclude,
  1878. const std::vector<Filter>& where,
  1879. const std::optional<Sort>& defaultSort) {
  1880. return impl_->createView(name, collection, include, exclude, where, defaultSort);
  1881. }
  1882. bool Client::dropView(const std::string& name) {
  1883. return impl_->dropView(name);
  1884. }
  1885. bool Client::createIndex(const std::string& collection, const std::string& field) {
  1886. return impl_->createIndex(collection, field, nullptr);
  1887. }
  1888. bool Client::createIndex(const std::string& collection, const std::string& field,
  1889. uint64_t& rowsIndexed) {
  1890. return impl_->createIndex(collection, field, &rowsIndexed);
  1891. }
  1892. bool Client::dropIndex(const std::string& collection, const std::string& field) {
  1893. return impl_->dropIndex(collection, field);
  1894. }
  1895. std::vector<Client::IndexDefinition> Client::listIndexes(const std::string& collection) {
  1896. return impl_->listIndexes(collection);
  1897. }
  1898. std::vector<Client::IndexValue> Client::indexValues(const std::string& collection,
  1899. const std::string& field,
  1900. uint32_t limit, bool ascending) {
  1901. return impl_->indexValues(collection, field, limit, ascending);
  1902. }
  1903. std::vector<Client::ViewDefinition> Client::listViews() {
  1904. return impl_->listViews();
  1905. }
  1906. std::optional<Client::ViewDefinition> Client::getViewInfo(const std::string& name) {
  1907. return impl_->getViewInfo(name);
  1908. }
  1909. std::shared_ptr<void> Client::subscribe(const std::vector<std::string>& collections, EventCallback callback) {
  1910. return impl_->subscribe(collections, std::move(callback));
  1911. }
  1912. bool Client::healthCheck() {
  1913. return impl_->healthCheck();
  1914. }
  1915. std::optional<Client::HealthInfo> Client::getHealthInfo() {
  1916. return impl_->getHealthInfo();
  1917. }
  1918. std::optional<Client::StatsInfo> Client::getStats() {
  1919. return impl_->getStats();
  1920. }
  1921. Client::MemoryStats Client::getMemoryStats() {
  1922. return impl_->getMemoryStats();
  1923. }
  1924. bool Client::setReadOnly(bool readOnly) {
  1925. return impl_->setReadOnly(readOnly);
  1926. }
  1927. Client::ReadOnlyStatus Client::getReadOnlyStatus() {
  1928. return impl_->getReadOnlyStatus();
  1929. }
  1930. Client::FileUploadResult Client::uploadFile(const std::vector<uint8_t>& data, const FileUploadMeta& meta) {
  1931. return impl_->uploadFile(data, meta);
  1932. }
  1933. Client::FileUploadResult Client::uploadFile(const std::vector<uint8_t>& data,
  1934. const FileUploadMeta& meta,
  1935. uint32_t ttlSeconds) {
  1936. return impl_->uploadFile(data, meta, ttlSeconds, /*explicitTtl=*/true);
  1937. }
  1938. std::vector<uint8_t> Client::downloadFile(const std::string& id) {
  1939. return impl_->downloadFile(id);
  1940. }
  1941. std::optional<Client::FileRecord> Client::getFileInfo(const std::string& id) {
  1942. return impl_->getFileInfo(id);
  1943. }
  1944. bool Client::deleteFile(const std::string& id) {
  1945. return impl_->deleteFile(id);
  1946. }
  1947. bool Client::setFileTtl(const std::string& id, uint32_t ttlSeconds) {
  1948. return impl_->setFileTtl(id, ttlSeconds);
  1949. }
  1950. Client::FileListResult Client::listFiles(const std::string& file_type,
  1951. const std::string& related_id,
  1952. uint32_t limit, uint32_t offset,
  1953. const std::string& checksum,
  1954. const std::string& name) {
  1955. return impl_->listFiles(file_type, related_id, limit, offset, checksum, name);
  1956. }
  1957. } // namespace smartbotic::database