add nativelink example to github actions (facebook#930)

Summary: Pull Request resolved: facebook#930

Differential Revision: D74001305

Author: samkevich

.github/actions/build_example_nativelink/action.yml

@@ -9,10 +9,13 @@ runs:
   - name: Build examples/remote_execution/nativelink directory using remote execution
     run: |-
       {
-      echo "[buck2_re_client]
-      engine_address       = grpc://scheduler-buck2.build-faster.nativelink.net:443
-      action_cache_address = grpc://cas-buck2.build-faster.nativelink.net:443
-      cas_address          = grpc://cas-buck2.build-faster.nativelink.net:443
+      echo "[buck2]
+      digest_algorithms = SHA256
+
+      [buck2_re_client]
+      engine_address       = grpcs://scheduler-buck2.build-faster.nativelink.net:443
+      action_cache_address = grpcs://cas-buck2.build-faster.nativelink.net:443
+      cas_address          = grpcs://cas-buck2.build-faster.nativelink.net:443
       http_headers         = x-nativelink-api-key:$NATIVELINK_HEADER_RW_KEY
       tls = true
       instance_name = main"

.github/workflows/build-and-test.yml

@@ -73,6 +73,9 @@ jobs:
     - uses: ./.github/actions/build_example_persistent_worker
       with:
         buildbuddyApiKey: ${{ secrets.BUILDBUDDY_API_KEY }}
+    - uses: ./.github/actions/build_example_nativelink
+      with:
+        NATIVELINK_HEADER_RW_KEY_SECRET: ${{ secrets.NATIVELINK_HEADER_RW_KEY_SECRET }}
     - uses: ./.github/actions/setup_reindeer
     - uses: ./.github/actions/build_bootstrap
   windows-build-examples:

examples/remote_execution/nativelink/platforms/defs.bzl

@@ -21,7 +21,7 @@ def _platforms(ctx):
             # Set those up based on what workers you've registered with NativeLink.
             remote_execution_properties = {
                 "OSFamily": "linux",
-                "container-image": "docker://buck2-github:latest",
+                "container-image": "docker://nativelink-toolchain-buck2:latest",
             },
             remote_execution_use_case = "buck2-default",
             remote_output_paths = "output_paths",

remote_execution/oss/re_grpc/src/client.rs

@@ -185,7 +185,7 @@ fn prepare_uri(uri: Uri, tls: bool) -> anyhow::Result<Uri> {
     // Many people (including Bazel), use grpc://, so we tolerate it.
 
     match uri.scheme_str() {
-        Some("grpc") | Some("dns") | Some("ipv4") | Some("ipv6") | None => {}
+        Some("grpc") | Some("grpcs") | Some("dns") | Some("ipv4") | Some("ipv6") | None => {}
         Some(scheme) => {
             return Err(anyhow::anyhow!(
                 "Invalid URI scheme: `{}` for `{}` (you should omit it)",
@@ -285,7 +285,7 @@ impl REClientBuilder {
             create_channel(opts.cas_address.clone()),
             create_channel(opts.engine_address.clone()),
         )
-        .await;
+            .await;
 
         let interceptor = InjectHeadersInterceptor::new(&opts.http_headers)?;
 
@@ -307,7 +307,7 @@ impl REClientBuilder {
                 &instance_name,
                 opts.max_total_batch_size,
             )
-            .await?
+                .await?
         } else {
             RECapabilities {
                 exec_enabled: true,
@@ -334,7 +334,7 @@ impl REClientBuilder {
                 cas.context("Error creating CAS client")?,
                 interceptor.dupe(),
             )
-            .max_decoding_message_size(max_decoding_msg_size),
+                .max_decoding_message_size(max_decoding_msg_size),
             execution_client: ExecutionClient::with_interceptor(
                 execution.context("Error creating Execution client")?,
                 interceptor.dupe(),
@@ -347,7 +347,7 @@ impl REClientBuilder {
                 bytestream.context("Error creating Bytestream client")?,
                 interceptor.dupe(),
             )
-            .max_decoding_message_size(max_decoding_msg_size),
+                .max_decoding_message_size(max_decoding_msg_size),
         };
 
         Ok(REClient::new(
@@ -672,7 +672,7 @@ impl REClient {
                             message: rpc_status.message,
                             group: TCodeReasonGroup::UNKNOWN,
                         }
-                        .into());
+                            .into());
                     }
                     OpResult::Response(any) => {
                         let execute_response_grpc: GExecuteResponse =
@@ -735,9 +735,9 @@ impl REClient {
         let stream = stream.map(move |mut r| {
             match &mut r {
                 Ok(ExecuteWithProgressResponse {
-                    execute_response: Some(ref mut response),
-                    ..
-                }) => {
+                       execute_response: Some(ref mut response),
+                       ..
+                   }) => {
                     response.action_digest = std::mem::take(&mut execute_request.action_digest);
                 }
                 _ => {}
@@ -786,7 +786,7 @@ impl REClient {
                 Ok(resp.into_inner())
             },
         )
-        .await
+            .await
     }
 
     pub async fn upload_blob_with_digest(
@@ -810,7 +810,7 @@ impl REClient {
                 ..Default::default()
             },
         )
-        .await?;
+            .await?;
         Ok(digest)
     }
 
@@ -851,7 +851,7 @@ impl REClient {
                 }
             },
         )
-        .await
+            .await
     }
 
     pub async fn get_digests_ttl(
@@ -1148,9 +1148,9 @@ async fn download_impl<Byt, BytRet, Cas>(
     bystream_fut: impl Fn(ReadRequest) -> Byt + Sync + Send + Copy,
 ) -> anyhow::Result<DownloadResponse>
 where
-    Byt: Future<Output = anyhow::Result<Pin<Box<BytRet>>>>,
-    BytRet: Stream<Item = Result<ReadResponse, tonic::Status>>,
-    Cas: Future<Output = anyhow::Result<BatchReadBlobsResponse>>,
+    Byt: Future<Output=anyhow::Result<Pin<Box<BytRet>>>>,
+    BytRet: Stream<Item=Result<ReadResponse, tonic::Status>>,
+    Cas: Future<Output=anyhow::Result<BatchReadBlobsResponse>>,
 {
     let bystream_fut = |digest: TDigest| async move {
         let hash = digest.hash;
@@ -1168,8 +1168,8 @@ where
             read_offset: 0,
             read_limit: 0,
         })
-        .await
-        .with_context(|| format!("Failed to read {} from Bytestream service", resource_name))
+            .await
+            .with_context(|| format!("Failed to read {} from Bytestream service", resource_name))
     };
 
     let inlined_digests = request.inlined_digests.unwrap_or_default();
@@ -1323,8 +1323,8 @@ async fn upload_impl<Byt, Cas>(
     bystream_fut: impl Fn(Vec<WriteRequest>) -> Byt + Sync + Send + Copy,
 ) -> anyhow::Result<UploadResponse>
 where
-    Cas: Future<Output = anyhow::Result<BatchUpdateBlobsResponse>> + Send,
-    Byt: Future<Output = anyhow::Result<WriteResponse>> + Send,
+    Cas: Future<Output=anyhow::Result<BatchUpdateBlobsResponse>> + Send,
+    Byt: Future<Output=anyhow::Result<WriteResponse>> + Send,
 {
     // NOTE if we stop recording blob_hashes, we can drop out a lot of allocations.
     let mut upload_futures: Vec<BoxFuture<anyhow::Result<Vec<String>>>> = vec![];
@@ -1562,8 +1562,8 @@ fn with_re_metadata<T>(
             platform: metadata.platform,
             use_case_id: Some(metadata.use_case_id),
         }
-        .encode(&mut encoded)
-        .expect("Encoding into a Vec cannot not fail");
+            .encode(&mut encoded)
+            .expect("Encoding into a Vec cannot not fail");
 
         msg.metadata_mut()
             .insert_bin("re-metadata-bin", MetadataValue::from_bytes(&encoded));
@@ -1584,8 +1584,8 @@ fn with_re_metadata<T>(
             target_id: "".to_owned(),
             configuration_id: "".to_owned(),
         }
-        .encode(&mut encoded)
-        .expect("Encoding into a Vec cannot not fail");
+            .encode(&mut encoded)
+            .expect("Encoding into a Vec cannot not fail");
 
         msg.metadata_mut().insert_bin(
             "build.bazel.remote.execution.v2.requestmetadata-bin",
@@ -1713,7 +1713,7 @@ mod tests {
             },
             |_digest| async move { anyhow::Ok(Box::pin(futures::stream::iter(vec![]))) },
         )
-        .await?;
+            .await?;
 
         assert_eq!(tokio::fs::read(&path1).await?, vec![1, 2, 3]);
         assert_eq!(tokio::fs::read(&path2).await?, vec![4, 5, 6]);
@@ -1827,7 +1827,7 @@ mod tests {
                 }
             },
         )
-        .await?;
+            .await?;
 
         assert_eq!(tokio::fs::read(&path1).await?, vec![1, 2, 3]);
         assert_eq!(tokio::fs::read(&path2).await?, blob_data);
@@ -1900,7 +1900,7 @@ mod tests {
             },
             |_digest| async move { anyhow::Ok(Box::pin(futures::stream::iter(vec![]))) },
         )
-        .await?;
+            .await?;
 
         let inlined_blobs = res.inlined_blobs.unwrap();
 
@@ -1986,7 +1986,7 @@ mod tests {
             },
             |_digest| async move { anyhow::Ok(Box::pin(futures::stream::iter(vec![]))) },
         )
-        .await?;
+            .await?;
 
         let inlined_blobs = res.inlined_blobs.unwrap();
 
@@ -2062,7 +2062,7 @@ mod tests {
                 }
             },
         )
-        .await?;
+            .await?;
 
         let inlined_blobs = res.inlined_blobs.unwrap();
 
@@ -2105,7 +2105,7 @@ mod tests {
             },
             |_digest| async move { anyhow::Ok(Box::pin(futures::stream::iter(vec![]))) },
         )
-        .await?;
+            .await?;
 
         let inlined_blobs = res.inlined_blobs.unwrap();
 
@@ -2140,7 +2140,7 @@ mod tests {
                 anyhow::Ok(Box::pin(futures::stream::iter(vec![])))
             },
         )
-        .await?;
+            .await?;
 
         Ok(())
     }
@@ -2219,7 +2219,7 @@ mod tests {
             },
             |_req| async { panic!("A Bytestream upload should not be triggered") },
         )
-        .await?;
+            .await?;
 
         Ok(())
     }
@@ -2311,7 +2311,7 @@ mod tests {
                 }
             },
         )
-        .await?;
+            .await?;
         Ok(())
     }
 
@@ -2386,7 +2386,7 @@ mod tests {
                 }
             },
         )
-        .await?;
+            .await?;
         Ok(())
     }
 
@@ -2430,7 +2430,7 @@ mod tests {
                 anyhow::Ok(WriteResponse { committed_size: 10 })
             },
         )
-        .await;
+            .await;
 
         let err: anyhow::Error = resp.unwrap_err();
         // can't compare the full message because tempfile is used
@@ -2492,7 +2492,7 @@ mod tests {
                 anyhow::Ok(WriteResponse { committed_size: 6 })
             },
         )
-        .await?;
+            .await?;
         Ok(())
     }
 
@@ -2531,7 +2531,7 @@ mod tests {
                 panic!("Not called");
             },
         )
-        .await;
+            .await;
 
         assert!(res.is_err()); // Should not panic.
 
@@ -2579,7 +2579,7 @@ mod tests {
                 anyhow::Ok(WriteResponse { committed_size: 3 })
             },
         )
-        .await?;
+            .await?;
 
         Ok(())
     }