Send ReadOnlyIntent when ApplicationIntent=ReadOnly specified (#297)

prisma · Jun 2, 2023 · 98943b2 · 98943b2
1 parent 3110856
commit 98943b2
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 0 deletions.
diff --git a/src/client/config.rs b/src/client/config.rs
@@ -31,6 +31,7 @@ pub struct Config {
     pub(crate) encryption: EncryptionLevel,
     pub(crate) trust: TrustConfig,
     pub(crate) auth: AuthMethod,
+    pub(crate) readonly: bool,
 }
 
 #[derive(Clone, Debug)]
@@ -62,6 +63,7 @@ impl Default for Config {
             encryption: EncryptionLevel::NotSupported,
             trust: TrustConfig::Default,
             auth: AuthMethod::None,
+            readonly: false,
         }
     }
 }
@@ -161,6 +163,13 @@ impl Config {
         self.auth = auth;
     }
 
+    /// Sets ApplicationIntent readonly.
+    ///
+    /// - Defaults to `false`.
+    pub fn readonly(&mut self, readnoly: bool) {
+        self.readonly = readnoly;
+    }
+
     pub(crate) fn get_host(&self) -> &str {
         self.host
             .as_deref()
@@ -257,6 +266,8 @@ impl Config {
 
         builder.encryption(s.encrypt()?);
 
+        builder.readonly(s.readonly());
+
         Ok(builder)
     }
 }
@@ -369,4 +380,11 @@ pub(crate) trait ConfigString {
             )),
         }
     }
+
+    fn readonly(&self) -> bool {
+        self.dict()
+            .get("applicationintent")
+            .filter(|val| *val == "ReadOnly")
+            .is_some()
+    }
 }
diff --git a/src/client/connection.rs b/src/client/connection.rs
@@ -105,6 +105,7 @@ impl<S: AsyncRead + AsyncWrite + Unpin + Send> Connection<S> {
                 config.database,
                 config.host,
                 config.application_name,
+                config.readonly,
                 prelogin,
             )
             .await?;
@@ -290,6 +291,7 @@ impl<S: AsyncRead + AsyncWrite + Unpin + Send> Connection<S> {
         db: Option<String>,
         server_name: Option<String>,
         application_name: Option<String>,
+        readonly: bool,
         prelogin: PreloginMessage,
     ) -> crate::Result<Self> {
         let mut login_message = LoginMessage::new();
@@ -306,6 +308,8 @@ impl<S: AsyncRead + AsyncWrite + Unpin + Send> Connection<S> {
             login_message.app_name(app_name);
         }
 
+        login_message.readonly(readonly);
+
         match auth {
             #[cfg(all(windows, feature = "winauth"))]
             AuthMethod::Integrated => {

diff --git a/src/tds/codec/login.rs b/src/tds/codec/login.rs
@@ -232,6 +232,14 @@ impl<'a> LoginMessage<'a> {
             nonce,
         })
     }
+
+    pub fn readonly(&mut self, readonly: bool) {
+        if readonly {
+            self.type_flags.insert(LoginTypeFlag::ReadOnlyIntent);
+        } else {
+            self.type_flags.remove(LoginTypeFlag::ReadOnlyIntent);
+        }
+    }
 }
 
 impl<'a> Encode<BytesMut> for LoginMessage<'a> {