faiss server

scripts/inspect_ai_redis.sh

@@ -0,0 +1,178 @@
+#!/usr/bin/env bash
+
+set -euo pipefail
+
+REDIS_HOST="${REDIS_HOST:-127.0.0.1}"
+REDIS_PORT="${REDIS_PORT:-8888}"
+REDIS_DB="${REDIS_DB:-0}"
+REDIS_PASSWORD="${REDIS_PASSWORD:-}"
+PATTERNS=("ai_chat_service_*")
+
+usage() {
+  cat <<'EOF'
+Usage:
+  inspect_ai_redis.sh [options]
+
+Options:
+  --host <host>         Redis host, default: 127.0.0.1
+  --port <port>         Redis port, default: 8888
+  --db <db>             Redis db index, default: 0
+  --password <pwd>      Redis password
+  --pattern <pattern>   Key pattern, can be repeated. Default: ai_chat_service_*
+  --help                Show this help
+
+Environment:
+  REDIS_HOST
+  REDIS_PORT
+  REDIS_DB
+  REDIS_PASSWORD
+
+Examples:
+  ./scripts/inspect_ai_redis.sh
+  ./scripts/inspect_ai_redis.sh --pattern 'ai_chat_service_*' --pattern 'foo_*'
+  REDIS_PORT=6379 ./scripts/inspect_ai_redis.sh --host 10.0.0.8
+EOF
+}
+
+while [[ $# -gt 0 ]]; do
+  case "$1" in
+    --host)
+      REDIS_HOST="$2"
+      shift 2
+      ;;
+    --port)
+      REDIS_PORT="$2"
+      shift 2
+      ;;
+    --db)
+      REDIS_DB="$2"
+      shift 2
+      ;;
+    --password)
+      REDIS_PASSWORD="$2"
+      shift 2
+      ;;
+    --pattern)
+      if [[ "${PATTERNS[*]}" == "ai_chat_service_*" && "${#PATTERNS[@]}" -eq 1 ]]; then
+        PATTERNS=()
+      fi
+      PATTERNS+=("$2")
+      shift 2
+      ;;
+    --help|-h)
+      usage
+      exit 0
+      ;;
+    *)
+      echo "Unknown argument: $1" >&2
+      usage >&2
+      exit 1
+      ;;
+  esac
+done
+
+if ! command -v redis-cli >/dev/null 2>&1; then
+  echo "redis-cli not found in PATH" >&2
+  exit 1
+fi
+
+redis_cmd() {
+  if [[ -n "$REDIS_PASSWORD" ]]; then
+    redis-cli -h "$REDIS_HOST" -p "$REDIS_PORT" -n "$REDIS_DB" -a "$REDIS_PASSWORD" --raw "$@"
+  else
+    redis-cli -h "$REDIS_HOST" -p "$REDIS_PORT" -n "$REDIS_DB" --raw "$@"
+  fi
+}
+
+print_string() {
+  local key="$1"
+  redis_cmd GET "$key"
+}
+
+print_hash() {
+  local key="$1"
+  redis_cmd HGETALL "$key" | awk 'NR % 2 == 1 { printf "%s: ", $0; next } { print $0 }'
+}
+
+print_list() {
+  local key="$1"
+  redis_cmd LRANGE "$key" 0 -1
+}
+
+print_set() {
+  local key="$1"
+  redis_cmd SMEMBERS "$key"
+}
+
+print_zset() {
+  local key="$1"
+  redis_cmd ZRANGE "$key" 0 -1 WITHSCORES | awk 'NR % 2 == 1 { printf "%s => ", $0; next } { print $0 }'
+}
+
+print_stream() {
+  local key="$1"
+  redis_cmd XRANGE "$key" - +
+}
+
+if ! redis_cmd PING >/dev/null 2>&1; then
+  echo "Failed to connect to Redis at ${REDIS_HOST}:${REDIS_PORT}, db=${REDIS_DB}" >&2
+  exit 1
+fi
+
+declare -A seen_keys=()
+keys=()
+
+for pattern in "${PATTERNS[@]}"; do
+  while IFS= read -r key; do
+    [[ -z "$key" ]] && continue
+    if [[ -z "${seen_keys[$key]:-}" ]]; then
+      seen_keys["$key"]=1
+      keys+=("$key")
+    fi
+  done < <(redis_cmd --scan --pattern "$pattern")
+done
+
+if [[ "${#keys[@]}" -eq 0 ]]; then
+  echo "No keys matched patterns: ${PATTERNS[*]}"
+  exit 0
+fi
+
+printf 'Redis: %s:%s db=%s\n' "$REDIS_HOST" "$REDIS_PORT" "$REDIS_DB"
+printf 'Patterns: %s\n' "${PATTERNS[*]}"
+printf 'Matched keys: %s\n\n' "${#keys[@]}"
+
+for key in "${keys[@]}"; do
+  key_type="$(redis_cmd TYPE "$key" | tr -d '\r')"
+  ttl="$(redis_cmd TTL "$key" | tr -d '\r')"
+
+  echo "KEY: $key"
+  echo "TYPE: $key_type"
+  echo "TTL:  $ttl"
+  echo "VALUE:"
+
+  case "$key_type" in
+    string)
+      print_string "$key"
+      ;;
+    hash)
+      print_hash "$key"
+      ;;
+    list)
+      print_list "$key"
+      ;;
+    set)
+      print_set "$key"
+      ;;
+    zset)
+      print_zset "$key"
+      ;;
+    stream)
+      print_stream "$key"
+      ;;
+    *)
+      echo "(unsupported type: $key_type)"
+      ;;
+  esac
+
+  echo
+done