Skip to content

Commit ee5b6c8

Browse files
chore: minor cleanup
1 parent 3739fbb commit ee5b6c8

1 file changed

Lines changed: 14 additions & 12 deletions

File tree

sdk/rust/src/dstack_client.rs

Lines changed: 14 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -156,13 +156,10 @@ impl BaseClient for DstackClient {}
156156
impl DstackClient {
157157
pub fn new(endpoint: Option<&str>) -> Self {
158158
let endpoint = get_endpoint(endpoint);
159-
let (base_url, client) = if
160-
endpoint.starts_with("http://") ||
161-
endpoint.starts_with("https://")
162-
{
163-
(endpoint.clone(), ClientKind::Http)
164-
} else {
165-
("http://localhost".to_string(), ClientKind::Unix)
159+
let (base_url, client) = match endpoint {
160+
ref e if e.starts_with("http://") || e.starts_with("https://") =>
161+
(e.to_string(), ClientKind::Http),
162+
_ => ("http://localhost".to_string(), ClientKind::Unix),
166163
};
167164

168165
DstackClient { base_url, endpoint, client }
@@ -177,7 +174,12 @@ impl DstackClient {
177174
self.base_url.trim_end_matches('/'),
178175
path.trim_start_matches('/')
179176
);
180-
let res = client.post(&url).json(payload).send().await?.error_for_status()?;
177+
let res = client
178+
.post(&url)
179+
.json(payload)
180+
.header("Content-Type", "application/json")
181+
.send().await?
182+
.error_for_status()?;
181183
Ok(res.json().await?)
182184
}
183185
ClientKind::Unix => {
@@ -200,11 +202,11 @@ impl DstackClient {
200202
) -> Result<GetKeyResponse, Box<dyn Error>> {
201203
let data =
202204
json!({
203-
"path": path.clone().unwrap_or_default(),
205+
"path": path.unwrap_or_default(),
204206
"purpose": purpose.unwrap_or_default(),
205207
});
206208
let response = self.send_rpc_request("/GetKey", &data).await?;
207-
let response = serde_json::from_value::<GetKeyResponse>(response.clone())?;
209+
let response = serde_json::from_value::<GetKeyResponse>(response)?;
208210

209211
Ok(response)
210212
}
@@ -219,7 +221,7 @@ impl DstackClient {
219221
let hex_data = hex_encode(report_data);
220222
let data = json!({ "report_data": hex_data });
221223
let response = self.send_rpc_request("/GetQuote", &data).await?;
222-
let response = serde_json::from_value::<GetQuoteResponse>(response.clone())?;
224+
let response = serde_json::from_value::<GetQuoteResponse>(response)?;
223225

224226
Ok(response)
225227
}
@@ -256,7 +258,7 @@ impl DstackClient {
256258
"alt_names": alt_names.unwrap_or_default(),
257259
});
258260
let response = self.send_rpc_request("/GetTlsKey", &data).await?;
259-
let response = serde_json::from_value::<GetTlsKeyResponse>(response.clone())?;
261+
let response = serde_json::from_value::<GetTlsKeyResponse>(response)?;
260262

261263
Ok(response)
262264
}

0 commit comments

Comments
 (0)