A simple PreStop script for BuildKit that ensures all ongoing build complete before Kubernetes stops the pod.

BuildKit does not (currently) support graceful shutdowns when it receives a SIGTERM signal. This means that if a BuildKit pod is stopped while any builds are in progress, those builds will be terminated immediately, resulting in failed builds.

We can mitigate this by providing a script that will wait for all ongoing builds to finish before allowing the pod to stop.

1. Add the buildkit-prestop.sh script to your BuildKit pod's container image. For example:

   FROM moby/buildkit:v0.15.1-rootless
   
   ADD --chmod=755 https://raw.githubusercontent.com/seatgeek/buildkit-prestop-script/main/buildkit-prestop.sh /usr/local/bin/buildkit-prestop.sh

2. Add the following to your BuildKit pod's spec:

   lifecycle:
     preStop:
       exec:
         command: ["/usr/local/bin/buildkit-prestop.sh"]

Instead of extending the buildkit docker image, it's also possible to mount the preStop script inside the pod via a ConfigMap:

1. Download the buildkit-prestop.sh script to your local machine.

2. Create a ConfigMap from that file:

   kubectl create configmap buildkit-prestop-script --from-file=buildkit-prestop.sh

   Which should create a ConfigMap named buildkit-prestop-script:

   apiVersion: v1
   kind: ConfigMap
   metadata:
     name: buildkit-prestop-script
   data:
     buildkit-prestop.sh: |
        #!/bin/bash
        ...
   

3. Update your BuildKit Deployment manifest to mount the ConfigMap as a volume and reference the script:

    apiVersion: apps/v1
    kind: Deployment
    metadata:
      labels:
        app: buildkitd
      name: buildkitd
    spec:
      strategy:
        type: RollingUpdate
        rollingUpdate:
          maxUnavailable: 0
      template:
        spec:
          terminationGracePeriodSeconds: 1200
          containers:
            - name: buildkitd
              image: moby/buildkit:v0.16.0
              lifecycle:
                preStop:
                  exec:
                    command: ["/bin/sh", "/usr/local/bin/buildkit-prestop.sh"]
              volumeMounts:
                - name: prestop-script-volume
                  mountPath: /usr/local/bin/buildkit-prestop.sh
                  subPath: buildkit-prestop.sh
          volumes:
          - name: prestop-script-volume
            configMap:
              name: buildkit-prestop-script
              defaultMode: 0777  # Ensure the script is executable

BuildKit does not provide any built-in API for querying the count and status of ongoing builds, forcing us to rely on external observations to determine when it is safe to stop the pod.

We found that the most reliable method is to check whether any clients are connected to the BuildKit daemon via netstat. If we see any established connections then we assume that there is an ongoing build process.

The script contains a few configuration options that you can adjust to suit your needs. See the script for details.

This project is licensed under the Apache License, Version 2.0. See LICENSE for the full license text.