mirror of
https://github.com/agent-substrate/substrate.git
synced 2026-10-02 03:24:42 +08:00
Michelle Au
9300387fef
Initial part of #232. Extends the ActorTemplate API to add support for per-actor external volumes and adds control plane and atelet hooks following the Actor lifecycle. Callouts to volume operations are abstracted with a Volume interface. Right now, only a mock volume plugin for testing has been implemented, but we will add CSI support next. - [x] Tests pass - [ ] Appropriate changes to documentation are included in the PR - Not going to update documentation until we add CSI support.
79 lines
3.1 KiB
Bash
79 lines
3.1 KiB
Bash
#!/usr/bin/env bash
|
|
|
|
# Copyright 2026 Google LLC
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
#
|
|
# This is sourced as part of install-ate.sh. Do not run directly.
|
|
|
|
ATE_DEMOS+=(demo-counter) # register demo-counter
|
|
|
|
demo-counter_usage() {
|
|
echo " --deploy-demo-counter-with-external-volume Deploy demo-counter with external volume validation"
|
|
}
|
|
|
|
demo-counter_cmdline() {
|
|
case "${1}" in
|
|
--deploy-demo-counter) demo-counter_deploy "false" ;;
|
|
--deploy-demo-counter-with-external-volume) demo-counter_deploy "true" ;;
|
|
--delete-demo-counter) demo-counter_delete ;;
|
|
*)
|
|
return 1
|
|
;;
|
|
esac
|
|
return 0
|
|
}
|
|
|
|
demo-counter_deploy() {
|
|
local with_external_volume="${1:-false}"
|
|
log_step "demo-counter_deploy (with_external_volume=${with_external_volume})"
|
|
ensure_crds
|
|
|
|
local validate_cmd=("-e" "/\${VALIDATE_EXISTING_FILE_PATH_ARG}/d")
|
|
local ext_vol_mount_cmd=("-e" "/\${EXTERNAL_VOLUME_MOUNTS}/d")
|
|
local ext_vol_spec_cmd=("-e" "/\${EXTERNAL_VOLUMES}/d")
|
|
if [[ "${with_external_volume}" == "true" ]]; then
|
|
validate_cmd=("-e" "s|\${VALIDATE_EXISTING_FILE_PATH_ARG}| - --validate-existing-file-path=/external-data/test.txt|g")
|
|
ext_vol_mount_cmd=("-e" "s|\${EXTERNAL_VOLUME_MOUNTS}| - name: external-data\n mountPath: /external-data|g")
|
|
ext_vol_spec_cmd=("-e" "s|\${EXTERNAL_VOLUMES}| - name: external-data\n externalVolumeTemplate:\n capacity: 1Gi\n storageClassName: standard|g")
|
|
fi
|
|
|
|
sed -e "s|\${BUCKET_NAME}|${BUCKET_NAME}|g" \
|
|
"${validate_cmd[@]}" \
|
|
"${ext_vol_mount_cmd[@]}" \
|
|
"${ext_vol_spec_cmd[@]}" \
|
|
demos/counter/counter.yaml.tmpl \
|
|
| run_ko apply -f -
|
|
|
|
# Wait for the demo to be fully ready before returning. On a cold cluster the
|
|
# first ActorTemplate golden snapshot pays one-time costs (downloading the
|
|
# gVisor runsc binary, first gVisor pod start, image pulls). Blocking here
|
|
# means callers -- notably the e2e suite, which creates its own ActorTemplate
|
|
# with a tight readiness deadline -- run against an already-warm node instead
|
|
# of racing that cold-start work.
|
|
log_step "Waiting for counter demo to be ready..."
|
|
run_kubectl rollout status deployment/counter -n ate-demo-counter --timeout=300s
|
|
run_kubectl wait --for=condition=Ready actortemplate/counter -n ate-demo-counter --timeout=300s
|
|
}
|
|
|
|
demo-counter_delete() {
|
|
log_step "demo-counter_delete"
|
|
delete_demo_actors ate-demo-counter counter
|
|
sed -e "s|\${BUCKET_NAME}|${BUCKET_NAME}|g" \
|
|
-e "/\${VALIDATE_EXISTING_FILE_PATH_ARG}/d" \
|
|
-e "/\${EXTERNAL_VOLUME_MOUNTS}/d" \
|
|
-e "/\${EXTERNAL_VOLUMES}/d" \
|
|
demos/counter/counter.yaml.tmpl \
|
|
| run_kubectl delete --ignore-not-found -f -
|
|
}
|