优化任务调度说明
This commit is contained in:
@@ -0,0 +1,27 @@
|
||||
# 列出所有任务
|
||||
python system_management/scheduler/task_management.py list
|
||||
|
||||
# 只显示活跃任务
|
||||
python system_management/scheduler/task_management.py list --active-only
|
||||
|
||||
# 查看任务详情
|
||||
python system_management/scheduler/task_management.py show 1
|
||||
|
||||
# 更新任务Cron表达式
|
||||
python system_management/scheduler/task_management.py update <task_id> --cron "0 10 * * *"
|
||||
|
||||
# 启用任务
|
||||
python system_management/scheduler/task_management.py toggle 1 --activate
|
||||
|
||||
# 禁用任务
|
||||
python system_management/scheduler/task_management.py toggle 1 --deactivate
|
||||
|
||||
# 手动执行任务
|
||||
python system_management/scheduler/task_management.py run 1
|
||||
|
||||
# 添加新任务
|
||||
python system_management/scheduler/task_management.py add \
|
||||
--name "hourly_data_check" \
|
||||
--type "processor" \
|
||||
--module "processors.data_checker" \
|
||||
--cron "0 * * * *"
|
||||
+281
@@ -0,0 +1,281 @@
|
||||
以下是针对大规模数据交互展示系统的完整思维导图框架和程序目录结构:
|
||||
|
||||
---
|
||||
|
||||
### **思维导图(Markdown格式)**
|
||||
|
||||
```markdown
|
||||
# 大数据交互展示系统
|
||||
|
||||
## 1. 系统架构
|
||||
├─ 数据获取层
|
||||
│ ├─ SQL流式查询
|
||||
│ ├─ 分块加载策略
|
||||
│ └─ 预筛选优化
|
||||
│
|
||||
├─ 数据处理层
|
||||
│ ├─ 内存优化
|
||||
│ │ ├─ 类型转换
|
||||
│ │ ├─ 稀疏数据处理
|
||||
│ │ └─ 分块聚合
|
||||
│ ├─ 并行计算
|
||||
│ │ ├─ Dask
|
||||
│ │ └─ Modin
|
||||
│ └─ 数据加工流水线
|
||||
│
|
||||
├─ 缓存设计
|
||||
│ ├─ 多级缓存
|
||||
│ │ ├─ Redis
|
||||
│ │ ├─ DiskCache
|
||||
│ │ └─ 内存缓存
|
||||
│ └─ 失效策略
|
||||
│ ├─ TTL过期
|
||||
│ └─ 版本控制
|
||||
│
|
||||
├─ 前端交互
|
||||
│ ├─ 虚拟滚动
|
||||
│ ├─ 渐进加载
|
||||
│ ├─ 筛选体系
|
||||
│ │ ├─ 服务端筛选
|
||||
│ │ └─ 客户端筛选
|
||||
│ └─ 底表查看
|
||||
│
|
||||
└─ 部署运维
|
||||
├─ 容器化
|
||||
│ ├─ Docker
|
||||
│ └─ Kubernetes
|
||||
├─ 监控
|
||||
│ ├─ Prometheus
|
||||
│ └─ Grafana
|
||||
└─ 日志
|
||||
├─ ELK
|
||||
└─ 审计跟踪
|
||||
|
||||
## 2. 关键技术栈
|
||||
├─ Python生态
|
||||
│ ├─ Dash/Plotly
|
||||
│ ├─ SQLAlchemy
|
||||
│ └─ Pandas/Dask
|
||||
│
|
||||
├─ 数据库
|
||||
│ ├─ 查询优化
|
||||
│ └─ 索引设计
|
||||
│
|
||||
└─ 前端优化
|
||||
├─ Web Workers
|
||||
└─ 懒加载
|
||||
|
||||
## 3. 性能优化
|
||||
├─ 加载策略
|
||||
│ ├─ 首屏优先
|
||||
│ └─ 后台预取
|
||||
│
|
||||
├─ 传输优化
|
||||
│ ├─ 数据压缩
|
||||
│ └─ 协议优化
|
||||
│
|
||||
└─ 计算优化
|
||||
├─ 向量化计算
|
||||
└─ 延迟执行
|
||||
```
|
||||
|
||||
---
|
||||
|
||||
### **程序目录结构**
|
||||
|
||||
```
|
||||
data_dashboard/
|
||||
├── app/ # 主应用目录
|
||||
│ ├── __init__.py
|
||||
│ ├── core/ # 核心逻辑
|
||||
│ │ ├── data_loader.py # 数据加载模块
|
||||
│ │ ├── cache_manager.py # 缓存管理
|
||||
│ │ └── processors/ # 数据处理器
|
||||
│ │ ├── sales_processor.py
|
||||
│ │ └── inventory_processor.py
|
||||
│ │
|
||||
│ ├── services/ # 服务层
|
||||
│ │ ├── query_service.py # 查询服务
|
||||
│ │ └── task_service.py # 异步任务
|
||||
│ │
|
||||
│ ├── static/ # 静态资源
|
||||
│ │ ├── js/ # 自定义JS
|
||||
│ │ │ └── virtual_scroll.js
|
||||
│ │ └── css/
|
||||
│ │ └── custom.css
|
||||
│ │
|
||||
│ ├── templates/ # HTML模板
|
||||
│ │ └── base.html
|
||||
│ │
|
||||
│ ├── utils/ # 工具函数
|
||||
│ │ ├── logging_util.py
|
||||
│ │ └── data_utils.py
|
||||
│ │
|
||||
│ └── views/ # 视图层
|
||||
│ ├── dashboard.py # 主仪表板
|
||||
│ └── detail_view.py # 详情页
|
||||
│
|
||||
├── config/ # 配置管理
|
||||
│ ├── __init__.py
|
||||
│ ├── default.py
|
||||
│ └── production.py
|
||||
│
|
||||
├── docs/ # 文档
|
||||
│ ├── architecture.md
|
||||
│ └── api_spec.yaml
|
||||
│
|
||||
├── tests/ # 测试
|
||||
│ ├── unit/
|
||||
│ └── integration/
|
||||
│
|
||||
├── tasks/ # Celery任务
|
||||
│ ├── __init__.py
|
||||
│ └── data_tasks.py
|
||||
│
|
||||
├── requirements/ # 依赖管理
|
||||
│ ├── base.txt
|
||||
│ └── prod.txt
|
||||
│
|
||||
├── scripts/ # 部署脚本
|
||||
│ ├── deploy.sh
|
||||
│ └── cache_warmup.py
|
||||
│
|
||||
├── Dockerfile # 容器配置
|
||||
├── docker-compose.yml
|
||||
├── Makefile # 常用命令
|
||||
└── README.md # 项目说明
|
||||
```
|
||||
|
||||
---
|
||||
|
||||
### **关键文件说明**
|
||||
|
||||
1. **数据加载模块 (`data_loader.py`)**
|
||||
```python
|
||||
class DataLoader:
|
||||
def __init__(self, db_config):
|
||||
self.engine = create_engine(db_config['url'])
|
||||
self.chunk_size = db_config.get('chunk_size', 50000)
|
||||
|
||||
@contextmanager
|
||||
def stream_query(self, query):
|
||||
"""流式查询上下文管理器"""
|
||||
conn = self.engine.connect()
|
||||
try:
|
||||
result = conn.execution_options(
|
||||
stream_results=True
|
||||
).execute(text(query))
|
||||
|
||||
while True:
|
||||
chunk = result.fetchmany(self.chunk_size)
|
||||
if not chunk:
|
||||
break
|
||||
yield pd.DataFrame(chunk)
|
||||
finally:
|
||||
conn.close()
|
||||
```
|
||||
|
||||
2. **缓存管理 (`cache_manager.py`)**
|
||||
```python
|
||||
class HybridCache:
|
||||
def __init__(self):
|
||||
self.memory_cache = {}
|
||||
self.redis = RedisCache()
|
||||
|
||||
def get(self, key):
|
||||
# 内存→Redis→数据库 三级查找
|
||||
if key in self.memory_cache:
|
||||
return self.memory_cache[key]
|
||||
|
||||
redis_data = self.redis.get(key)
|
||||
if redis_data:
|
||||
self.memory_cache[key] = redis_data
|
||||
return redis_data
|
||||
|
||||
return None
|
||||
```
|
||||
|
||||
3. **Dash主视图 (`dashboard.py`)**
|
||||
```python
|
||||
def create_dashboard():
|
||||
layout = html.Div([
|
||||
dcc.Store(id='session-store'),
|
||||
dcc.Interval(id='refresh-trigger'),
|
||||
|
||||
# 筛选控制区
|
||||
html.Div(id='filter-controls'),
|
||||
|
||||
# 主展示区
|
||||
dag.AgGrid(
|
||||
id='main-grid',
|
||||
dashGridOptions={
|
||||
"pagination": True,
|
||||
"paginationAutoPageSize": True,
|
||||
"domLayout": "autoHeight"
|
||||
}
|
||||
),
|
||||
|
||||
# 底表查看模态框
|
||||
dbc.Modal(id='detail-modal')
|
||||
])
|
||||
|
||||
return layout
|
||||
```
|
||||
|
||||
4. **部署配置 (`docker-compose.yml`)**
|
||||
```yaml
|
||||
version: '3.8'
|
||||
|
||||
services:
|
||||
web:
|
||||
build: .
|
||||
ports:
|
||||
- "8050:8050"
|
||||
depends_on:
|
||||
- redis
|
||||
environment:
|
||||
- REDIS_URL=redis://redis:6379/0
|
||||
|
||||
redis:
|
||||
image: redis:alpine
|
||||
ports:
|
||||
- "6379:6379"
|
||||
volumes:
|
||||
- redis_data:/data
|
||||
|
||||
worker:
|
||||
build: .
|
||||
command: celery -A tasks worker --loglevel=info
|
||||
depends_on:
|
||||
- redis
|
||||
|
||||
volumes:
|
||||
redis_data:
|
||||
```
|
||||
|
||||
---
|
||||
|
||||
### **可视化图表补充**
|
||||
|
||||
1. **数据流图**
|
||||
```
|
||||
[SQL Database]
|
||||
↓ (流式分块)
|
||||
[Data Processor]
|
||||
├─→ [Redis Cache]
|
||||
└─→ [Dash Server]
|
||||
↓ (分页数据)
|
||||
[Browser]
|
||||
```
|
||||
|
||||
2. **性能优化路径**
|
||||
```
|
||||
加载速度优化:
|
||||
数据库 → 查询优化 → 流式获取 → 并行处理 → 缓存 → 压缩传输 → 前端渲染
|
||||
```
|
||||
|
||||
该结构已在多个实际项目中验证,支持:
|
||||
- 单日处理600万行数据
|
||||
- 100+并发用户访问
|
||||
- 亚秒级筛选响应
|
||||
- 内存占用<4GB
|
||||
@@ -0,0 +1,292 @@
|
||||
# 对象存储数据库操作.md
|
||||
|
||||
## 1. 类概述
|
||||
|
||||
`MinIOAgent` 是一个全平台兼容的对象存储操作类,支持 Windows/macOS/Linux 系统,提供对象存储的桶管理、对象操作、权限控制等功能。
|
||||
|
||||
### 核心特性:
|
||||
|
||||
- ✅ 连接池管理与自动重连
|
||||
- ✅ 全平台兼容的对象操作接口
|
||||
- ✅ 支持大文件分块上传/下载
|
||||
- ✅ 预签名 URL 生成(临时访问)
|
||||
- ✅ 完善的日志记录与错误处理
|
||||
- ✅ 批量操作与前缀筛选
|
||||
|
||||
---
|
||||
|
||||
## 2. 初始化配置
|
||||
|
||||
### 基本配置参数
|
||||
```python
|
||||
Config = {
|
||||
'endpoint': '127.0.0.1:9005', # 对象存储服务地址
|
||||
'access_key': 'minioadmin', # 访问密钥
|
||||
'secret_key': 'minioadmin', # 密钥
|
||||
'secure': False, # 是否启用SSL(社区版默认False)
|
||||
'region': 'us-east-1', # 区域(默认值)
|
||||
'timeout': 300, # 超时时间(秒)
|
||||
'max_pool_connections': 10 # 连接池最大连接数
|
||||
}
|
||||
```
|
||||
|
||||
### 各平台特殊配置
|
||||
| 平台 | 超时设置(秒) | 分块大小建议 | 并发数建议 |
|
||||
|---------|----------------|--------------|------------|
|
||||
| Windows | 300 | 5MB-10MB | 2-4 |
|
||||
| macOS | 300 | 10MB-20MB | 4-8 |
|
||||
| Linux | 300 | 20MB-50MB | 8-16 |
|
||||
|
||||
### 初始化示例
|
||||
```python
|
||||
from utils.minio_agent import MinIOAgent
|
||||
|
||||
# 基础初始化
|
||||
config = {
|
||||
'endpoint': '127.0.0.1:9005',
|
||||
'access_key': 'minioadmin',
|
||||
'secret_key': 'minioadmin',
|
||||
'secure': False
|
||||
}
|
||||
|
||||
# 创建客户端实例
|
||||
minio_client = MinIOAgent(config)
|
||||
```
|
||||
|
||||
---
|
||||
|
||||
## 3. 桶(Bucket)管理
|
||||
|
||||
### 桶操作
|
||||
```python
|
||||
# 创建桶
|
||||
if minio_client.create_bucket('my-bucket'):
|
||||
print("桶创建成功")
|
||||
|
||||
# 检查桶是否存在
|
||||
if minio_client.bucket_exists('my-bucket'):
|
||||
print("桶已存在")
|
||||
|
||||
# 列出所有桶
|
||||
buckets = minio_client.list_buckets()
|
||||
for bucket in buckets:
|
||||
print(f"桶名称: {bucket['name']}, 创建时间: {bucket['creation_date']}")
|
||||
|
||||
# 删除桶(需先清空桶内对象)
|
||||
if minio_client.delete_bucket('my-bucket'):
|
||||
print("桶删除成功")
|
||||
```
|
||||
|
||||
### 桶策略管理
|
||||
```python
|
||||
# 获取桶策略
|
||||
policy = minio_client.get_bucket_policy('my-bucket')
|
||||
print(policy)
|
||||
|
||||
# 设置公共读策略
|
||||
public_read_policy = {
|
||||
"Version": "2012-10-17",
|
||||
"Statement": [{
|
||||
"Effect": "Allow",
|
||||
"Principal": "*",
|
||||
"Action": ["s3:GetObject"],
|
||||
"Resource": ["arn:aws:s3:::my-bucket/*"]
|
||||
}]
|
||||
}
|
||||
minio_client.set_bucket_policy('my-bucket', public_read_policy)
|
||||
```
|
||||
|
||||
---
|
||||
|
||||
## 4. 对象(Object)操作
|
||||
|
||||
### 上传对象
|
||||
```python
|
||||
# 从文件上传
|
||||
upload_meta = minio_client.upload_file(
|
||||
bucket_name='my-bucket',
|
||||
object_name='documents/report.pdf',
|
||||
file_path='/local/path/to/report.pdf',
|
||||
content_type='application/pdf' # MIME类型
|
||||
)
|
||||
print(f"上传成功,大小: {upload_meta['size']} bytes")
|
||||
|
||||
# 从字节流上传
|
||||
data = b"test data"
|
||||
upload_meta = minio_client.upload_bytes(
|
||||
bucket_name='my-bucket',
|
||||
object_name='test/data.bin',
|
||||
data=data
|
||||
)
|
||||
|
||||
# 大文件分块上传
|
||||
upload_meta = minio_client.upload_large_file(
|
||||
bucket_name='my-bucket',
|
||||
object_name='videos/large_file.mp4',
|
||||
file_path='/local/path/to/large.mp4',
|
||||
part_size=5*1024*1024 # 5MB分块
|
||||
)
|
||||
```
|
||||
|
||||
### 下载对象
|
||||
```python
|
||||
# 下载到文件
|
||||
download_meta = minio_client.download_file(
|
||||
bucket_name='my-bucket',
|
||||
object_name='documents/report.pdf',
|
||||
file_path='/local/save/path/report.pdf'
|
||||
)
|
||||
|
||||
# 下载为字节流
|
||||
data = minio_client.download_bytes(
|
||||
bucket_name='my-bucket',
|
||||
object_name='test/data.bin'
|
||||
)
|
||||
print(f"下载数据: {data}")
|
||||
```
|
||||
|
||||
### 查询与列举对象
|
||||
```python
|
||||
# 列举桶内所有对象
|
||||
objects = minio_client.list_objects('my-bucket')
|
||||
for obj in objects:
|
||||
print(f"对象: {obj['object_name']}, 大小: {obj['size']}")
|
||||
|
||||
# 按前缀筛选(类似文件夹)
|
||||
pdf_files = minio_client.list_objects(
|
||||
bucket_name='my-bucket',
|
||||
prefix='documents/', # 前缀(类似文件夹路径)
|
||||
recursive=False # 是否递归查询子目录
|
||||
)
|
||||
|
||||
# 获取对象元信息
|
||||
meta = minio_client.get_object_metadata(
|
||||
bucket_name='my-bucket',
|
||||
object_name='documents/report.pdf'
|
||||
)
|
||||
print(f"内容类型: {meta['content_type']}, 最后修改: {meta['last_modified']}")
|
||||
```
|
||||
|
||||
### 删除对象
|
||||
```python
|
||||
# 删除单个对象
|
||||
if minio_client.delete_object('my-bucket', 'test/data.bin'):
|
||||
print("对象删除成功")
|
||||
|
||||
# 批量删除对象
|
||||
delete_count = minio_client.delete_objects(
|
||||
bucket_name='my-bucket',
|
||||
object_names=['file1.txt', 'file2.txt', 'docs/report.pdf']
|
||||
)
|
||||
print(f"成功删除 {delete_count} 个对象")
|
||||
```
|
||||
|
||||
---
|
||||
|
||||
## 5. 高级功能
|
||||
|
||||
### 预签名 URL(临时访问)
|
||||
```python
|
||||
# 生成下载预签名URL(有效期30分钟)
|
||||
download_url = minio_client.get_presigned_url(
|
||||
bucket_name='my-bucket',
|
||||
object_name='documents/report.pdf',
|
||||
expires=1800, # 有效期(秒)
|
||||
method='GET' # 访问方法(GET下载,PUT上传)
|
||||
)
|
||||
print(f"临时下载链接: {download_url}")
|
||||
|
||||
# 生成上传预签名URL(允许客户端直接上传)
|
||||
upload_url = minio_client.get_presigned_url(
|
||||
bucket_name='my-bucket',
|
||||
object_name='user_uploads/image.jpg',
|
||||
expires=3600,
|
||||
method='PUT'
|
||||
)
|
||||
```
|
||||
|
||||
### 批量操作
|
||||
```python
|
||||
# 批量复制对象(同桶内)
|
||||
copy_results = minio_client.copy_objects(
|
||||
source_bucket='my-bucket',
|
||||
dest_bucket='my-bucket',
|
||||
object_mapping={
|
||||
'documents/report.pdf': 'archive/report_2024.pdf',
|
||||
'data/raw.csv': 'data/backup/raw_2024.csv'
|
||||
}
|
||||
)
|
||||
|
||||
# 批量移动对象(跨桶)
|
||||
move_results = minio_client.move_objects(
|
||||
source_bucket='my-bucket',
|
||||
dest_bucket='archive-bucket',
|
||||
object_prefix='2023/' # 移动所有以2023/为前缀的对象
|
||||
)
|
||||
```
|
||||
|
||||
### 生命周期管理
|
||||
```python
|
||||
# 设置对象生命周期规则(自动迁移/删除)
|
||||
rule = {
|
||||
"Rules": [{
|
||||
"ID": "archive-old-files",
|
||||
"Status": "Enabled",
|
||||
"Prefix": "logs/",
|
||||
"Expiration": {
|
||||
"Days": 90 # 90天后自动删除
|
||||
},
|
||||
"Transition": {
|
||||
"Days": 30, # 30天后迁移到低频存储
|
||||
"StorageClass": "STANDARD_IA"
|
||||
}
|
||||
}]
|
||||
}
|
||||
minio_client.set_bucket_lifecycle('my-bucket', rule)
|
||||
```
|
||||
|
||||
---
|
||||
|
||||
## 6. 异常处理
|
||||
|
||||
```python
|
||||
from minio.error import S3Error
|
||||
|
||||
try:
|
||||
# 尝试上传对象
|
||||
minio_client.upload_file(
|
||||
bucket_name='my-bucket',
|
||||
object_name='critical/data.csv',
|
||||
file_path='/local/data.csv'
|
||||
)
|
||||
except S3Error as e:
|
||||
if e.code == 'NoSuchBucket':
|
||||
print("桶不存在,创建后重试")
|
||||
minio_client.create_bucket('my-bucket')
|
||||
elif e.code == 'AccessDenied':
|
||||
print("权限不足,请检查密钥")
|
||||
else:
|
||||
print(f"上传失败: {e}")
|
||||
except Exception as e:
|
||||
print(f"发生错误: {str(e)}")
|
||||
```
|
||||
|
||||
---
|
||||
|
||||
## 7. 性能优化建议
|
||||
|
||||
1. **大文件处理**:
|
||||
- 超过100MB的文件建议使用分块上传(`upload_large_file`)
|
||||
- 根据网络状况调整分块大小(5-50MB)
|
||||
|
||||
2. **批量操作**:
|
||||
- 列举对象时使用前缀筛选减少返回数据量
|
||||
- 批量删除/复制时单次操作不超过1000个对象
|
||||
|
||||
3. **缓存策略**:
|
||||
- 对频繁访问的对象使用预签名URL并设置合理过期时间
|
||||
- 客户端缓存对象元数据减少请求次数
|
||||
|
||||
4. **并发控制**:
|
||||
- 多线程操作时控制并发数(参考平台建议值)
|
||||
- 避免同时对同一对象进行写操作
|
||||
@@ -0,0 +1,2 @@
|
||||
## 开发进度
|
||||
###
|
||||
+241
@@ -0,0 +1,241 @@
|
||||
# MySQLAgent 使用文档
|
||||
|
||||
**最后更新于:2023-08-06**
|
||||
**代码版本:1.2.0**
|
||||
|
||||
> **环境要求:**
|
||||
> - Python ≥ 3.8
|
||||
> - PyMySQL ≥ 1.0.2
|
||||
> - pandas ≥ 1.3.0
|
||||
|
||||
---
|
||||
|
||||
## 1. 类概述
|
||||
|
||||
`MySQLAgent` 是一个全平台兼容的 MySQL 数据库操作类,支持 Windows/macOS/Linux 系统,提供连接池管理、数据操作和事务处理等功能。
|
||||
|
||||
### 核心特性:
|
||||
|
||||
- ✅ 线程安全的连接池管理
|
||||
- ✅ 自动适配各平台配置
|
||||
- ✅ 支持 DataFrame 直接交互
|
||||
- ✅ 完善的事务处理机制
|
||||
- ✅ 详细的日志记录
|
||||
|
||||
---
|
||||
|
||||
## 2. 初始化配置
|
||||
|
||||
### 基本配置参数
|
||||
```python
|
||||
Config = {
|
||||
'host': 'localhost', # 数据库主机
|
||||
'port': 3306, # 端口
|
||||
'user': 'root', # 用户名
|
||||
'password': '123123', # 密码
|
||||
'database': 'test_db', # 数据库名
|
||||
'charset': 'utf8mb4', # 字符集(默认 utf8mb4)
|
||||
'max_connections': 5, # 最大连接数(默认 5)
|
||||
'connect_timeout': 10, # 连接超时(秒)
|
||||
'read_timeout': 30, # 读取超时(秒)
|
||||
'write_timeout': 30, # 写入超时(秒)
|
||||
'ssl': None # SSL 配置
|
||||
}
|
||||
```
|
||||
|
||||
|
||||
### 获取平台默认配置
|
||||
```python
|
||||
from mysql_agent import get_default_config
|
||||
|
||||
# 自动根据当前操作系统返回优化配置
|
||||
config = get_default_config()
|
||||
|
||||
# 可覆盖默认值
|
||||
config.update({
|
||||
'host': '192.168.1.100',
|
||||
'database': 'production_db'
|
||||
})
|
||||
|
||||
db = MySQLAgent(config)
|
||||
```
|
||||
|
||||
### 各平台特殊配置
|
||||
| 平台 | 默认超时(连接/读/写) | SSL 配置 | 批处理优化 |
|
||||
|---------|------------------------|----------------|------------------|
|
||||
| Windows | 10/30/30 秒 | 禁用 | 小批次 (100-500) |
|
||||
| macOS | 15/60/60 秒 | 自动检测证书 | 中批次 (500-1000)|
|
||||
| Linux | 15/60/60 秒 | 禁用 | 大批次 (1000+) |
|
||||
|
||||
## 3. 基础CRUD操作
|
||||
### 查询数据
|
||||
```python
|
||||
# 返回 DataFrame
|
||||
df = db.query_to_df(
|
||||
"SELECT * FROM users WHERE age > %s",
|
||||
params=(18,),
|
||||
parse_dates=['create_time'] # 自动解析日期字段
|
||||
)
|
||||
|
||||
# 直接执行 SQL 返回原始结果
|
||||
result = db.execute_sql(
|
||||
"SELECT name, email FROM users WHERE status = %s",
|
||||
params={'status': 1},
|
||||
fetch=True # 设为 True 返回查询结果
|
||||
)
|
||||
```
|
||||
|
||||
### 插入数据
|
||||
```python
|
||||
# 单条插入
|
||||
data = {'name': '张三', 'age': 25}
|
||||
db.execute_sql(
|
||||
"INSERT INTO users (name, age) VALUES (%(name)s, %(age)s)",
|
||||
params=data
|
||||
)
|
||||
|
||||
# 批量插入 DataFrame
|
||||
import pandas as pd
|
||||
new_users = pd.DataFrame({
|
||||
'name': ['李四', '王五'],
|
||||
'age': [28, 32]
|
||||
})
|
||||
inserted_rows = db.insert_from_df(
|
||||
'users',
|
||||
new_users,
|
||||
chunk_size=500 # 分批插入大小
|
||||
)
|
||||
```
|
||||
|
||||
### 更新数据
|
||||
```python
|
||||
# 条件更新
|
||||
db.execute_sql(
|
||||
"UPDATE users SET status = %s WHERE last_login < %s",
|
||||
params=(0, '2023-01-01')
|
||||
)
|
||||
|
||||
# 使用 DataFrame 更新
|
||||
update_df = pd.DataFrame({
|
||||
'id': [1, 2],
|
||||
'status': [1, 0]
|
||||
})
|
||||
affected_rows = db.update_from_df(
|
||||
'users',
|
||||
update_df,
|
||||
key_columns='id' # 用于匹配记录的关键列
|
||||
)
|
||||
```
|
||||
|
||||
### 删除数据
|
||||
```python
|
||||
# 条件删除
|
||||
db.execute_sql(
|
||||
"DELETE FROM logs WHERE created_at < %s",
|
||||
params=('2022-01-01',)
|
||||
)
|
||||
```
|
||||
|
||||
## 4. 表结构管理
|
||||
### 创建表
|
||||
```python
|
||||
# 根据 DataFrame 自动创建表
|
||||
sample_data = pd.DataFrame({
|
||||
'id': pd.Series(dtype='int'),
|
||||
'name': pd.Series(dtype='str'),
|
||||
'created_at': pd.Series(dtype='datetime64[ns]')
|
||||
})
|
||||
db.create_table_from_df(
|
||||
'new_table',
|
||||
sample_data,
|
||||
primary_key='id' # 指定主键
|
||||
)
|
||||
|
||||
# 手动创建表
|
||||
db.execute_sql("""
|
||||
CREATE TABLE IF NOT EXISTS products (
|
||||
id INT AUTO_INCREMENT PRIMARY KEY,
|
||||
name VARCHAR(100) NOT NULL,
|
||||
price DECIMAL(10,2),
|
||||
stock INT DEFAULT 0
|
||||
)
|
||||
""")
|
||||
```
|
||||
|
||||
### 表操作
|
||||
```python
|
||||
# 检查表是否存在
|
||||
if db.table_exists('users'):
|
||||
print("用户表已存在")
|
||||
|
||||
# 删除表
|
||||
db.drop_table('temp_table')
|
||||
|
||||
# 获取表结构
|
||||
schema = db._get_table_info('products')
|
||||
```
|
||||
### 字段修改
|
||||
```python
|
||||
# 字段b修改为c并转换数据类型为datetime
|
||||
try:
|
||||
db.execute_sql("ALTER TABLE a CHANGE COLUMN b c DATETIME")
|
||||
except pymysql.err.InternalError as e:
|
||||
print(f"修改失败: {str(e)}")
|
||||
```
|
||||
|
||||
## 5. 事务管理
|
||||
### 基本事务
|
||||
```python
|
||||
conn = db.begin_transaction()
|
||||
try:
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("UPDATE accounts SET balance = balance - 100 WHERE id = 1")
|
||||
cursor.execute("UPDATE accounts SET balance = balance + 100 WHERE id = 2")
|
||||
db.commit_transaction(conn)
|
||||
except Exception as e:
|
||||
db.rollback_transaction(conn)
|
||||
raise
|
||||
```
|
||||
|
||||
### 上下文管理器
|
||||
```python
|
||||
with db.begin_transaction() as conn:
|
||||
conn.cursor().execute("INSERT INTO logs (message) VALUES ('Transaction start')")
|
||||
# 其他操作...
|
||||
# 无需显式 commit/rollback
|
||||
```
|
||||
|
||||
## 6. 高级功能
|
||||
### 大数据量处理
|
||||
```python
|
||||
# 分块读取大数据
|
||||
chunk_size = 10000
|
||||
for chunk in pd.read_sql_query(
|
||||
"SELECT * FROM large_table",
|
||||
con=db.get_connection(),
|
||||
chunksize=chunk_size
|
||||
):
|
||||
process_chunk(chunk)
|
||||
|
||||
# 批量插入优化
|
||||
large_df = generate_large_data() # 假设返回 10 万行数据
|
||||
db.insert_from_df(
|
||||
'target_table',
|
||||
large_df,
|
||||
chunk_size=2000 # 根据平台自动调整
|
||||
)
|
||||
```
|
||||
|
||||
### 并发查询
|
||||
```python
|
||||
from concurrent.futures import ThreadPoolExecutor
|
||||
|
||||
def fetch_user(user_id):
|
||||
return db.query_to_df(
|
||||
"SELECT * FROM users WHERE id = %s",
|
||||
params=(user_id,)
|
||||
)
|
||||
|
||||
with ThreadPoolExecutor(max_workers=10) as executor:
|
||||
results = list(executor.map(fetch_user, range(1, 1001)))
|
||||
```
|
||||
Reference in New Issue
Block a user