feat: add scripts for init db

Signed-off-by: Xin Liu <[email protected]>

Author: apepkuss

.gitignore

@@ -1 +1,5 @@
 /target
+
+*.db
+*.db-shm
+*.db-wal

README.md

@@ -0,0 +1,120 @@
+# ss-proxy
+
+一个使用 SQLite 存储会话信息的代理服务。
+
+## 数据库初始化
+
+### 方式 1：使用 Shell 脚本（推荐）
+
+```bash
+# 给脚本添加执行权限
+chmod +x init_db.sh
+
+# 执行初始化（默认创建 ./sessions.db）
+./init_db.sh
+
+# 或指定自定义数据库路径
+./init_db.sh /path/to/custom.db
+```
+
+### 方式 2：直接使用 sqlite3 命令
+
+```bash
+# 创建数据库并执行初始化脚本
+sqlite3 sessions.db < migrations/init.sql
+
+# 或指定自定义路径
+sqlite3 /path/to/custom.db < migrations/init.sql
+```
+
+## 数据库结构
+
+### sessions 表
+
+| 字段名 | 类型 | 约束 | 说明 |
+|--------|------|------|------|
+| `session_id` | TEXT | PRIMARY KEY | 会话ID（主键） |
+| `downstream_server_url` | TEXT | NOT NULL | 下游服务器URL |
+| `downstream_server_status` | TEXT | NOT NULL | 下游服务器状态 |
+| `created_at` | DATETIME | DEFAULT CURRENT_TIMESTAMP | 创建时间 |
+| `updated_at` | DATETIME | DEFAULT CURRENT_TIMESTAMP | 更新时间 |
+
+### 索引
+
+- `idx_session_status`: 基于 `downstream_server_status` 的索引
+- `idx_created_at`: 基于 `created_at` 的索引
+
+## 常用数据库操作
+
+### 交互式操作（推荐）
+
+进入 SQLite 交互式命令行：
+
+```bash
+sqlite3 sessions.db
+```
+
+在交互式环境中执行操作：
+
+```sql
+-- 查询所有会话
+SELECT * FROM sessions;
+
+-- 根据 session_id 查询
+SELECT * FROM sessions WHERE session_id = 'your-session-id';
+
+-- 查询特定状态的会话
+SELECT * FROM sessions WHERE downstream_server_status = 'active';
+
+-- 插入数据
+INSERT INTO sessions (session_id, downstream_server_url, downstream_server_status)
+VALUES ('session-001', 'http://localhost:8080', 'active');
+
+-- 更新数据
+UPDATE sessions SET downstream_server_status = 'inactive'
+WHERE session_id = 'session-001';
+
+-- 删除数据
+DELETE FROM sessions WHERE session_id = 'session-001';
+
+-- 退出
+.quit
+```
+
+### 使用 SQL 文件（推荐用于批量操作）
+
+创建 SQL 文件（例如 `query.sql`）：
+
+```sql
+SELECT * FROM sessions WHERE downstream_server_status = 'active';
+```
+
+执行 SQL 文件：
+
+```bash
+sqlite3 sessions.db < query.sql
+```
+
+### 单行命令（简单查询）
+
+对于简单的只读查询，可以使用单行命令：
+
+```bash
+# 查询所有会话（使用单引号）
+sqlite3 sessions.db 'SELECT * FROM sessions;'
+
+# 统计会话数量
+sqlite3 sessions.db 'SELECT COUNT(*) FROM sessions;'
+```
+
+**注意**：对于包含复杂 SQL 语句（特别是带逗号的 INSERT/UPDATE 语句），建议使用交互式模式或 SQL 文件方式，以避免 shell 解析问题。
+
+## 开发
+
+```bash
+# 构建项目
+cargo build
+
+# 运行项目
+cargo run
+```

init_db.sh

@@ -0,0 +1,51 @@
+#!/bin/bash
+
+# ss-proxy 数据库初始化脚本
+# 用途：执行 SQL 脚本创建数据库表结构
+
+set -e  # 遇到错误立即退出
+
+# 默认数据库路径
+DB_PATH="${1:-./sessions.db}"
+
+echo "================================================"
+echo "  ss-proxy 数据库初始化工具"
+echo "================================================"
+echo ""
+echo "数据库路径: $DB_PATH"
+echo ""
+
+# 检查 sqlite3 是否安装
+if ! command -v sqlite3 &> /dev/null; then
+    echo "❌ 错误: 未找到 sqlite3 命令"
+    echo "请先安装 SQLite: brew install sqlite"
+    exit 1
+fi
+
+# 检查 SQL 脚本是否存在
+if [ ! -f "migrations/init.sql" ]; then
+    echo "❌ 错误: 未找到 migrations/init.sql 文件"
+    exit 1
+fi
+
+# 执行 SQL 脚本
+echo "正在执行初始化脚本..."
+if sqlite3 "$DB_PATH" < migrations/init.sql; then
+    echo ""
+    echo "================================================"
+    echo "✅ 数据库初始化成功！"
+    echo "================================================"
+    echo ""
+    echo "数据库位置: $DB_PATH"
+    echo ""
+    echo "查看表结构:"
+    echo "  sqlite3 $DB_PATH '.schema sessions'"
+    echo ""
+    echo "查询数据:"
+    echo "  sqlite3 $DB_PATH 'SELECT * FROM sessions;'"
+    echo ""
+else
+    echo ""
+    echo "❌ 数据库初始化失败"
+    exit 1
+fi

migrations/init.sql

@@ -0,0 +1,25 @@
+-- ss-proxy 数据库初始化脚本
+-- 功能：创建 sessions 表用于存储会话和下游服务器信息
+
+-- 创建 sessions 表
+CREATE TABLE IF NOT EXISTS sessions (
+    session_id TEXT PRIMARY KEY NOT NULL,
+    downstream_server_url TEXT NOT NULL,
+    downstream_server_status TEXT NOT NULL,
+    created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+    updated_at DATETIME DEFAULT CURRENT_TIMESTAMP
+);
+
+-- 创建索引以提高查询性能
+CREATE INDEX IF NOT EXISTS idx_session_status
+ON sessions(downstream_server_status);
+
+-- 创建索引以提高按创建时间查询的性能
+CREATE INDEX IF NOT EXISTS idx_created_at
+ON sessions(created_at);
+
+-- 显示创建成功的信息
+SELECT '✅ sessions 表创建成功' AS status;
+
+-- 显示表结构
+.schema sessions