Struct Mockgres 

pub struct Mockgres {
    pub db: Arc<RwLock<Db>>,
    pub txn_manager: Arc<TransactionManager>,
    /* private fields */
}

Fields

db: Arc<RwLock<Db>>

txn_manager: Arc<TransactionManager>

Implementations

impl Mockgres

pub fn new(db: Arc<RwLock<Db>>) -> Self

pub fn new_with_config(db: Arc<RwLock<Db>>, config: ServerConfig) -> Self

pub fn with_config(config: ServerConfig) -> Self

pub fn config(&self) -> &ServerConfig

pub async fn serve(self: Arc<Self>, addr: SocketAddr) -> Result<()>

Trait Implementations

impl Clone for Mockgres

fn clone(&self) -> Mockgres

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Default for Mockgres

fn default() -> Self

Returns the “default value” for a type.

impl ExtendedQueryHandler for Mockgres

type Statement = StatementPlan

type QueryParser = PgQueryParserAdapter

fn query_parser(&self) -> Arc<Self::QueryParser>

Get a reference to associated QueryParser implementation

fn on_execute<'life0, 'life1, 'async_trait, C>( &'life0 self, client: &'life1 mut C, message: Execute, ) -> Pin<Box<dyn Future<Output = PgWireResult<()>> + Send + 'async_trait>>

where C: ClientInfo + ClientPortalStore + Sink<PgWireBackendMessage> + Unpin + Send + Sync + 'async_trait, C::PortalStore: PortalStore<Statement = Self::Statement>, C::Error: Debug, PgWireError: From<<C as Sink<PgWireBackendMessage>>::Error>, Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Called when client sends execute command.

fn do_describe_statement<'life0, 'life1, 'life2, 'async_trait, C>( &'life0 self, client: &'life1 mut C, target: &'life2 StoredStatement<Self::Statement>, ) -> Pin<Box<dyn Future<Output = PgWireResult<DescribeStatementResponse>> + Send + 'async_trait>>

where C: ClientInfo + ClientPortalStore + Sink<PgWireBackendMessage> + Unpin + Send + Sync + 'async_trait, C::PortalStore: PortalStore<Statement = Self::Statement>, C::Error: Debug, PgWireError: From<<C as Sink<PgWireBackendMessage>>::Error>, Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait,

Return resultset metadata without actually executing statement

fn do_describe_portal<'life0, 'life1, 'life2, 'async_trait, C>( &'life0 self, client: &'life1 mut C, portal: &'life2 Portal<Self::Statement>, ) -> Pin<Box<dyn Future<Output = PgWireResult<DescribePortalResponse>> + Send + 'async_trait>>

where C: ClientInfo + ClientPortalStore + Sink<PgWireBackendMessage> + Unpin + Send + Sync + 'async_trait, C::PortalStore: PortalStore<Statement = Self::Statement>, C::Error: Debug, PgWireError: From<<C as Sink<PgWireBackendMessage>>::Error>, Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait,

Return resultset metadata without actually executing portal

fn do_query<'life0, 'life1, 'life2, 'async_trait, C>( &'life0 self, client: &'life1 mut C, portal: &'life2 Portal<Self::Statement>, _max_rows: usize, ) -> Pin<Box<dyn Future<Output = PgWireResult<Response>> + Send + 'async_trait>>

where C: ClientInfo + ClientPortalStore + Sink<PgWireBackendMessage> + Unpin + Send + Sync + 'async_trait, C::PortalStore: PortalStore<Statement = Self::Statement>, C::Error: Debug, PgWireError: From<<C as Sink<PgWireBackendMessage>>::Error>, Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait,

This is the main implementation for query execution. Context has been provided:

fn on_parse<'life0, 'life1, 'async_trait, C>( &'life0 self, client: &'life1 mut C, message: Parse, ) -> Pin<Box<dyn Future<Output = Result<(), PgWireError>> + Send + 'async_trait>>

where 'life0: 'async_trait, 'life1: 'async_trait, C: ClientInfo + ClientPortalStore + Sink<PgWireBackendMessage> + Unpin + Send + Sync + 'async_trait, <C as ClientPortalStore>::PortalStore: PortalStore<Statement = Self::Statement>, <C as Sink<PgWireBackendMessage>>::Error: Debug, PgWireError: From<<C as Sink<PgWireBackendMessage>>::Error>, Self: 'async_trait,

Called when client sends parse command.

fn on_bind<'life0, 'life1, 'async_trait, C>( &'life0 self, client: &'life1 mut C, message: Bind, ) -> Pin<Box<dyn Future<Output = Result<(), PgWireError>> + Send + 'async_trait>>

where 'life0: 'async_trait, 'life1: 'async_trait, C: ClientInfo + ClientPortalStore + Sink<PgWireBackendMessage> + Unpin + Send + Sync + 'async_trait, <C as ClientPortalStore>::PortalStore: PortalStore<Statement = Self::Statement>, <C as Sink<PgWireBackendMessage>>::Error: Debug, PgWireError: From<<C as Sink<PgWireBackendMessage>>::Error>, Self: 'async_trait,

Called when client sends bind command.

fn _on_execute<'life0, 'life1, 'async_trait, C>( &'life0 self, client: &'life1 mut C, message: Execute, ) -> Pin<Box<dyn Future<Output = Result<(), PgWireError>> + Send + 'async_trait>>

where 'life0: 'async_trait, 'life1: 'async_trait, C: ClientInfo + ClientPortalStore + Sink<PgWireBackendMessage> + Unpin + Send + Sync + 'async_trait, <C as ClientPortalStore>::PortalStore: PortalStore<Statement = Self::Statement>, <C as Sink<PgWireBackendMessage>>::Error: Debug, PgWireError: From<<C as Sink<PgWireBackendMessage>>::Error>, Self: 'async_trait,

The default implementation of on_execute.

fn on_describe<'life0, 'life1, 'async_trait, C>( &'life0 self, client: &'life1 mut C, message: Describe, ) -> Pin<Box<dyn Future<Output = Result<(), PgWireError>> + Send + 'async_trait>>

where 'life0: 'async_trait, 'life1: 'async_trait, C: ClientInfo + ClientPortalStore + Sink<PgWireBackendMessage> + Unpin + Send + Sync + 'async_trait, <C as ClientPortalStore>::PortalStore: PortalStore<Statement = Self::Statement>, <C as Sink<PgWireBackendMessage>>::Error: Debug, PgWireError: From<<C as Sink<PgWireBackendMessage>>::Error>, Self: 'async_trait,

Called when client sends describe command.

fn _on_describe<'life0, 'life1, 'async_trait, C>( &'life0 self, client: &'life1 mut C, message: Describe, ) -> Pin<Box<dyn Future<Output = Result<(), PgWireError>> + Send + 'async_trait>>

where 'life0: 'async_trait, 'life1: 'async_trait, C: ClientInfo + ClientPortalStore + Sink<PgWireBackendMessage> + Unpin + Send + Sync + 'async_trait, <C as ClientPortalStore>::PortalStore: PortalStore<Statement = Self::Statement>, <C as Sink<PgWireBackendMessage>>::Error: Debug, PgWireError: From<<C as Sink<PgWireBackendMessage>>::Error>, Self: 'async_trait,

The default implementation of on_describe

fn on_flush<'life0, 'life1, 'async_trait, C>( &'life0 self, client: &'life1 mut C, _message: Flush, ) -> Pin<Box<dyn Future<Output = Result<(), PgWireError>> + Send + 'async_trait>>

where 'life0: 'async_trait, 'life1: 'async_trait, C: ClientInfo + Sink<PgWireBackendMessage> + Unpin + Send + Sync + 'async_trait, <C as Sink<PgWireBackendMessage>>::Error: Debug, PgWireError: From<<C as Sink<PgWireBackendMessage>>::Error>, Self: 'async_trait,

Called when client sends flush command.

fn on_sync<'life0, 'life1, 'async_trait, C>( &'life0 self, client: &'life1 mut C, _message: Sync, ) -> Pin<Box<dyn Future<Output = Result<(), PgWireError>> + Send + 'async_trait>>

where 'life0: 'async_trait, 'life1: 'async_trait, C: ClientInfo + Sink<PgWireBackendMessage> + Unpin + Send + Sync + 'async_trait, <C as Sink<PgWireBackendMessage>>::Error: Debug, PgWireError: From<<C as Sink<PgWireBackendMessage>>::Error>, Self: 'async_trait,

Called when client sends sync command.

fn on_close<'life0, 'life1, 'async_trait, C>( &'life0 self, client: &'life1 mut C, message: Close, ) -> Pin<Box<dyn Future<Output = Result<(), PgWireError>> + Send + 'async_trait>>

where 'life0: 'async_trait, 'life1: 'async_trait, C: ClientInfo + ClientPortalStore + Sink<PgWireBackendMessage> + Unpin + Send + Sync + 'async_trait, <C as ClientPortalStore>::PortalStore: PortalStore<Statement = Self::Statement>, <C as Sink<PgWireBackendMessage>>::Error: Debug, PgWireError: From<<C as Sink<PgWireBackendMessage>>::Error>, Self: 'async_trait,

Called when client sends close command.

impl NoopStartupHandler for Mockgres

fn post_startup<'life0, 'life1, 'async_trait, C>( &'life0 self, client: &'life1 mut C, message: PgWireFrontendMessage, ) -> Pin<Box<dyn Future<Output = PgWireResult<()>> + Send + 'async_trait>>

where C: ClientInfo + Sink<PgWireBackendMessage> + Unpin + Send + 'async_trait, C::Error: Debug, PgWireError: From<<C as Sink<PgWireBackendMessage>>::Error>, Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

impl PgWireServerHandlers for Mockgres

fn simple_query_handler(&self) -> Arc<impl SimpleQueryHandler>

fn extended_query_handler(&self) -> Arc<impl ExtendedQueryHandler>

fn startup_handler(&self) -> Arc<impl StartupHandler>

fn copy_handler(&self) -> Arc<impl CopyHandler>

fn error_handler(&self) -> Arc<impl ErrorHandler>

fn cancel_handler(&self) -> Arc<impl CancelHandler>

impl SimpleQueryHandler for Mockgres

fn do_query<'life0, 'life1, 'life2, 'async_trait, C>( &'life0 self, client: &'life1 mut C, query: &'life2 str, ) -> Pin<Box<dyn Future<Output = PgWireResult<Vec<Response>>> + Send + 'async_trait>>

where C: ClientInfo + Sink<PgWireBackendMessage> + Unpin + Send + Sync + 'async_trait, C::Error: Debug, PgWireError: From<<C as Sink<PgWireBackendMessage>>::Error>, Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait,

Provide your query implementation using the incoming query string.

fn on_query<'life0, 'life1, 'async_trait, C>( &'life0 self, client: &'life1 mut C, query: Query, ) -> Pin<Box<dyn Future<Output = Result<(), PgWireError>> + Send + 'async_trait>>

where 'life0: 'async_trait, 'life1: 'async_trait, C: ClientInfo + ClientPortalStore + Sink<PgWireBackendMessage> + Unpin + Send + Sync + 'async_trait, <C as Sink<PgWireBackendMessage>>::Error: Debug, PgWireError: From<<C as Sink<PgWireBackendMessage>>::Error>, Self: 'async_trait,

Executed on Query request arrived. This is how postgres respond to simple query. The default implementation calls do_query with the incoming query string.

fn _on_query<'life0, 'life1, 'async_trait, C>( &'life0 self, client: &'life1 mut C, query: Query, ) -> Pin<Box<dyn Future<Output = Result<(), PgWireError>> + Send + 'async_trait>>

where 'life0: 'async_trait, 'life1: 'async_trait, C: ClientInfo + ClientPortalStore + Sink<PgWireBackendMessage> + Unpin + Send + Sync + 'async_trait, <C as Sink<PgWireBackendMessage>>::Error: Debug, PgWireError: From<<C as Sink<PgWireBackendMessage>>::Error>, Self: 'async_trait,

This is the default implementation of on_query. If you want to override on_query with your own pre/post processing logic, you can call this function.