send the runtime user agent when polling for events (#180)

Author: bmoffatt

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "lambda_runtime_client"
-version = "0.2.2"
+version = "0.2.3"
 authors = ["Stefano Buliani", "David Barsky"]
 edition = "2018"
 description = "Client SDK for AWS Lambda's runtime APIs"
@@ -27,4 +27,4 @@ lambda_runtime_errors = { path = "../lambda-runtime-errors", version = "^0.1" }
 failure = "^0.1"
 
 [dev-dependencies]
-chrono = "^0.4"
+chrono = "^0.4"

lambda-runtime-client/Cargo.toml

@@ -171,12 +171,19 @@ impl<'ev> RuntimeClient {
     pub fn next_event(&self) -> Result<(Vec<u8>, EventContext), ApiError> {
         trace!("Polling for next event");
 
+        let req = Request::builder()
+            .method(Method::GET)
+            .uri(self.next_endpoint.clone())
+            .header(header::USER_AGENT, self.runtime_agent.clone())
+            .body(Body::from(""))
+            .unwrap();
+
         // We wait instead of processing the future asynchronously because AWS Lambda
         // itself enforces only one event per container at a time. No point in taking on
         // the additional complexity.
         let resp = self
             .http_client
-            .get(self.next_endpoint.clone())
+            .request(req)
             .wait()
             .context(ApiErrorKind::Unrecoverable("Could not fetch next event".to_string()))?;