Skip to content

Commit f3b54f5

Browse files
committed
(fix) revert changes for v1 username and password
1 parent 7e8c712 commit f3b54f5

File tree

1 file changed

+12
-16
lines changed

1 file changed

+12
-16
lines changed

influxdb/src/client/mod.rs

Lines changed: 12 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,6 @@ pub struct Client {
3030
pub(crate) url: Arc<String>,
3131
pub(crate) parameters: Arc<HashMap<&'static str, String>>,
3232
pub(crate) client: SurfClient,
33-
pub(crate) username: Option<String>,
34-
pub(crate) password: Option<String>,
3533
pub(crate) token: Option<String>,
3634
}
3735

@@ -61,8 +59,6 @@ impl Client {
6159
url: Arc::new(url.into()),
6260
parameters: Arc::new(parameters),
6361
client: SurfClient::new(),
64-
username: None,
65-
password: None,
6662
token: None,
6763
}
6864
}
@@ -86,8 +82,10 @@ impl Client {
8682
S1: Into<String>,
8783
S2: Into<String>,
8884
{
89-
self.username = Some(username.into());
90-
self.password = Some(password.into());
85+
let mut with_auth = self.parameters.as_ref().clone();
86+
with_auth.insert("u", username.into());
87+
with_auth.insert("p", password.into());
88+
self.parameters = Arc::new(with_auth);
9189
self
9290
}
9391

@@ -181,11 +179,11 @@ impl Client {
181179
error: err.to_string(),
182180
})?;
183181

182+
let mut parameters = self.parameters.as_ref().clone();
184183
let request_builder = match q.get_type() {
185184
QueryType::ReadQuery => {
186185
let read_query = query.get();
187186
let url = &format!("{}/query", &self.url);
188-
let mut parameters = self.parameters.as_ref().clone();
189187
parameters.insert("q", read_query.clone());
190188

191189
if read_query.contains("SELECT") || read_query.contains("SHOW") {
@@ -206,7 +204,7 @@ impl Client {
206204
error: err.to_string(),
207205
})?;
208206

209-
let request = self.try_authed(request_builder).build();
207+
let request = self.auth_if_needed(request_builder).build();
210208
let mut res = self
211209
.client
212210
.send(request)
@@ -238,10 +236,8 @@ impl Client {
238236
Ok(s)
239237
}
240238

241-
fn try_authed(&self, rb: RequestBuilder) -> RequestBuilder {
242-
if let (Some(ref username), Some(ref password)) = (&self.username, &self.password) {
243-
rb.header("Authorization", format!("Basic {}:{}", username, password))
244-
} else if let Some(ref token) = self.token {
239+
fn auth_if_needed(&self, rb: RequestBuilder) -> RequestBuilder {
240+
if let Some(ref token) = self.token {
245241
rb.header("Authorization", format!("Token {}", token))
246242
} else {
247243
rb
@@ -263,14 +259,14 @@ mod tests {
263259
#[test]
264260
fn test_with_auth() {
265261
let client = Client::new("http://localhost:8068", "database");
266-
assert_eq!(client.parameters.len(), 1);
262+
assert_eq!(client.parameters.len(), 3);
267263
assert_eq!(client.parameters.get("db").unwrap(), "database");
268264

269265
let with_auth = client.with_auth("username", "password");
270-
assert_eq!(with_auth.parameters.len(), 1);
266+
assert_eq!(with_auth.parameters.len(), 3);
271267
assert_eq!(with_auth.parameters.get("db").unwrap(), "database");
272-
assert_eq!(with_auth.username.unwrap(), "username");
273-
assert_eq!(with_auth.password.unwrap(), "password");
268+
assert_eq!(with_auth.parameters.get("u").unwrap(), "username");
269+
assert_eq!(with_auth.parameters.get("p").unwrap(), "password");
274270

275271
let client = Client::new("http://localhost:8068", "database");
276272
let with_auth = client.with_token("token");

0 commit comments

Comments
 (0)