177 lines
8.4 KiB
Python
177 lines
8.4 KiB
Python
"""量测数据服务"""
|
||
from typing import List, Dict, Tuple
|
||
from sqlalchemy import text
|
||
from sqlalchemy.orm import Session
|
||
from app.core.logging_config import get_logger
|
||
from app.models.measurement_data import MeasurementData
|
||
from app.models.checkpoint import Checkpoint
|
||
from app.models.section_data import SectionData
|
||
from app.schemas.measurement_data import MeasurementDataCreate, MeasurementDataQuery
|
||
from app.schemas.common import BatchImportResponse
|
||
from .table_manager import TableManager
|
||
|
||
logger = get_logger(__name__)
|
||
|
||
class MeasurementDataService:
|
||
"""量测数据服务"""
|
||
|
||
@staticmethod
|
||
def batch_import(db: Session, account_id: int, data: List[MeasurementDataCreate]) -> BatchImportResponse:
|
||
"""批量导入量测数据"""
|
||
table_name = MeasurementData.get_table_name(account_id)
|
||
|
||
if not TableManager.ensure_table_exists(db, "measurement_data", account_id):
|
||
return BatchImportResponse(
|
||
success=False, total=len(data), inserted=0, skipped=0,
|
||
message="创建表失败"
|
||
)
|
||
|
||
# 量测数据不检查重复，直接插入
|
||
if data:
|
||
try:
|
||
values = []
|
||
params = {}
|
||
for i, item in enumerate(data):
|
||
values.append(f"(:monitoring_time_{i}, :upload_time_{i}, :monitoring_value_{i}, "
|
||
f":deformation_value_{i}, :time_interval_{i}, :cumulative_deformation_{i}, "
|
||
f":deformation_rate_{i}, :distance_working_face_{i}, :point_id_{i})")
|
||
params[f"monitoring_time_{i}"] = item.monitoring_time
|
||
params[f"upload_time_{i}"] = item.upload_time
|
||
params[f"monitoring_value_{i}"] = item.monitoring_value
|
||
params[f"deformation_value_{i}"] = item.deformation_value
|
||
params[f"time_interval_{i}"] = item.time_interval
|
||
params[f"cumulative_deformation_{i}"] = item.cumulative_deformation
|
||
params[f"deformation_rate_{i}"] = item.deformation_rate
|
||
params[f"distance_working_face_{i}"] = item.distance_working_face
|
||
params[f"point_id_{i}"] = item.point_id
|
||
|
||
sql = f"""INSERT INTO {table_name}
|
||
(monitoring_time, upload_time, monitoring_value, deformation_value,
|
||
time_interval, cumulative_deformation, deformation_rate, distance_working_face, point_id)
|
||
VALUES {','.join(values)}"""
|
||
db.execute(text(sql), params)
|
||
db.commit()
|
||
logger.info(f"量测数据导入成功: account_id={account_id}, 插入={len(data)}")
|
||
except Exception as e:
|
||
db.rollback()
|
||
logger.error(f"量测数据导入失败: {e}")
|
||
return BatchImportResponse(
|
||
success=False, total=len(data), inserted=0, skipped=0,
|
||
message=f"插入失败: {str(e)}"
|
||
)
|
||
|
||
return BatchImportResponse(
|
||
success=True, total=len(data), inserted=len(data), skipped=0,
|
||
message="导入成功"
|
||
)
|
||
|
||
@staticmethod
|
||
def query(db: Session, params: MeasurementDataQuery) -> Tuple[List[Dict], int]:
|
||
"""查询量测数据"""
|
||
table_name = MeasurementData.get_table_name(params.account_id)
|
||
|
||
if not TableManager.ensure_table_exists(db, "measurement_data", params.account_id):
|
||
return [], 0
|
||
|
||
conditions = []
|
||
query_params = {}
|
||
|
||
if params.point_id:
|
||
conditions.append("point_id = :point_id")
|
||
query_params["point_id"] = params.point_id
|
||
if params.monitoring_time_start:
|
||
conditions.append("monitoring_time >= :monitoring_time_start")
|
||
query_params["monitoring_time_start"] = params.monitoring_time_start
|
||
if params.monitoring_time_end:
|
||
conditions.append("monitoring_time <= :monitoring_time_end")
|
||
query_params["monitoring_time_end"] = params.monitoring_time_end
|
||
|
||
where_clause = " AND ".join(conditions) if conditions else "1=1"
|
||
|
||
count_sql = f"SELECT COUNT(*) FROM {table_name} WHERE {where_clause}"
|
||
total = db.execute(text(count_sql), query_params).scalar()
|
||
|
||
offset = (params.page - 1) * params.page_size
|
||
query_params["limit"] = params.page_size
|
||
query_params["offset"] = offset
|
||
|
||
data_sql = f"SELECT * FROM {table_name} WHERE {where_clause} ORDER BY monitoring_time DESC LIMIT :limit OFFSET :offset"
|
||
result = db.execute(text(data_sql), query_params)
|
||
items = [dict(row._mapping) for row in result.fetchall()]
|
||
|
||
return items, total
|
||
|
||
@staticmethod
|
||
def query_by_department(db: Session, account_id: int, department_id: str,
|
||
page: int = 1, page_size: int = 20,
|
||
monitoring_time_start=None, monitoring_time_end=None) -> Tuple[List[Dict], int]:
|
||
"""根据department_id查询量测数据（包含断面和观测点信息）"""
|
||
measurement_table = MeasurementData.get_table_name(account_id)
|
||
checkpoint_table = Checkpoint.get_table_name(account_id)
|
||
section_table = SectionData.get_table_name(account_id)
|
||
|
||
for table_type in ["measurement_data", "checkpoint", "section_data"]:
|
||
if not TableManager.ensure_table_exists(db, table_type, account_id):
|
||
return [], 0
|
||
|
||
# 获取department下的section信息
|
||
section_sql = f"SELECT section_id, mileage, rock_mass_classification FROM {section_table} WHERE department_id = :department_id"
|
||
section_result = db.execute(text(section_sql), {"department_id": department_id})
|
||
section_map = {row[0]: {"mileage": row[1], "rock_mass_classification": row[2]} for row in section_result.fetchall()}
|
||
|
||
if not section_map:
|
||
return [], 0
|
||
|
||
section_ids = list(section_map.keys())
|
||
section_placeholders = ",".join([f":sid_{i}" for i in range(len(section_ids))])
|
||
params = {f"sid_{i}": sid for i, sid in enumerate(section_ids)}
|
||
|
||
# 获取这些section下的checkpoint信息
|
||
checkpoint_sql = f"SELECT point_id, name, section_id FROM {checkpoint_table} WHERE section_id IN ({section_placeholders})"
|
||
checkpoint_result = db.execute(text(checkpoint_sql), params)
|
||
checkpoint_map = {row[0]: {"name": row[1], "section_id": row[2]} for row in checkpoint_result.fetchall()}
|
||
|
||
if not checkpoint_map:
|
||
return [], 0
|
||
|
||
point_ids = list(checkpoint_map.keys())
|
||
point_placeholders = ",".join([f":pid_{i}" for i in range(len(point_ids))])
|
||
params = {f"pid_{i}": pid for i, pid in enumerate(point_ids)}
|
||
|
||
# 构建时间条件
|
||
time_conditions = []
|
||
if monitoring_time_start:
|
||
time_conditions.append("monitoring_time >= :time_start")
|
||
params["time_start"] = monitoring_time_start
|
||
if monitoring_time_end:
|
||
time_conditions.append("monitoring_time <= :time_end")
|
||
params["time_end"] = monitoring_time_end
|
||
|
||
time_clause = " AND " + " AND ".join(time_conditions) if time_conditions else ""
|
||
|
||
# 查询总数
|
||
count_sql = f"SELECT COUNT(*) FROM {measurement_table} WHERE point_id IN ({point_placeholders}){time_clause}"
|
||
total = db.execute(text(count_sql), params).scalar()
|
||
|
||
# 分页查询
|
||
offset = (page - 1) * page_size
|
||
params["limit"] = page_size
|
||
params["offset"] = offset
|
||
|
||
data_sql = f"SELECT * FROM {measurement_table} WHERE point_id IN ({point_placeholders}){time_clause} ORDER BY monitoring_time DESC LIMIT :limit OFFSET :offset"
|
||
result = db.execute(text(data_sql), params)
|
||
|
||
items = []
|
||
for row in result.fetchall():
|
||
item = dict(row._mapping)
|
||
checkpoint_info = checkpoint_map.get(item.get("point_id"), {})
|
||
item["point_name"] = checkpoint_info.get("name")
|
||
|
||
section_id = checkpoint_info.get("section_id")
|
||
section_info = section_map.get(section_id, {})
|
||
item["section_mileage"] = section_info.get("mileage")
|
||
item["rock_mass_classification"] = section_info.get("rock_mass_classification")
|
||
items.append(item)
|
||
|
||
return items, total
|