@@ -72,6 +72,21 @@ pub enum LoadBalanceHosts {
7272Random ,
7373}
7474
75+ /// Replication mode configuration.
76+ ///
77+ /// It is recommended that you use a PostgreSQL server patch version
78+ /// of at least: 14.0, 13.2, 12.6, 11.11, 10.16, 9.6.21, or
79+ /// 9.5.25. Earlier patch levels have a bug that doesn't properly
80+ /// handle pipelined requests after streaming has stopped.
81+ #[ derive( Debug , Copy , Clone , PartialEq , Eq ) ]
82+ #[ non_exhaustive]
83+ pub enum ReplicationMode {
84+ /// Physical replication.
85+ Physical ,
86+ /// Logical replication.
87+ Logical ,
88+ }
89+
7590/// A host specification.
7691#[ derive( Debug , Clone , PartialEq , Eq ) ]
7792pub enum Host {
@@ -209,6 +224,7 @@ pub struct Config {
209224pub ( crate ) target_session_attrs : TargetSessionAttrs ,
210225pub ( crate ) channel_binding : ChannelBinding ,
211226pub ( crate ) load_balance_hosts : LoadBalanceHosts ,
227+ pub ( crate ) replication_mode : Option < ReplicationMode > ,
212228}
213229
214230impl Default for Config {
@@ -242,6 +258,7 @@ impl Config {
242258target_session_attrs : TargetSessionAttrs :: Any ,
243259channel_binding : ChannelBinding :: Prefer ,
244260load_balance_hosts : LoadBalanceHosts :: Disable ,
261+ replication_mode : None ,
245262}
246263}
247264
@@ -524,6 +541,22 @@ impl Config {
524541self . load_balance_hosts
525542}
526543
544+ /// Set replication mode.
545+ ///
546+ /// It is recommended that you use a PostgreSQL server patch version
547+ /// of at least: 14.0, 13.2, 12.6, 11.11, 10.16, 9.6.21, or
548+ /// 9.5.25. Earlier patch levels have a bug that doesn't properly
549+ /// handle pipelined requests after streaming has stopped.
550+ pub fn replication_mode ( & mut self , replication_mode : ReplicationMode ) ->& mut Config {
551+ self . replication_mode =Some ( replication_mode) ;
552+ self
553+ }
554+
555+ /// Get replication mode.
556+ pub fn get_replication_mode ( & self ) ->Option < ReplicationMode > {
557+ self . replication_mode
558+ }
559+
527560fn param ( & mut self , key : & str , value : & str ) ->Result < ( ) , Error > {
528561match key{
529562"user" =>{
@@ -660,6 +693,17 @@ impl Config {
660693} ;
661694self . load_balance_hosts ( load_balance_hosts) ;
662695}
696+ "replication" =>{
697+ let mode =match value{
698+ "off" =>None ,
699+ "true" =>Some ( ReplicationMode :: Physical ) ,
700+ "database" =>Some ( ReplicationMode :: Logical ) ,
701+ _ =>return Err ( Error :: config_parse ( Box :: new ( InvalidValue ( "replication" ) ) ) ) ,
702+ } ;
703+ if let Some ( mode) = mode{
704+ self . replication_mode ( mode) ;
705+ }
706+ }
663707 key =>{
664708return Err ( Error :: config_parse ( Box :: new ( UnknownOption (
665709 key. to_string ( ) ,
@@ -744,6 +788,7 @@ impl fmt::Debug for Config {
744788 config_dbg
745789. field ( "target_session_attrs" , & self . target_session_attrs )
746790. field ( "channel_binding" , & self . channel_binding )
791+ . field ( "replication" , & self . replication_mode )
747792. finish ( )
748793}
749794}