Add Bigtable infrastructure scripts and Makefile

Author: durch

Makefile

@@ -0,0 +1,55 @@
+.PHONY: build test test-integration test-all clean doc bigtable-up bigtable-down help
+
+# Default target
+help:
+	@echo "rust-bigtable Makefile"
+	@echo ""
+	@echo "Build targets:"
+	@echo "  make build            - Build the library"
+	@echo "  make check            - Check compilation without building"
+	@echo "  make doc              - Generate documentation"
+	@echo "  make clean            - Clean build artifacts"
+	@echo ""
+	@echo "Test targets:"
+	@echo "  make test             - Run doc tests"
+	@echo "  make test-integration - Run integration tests (requires Bigtable)"
+	@echo "  make test-all         - Run all tests"
+	@echo ""
+	@echo "Bigtable infrastructure:"
+	@echo "  make bigtable-up      - Create Bigtable instance and table"
+	@echo "  make bigtable-down    - Delete Bigtable instance"
+	@echo ""
+	@echo "Environment variables:"
+	@echo "  PROJECT_ID     - GCP project ID (default: gen-lang-client-0421059902)"
+	@echo "  INSTANCE_ID    - Bigtable instance ID (default: test-inst)"
+	@echo "  TABLE_NAME     - Table name (default: my-table)"
+	@echo "  COLUMN_FAMILY  - Column family name (default: cf1)"
+
+# Build targets
+build:
+	cargo build
+
+check:
+	cargo check
+
+doc:
+	cargo doc --no-deps --open
+
+clean:
+	cargo clean
+
+# Test targets
+test:
+	cargo test
+
+test-integration:
+	cargo test --test integration_tests -- --ignored --test-threads=1
+
+test-all: test test-integration
+
+# Bigtable infrastructure
+bigtable-up:
+	./scripts/bigtable-up.sh
+
+bigtable-down:
+	./scripts/bigtable-down.sh

scripts/bigtable-down.sh

@@ -0,0 +1,28 @@
+#!/bin/bash
+# Deletes the Bigtable instance (and all tables within it)
+set -e
+
+PROJECT_ID="${PROJECT_ID:-gen-lang-client-0421059902}"
+INSTANCE_ID="${INSTANCE_ID:-test-inst}"
+
+echo "Deleting Bigtable instance..."
+echo "  Project: $PROJECT_ID"
+echo "  Instance: $INSTANCE_ID"
+echo ""
+
+# Check if instance exists
+if ! gcloud bigtable instances describe "$INSTANCE_ID" --project="$PROJECT_ID" &>/dev/null; then
+    echo "Instance '$INSTANCE_ID' does not exist, nothing to delete."
+    exit 0
+fi
+
+read -p "Are you sure you want to delete instance '$INSTANCE_ID'? This will delete ALL tables. [y/N] " -n 1 -r
+echo ""
+
+if [[ $REPLY =~ ^[Yy]$ ]]; then
+    gcloud bigtable instances delete "$INSTANCE_ID" --project="$PROJECT_ID" --quiet
+    echo "Instance '$INSTANCE_ID' deleted."
+else
+    echo "Aborted."
+    exit 1
+fi

scripts/bigtable-up.sh

@@ -0,0 +1,53 @@
+#!/bin/bash
+# Creates a Bigtable instance, table, and column family for testing
+set -e
+
+PROJECT_ID="${PROJECT_ID:-gen-lang-client-0421059902}"
+INSTANCE_ID="${INSTANCE_ID:-test-inst}"
+CLUSTER_ID="${CLUSTER_ID:-test-cluster}"
+ZONE="${ZONE:-us-central1-c}"
+TABLE_NAME="${TABLE_NAME:-my-table}"
+COLUMN_FAMILY="${COLUMN_FAMILY:-cf1}"
+
+echo "Creating Bigtable instance..."
+echo "  Project: $PROJECT_ID"
+echo "  Instance: $INSTANCE_ID"
+echo "  Cluster: $CLUSTER_ID"
+echo "  Zone: $ZONE"
+
+# Check if instance already exists
+if gcloud bigtable instances describe "$INSTANCE_ID" --project="$PROJECT_ID" &>/dev/null; then
+    echo "Instance '$INSTANCE_ID' already exists, skipping creation."
+else
+    gcloud bigtable instances create "$INSTANCE_ID" \
+        --project="$PROJECT_ID" \
+        --cluster="$CLUSTER_ID" \
+        --cluster-zone="$ZONE" \
+        --display-name="Test Instance"
+    echo "Instance created."
+fi
+
+echo ""
+echo "Creating table '$TABLE_NAME' with column family '$COLUMN_FAMILY'..."
+
+# Check if table exists
+if cbt -project="$PROJECT_ID" -instance="$INSTANCE_ID" ls 2>/dev/null | grep -q "^$TABLE_NAME$"; then
+    echo "Table '$TABLE_NAME' already exists, skipping creation."
+else
+    cbt -project="$PROJECT_ID" -instance="$INSTANCE_ID" createtable "$TABLE_NAME"
+    echo "Table created."
+fi
+
+# Check if column family exists
+if cbt -project="$PROJECT_ID" -instance="$INSTANCE_ID" ls "$TABLE_NAME" 2>/dev/null | grep -q "$COLUMN_FAMILY"; then
+    echo "Column family '$COLUMN_FAMILY' already exists, skipping creation."
+else
+    cbt -project="$PROJECT_ID" -instance="$INSTANCE_ID" createfamily "$TABLE_NAME" "$COLUMN_FAMILY"
+    echo "Column family created."
+fi
+
+echo ""
+echo "Bigtable setup complete!"
+echo "  Instance: $INSTANCE_ID"
+echo "  Table: $TABLE_NAME"
+echo "  Column Family: $COLUMN_FAMILY"