From fe86fb47c027c8da6cc82791b09286bdf26d9f04 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=8E=8B=E5=BD=A6=E6=9C=8B=20Penn=20Wang?=
 <penn.wang@digitwin.com.cn>
Date: Wed, 8 Nov 2023 15:35:54 +0800
Subject: [PATCH] feat: Add improve docker-compose.yml and support fast startup

---
 .gitignore         |  3 ++-
 README.md          | 14 ++++++++++++++
 docker-compose.yml | 21 ++++++++++++++++++---
 3 files changed, 34 insertions(+), 4 deletions(-)

diff --git a/.gitignore b/.gitignore
index 1b2cf071..60abb13e 100644
--- a/.gitignore
+++ b/.gitignore
@@ -5,4 +5,5 @@ upload
 *.db
 build
 *.db-journal
-logs
\ No newline at end of file
+logs
+data
\ No newline at end of file
diff --git a/README.md b/README.md
index 79229a94..47c437c2 100644
--- a/README.md
+++ b/README.md
@@ -107,6 +107,7 @@ _✨ 通过标准的 OpenAI API 格式访问所有的大模型，开箱即用 
     + 微信公众号授权（需要额外部署 [WeChat Server](https://github.com/songquanpeng/wechat-server)）。
 
 ## 部署
+
 ### 基于 Docker 进行部署
 ```shell
 # 使用 SQLite 的部署命令：
@@ -160,6 +161,19 @@ sudo service nginx restart
 
 初始账号用户名为 `root`，密码为 `123456`。
 
+
+### 基于 Docker-compose 进行部署（快速启动）
+
+> 仅启动方式不同，参数设置不变，请参考Docker部署部分。
+
+```shell
+# 目前支持 MySQL 启动，数据存储在 ./data/mysql 文件夹内。
+docker-compose up -d
+
+# 查看部署状态
+docker-compose ps
+```
+
 ### 手动部署
 1. 从 [GitHub Releases](https://github.com/songquanpeng/one-api/releases/latest) 下载可执行文件或者从源码编译：
    ```shell
diff --git a/docker-compose.yml b/docker-compose.yml
index 9b814a03..ecfa73cd 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -9,19 +9,19 @@ services:
     ports:
       - "3000:3000"
     volumes:
-      - ./data:/data
+      - ./data/oneapi:/data
       - ./logs:/app/logs
     environment:
-      - SQL_DSN=root:123456@tcp(host.docker.internal:3306)/one-api  # 修改此行，或注释掉以使用 SQLite 作为数据库
+      - SQL_DSN=oneapi:123456@tcp(db:3306)/one-api  # 修改此行，或注释掉以使用 SQLite 作为数据库
       - REDIS_CONN_STRING=redis://redis
       - SESSION_SECRET=random_string  # 修改为随机字符串
       - TZ=Asia/Shanghai
 #      - NODE_TYPE=slave  # 多机部署时从节点取消注释该行
 #      - SYNC_FREQUENCY=60  # 需要定期从数据库加载数据时取消注释该行
 #      - FRONTEND_BASE_URL=https://openai.justsong.cn  # 多机部署时从节点取消注释该行
-
     depends_on:
       - redis
+      - db
     healthcheck:
       test: [ "CMD-SHELL", "wget -q -O - http://localhost:3000/api/status | grep -o '\"success\":\\s*true' | awk -F: '{print $2}'" ]
       interval: 30s
@@ -32,3 +32,18 @@ services:
     image: redis:latest
     container_name: redis
     restart: always
+
+  db:
+    image: mysql:8.2.0 # mysql版本
+    restart: always
+    container_name: mysql  # 容器名
+    volumes:
+      - ./data/mysql:/var/lib/mysql  #挂载目录，持久化存储
+    ports:
+      - '3306:3306'
+    environment:
+      TZ: Asia/Shanghai   # 设置时区
+      MYSQL_ROOT_PASSWORD: 'One@123456'   # 设置root用户的密码
+      MYSQL_USER: oneapi   # 创建专用用户
+      MYSQL_PASSWORD: '123456'    # 设置专用用户密码
+      MYSQL_DATABASE: one-api   # 自动创建数据库
\ No newline at end of file