324 lines
14 KiB
Rust
324 lines
14 KiB
Rust
//! 提示词模板服务层
|
|
|
|
use sqlx::PgPool;
|
|
use crate::error::{SaasError, SaasResult};
|
|
use crate::common::PaginatedResponse;
|
|
use crate::common::normalize_pagination;
|
|
use super::types::*;
|
|
|
|
/// 创建提示词模板 + 初始版本
|
|
pub async fn create_template(
|
|
db: &PgPool,
|
|
name: &str,
|
|
category: &str,
|
|
description: Option<&str>,
|
|
source: &str,
|
|
system_prompt: &str,
|
|
user_prompt_template: Option<&str>,
|
|
variables: Option<serde_json::Value>,
|
|
min_app_version: Option<&str>,
|
|
) -> SaasResult<PromptTemplateInfo> {
|
|
let id = uuid::Uuid::new_v4().to_string();
|
|
let version_id = uuid::Uuid::new_v4().to_string();
|
|
let now = chrono::Utc::now().to_rfc3339();
|
|
let vars_json = variables.unwrap_or(serde_json::json!([])).to_string();
|
|
|
|
// 插入模板
|
|
sqlx::query(
|
|
"INSERT INTO prompt_templates (id, name, category, description, source, current_version, status, created_at, updated_at)
|
|
VALUES ($1, $2, $3, $4, $5, 1, 'active', $6, $6)"
|
|
)
|
|
.bind(&id).bind(name).bind(category).bind(description).bind(source).bind(&now)
|
|
.execute(db).await.map_err(|e| {
|
|
if e.to_string().contains("unique") {
|
|
SaasError::AlreadyExists(format!("提示词模板 '{}' 已存在", name))
|
|
} else {
|
|
SaasError::Database(e)
|
|
}
|
|
})?;
|
|
|
|
// 插入 v1
|
|
sqlx::query(
|
|
"INSERT INTO prompt_versions (id, template_id, version, system_prompt, user_prompt_template, variables, min_app_version, created_at)
|
|
VALUES ($1, $2, 1, $3, $4, $5, $6, $7)"
|
|
)
|
|
.bind(&version_id).bind(&id).bind(system_prompt).bind(user_prompt_template).bind(&vars_json).bind(min_app_version).bind(&now)
|
|
.execute(db).await?;
|
|
|
|
get_template(db, &id).await
|
|
}
|
|
|
|
/// 获取单个模板
|
|
pub async fn get_template(db: &PgPool, id: &str) -> SaasResult<PromptTemplateInfo> {
|
|
let row: Option<(String, String, String, Option<String>, String, i32, String, String, String)> =
|
|
sqlx::query_as(
|
|
"SELECT id, name, category, description, source, current_version, status, created_at, updated_at
|
|
FROM prompt_templates WHERE id = $1"
|
|
).bind(id).fetch_optional(db).await?;
|
|
|
|
let (id, name, category, description, source, current_version, status, created_at, updated_at) =
|
|
row.ok_or_else(|| SaasError::NotFound(format!("提示词模板 {} 不存在", id)))?;
|
|
|
|
Ok(PromptTemplateInfo { id, name, category, description, source, current_version, status, created_at, updated_at })
|
|
}
|
|
|
|
/// 按名称获取模板
|
|
pub async fn get_template_by_name(db: &PgPool, name: &str) -> SaasResult<PromptTemplateInfo> {
|
|
let row: Option<(String, String, String, Option<String>, String, i32, String, String, String)> =
|
|
sqlx::query_as(
|
|
"SELECT id, name, category, description, source, current_version, status, created_at, updated_at
|
|
FROM prompt_templates WHERE name = $1"
|
|
).bind(name).fetch_optional(db).await?;
|
|
|
|
let (id, name, category, description, source, current_version, status, created_at, updated_at) =
|
|
row.ok_or_else(|| SaasError::NotFound(format!("提示词模板 '{}' 不存在", name)))?;
|
|
|
|
Ok(PromptTemplateInfo { id, name, category, description, source, current_version, status, created_at, updated_at })
|
|
}
|
|
|
|
/// 列表模板
|
|
pub async fn list_templates(
|
|
db: &PgPool,
|
|
query: &PromptListQuery,
|
|
) -> SaasResult<PaginatedResponse<PromptTemplateInfo>> {
|
|
let (page, page_size, offset) = normalize_pagination(query.page, query.page_size);
|
|
|
|
let mut where_clauses = vec!["1=1".to_string()];
|
|
let mut count_sql = String::from("SELECT COUNT(*) FROM prompt_templates WHERE ");
|
|
let mut data_sql = String::from(
|
|
"SELECT id, name, category, description, source, current_version, status, created_at, updated_at
|
|
FROM prompt_templates WHERE "
|
|
);
|
|
|
|
if let Some(ref cat) = query.category {
|
|
where_clauses.push(format!("category = '{}'", cat.replace('\'', "''")));
|
|
}
|
|
if let Some(ref src) = query.source {
|
|
where_clauses.push(format!("source = '{}'", src.replace('\'', "''")));
|
|
}
|
|
if let Some(ref st) = query.status {
|
|
where_clauses.push(format!("status = '{}'", st.replace('\'', "''")));
|
|
}
|
|
|
|
let where_clause = where_clauses.join(" AND ");
|
|
count_sql.push_str(&where_clause);
|
|
data_sql.push_str(&where_clause);
|
|
data_sql.push_str(&format!(" ORDER BY updated_at DESC LIMIT {} OFFSET {}", page_size, offset));
|
|
|
|
let total: i64 = sqlx::query_scalar(&count_sql).fetch_one(db).await?;
|
|
|
|
let rows: Vec<(String, String, String, Option<String>, String, i32, String, String, String)> =
|
|
sqlx::query_as(&data_sql).fetch_all(db).await?;
|
|
|
|
let items: Vec<PromptTemplateInfo> = rows.into_iter().map(|(id, name, category, description, source, current_version, status, created_at, updated_at)| {
|
|
PromptTemplateInfo { id, name, category, description, source, current_version, status, created_at, updated_at }
|
|
}).collect();
|
|
|
|
Ok(PaginatedResponse { items, total, page, page_size })
|
|
}
|
|
|
|
/// 更新模板元数据（不修改内容）
|
|
pub async fn update_template(
|
|
db: &PgPool,
|
|
id: &str,
|
|
description: Option<&str>,
|
|
status: Option<&str>,
|
|
) -> SaasResult<PromptTemplateInfo> {
|
|
let now = chrono::Utc::now().to_rfc3339();
|
|
|
|
if let Some(desc) = description {
|
|
sqlx::query("UPDATE prompt_templates SET description = $1, updated_at = $2 WHERE id = $3")
|
|
.bind(desc).bind(&now).bind(id).execute(db).await?;
|
|
}
|
|
if let Some(st) = status {
|
|
let valid = ["active", "deprecated", "archived"];
|
|
if !valid.contains(&st) {
|
|
return Err(SaasError::InvalidInput(format!("无效状态: {}，允许: {}", st, valid.join(", "))));
|
|
}
|
|
sqlx::query("UPDATE prompt_templates SET status = $1, updated_at = $2 WHERE id = $3")
|
|
.bind(st).bind(&now).bind(id).execute(db).await?;
|
|
}
|
|
|
|
get_template(db, id).await
|
|
}
|
|
|
|
/// 发布新版本
|
|
pub async fn create_version(
|
|
db: &PgPool,
|
|
template_id: &str,
|
|
system_prompt: &str,
|
|
user_prompt_template: Option<&str>,
|
|
variables: Option<serde_json::Value>,
|
|
changelog: Option<&str>,
|
|
min_app_version: Option<&str>,
|
|
) -> SaasResult<PromptVersionInfo> {
|
|
let tmpl = get_template(db, template_id).await?;
|
|
|
|
let new_version = tmpl.current_version + 1;
|
|
let version_id = uuid::Uuid::new_v4().to_string();
|
|
let now = chrono::Utc::now().to_rfc3339();
|
|
let vars_json = variables.unwrap_or(serde_json::json!([])).to_string();
|
|
|
|
sqlx::query(
|
|
"INSERT INTO prompt_versions (id, template_id, version, system_prompt, user_prompt_template, variables, changelog, min_app_version, created_at)
|
|
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9)"
|
|
)
|
|
.bind(&version_id).bind(template_id).bind(new_version)
|
|
.bind(system_prompt).bind(user_prompt_template).bind(&vars_json).bind(changelog).bind(min_app_version).bind(&now)
|
|
.execute(db).await?;
|
|
|
|
// 更新模板的 current_version
|
|
sqlx::query("UPDATE prompt_templates SET current_version = $1, updated_at = $2 WHERE id = $3")
|
|
.bind(new_version).bind(&now).bind(template_id)
|
|
.execute(db).await?;
|
|
|
|
get_version(db, &version_id).await
|
|
}
|
|
|
|
/// 获取特定版本
|
|
pub async fn get_version(db: &PgPool, version_id: &str) -> SaasResult<PromptVersionInfo> {
|
|
let row: Option<(String, String, i32, String, Option<String>, String, Option<String>, Option<String>, String)> =
|
|
sqlx::query_as(
|
|
"SELECT id, template_id, version, system_prompt, user_prompt_template, variables, changelog, min_app_version, created_at
|
|
FROM prompt_versions WHERE id = $1"
|
|
).bind(version_id).fetch_optional(db).await?;
|
|
|
|
let (id, template_id, version, system_prompt, user_prompt_template, variables_str, changelog, min_app_version, created_at) =
|
|
row.ok_or_else(|| SaasError::NotFound(format!("提示词版本 {} 不存在", version_id)))?;
|
|
|
|
let variables: serde_json::Value = serde_json::from_str(&variables_str).unwrap_or(serde_json::json!([]));
|
|
|
|
Ok(PromptVersionInfo { id, template_id, version, system_prompt, user_prompt_template, variables, changelog, min_app_version, created_at })
|
|
}
|
|
|
|
/// 获取模板的当前版本内容
|
|
pub async fn get_current_version(db: &PgPool, template_name: &str) -> SaasResult<PromptVersionInfo> {
|
|
let tmpl = get_template_by_name(db, template_name).await?;
|
|
|
|
let row: Option<(String, String, i32, String, Option<String>, String, Option<String>, Option<String>, String)> =
|
|
sqlx::query_as(
|
|
"SELECT id, template_id, version, system_prompt, user_prompt_template, variables, changelog, min_app_version, created_at
|
|
FROM prompt_versions WHERE template_id = $1 AND version = $2"
|
|
).bind(&tmpl.id).bind(tmpl.current_version).fetch_optional(db).await?;
|
|
|
|
let (id, template_id, version, system_prompt, user_prompt_template, variables_str, changelog, min_app_version, created_at) =
|
|
row.ok_or_else(|| SaasError::NotFound(format!("提示词 '{}' 的版本 {} 不存在", template_name, tmpl.current_version)))?;
|
|
|
|
let variables: serde_json::Value = serde_json::from_str(&variables_str).unwrap_or(serde_json::json!([]));
|
|
|
|
Ok(PromptVersionInfo { id, template_id, version, system_prompt, user_prompt_template, variables, changelog, min_app_version, created_at })
|
|
}
|
|
|
|
/// 列出模板的所有版本
|
|
pub async fn list_versions(
|
|
db: &PgPool,
|
|
template_id: &str,
|
|
) -> SaasResult<Vec<PromptVersionInfo>> {
|
|
let rows: Vec<(String, String, i32, String, Option<String>, String, Option<String>, Option<String>, String)> =
|
|
sqlx::query_as(
|
|
"SELECT id, template_id, version, system_prompt, user_prompt_template, variables, changelog, min_app_version, created_at
|
|
FROM prompt_versions WHERE template_id = $1 ORDER BY version DESC"
|
|
).bind(template_id).fetch_all(db).await?;
|
|
|
|
Ok(rows.into_iter().map(|(id, template_id, version, system_prompt, user_prompt_template, variables_str, changelog, min_app_version, created_at)| {
|
|
let variables = serde_json::from_str(&variables_str).unwrap_or(serde_json::json!([]));
|
|
PromptVersionInfo { id, template_id, version, system_prompt, user_prompt_template, variables, changelog, min_app_version, created_at }
|
|
}).collect())
|
|
}
|
|
|
|
/// 回退到指定版本
|
|
pub async fn rollback_version(
|
|
db: &PgPool,
|
|
template_id: &str,
|
|
target_version: i32,
|
|
) -> SaasResult<PromptTemplateInfo> {
|
|
// 验证目标版本存在
|
|
let exists: (bool,) = sqlx::query_as(
|
|
"SELECT EXISTS(SELECT 1 FROM prompt_versions WHERE template_id = $1 AND version = $2)"
|
|
).bind(template_id).bind(target_version).fetch_one(db).await?;
|
|
|
|
if !exists.0 {
|
|
return Err(SaasError::NotFound(format!("版本 {} 不存在", target_version)));
|
|
}
|
|
|
|
let now = chrono::Utc::now().to_rfc3339();
|
|
sqlx::query("UPDATE prompt_templates SET current_version = $1, updated_at = $2 WHERE id = $3")
|
|
.bind(target_version).bind(&now).bind(template_id)
|
|
.execute(db).await?;
|
|
|
|
get_template(db, template_id).await
|
|
}
|
|
|
|
/// OTA 批量检查更新
|
|
pub async fn check_updates(
|
|
db: &PgPool,
|
|
device_id: &str,
|
|
client_versions: &std::collections::HashMap<String, i32>,
|
|
) -> SaasResult<PromptCheckResponse> {
|
|
let mut updates = Vec::new();
|
|
|
|
for (name, client_ver) in client_versions {
|
|
let tmpl = match get_template_by_name(db, name).await {
|
|
Ok(t) if t.status == "active" => t,
|
|
_ => continue,
|
|
};
|
|
|
|
if tmpl.current_version > *client_ver {
|
|
// 获取最新版本内容
|
|
if let Ok(ver) = get_current_version(db, name).await {
|
|
updates.push(PromptUpdatePayload {
|
|
name: tmpl.name.clone(),
|
|
version: ver.version,
|
|
system_prompt: ver.system_prompt,
|
|
user_prompt_template: ver.user_prompt_template,
|
|
variables: ver.variables,
|
|
source: tmpl.source.clone(),
|
|
min_app_version: ver.min_app_version,
|
|
changelog: ver.changelog,
|
|
});
|
|
|
|
// 更新同步状态
|
|
let now = chrono::Utc::now().to_rfc3339();
|
|
sqlx::query(
|
|
"INSERT INTO prompt_sync_status (device_id, template_id, synced_version, synced_at)
|
|
VALUES ($1, $2, $3, $4)
|
|
ON CONFLICT (device_id, template_id) DO UPDATE SET synced_version = $3, synced_at = $4"
|
|
)
|
|
.bind(device_id).bind(&tmpl.id).bind(ver.version).bind(&now)
|
|
.execute(db).await.ok(); // 非关键，忽略错误
|
|
}
|
|
}
|
|
}
|
|
|
|
// 首次连接：返回所有 active 的 builtin 模板
|
|
if client_versions.is_empty() {
|
|
let all_templates = list_templates(db, &PromptListQuery {
|
|
source: Some("builtin".into()),
|
|
status: Some("active".into()),
|
|
category: None,
|
|
page: None,
|
|
page_size: Some(100),
|
|
}).await?;
|
|
|
|
for tmpl in &all_templates.items {
|
|
if let Ok(ver) = get_current_version(db, &tmpl.name).await {
|
|
updates.push(PromptUpdatePayload {
|
|
name: tmpl.name.clone(),
|
|
version: ver.version,
|
|
system_prompt: ver.system_prompt,
|
|
user_prompt_template: ver.user_prompt_template,
|
|
variables: ver.variables,
|
|
source: tmpl.source.clone(),
|
|
min_app_version: ver.min_app_version,
|
|
changelog: ver.changelog,
|
|
});
|
|
}
|
|
}
|
|
}
|
|
|
|
Ok(PromptCheckResponse {
|
|
updates,
|
|
server_time: chrono::Utc::now().to_rfc3339(),
|
|
})
|
|
}
|