239 lines
8.8 KiB
Bash
Executable File
239 lines
8.8 KiB
Bash
Executable File
#!/bin/bash
|
||
|
||
# HeroDB Tantivy Search Demo
|
||
# This script demonstrates full-text search capabilities using Redis commands
|
||
# HeroDB server should be running on port 6381
|
||
|
||
set -e # Exit on any error
|
||
|
||
# Configuration
|
||
REDIS_HOST="localhost"
|
||
REDIS_PORT="6382"
|
||
REDIS_CLI="redis-cli -h $REDIS_HOST -p $REDIS_PORT"
|
||
|
||
# Start the herodb server in the background
|
||
echo "Starting herodb server..."
|
||
cargo run -p herodb -- --dir /tmp/herodbtest --port ${REDIS_PORT} --debug &
|
||
SERVER_PID=$!
|
||
echo
|
||
sleep 2 # Give the server a moment to start
|
||
|
||
# Colors for output
|
||
RED='\033[0;31m'
|
||
GREEN='\033[0;32m'
|
||
BLUE='\033[0;34m'
|
||
YELLOW='\033[1;33m'
|
||
NC='\033[0m' # No Color
|
||
|
||
# Function to print colored output
|
||
print_header() {
|
||
echo -e "${BLUE}=== $1 ===${NC}"
|
||
}
|
||
|
||
print_success() {
|
||
echo -e "${GREEN}✓ $1${NC}"
|
||
}
|
||
|
||
print_info() {
|
||
echo -e "${YELLOW}ℹ $1${NC}"
|
||
}
|
||
|
||
print_error() {
|
||
echo -e "${RED}✗ $1${NC}"
|
||
}
|
||
|
||
# Function to check if HeroDB is running
|
||
check_herodb() {
|
||
print_info "Checking if HeroDB is running on port $REDIS_PORT..."
|
||
if ! $REDIS_CLI ping > /dev/null 2>&1; then
|
||
print_error "HeroDB is not running on port $REDIS_PORT"
|
||
print_info "Please start HeroDB with: cargo run -- --port $REDIS_PORT"
|
||
exit 1
|
||
fi
|
||
print_success "HeroDB is running and responding"
|
||
}
|
||
|
||
# Function to execute Redis command with error handling
|
||
execute_cmd() {
|
||
local cmd="$1"
|
||
local description="$2"
|
||
|
||
echo -e "${YELLOW}Command:${NC} $cmd"
|
||
if result=$($REDIS_CLI $cmd 2>&1); then
|
||
echo -e "${GREEN}Result:${NC} $result"
|
||
return 0
|
||
else
|
||
print_error "Failed: $description"
|
||
echo "Error: $result"
|
||
return 1
|
||
fi
|
||
}
|
||
|
||
# Function to pause for readability
|
||
pause() {
|
||
echo
|
||
read -p "Press Enter to continue..."
|
||
echo
|
||
}
|
||
|
||
# Main demo function
|
||
main() {
|
||
clear
|
||
print_header "HeroDB Tantivy Search Demonstration"
|
||
echo "This demo shows full-text search capabilities using Redis commands"
|
||
echo "HeroDB runs on port $REDIS_PORT (instead of Redis default 6379)"
|
||
echo
|
||
|
||
# Check if HeroDB is running
|
||
check_herodb
|
||
echo
|
||
|
||
print_header "Step 1: Create Search Index"
|
||
print_info "Creating a product catalog search index with various field types"
|
||
|
||
# Create search index with schema
|
||
execute_cmd "FT.CREATE product_catalog SCHEMA title TEXT description TEXT category TAG price NUMERIC rating NUMERIC location GEO" \
|
||
"Creating search index"
|
||
|
||
print_success "Search index 'product_catalog' created successfully"
|
||
pause
|
||
|
||
print_header "Step 2: Add Sample Products"
|
||
print_info "Adding sample products to demonstrate different search scenarios"
|
||
|
||
# Add sample products using FT.ADD
|
||
execute_cmd "FT.ADD product_catalog product:1 1.0 title 'Wireless Bluetooth Headphones' description 'Premium noise-canceling headphones with 30-hour battery life' category 'electronics,audio' price 299.99 rating 4.5 location '-122.4194,37.7749'" "Adding product 1"
|
||
execute_cmd "FT.ADD product_catalog product:2 1.0 title 'Organic Coffee Beans' description 'Single-origin Ethiopian coffee beans, medium roast' category 'food,beverages,organic' price 24.99 rating 4.8 location '-74.0060,40.7128'" "Adding product 2"
|
||
execute_cmd "FT.ADD product_catalog product:3 1.0 title 'Yoga Mat Premium' description 'Eco-friendly yoga mat with superior grip and cushioning' category 'fitness,wellness,eco-friendly' price 89.99 rating 4.3 location '-118.2437,34.0522'" "Adding product 3"
|
||
execute_cmd "FT.ADD product_catalog product:4 1.0 title 'Smart Home Speaker' description 'Voice-controlled smart speaker with AI assistant' category 'electronics,smart-home' price 149.99 rating 4.2 location '-87.6298,41.8781'" "Adding product 4"
|
||
execute_cmd "FT.ADD product_catalog product:5 1.0 title 'Organic Green Tea' description 'Premium organic green tea leaves from Japan' category 'food,beverages,organic,tea' price 18.99 rating 4.7 location '139.6503,35.6762'" "Adding product 5"
|
||
execute_cmd "FT.ADD product_catalog product:6 1.0 title 'Wireless Gaming Mouse' description 'High-precision gaming mouse with RGB lighting' category 'electronics,gaming' price 79.99 rating 4.4 location '-122.3321,47.6062'" "Adding product 6"
|
||
execute_cmd "FT.ADD product_catalog product:7 1.0 title 'Comfortable meditation cushion for mindfulness practice' description 'Meditation cushion with premium materials' category 'wellness,meditation' price 45.99 rating 4.6 location '-122.4194,37.7749'" "Adding product 7"
|
||
execute_cmd "FT.ADD product_catalog product:8 1.0 title 'Bluetooth Earbuds' description 'True wireless earbuds with active noise cancellation' category 'electronics,audio' price 199.99 rating 4.1 location '-74.0060,40.7128'" "Adding product 8"
|
||
|
||
print_success "Added 8 products to the index"
|
||
pause
|
||
|
||
print_header "Step 3: Basic Text Search"
|
||
print_info "Searching for 'wireless' products"
|
||
|
||
execute_cmd "FT.SEARCH product_catalog wireless" "Basic text search"
|
||
pause
|
||
|
||
print_header "Step 4: Search with Filters"
|
||
print_info "Searching for 'organic' products"
|
||
|
||
execute_cmd "FT.SEARCH product_catalog organic" "Filtered search"
|
||
pause
|
||
|
||
print_header "Step 5: Numeric Range Search"
|
||
print_info "Searching for 'premium' products"
|
||
|
||
execute_cmd "FT.SEARCH product_catalog premium" "Text search"
|
||
pause
|
||
|
||
print_header "Step 6: Sorting Results"
|
||
print_info "Searching for electronics"
|
||
|
||
execute_cmd "FT.SEARCH product_catalog electronics" "Category search"
|
||
pause
|
||
|
||
print_header "Step 7: Limiting Results"
|
||
print_info "Searching for wireless products with limit"
|
||
|
||
execute_cmd "FT.SEARCH product_catalog wireless LIMIT 0 3" "Limited results"
|
||
pause
|
||
|
||
print_header "Step 8: Complex Query"
|
||
print_info "Finding audio products with noise cancellation"
|
||
|
||
execute_cmd "FT.SEARCH product_catalog 'noise cancellation'" "Complex query"
|
||
pause
|
||
|
||
print_header "Step 9: Geographic Search"
|
||
print_info "Searching for meditation products"
|
||
|
||
execute_cmd "FT.SEARCH product_catalog meditation" "Text search"
|
||
pause
|
||
|
||
print_header "Step 10: Aggregation Example"
|
||
print_info "Getting index information and statistics"
|
||
|
||
execute_cmd "FT.INFO product_catalog" "Index information"
|
||
pause
|
||
|
||
print_header "Step 11: Search Comparison"
|
||
print_info "Comparing Tantivy search vs simple key matching"
|
||
|
||
echo -e "${YELLOW}Tantivy Full-Text Search:${NC}"
|
||
execute_cmd "FT.SEARCH product_catalog 'battery life'" "Full-text search for 'battery life'"
|
||
|
||
echo
|
||
echo -e "${YELLOW}Simple Key Pattern Matching:${NC}"
|
||
execute_cmd "KEYS *battery*" "Simple pattern matching for 'battery'"
|
||
|
||
print_info "Notice how full-text search finds relevant results even when exact words don't match keys"
|
||
pause
|
||
|
||
print_header "Step 12: Fuzzy Search"
|
||
print_info "Searching for headphones"
|
||
|
||
execute_cmd "FT.SEARCH product_catalog headphones" "Text search"
|
||
pause
|
||
|
||
print_header "Step 13: Phrase Search"
|
||
print_info "Searching for coffee products"
|
||
|
||
execute_cmd "FT.SEARCH product_catalog coffee" "Text search"
|
||
pause
|
||
|
||
print_header "Step 14: Boolean Queries"
|
||
print_info "Searching for gaming products"
|
||
|
||
execute_cmd "FT.SEARCH product_catalog gaming" "Text search"
|
||
echo
|
||
execute_cmd "FT.SEARCH product_catalog tea" "Text search"
|
||
pause
|
||
|
||
print_header "Step 15: Cleanup"
|
||
print_info "Removing test data"
|
||
|
||
# Delete the search index
|
||
execute_cmd "FT.DROP product_catalog" "Dropping search index"
|
||
|
||
# Clean up documents from search index
|
||
for i in {1..8}; do
|
||
execute_cmd "FT.DEL product_catalog product:$i" "Deleting product:$i from index"
|
||
done
|
||
|
||
print_success "Cleanup completed"
|
||
echo
|
||
|
||
print_header "Demo Summary"
|
||
echo "This demonstration showed:"
|
||
echo "• Creating search indexes with different field types"
|
||
echo "• Adding documents to the search index"
|
||
echo "• Basic and advanced text search queries"
|
||
echo "• Filtering by categories and numeric ranges"
|
||
echo "• Sorting and limiting results"
|
||
echo "• Geographic searches"
|
||
echo "• Fuzzy matching and phrase searches"
|
||
echo "• Boolean query operators"
|
||
echo "• Comparison with simple pattern matching"
|
||
echo
|
||
print_success "HeroDB Tantivy search demo completed successfully!"
|
||
echo
|
||
print_info "Key advantages of Tantivy full-text search:"
|
||
echo " - Relevance scoring and ranking"
|
||
echo " - Fuzzy matching and typo tolerance"
|
||
echo " - Complex boolean queries"
|
||
echo " - Field-specific searches and filters"
|
||
echo " - Geographic and numeric range queries"
|
||
echo " - Much faster than pattern matching on large datasets"
|
||
echo
|
||
print_info "To run HeroDB server: cargo run -- --port 6381"
|
||
print_info "To connect with redis-cli: redis-cli -h localhost -p 6381"
|
||
}
|
||
|
||
# Run the demo
|
||
main "$@" |