blob: 8f1a965f5c0f0166b799bd6eaf5d37554ff83341 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
|
#!/bin/bash
# Wait for Services Script for Kafka Integration Tests
set -e
echo "Waiting for services to be ready..."
# Configuration
KAFKA_HOST=${KAFKA_HOST:-localhost}
KAFKA_PORT=${KAFKA_PORT:-9092}
SCHEMA_REGISTRY_URL=${SCHEMA_REGISTRY_URL:-http://localhost:8081}
KAFKA_GATEWAY_HOST=${KAFKA_GATEWAY_HOST:-localhost}
KAFKA_GATEWAY_PORT=${KAFKA_GATEWAY_PORT:-9093}
SEAWEEDFS_MASTER_URL=${SEAWEEDFS_MASTER_URL:-http://localhost:9333}
MAX_WAIT=${MAX_WAIT:-300} # 5 minutes
# Colors
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color
# Helper function to wait for a service
wait_for_service() {
local service_name=$1
local check_command=$2
local timeout=${3:-60}
echo -e "${BLUE}Waiting for ${service_name}...${NC}"
local count=0
while [ $count -lt $timeout ]; do
if eval "$check_command" > /dev/null 2>&1; then
echo -e "${GREEN}[OK] ${service_name} is ready${NC}"
return 0
fi
if [ $((count % 10)) -eq 0 ]; then
echo -e "${YELLOW}Still waiting for ${service_name}... (${count}s)${NC}"
fi
sleep 1
count=$((count + 1))
done
echo -e "${RED}[FAIL] ${service_name} failed to start within ${timeout} seconds${NC}"
return 1
}
# Wait for Zookeeper
echo "=== Checking Zookeeper ==="
wait_for_service "Zookeeper" "nc -z localhost 2181" 30
# Wait for Kafka
echo "=== Checking Kafka ==="
wait_for_service "Kafka" "nc -z ${KAFKA_HOST} ${KAFKA_PORT}" 60
# Test Kafka broker API
echo "=== Testing Kafka API ==="
wait_for_service "Kafka API" "timeout 5 kafka-broker-api-versions --bootstrap-server ${KAFKA_HOST}:${KAFKA_PORT}" 30
# Wait for Schema Registry
echo "=== Checking Schema Registry ==="
wait_for_service "Schema Registry" "curl -f ${SCHEMA_REGISTRY_URL}/subjects" 60
# Wait for SeaweedFS Master
echo "=== Checking SeaweedFS Master ==="
wait_for_service "SeaweedFS Master" "curl -f ${SEAWEEDFS_MASTER_URL}/cluster/status" 30
# Wait for SeaweedFS Volume
echo "=== Checking SeaweedFS Volume ==="
wait_for_service "SeaweedFS Volume" "curl -f http://localhost:8080/status" 30
# Wait for SeaweedFS Filer
echo "=== Checking SeaweedFS Filer ==="
wait_for_service "SeaweedFS Filer" "curl -f http://localhost:8888/" 30
# Wait for SeaweedFS MQ Broker
echo "=== Checking SeaweedFS MQ Broker ==="
wait_for_service "SeaweedFS MQ Broker" "nc -z localhost 17777" 30
# Wait for SeaweedFS MQ Agent
echo "=== Checking SeaweedFS MQ Agent ==="
wait_for_service "SeaweedFS MQ Agent" "nc -z localhost 16777" 30
# Wait for Kafka Gateway
echo "=== Checking Kafka Gateway ==="
wait_for_service "Kafka Gateway" "nc -z ${KAFKA_GATEWAY_HOST} ${KAFKA_GATEWAY_PORT}" 60
# Final verification
echo "=== Final Verification ==="
# Test Kafka topic creation
echo "Testing Kafka topic operations..."
TEST_TOPIC="health-check-$(date +%s)"
if kafka-topics --create --topic "$TEST_TOPIC" --bootstrap-server "${KAFKA_HOST}:${KAFKA_PORT}" --partitions 1 --replication-factor 1 > /dev/null 2>&1; then
echo -e "${GREEN}[OK] Kafka topic creation works${NC}"
kafka-topics --delete --topic "$TEST_TOPIC" --bootstrap-server "${KAFKA_HOST}:${KAFKA_PORT}" > /dev/null 2>&1 || true
else
echo -e "${RED}[FAIL] Kafka topic creation failed${NC}"
exit 1
fi
# Test Schema Registry
echo "Testing Schema Registry..."
if curl -f "${SCHEMA_REGISTRY_URL}/subjects" > /dev/null 2>&1; then
echo -e "${GREEN}[OK] Schema Registry is accessible${NC}"
else
echo -e "${RED}[FAIL] Schema Registry is not accessible${NC}"
exit 1
fi
# Test Kafka Gateway connectivity
echo "Testing Kafka Gateway..."
if nc -z "${KAFKA_GATEWAY_HOST}" "${KAFKA_GATEWAY_PORT}"; then
echo -e "${GREEN}[OK] Kafka Gateway is accessible${NC}"
else
echo -e "${RED}[FAIL] Kafka Gateway is not accessible${NC}"
exit 1
fi
echo -e "${GREEN}All services are ready!${NC}"
echo ""
echo "Service endpoints:"
echo " Kafka: ${KAFKA_HOST}:${KAFKA_PORT}"
echo " Schema Registry: ${SCHEMA_REGISTRY_URL}"
echo " Kafka Gateway: ${KAFKA_GATEWAY_HOST}:${KAFKA_GATEWAY_PORT}"
echo " SeaweedFS Master: ${SEAWEEDFS_MASTER_URL}"
echo " SeaweedFS Filer: http://localhost:8888"
echo " SeaweedFS MQ Broker: localhost:17777"
echo " SeaweedFS MQ Agent: localhost:16777"
echo ""
echo "Ready to run integration tests!"
|
