mirror of
https://github.com/hexedtech/codemp.git
synced 2024-12-23 13:24:54 +01:00
feat: more capillar errors
so its clear that some functions wont return a transport error or an rpc error
This commit is contained in:
parent
0b1a542ed5
commit
b98be22a8b
10 changed files with 145 additions and 156 deletions
|
@ -3,7 +3,7 @@
|
|||
//! an bidirectional stream handler to easily manage async operations across local buffers and the
|
||||
//! server
|
||||
|
||||
use crate::Result;
|
||||
use crate::errors::ControllerResult;
|
||||
|
||||
#[async_trait::async_trait]
|
||||
pub(crate) trait ControllerWorker<T : Sized + Send + Sync> {
|
||||
|
@ -15,6 +15,9 @@ pub(crate) trait ControllerWorker<T : Sized + Send + Sync> {
|
|||
async fn work(self, tx: Self::Tx, rx: Self::Rx);
|
||||
}
|
||||
|
||||
// note that we don't use thiserror's #[from] because we don't want the error structs to contain
|
||||
// these foreign types, and also we want these to be easily constructable
|
||||
|
||||
/// async and threadsafe handle to a generic bidirectional stream
|
||||
///
|
||||
/// this generic trait is implemented by actors managing stream procedures.
|
||||
|
@ -29,14 +32,14 @@ pub trait Controller<T : Sized + Send + Sync> : Sized + Send + Sync {
|
|||
///
|
||||
/// success or failure of this function does not imply validity of sent operation,
|
||||
/// because it's integrated asynchronously on the background worker
|
||||
async fn send(&self, x: T) -> Result<()>;
|
||||
async fn send(&self, x: T) -> ControllerResult<()>;
|
||||
|
||||
/// get next value from other users, blocking until one is available
|
||||
///
|
||||
/// this is just an async trait function wrapped by `async_trait`:
|
||||
///
|
||||
/// `async fn recv(&self) -> codemp::Result<T>;`
|
||||
async fn recv(&self) -> Result<T> {
|
||||
/// `async fn recv(&self) -> codemp::ControllerResult<T>;`
|
||||
async fn recv(&self) -> ControllerResult<T> {
|
||||
loop {
|
||||
self.poll().await?;
|
||||
if let Some(x) = self.try_recv().await? {
|
||||
|
@ -57,11 +60,11 @@ pub trait Controller<T : Sized + Send + Sync> : Sized + Send + Sync {
|
|||
///
|
||||
/// this is just an async trait function wrapped by `async_trait`:
|
||||
///
|
||||
/// `async fn poll(&self) -> codemp::Result<()>;`
|
||||
async fn poll(&self) -> Result<()>;
|
||||
/// `async fn poll(&self) -> codemp::ControllerResult<()>;`
|
||||
async fn poll(&self) -> ControllerResult<()>;
|
||||
|
||||
/// attempt to receive a value without blocking, return None if nothing is available
|
||||
async fn try_recv(&self) -> Result<Option<T>>;
|
||||
async fn try_recv(&self) -> ControllerResult<Option<T>>;
|
||||
|
||||
/// stop underlying worker
|
||||
///
|
||||
|
@ -75,7 +78,7 @@ pub trait Controller<T : Sized + Send + Sync> : Sized + Send + Sync {
|
|||
|
||||
|
||||
/// type wrapper for Boxed dyn callback
|
||||
pub struct ControllerCallback<T>(Box<dyn Sync + Send + Fn(T)>);
|
||||
pub struct ControllerCallback<T>(pub Box<dyn Sync + Send + Fn(T)>);
|
||||
|
||||
impl<T> ControllerCallback<T> {
|
||||
pub fn call(&self, x: T) {
|
||||
|
|
|
@ -8,11 +8,9 @@ use diamond_types::LocalVersion;
|
|||
use tokio::sync::{mpsc, oneshot, watch};
|
||||
use tonic::async_trait;
|
||||
|
||||
use crate::api::controller::ControllerCallback;
|
||||
use crate::api::Controller;
|
||||
|
||||
use crate::api::controller::{Controller, ControllerCallback};
|
||||
use crate::api::TextChange;
|
||||
|
||||
use crate::errors::ControllerResult;
|
||||
use crate::ext::InternallyMutable;
|
||||
|
||||
use super::worker::DeltaRequest;
|
||||
|
@ -36,7 +34,7 @@ impl BufferController {
|
|||
}
|
||||
|
||||
/// return buffer whole content, updating internal buffer previous state
|
||||
pub async fn content(&self) -> crate::Result<String> {
|
||||
pub async fn content(&self) -> ControllerResult<String> {
|
||||
let (tx, rx) = oneshot::channel();
|
||||
self.0.content_request.send(tx).await?;
|
||||
let content = rx.await?;
|
||||
|
@ -64,20 +62,19 @@ pub(crate) struct BufferControllerInner {
|
|||
impl Controller<TextChange> for BufferController {
|
||||
/// block until a text change is available
|
||||
/// this returns immediately if one is already available
|
||||
async fn poll(&self) -> crate::Result<()> {
|
||||
async fn poll(&self) -> ControllerResult<()> {
|
||||
if self.0.last_update.get() != *self.0.latest_version.borrow() {
|
||||
return Ok(());
|
||||
}
|
||||
|
||||
let (tx, rx) = oneshot::channel::<()>();
|
||||
self.0.poller.send(tx)?;
|
||||
rx.await
|
||||
.map_err(|_| crate::Error::Channel { send: false })?;
|
||||
rx.await?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// if a text change is available, return it immediately
|
||||
async fn try_recv(&self) -> crate::Result<Option<TextChange>> {
|
||||
async fn try_recv(&self) -> ControllerResult<Option<TextChange>> {
|
||||
let last_update = self.0.last_update.get();
|
||||
let latest_version = self.0.latest_version.borrow().clone();
|
||||
|
||||
|
@ -94,7 +91,7 @@ impl Controller<TextChange> for BufferController {
|
|||
|
||||
/// enqueue a text change for processing
|
||||
/// this also updates internal buffer previous state
|
||||
async fn send(&self, op: TextChange) -> crate::Result<()> {
|
||||
async fn send(&self, op: TextChange) -> ControllerResult<()> {
|
||||
// we let the worker do the updating to the last version and send it back.
|
||||
let (tx, rx) = oneshot::channel();
|
||||
self.0.ops_in.send((op, tx))?;
|
||||
|
|
|
@ -7,7 +7,7 @@ use std::sync::Arc;
|
|||
use dashmap::DashMap;
|
||||
use tonic::{service::interceptor::InterceptedService, transport::{Channel, Endpoint}};
|
||||
|
||||
use crate::{api::User, ext::InternallyMutable, workspace::Workspace};
|
||||
use crate::{api::User, errors::{ConnectionResult, ProcedureResult}, ext::InternallyMutable, workspace::Workspace};
|
||||
use codemp_proto::{
|
||||
auth::{auth_client::AuthClient, LoginRequest},
|
||||
common::{Empty, Token}, session::{session_client::SessionClient, InviteRequest, WorkspaceRequest},
|
||||
|
@ -42,7 +42,7 @@ impl Client {
|
|||
host: impl AsRef<str>,
|
||||
username: impl AsRef<str>,
|
||||
password: impl AsRef<str>,
|
||||
) -> crate::Result<Self> {
|
||||
) -> ConnectionResult<Self> {
|
||||
let host = if host.as_ref().starts_with("http") {
|
||||
host.as_ref().to_string()
|
||||
} else {
|
||||
|
@ -75,7 +75,7 @@ impl Client {
|
|||
}
|
||||
|
||||
/// refresh session token
|
||||
pub async fn refresh(&self) -> crate::Result<()> {
|
||||
pub async fn refresh(&self) -> ProcedureResult<()> {
|
||||
let new_token = self.0.auth.clone().refresh(self.0.claims.get())
|
||||
.await?
|
||||
.into_inner();
|
||||
|
@ -84,7 +84,7 @@ impl Client {
|
|||
}
|
||||
|
||||
/// attempts to create a new workspace with given name
|
||||
pub async fn create_workspace(&self, name: impl AsRef<str>) -> crate::Result<()> {
|
||||
pub async fn create_workspace(&self, name: impl AsRef<str>) -> ProcedureResult<()> {
|
||||
self.0.session
|
||||
.clone()
|
||||
.create_workspace(WorkspaceRequest { workspace: name.as_ref().to_string() })
|
||||
|
@ -93,7 +93,7 @@ impl Client {
|
|||
}
|
||||
|
||||
/// delete an existing workspace if possible
|
||||
pub async fn delete_workspace(&self, name: impl AsRef<str>) -> crate::Result<()> {
|
||||
pub async fn delete_workspace(&self, name: impl AsRef<str>) -> ProcedureResult<()> {
|
||||
self.0.session
|
||||
.clone()
|
||||
.delete_workspace(WorkspaceRequest { workspace: name.as_ref().to_string() })
|
||||
|
@ -102,7 +102,7 @@ impl Client {
|
|||
}
|
||||
|
||||
/// invite user associated with username to workspace, if possible
|
||||
pub async fn invite_to_workspace(&self, workspace_name: impl AsRef<str>, user_name: impl AsRef<str>) -> crate::Result<()> {
|
||||
pub async fn invite_to_workspace(&self, workspace_name: impl AsRef<str>, user_name: impl AsRef<str>) -> ProcedureResult<()> {
|
||||
self.0.session
|
||||
.clone()
|
||||
.invite_to_workspace(InviteRequest {
|
||||
|
@ -114,7 +114,7 @@ impl Client {
|
|||
}
|
||||
|
||||
/// list all available workspaces, filtering between those owned and those invited to
|
||||
pub async fn list_workspaces(&self, owned: bool, invited: bool) -> crate::Result<Vec<String>> {
|
||||
pub async fn list_workspaces(&self, owned: bool, invited: bool) -> ProcedureResult<Vec<String>> {
|
||||
let mut workspaces = self.0.session
|
||||
.clone()
|
||||
.list_workspaces(Empty {})
|
||||
|
@ -130,7 +130,7 @@ impl Client {
|
|||
}
|
||||
|
||||
/// join a workspace, returns [Workspace]
|
||||
pub async fn join_workspace(&self, workspace: impl AsRef<str>) -> crate::Result<Workspace> {
|
||||
pub async fn join_workspace(&self, workspace: impl AsRef<str>) -> ConnectionResult<Workspace> {
|
||||
let token = self.0.session
|
||||
.clone()
|
||||
.access_workspace(WorkspaceRequest { workspace: workspace.as_ref().to_string() })
|
||||
|
@ -184,7 +184,7 @@ impl tonic::service::Interceptor for SessionInterceptor {
|
|||
fn call(
|
||||
&mut self,
|
||||
mut request: tonic::Request<()>,
|
||||
) -> Result<tonic::Request<()>, tonic::Status> {
|
||||
) -> tonic::Result<tonic::Request<()>> {
|
||||
if let Ok(token) = self.0.borrow().token.parse() {
|
||||
request.metadata_mut().insert("session", token);
|
||||
}
|
||||
|
|
|
@ -9,8 +9,8 @@ use tokio::sync::{
|
|||
};
|
||||
use tonic::async_trait;
|
||||
|
||||
use crate::api::{controller::ControllerCallback, Controller, Cursor};
|
||||
use codemp_proto::cursor::{CursorEvent, CursorPosition};
|
||||
use crate::{api::{controller::ControllerCallback, Controller, Cursor}, errors::ControllerResult};
|
||||
/// the cursor controller implementation
|
||||
///
|
||||
/// this contains
|
||||
|
@ -40,7 +40,7 @@ pub(crate) struct CursorControllerInner {
|
|||
impl Controller<Cursor> for CursorController {
|
||||
/// enqueue a cursor event to be broadcast to current workspace
|
||||
/// will automatically invert cursor start/end if they are inverted
|
||||
async fn send(&self, mut cursor: Cursor) -> crate::Result<()> {
|
||||
async fn send(&self, mut cursor: Cursor) -> ControllerResult<()> {
|
||||
if cursor.start > cursor.end {
|
||||
std::mem::swap(&mut cursor.start, &mut cursor.end);
|
||||
}
|
||||
|
@ -48,7 +48,6 @@ impl Controller<Cursor> for CursorController {
|
|||
}
|
||||
|
||||
/// try to receive without blocking, but will still block on stream mutex
|
||||
async fn try_recv(&self) -> crate::Result<Option<Cursor>> {
|
||||
let mut stream = self.0.stream.lock().await;
|
||||
match stream.try_recv() {
|
||||
Ok(x) => Ok(Some(x.into())),
|
||||
|
@ -59,11 +58,12 @@ impl Controller<Cursor> for CursorController {
|
|||
Ok(stream.try_recv().map(|x| x.into()).ok())
|
||||
}
|
||||
}
|
||||
async fn try_recv(&self) -> ControllerResult<Option<Cursor>> {
|
||||
}
|
||||
|
||||
/// await for changed mutex and then next op change
|
||||
async fn poll(&self) -> crate::Result<()> {
|
||||
Ok(self.0.last_op.lock().await.changed().await?)
|
||||
async fn poll(&self) -> ControllerResult<()> {
|
||||
}
|
||||
|
||||
fn callback(&self, cb: impl Into<ControllerCallback<CursorController>>) {
|
||||
|
|
159
src/errors.rs
159
src/errors.rs
|
@ -1,120 +1,73 @@
|
|||
//! ### Errors
|
||||
//!
|
||||
//! library error helpers and types
|
||||
|
||||
use std::result::Result as StdResult;
|
||||
#[deprecated = "use underlying errors to provide more context on what errors could really be thrown"]
|
||||
#[allow(deprecated)]
|
||||
pub type Result<T> = std::result::Result<T, Error>;
|
||||
|
||||
use tracing::warn;
|
||||
|
||||
/// an error which can be ignored with just a warning entry
|
||||
pub trait IgnorableError {
|
||||
fn unwrap_or_warn(self, msg: &str);
|
||||
}
|
||||
|
||||
impl<T, E> IgnorableError for StdResult<T, E>
|
||||
where E : std::fmt::Debug {
|
||||
fn unwrap_or_warn(self, msg: &str) {
|
||||
match self {
|
||||
Ok(_) => {},
|
||||
Err(e) => warn!("{}: {:?}", msg, e),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/// an error which can be ignored with just a warning entry and returning the default value
|
||||
pub trait IgnorableDefaultableError<T> {
|
||||
fn unwrap_or_warn_default(self, msg: &str) -> T;
|
||||
}
|
||||
|
||||
impl<T, E> IgnorableDefaultableError<T> for StdResult<T, E>
|
||||
where E : std::fmt::Display, T: Default {
|
||||
fn unwrap_or_warn_default(self, msg: &str) -> T {
|
||||
match self {
|
||||
Ok(x) => x,
|
||||
Err(e) => {
|
||||
warn!("{}: {}", msg, e);
|
||||
T::default()
|
||||
},
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// result type for codemp errors
|
||||
pub type Result<T> = StdResult<T, Error>;
|
||||
|
||||
// TODO split this into specific errors for various parts of the library
|
||||
/// codemp error type for library issues
|
||||
#[deprecated = "use underlying errors to provide more context on what errors could really be thrown"]
|
||||
#[derive(Debug, thiserror::Error)]
|
||||
pub enum Error {
|
||||
/// errors caused by tonic http layer
|
||||
#[error("tonic error (status: {status}, message: {message})")]
|
||||
Transport {
|
||||
status: String,
|
||||
message: String,
|
||||
},
|
||||
#[error("connection error: {0}")]
|
||||
Connection(#[from] ConnectionError),
|
||||
|
||||
/// errors caused by async channels
|
||||
#[error("channel error, send: {send}")]
|
||||
Channel {
|
||||
send: bool
|
||||
},
|
||||
#[error("procedure error: {0}")]
|
||||
Procedure(#[from] ProcedureError),
|
||||
|
||||
/// errors caused by wrong usage of library objects
|
||||
#[error("invalid state error: {msg}")]
|
||||
InvalidState {
|
||||
msg: String,
|
||||
},
|
||||
|
||||
/// errors caused by wrong interlocking, safe to retry
|
||||
#[error("deadlocked error")]
|
||||
Deadlocked
|
||||
#[error("controller error: {0}")]
|
||||
Controller(#[from] ControllerError),
|
||||
}
|
||||
|
||||
impl From<tonic::Status> for Error {
|
||||
fn from(status: tonic::Status) -> Self {
|
||||
Error::Transport {
|
||||
status: status.code().to_string(),
|
||||
message: status.message().to_string()
|
||||
|
||||
|
||||
pub type ProcedureResult<T> = std::result::Result<T, ProcedureError>;
|
||||
|
||||
#[derive(Debug, thiserror::Error)]
|
||||
pub enum ProcedureError {
|
||||
#[error("server rejected procedure with error: {0}")]
|
||||
Server(#[from] tonic::Status)
|
||||
}
|
||||
|
||||
|
||||
|
||||
pub type ConnectionResult<T> = std::result::Result<T, ConnectionError>;
|
||||
|
||||
#[derive(Debug, thiserror::Error)]
|
||||
pub enum ConnectionError {
|
||||
#[error("network error: {0}")]
|
||||
Transport(#[from] tonic::transport::Error),
|
||||
|
||||
#[error("server rejected connection attempt: {0}")]
|
||||
Procedure(#[from] tonic::Status),
|
||||
}
|
||||
|
||||
impl From<ProcedureError> for ConnectionError {
|
||||
fn from(value: ProcedureError) -> Self {
|
||||
match value {
|
||||
ProcedureError::Server(x) => Self::Procedure(x)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl From<tonic::transport::Error> for Error {
|
||||
fn from(err: tonic::transport::Error) -> Self {
|
||||
Error::Transport {
|
||||
status: tonic::Code::Unknown.to_string(),
|
||||
message: format!("underlying transport error: {:?}", err)
|
||||
}
|
||||
|
||||
|
||||
pub type ControllerResult<T> = std::result::Result<T, ControllerError>;
|
||||
|
||||
#[derive(Debug, thiserror::Error)]
|
||||
pub enum ControllerError {
|
||||
#[error("worker is already stopped")]
|
||||
Stopped,
|
||||
|
||||
#[error("worker stopped before completing requested operation")]
|
||||
Unfulfilled,
|
||||
}
|
||||
|
||||
impl<T> From<tokio::sync::mpsc::error::SendError<T>> for ControllerError {
|
||||
fn from(_: tokio::sync::mpsc::error::SendError<T>) -> Self {
|
||||
Self::Stopped
|
||||
}
|
||||
}
|
||||
|
||||
impl<T> From<tokio::sync::mpsc::error::SendError<T>> for Error {
|
||||
fn from(_value: tokio::sync::mpsc::error::SendError<T>) -> Self {
|
||||
Error::Channel { send: true }
|
||||
}
|
||||
}
|
||||
|
||||
impl<T> From<tokio::sync::watch::error::SendError<T>> for Error {
|
||||
fn from(_value: tokio::sync::watch::error::SendError<T>) -> Self {
|
||||
Error::Channel { send: true }
|
||||
}
|
||||
}
|
||||
|
||||
impl From<tokio::sync::broadcast::error::RecvError> for Error {
|
||||
fn from(_value: tokio::sync::broadcast::error::RecvError) -> Self {
|
||||
Error::Channel { send: false }
|
||||
}
|
||||
}
|
||||
|
||||
impl From<tokio::sync::oneshot::error::RecvError> for Error {
|
||||
fn from(_value: tokio::sync::oneshot::error::RecvError) -> Self {
|
||||
Error::Channel { send: false }
|
||||
}
|
||||
}
|
||||
|
||||
impl From<tokio::sync::watch::error::RecvError> for Error {
|
||||
fn from(_value: tokio::sync::watch::error::RecvError) -> Self {
|
||||
Error::Channel { send: false }
|
||||
impl From<tokio::sync::oneshot::error::RecvError> for ControllerError {
|
||||
fn from(_: tokio::sync::oneshot::error::RecvError) -> Self {
|
||||
Self::Unfulfilled
|
||||
}
|
||||
}
|
||||
|
|
42
src/ext.rs
42
src/ext.rs
|
@ -1,4 +1,4 @@
|
|||
use crate::{Error, api::Controller};
|
||||
use crate::{api::Controller, errors::ControllerResult};
|
||||
use tokio::sync::mpsc;
|
||||
|
||||
/// invoke .poll() on all given buffer controllers and wait, returning the first one ready
|
||||
|
@ -15,7 +15,7 @@ pub async fn select_buffer(
|
|||
buffers: &[crate::buffer::Controller],
|
||||
timeout: Option<std::time::Duration>,
|
||||
runtime: &tokio::runtime::Runtime
|
||||
) -> crate::Result<Option<crate::buffer::Controller>> {
|
||||
) -> ControllerResult<Option<crate::buffer::Controller>> {
|
||||
let (tx, mut rx) = mpsc::unbounded_channel();
|
||||
let mut tasks = Vec::new();
|
||||
for buffer in buffers {
|
||||
|
@ -24,7 +24,7 @@ pub async fn select_buffer(
|
|||
tasks.push(runtime.spawn(async move {
|
||||
match _buffer.poll().await {
|
||||
Ok(()) => _tx.send(Ok(Some(_buffer))),
|
||||
Err(_) => _tx.send(Err(Error::Channel { send: true })),
|
||||
Err(e) => _tx.send(Err(e)),
|
||||
}
|
||||
}))
|
||||
}
|
||||
|
@ -37,7 +37,7 @@ pub async fn select_buffer(
|
|||
}
|
||||
loop {
|
||||
match rx.recv().await {
|
||||
None => return Err(Error::Channel { send: false }),
|
||||
None => return Err(crate::errors::ControllerError::Unfulfilled),
|
||||
Some(Err(_)) => continue, // TODO log errors maybe?
|
||||
Some(Ok(x)) => {
|
||||
for t in tasks {
|
||||
|
@ -99,3 +99,37 @@ impl<T> crate::api::controller::CallbackHandle for CallbackHandleWatch<T> {
|
|||
self.0.send_replace(None);
|
||||
}
|
||||
}*/
|
||||
|
||||
/// an error which can be ignored with just a warning entry
|
||||
pub trait IgnorableError {
|
||||
fn unwrap_or_warn(self, msg: &str);
|
||||
}
|
||||
|
||||
impl<T, E> IgnorableError for std::result::Result<T, E>
|
||||
where E : std::fmt::Debug {
|
||||
fn unwrap_or_warn(self, msg: &str) {
|
||||
match self {
|
||||
Ok(_) => {},
|
||||
Err(e) => tracing::warn!("{}: {:?}", msg, e),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/// an error which can be ignored with just a warning entry and returning the default value
|
||||
pub trait IgnorableDefaultableError<T> {
|
||||
fn unwrap_or_warn_default(self, msg: &str) -> T;
|
||||
}
|
||||
|
||||
impl<T, E> IgnorableDefaultableError<T> for std::result::Result<T, E>
|
||||
where E : std::fmt::Display, T: Default {
|
||||
fn unwrap_or_warn_default(self, msg: &str) -> T {
|
||||
match self {
|
||||
Ok(x) => x,
|
||||
Err(e) => {
|
||||
tracing::warn!("{}: {}", msg, e);
|
||||
T::default()
|
||||
},
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
@ -138,8 +138,7 @@ pub use client::Client;
|
|||
|
||||
/// crate error types and helpers
|
||||
pub mod errors;
|
||||
pub use errors::Error;
|
||||
pub use errors::Result;
|
||||
pub use errors::{Error, Result};
|
||||
|
||||
/// all-in-one imports : `use codemp::prelude::*;`
|
||||
pub mod prelude;
|
||||
|
|
|
@ -2,11 +2,6 @@
|
|||
//!
|
||||
//! all-in-one renamed imports with `use codemp::prelude::*`
|
||||
|
||||
pub use crate::{
|
||||
Error as CodempError,
|
||||
Result as CodempResult,
|
||||
};
|
||||
|
||||
pub use crate::api::{
|
||||
Controller as CodempController,
|
||||
TextChange as CodempTextChange,
|
||||
|
|
|
@ -7,6 +7,8 @@ use tonic::{
|
|||
transport::{Channel, Endpoint},
|
||||
};
|
||||
|
||||
use crate::errors::ConnectionResult;
|
||||
|
||||
type AuthedService = InterceptedService<Channel, WorkspaceInterceptor>;
|
||||
|
||||
#[derive(Debug)]
|
||||
|
@ -21,7 +23,7 @@ impl Services {
|
|||
dest: &str,
|
||||
session: tokio::sync::watch::Receiver<codemp_proto::common::Token>,
|
||||
workspace: tokio::sync::watch::Receiver<codemp_proto::common::Token>,
|
||||
) -> crate::Result<Self> {
|
||||
) -> ConnectionResult<Self> {
|
||||
let channel = Endpoint::from_shared(dest.to_string())?.connect().await?;
|
||||
let inter = WorkspaceInterceptor { session, workspace };
|
||||
Ok(Self {
|
||||
|
|
|
@ -1,5 +1,10 @@
|
|||
use crate::{
|
||||
api::{controller::ControllerWorker, Controller, Event, User}, buffer::{self, worker::BufferWorker}, cursor::{self, worker::CursorWorker}, ext::InternallyMutable, workspace::service::Services
|
||||
api::{controller::ControllerWorker, Controller, Event, User},
|
||||
buffer::{self, worker::BufferWorker},
|
||||
cursor::{self, worker::CursorWorker},
|
||||
errors::{ConnectionResult, ControllerResult, ProcedureResult},
|
||||
ext::InternallyMutable,
|
||||
workspace::service::Services
|
||||
};
|
||||
|
||||
use codemp_proto::{
|
||||
|
@ -47,7 +52,7 @@ impl Workspace {
|
|||
dest: &str,
|
||||
token: Token,
|
||||
claims: tokio::sync::watch::Receiver<codemp_proto::common::Token>, // TODO ughh receiving this
|
||||
) -> crate::Result<Self> {
|
||||
) -> ConnectionResult<Self> {
|
||||
let workspace_claim = InternallyMutable::new(token);
|
||||
let services = Services::try_new(dest, claims, workspace_claim.channel()).await?;
|
||||
let ws_stream = services.ws().attach(Empty {}).await?.into_inner();
|
||||
|
@ -141,7 +146,7 @@ impl Workspace {
|
|||
|
||||
impl Workspace {
|
||||
/// create a new buffer in current workspace
|
||||
pub async fn create(&self, path: &str) -> crate::Result<()> {
|
||||
pub async fn create(&self, path: &str) -> ProcedureResult<()> {
|
||||
let mut workspace_client = self.0.services.ws();
|
||||
workspace_client
|
||||
.create_buffer(tonic::Request::new(BufferNode {
|
||||
|
@ -162,7 +167,7 @@ impl Workspace {
|
|||
///
|
||||
/// to interact with such buffer use [crate::api::Controller::send] or
|
||||
/// [crate::api::Controller::recv] to exchange [crate::api::TextChange]
|
||||
pub async fn attach(&self, path: &str) -> crate::Result<buffer::Controller> {
|
||||
pub async fn attach(&self, path: &str) -> ConnectionResult<buffer::Controller> {
|
||||
let mut worskspace_client = self.0.services.ws();
|
||||
let request = tonic::Request::new(BufferNode {
|
||||
path: path.to_string(),
|
||||
|
@ -216,18 +221,19 @@ impl Workspace {
|
|||
}
|
||||
|
||||
/// await next workspace [crate::api::Event] and return it
|
||||
pub async fn event(&self) -> crate::Result<Event> {
|
||||
// TODO this method is weird and ugly, can we make it more standard?
|
||||
pub async fn event(&self) -> ControllerResult<Event> {
|
||||
self.0
|
||||
.events
|
||||
.lock()
|
||||
.await
|
||||
.recv()
|
||||
.await
|
||||
.ok_or(crate::Error::Channel { send: false })
|
||||
.ok_or(crate::errors::ControllerError::Unfulfilled)
|
||||
}
|
||||
|
||||
/// fetch a list of all buffers in a workspace
|
||||
pub async fn fetch_buffers(&self) -> crate::Result<()> {
|
||||
pub async fn fetch_buffers(&self) -> ProcedureResult<()> {
|
||||
let mut workspace_client = self.0.services.ws();
|
||||
let buffers = workspace_client
|
||||
.list_buffers(tonic::Request::new(Empty {}))
|
||||
|
@ -244,7 +250,7 @@ impl Workspace {
|
|||
}
|
||||
|
||||
/// fetch a list of all users in a workspace
|
||||
pub async fn fetch_users(&self) -> crate::Result<()> {
|
||||
pub async fn fetch_users(&self) -> ProcedureResult<()> {
|
||||
let mut workspace_client = self.0.services.ws();
|
||||
let users = BTreeSet::from_iter(
|
||||
workspace_client
|
||||
|
@ -267,7 +273,7 @@ impl Workspace {
|
|||
/// get a list of the users attached to a specific buffer
|
||||
///
|
||||
/// TODO: discuss implementation details
|
||||
pub async fn list_buffer_users(&self, path: &str) -> crate::Result<Vec<User>> {
|
||||
pub async fn list_buffer_users(&self, path: &str) -> ProcedureResult<Vec<User>> {
|
||||
let mut workspace_client = self.0.services.ws();
|
||||
let buffer_users = workspace_client
|
||||
.list_buffer_users(tonic::Request::new(BufferNode {
|
||||
|
@ -284,7 +290,7 @@ impl Workspace {
|
|||
}
|
||||
|
||||
/// delete a buffer
|
||||
pub async fn delete(&self, path: &str) -> crate::Result<()> {
|
||||
pub async fn delete(&self, path: &str) -> ProcedureResult<()> {
|
||||
let mut workspace_client = self.0.services.ws();
|
||||
workspace_client
|
||||
.delete_buffer(tonic::Request::new(BufferNode {
|
||||
|
|
Loading…
Reference in a new issue