reifydb-client 0.6.0

Official Rust client library for ReifyDB
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77

// SPDX-License-Identifier: MIT
// Copyright (c) 2026 ReifyDB

use std::collections::HashMap;

use reifydb_core::actors::server::{
	Operation, ServerAuthResponse, ServerLogoutResponse, ServerMessage, ServerResponse, ServerSubscribeResponse,
	build_server_message,
};
use reifydb_runtime::actor::{mailbox::ActorRef, reply::reply_channel, system::ActorSystem};
use reifydb_value::{params::Params, value::identity::IdentityId};

pub struct DstClient {
	actor_ref: ActorRef<ServerMessage>,
	system: ActorSystem,
}

impl DstClient {
	pub fn new(actor_ref: ActorRef<ServerMessage>, system: ActorSystem) -> Self {
		Self {
			actor_ref,
			system,
		}
	}

	fn send(&self, msg: ServerMessage) {
		self.actor_ref.send(msg).ok().expect("actor mailbox closed");
		self.system.run_until_idle();
	}

	pub fn query(&self, identity: IdentityId, rql: String, params: Params) -> ServerResponse {
		let (reply, receiver) = reply_channel();
		self.send(build_server_message(Operation::Query, identity, rql, params, reply));
		receiver.try_recv().expect("no reply from actor")
	}

	pub fn command(&self, identity: IdentityId, rql: String, params: Params) -> ServerResponse {
		let (reply, receiver) = reply_channel();
		self.send(build_server_message(Operation::Command, identity, rql, params, reply));
		receiver.try_recv().expect("no reply from actor")
	}

	pub fn admin(&self, identity: IdentityId, rql: String, params: Params) -> ServerResponse {
		let (reply, receiver) = reply_channel();
		self.send(build_server_message(Operation::Admin, identity, rql, params, reply));
		receiver.try_recv().expect("no reply from actor")
	}

	pub fn subscribe(&self, identity: IdentityId, rql: String) -> ServerSubscribeResponse {
		let (reply, receiver) = reply_channel();
		self.send(ServerMessage::Subscribe {
			identity,
			rql,
			reply,
		});
		receiver.try_recv().expect("no reply from actor")
	}

	pub fn authenticate(&self, method: String, credentials: HashMap<String, String>) -> ServerAuthResponse {
		let (reply, receiver) = reply_channel();
		self.send(ServerMessage::Authenticate {
			method,
			credentials,
			reply,
		});
		receiver.try_recv().expect("no reply from actor")
	}

	pub fn logout(&self, token: String) -> ServerLogoutResponse {
		let (reply, receiver) = reply_channel();
		self.send(ServerMessage::Logout {
			token,
			reply,
		});
		receiver.try_recv().expect("no reply from actor")
	}
}