# Set batch size (default: 100 records)
export USAGE_BATCH_SIZE=500

# Set batch timeout (default: 5 seconds)
export USAGE_BATCH_TIMEOUT=10

usage_batch_add "$customer_id" "api" 1 '{"endpoint":"/users","method":"GET"}'

usage_batch_flush

usage_batch_insert \
    "api:1:{\"endpoint\":\"/users\"}" \
    "api:1:{\"endpoint\":\"/posts\"}" \
    "database:1:{\"type\":\"SELECT\"}"

usage_track_api_request "/api/users" "GET" 200
usage_track_api_request "/api/posts" "POST" 201

# Track 5GB stored for 24 hours
bytes=5368709120  # 5GB
usage_track_storage "$bytes" 24

# Track 500MB egress
bytes=524288000
usage_track_bandwidth "$bytes" "egress"

# Track 3600 seconds (1 hour) of CPU time
cpu_seconds=3600
usage_track_compute "$cpu_seconds" '{"instance":"t3.medium"}'

usage_track_database_query "SELECT" 45  # 45ms duration
usage_track_database_query "INSERT" 120

usage_track_function "sendEmail" 120 256  # 120ms, 256MB memory
usage_track_function "processImage" 3500 512

billing_record_usage "service_name" quantity "metadata_json"

# Hourly aggregation
usage_aggregate hourly "$customer_id" "2026-01-01" "2026-01-31"

# Daily aggregation
usage_aggregate daily "$customer_id" "2026-01-01" "2026-01-31"

# Monthly aggregation
usage_aggregate monthly "$customer_id" "2026-01-01" "2026-12-31"

usage_refresh_summary

# Table format (default)
usage_get_all "2026-01-01" "2026-01-31"

# JSON format
usage_get_all "2026-01-01" "2026-01-31" "json"

# CSV format
usage_get_all "2026-01-01" "2026-01-31" "csv"

# Detailed breakdown
usage_get_all "2026-01-01" "2026-01-31" "table" "true"

# API usage for January
usage_get_service "api" "2026-01-01" "2026-01-31"

# Storage usage with detailed timeline
usage_get_service "storage" "2026-01-01" "2026-01-31" "table" "true"

# Bandwidth usage as JSON
usage_get_service "bandwidth" "2026-01-01" "2026-01-31" "json"

# Configure thresholds (percentage of quota)
export USAGE_ALERT_WARNING=75    # Default: 75%
export USAGE_ALERT_CRITICAL=90   # Default: 90%
export USAGE_ALERT_EXCEEDED=100  # Default: 100%

# Check all services
usage_check_alerts

# Check specific service
usage_check_service_alert "$customer_id" "api"

# Last 7 days (default)
usage_get_alerts

# Last 30 days
usage_get_alerts 30

# Export as CSV (default)
usage_export csv

# Export as JSON
usage_export json

# Export to specific file
usage_export csv "/path/to/usage.csv"

# Export with date range
usage_export csv "" "2026-01-01" "2026-01-31"

# Export specific service
usage_export json "" "2026-01-01" "2026-01-31" "api"

# Install csvkit (if needed)
pip install csvkit

# Export as XLSX
usage_export xlsx

# API statistics for last 30 days
usage_get_stats "api" 30

# Storage statistics for last 7 days
usage_get_stats "storage" 7

# All services, last 7 days
usage_get_trends

# Specific service, last 30 days
usage_get_trends "api" 30

# Top 10 hours for API
usage_get_peaks "api" "hourly" 10

# Top 5 days for storage
usage_get_peaks "storage" "daily" 5

# Archive records older than 90 days (default)
usage_archive

# Archive records older than 30 days
usage_archive 30

# Archive to specific file
usage_archive 90 "/path/to/archive.csv"

usage_cleanup_batch

#!/usr/bin/env bash
source "src/lib/billing/usage.sh"

# Track various API endpoints
usage_track_api_request "/api/users" "GET" 200
usage_track_api_request "/api/users/123" "GET" 200
usage_track_api_request "/api/posts" "POST" 201
usage_track_api_request "/api/invalid" "GET" 404

# View usage
usage_get_service "api" "$(date -u +"%Y-%m-%d 00:00:00")" "$(date -u +"%Y-%m-%d 23:59:59")"

#!/usr/bin/env bash
source "src/lib/billing/usage.sh"

# Initialize batch processing
usage_init_batch

# Get customer ID
customer_id=$(billing_get_customer_id)

# Track 1000 API requests (batched)
for i in {1..1000}; do
    usage_batch_add "$customer_id" "api" 1 "{\"endpoint\":\"/test\",\"i\":$i}"
done

# Batch auto-flushes at 100 records, but flush any remaining
usage_batch_flush

#!/usr/bin/env bash
source "src/lib/billing/usage.sh"

# Calculate storage usage
total_bytes=$(du -sb /data | cut -f1)
hours_since_last=$(( ($(date +%s) - $(stat -f %m /data/.last_check)) / 3600 ))

# Track usage
usage_track_storage "$total_bytes" "$hours_since_last"

# Update last check
touch /data/.last_check

# Check if approaching quota
usage_check_service_alert "$(billing_get_customer_id)" "storage"

#!/usr/bin/env bash
source "src/lib/billing/usage.sh"

# Get today's date range
today=$(date -u +"%Y-%m-%d")
start="${today} 00:00:00"
end="${today} 23:59:59"

# Generate report
printf "Daily Usage Report - %s\n\n" "$today"

usage_get_all "$start" "$end" "table" "true"

# Export to file
output_file="/var/reports/usage_${today}.csv"
usage_export csv "$output_file" "$start" "$end"

printf "\nExported to: %s\n" "$output_file"

#!/usr/bin/env bash
source "src/lib/billing/usage.sh"

# Analyze API usage trends
printf "API Usage Trends (Last 7 Days)\n"
printf "================================\n\n"

usage_get_trends "api" 7

# Find peak usage hours
printf "\nTop 5 Peak Hours\n"
printf "================\n\n"

usage_get_peaks "api" "hourly" 5

# Get detailed statistics
printf "\nDetailed Statistics (Last 30 Days)\n"
printf "==================================\n\n"

usage_get_stats "api" 30

#!/usr/bin/env bash
source "src/lib/billing/usage.sh"

# Check quotas for all services
printf "Checking usage quotas...\n\n"

usage_check_alerts 2>&1 | while IFS= read -r line; do
    if [[ "$line" =~ "EXCEEDED" ]]; then
        # Send urgent alert
        echo "$line" | mail -s "URGENT: Quota Exceeded" [email protected]
    elif [[ "$line" =~ "CRITICAL" ]]; then
        # Send warning
        echo "$line" | mail -s "WARNING: Quota Critical" [email protected]
    fi

    # Log all alerts
    printf "%s\n" "$line"
done

# Show recent alert history
printf "\nRecent Alerts (Last 7 Days)\n"
printf "===========================\n\n"
usage_get_alerts 7

#!/usr/bin/env bash
source "src/lib/billing/usage.sh"

# Get last month's date range
if date -v-1m >/dev/null 2>&1; then
    # macOS
    start=$(date -v-1m -v1d +"%Y-%m-01 00:00:00")
    end=$(date -v-1m -v+1m -v-1d +"%Y-%m-%d 23:59:59")
else
    # Linux
    start=$(date -d "last month" +"%Y-%m-01 00:00:00")
    end=$(date -d "$(date +%Y-%m-01) -1 day" +"%Y-%m-%d 23:59:59")
fi

month=$(date -d "$start" +"%Y-%m" 2>/dev/null || date -j -f "%Y-%m-%d" "$(echo $start | cut -d' ' -f1)" +"%Y-%m")

printf "Generating monthly usage report for %s...\n" "$month"

# Export in multiple formats
usage_export csv "/var/billing/usage_${month}.csv" "$start" "$end"
usage_export json "/var/billing/usage_${month}.json" "$start" "$end"

# Archive old records (keep last 90 days)
usage_archive 90 "/var/billing/archive/usage_${month}_archive.csv"

printf "\nMonthly export complete\n"