You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
266 lines
10 KiB
266 lines
10 KiB
package echomeet
|
|
|
|
import (
|
|
"yunyan/comm"
|
|
"yunyan/lego/core"
|
|
"yunyan/lego/core/cbase"
|
|
"yunyan/lego/sys/mysql"
|
|
"yunyan/lego/sys/postgres"
|
|
"yunyan/pb"
|
|
"fmt"
|
|
"path/filepath"
|
|
"runtime"
|
|
|
|
"gorm.io/gorm/clause"
|
|
)
|
|
|
|
// templatePrivateIdStart 模板 id 分段阈值（唯一的归属判定依据）：
|
|
// - id < 此值 → 公共模板，存公共数据库（Postgres），可有多语言版本；
|
|
// - id >= 此值 → 用户私有模板，存业务服务数据库（MySQL），单条、无多语言版本。
|
|
// 由建表时对私有模板表设置自增起点保证两段 id 不重叠（见 modelComp.Init）。
|
|
const templatePrivateIdStart = 10000
|
|
|
|
// 代理模型
|
|
type modelComp struct {
|
|
cbase.ModuleCompBase
|
|
module *Echomeet
|
|
}
|
|
|
|
// 组件初始化接口
|
|
func (this *modelComp) Init(service core.IService, module core.IModule, comp core.IModuleComp, opt core.IModuleOptions) (err error) {
|
|
this.ModuleCompBase.Init(service, module, comp, opt)
|
|
this.module = module.(*Echomeet)
|
|
if err = mysql.CreateTable(comm.TableEchomeetTemplate, &pb.DBEchoMeetTemplate{}); err == nil {
|
|
// 私有模板 id 从分段起点(10000)开始递增，确保与公共模板(公共库，id<10000)id 段不重叠，
|
|
// 从而 gettemplateforid 能仅凭 id 把请求路由到正确的库（仅空表生效）。
|
|
mysql.AutoIncrementStart(comm.TableEchomeetTemplate, "id", templatePrivateIdStart)
|
|
}
|
|
err = mysql.CreateTable(comm.TableEchomeetRecord, &pb.DBEchoMeetRecord{})
|
|
return
|
|
}
|
|
|
|
func (this *modelComp) getuser(uid string) (user *pb.DBUser, err error) {
|
|
user = &pb.DBUser{}
|
|
err = mysql.FindOne(comm.TableUser, user, "uid=?", uid)
|
|
return
|
|
}
|
|
|
|
func (this *modelComp) updateuser(user *pb.DBUser) (err error) {
|
|
err = mysql.Save(comm.TableUser, user)
|
|
return
|
|
}
|
|
|
|
// d读取模版
|
|
func (this *modelComp) getcommtemplates(language string) (models []*pb.DBEchoMeetTemplate, err error) {
|
|
models = make([]*pb.DBEchoMeetTemplate, 0)
|
|
if language == "" {
|
|
err = postgres.Find(comm.TableEchomeetTemplate, &models, "source = ?", "public")
|
|
return
|
|
}
|
|
if err = postgres.Find(comm.TableEchomeetTemplate, &models, "source = ? AND language = ?", "public", language); err != nil {
|
|
return
|
|
}
|
|
if len(models) > 0 {
|
|
return // 精确命中
|
|
}
|
|
// 回退：该区域 locale 无专属模板时，取同基础语言变体（裸 base 或区域变体，如 es-MX→es/es-ES），
|
|
// 按 tid 去重保留 id 最小一条，避免同一模板出现多条。
|
|
if base := baseLang(language); base != "" && base != language {
|
|
var variants []*pb.DBEchoMeetTemplate
|
|
if err = postgres.Find(comm.TableEchomeetTemplate, &variants, "source = ? AND (language = ? OR language LIKE ?)", "public", base, base+"-%"); err != nil {
|
|
return
|
|
}
|
|
models = pickBaseLangVariants(variants, base)
|
|
}
|
|
return
|
|
}
|
|
|
|
// pickBaseLangVariants 从候选模板里挑出与 base 同基础语言的，按 tid 去重保留 id 最小一条。
|
|
// 用于区域 locale（如 es-MX）无专属模板时回退到同基础语言那套（es / es-ES）做列表，避免同 tid 重复。
|
|
func pickBaseLangVariants(candidates []*pb.DBEchoMeetTemplate, base string) []*pb.DBEchoMeetTemplate {
|
|
byTid := make(map[string]*pb.DBEchoMeetTemplate)
|
|
for _, m := range candidates {
|
|
if baseLang(m.Language) != base {
|
|
continue
|
|
}
|
|
if cur, ok := byTid[m.Tid]; !ok || m.Id < cur.Id {
|
|
byTid[m.Tid] = m
|
|
}
|
|
}
|
|
out := make([]*pb.DBEchoMeetTemplate, 0, len(byTid))
|
|
for _, m := range byTid {
|
|
out = append(out, m)
|
|
}
|
|
return out
|
|
}
|
|
|
|
// 读取模版
|
|
func (this *modelComp) gettemplates(uid string, language string) (models []*pb.DBEchoMeetTemplate, err error) {
|
|
models = make([]*pb.DBEchoMeetTemplate, 0)
|
|
if language == "" {
|
|
err = mysql.Find(comm.TableEchomeetTemplate, &models, "source = ?", uid)
|
|
} else {
|
|
err = mysql.Find(comm.TableEchomeetTemplate, &models, "source = ? AND language = ?", uid, language)
|
|
}
|
|
return
|
|
}
|
|
|
|
/*
|
|
gettemplateforid 按主键读取模板，仅凭 id 段判定归属并路由到对应库：
|
|
- id < templatePrivateIdStart：公共模板 → 公共数据库（Postgres）
|
|
- id >= templatePrivateIdStart：用户私有模板 → 业务服务数据库（MySQL）
|
|
私有模板单条、无多语言版本，按主键直接取即可。
|
|
*/
|
|
func (this *modelComp) gettemplateforid(id uint64) (template *pb.DBEchoMeetTemplate, err error) {
|
|
template = &pb.DBEchoMeetTemplate{}
|
|
if id >= templatePrivateIdStart {
|
|
err = mysql.FindOne(comm.TableEchomeetTemplate, template, "id=?", id)
|
|
return
|
|
}
|
|
err = postgres.FindOne(comm.TableEchomeetTemplate, template, "id=?", id)
|
|
return
|
|
}
|
|
|
|
// resolvetemplate 私有模板（id>=起点）单条无多语言，按主键直取；
|
|
// 公共模板 tid 非空按 tid+language 取对应语言版本，否则按主键。
|
|
func (this *modelComp) resolvetemplate(tid string, templateid uint64, language string) (template *pb.DBEchoMeetTemplate, err error) {
|
|
if templateid >= templatePrivateIdStart {
|
|
return this.gettemplateforid(templateid)
|
|
}
|
|
if tid != "" {
|
|
return this.gettemplatefortid(tid, language)
|
|
}
|
|
return this.gettemplateforid(templateid)
|
|
}
|
|
|
|
// gettemplatefortid 按 tid + language 查找模板（用于公共模板）。
|
|
// 精确匹配不到时，回退到同一基础语言的任一变体（如 es-MX→es-ES、zh-TW→zh-CN），
|
|
// 避免区域 locale 没有专属模板时解析失败；FindOne→gorm First 默认按主键 id 升序，取最小一条保证结果稳定。
|
|
func (this *modelComp) gettemplatefortid(tid string, language string) (template *pb.DBEchoMeetTemplate, err error) {
|
|
template = &pb.DBEchoMeetTemplate{}
|
|
if err = postgres.FindOne(comm.TableEchomeetTemplate, template, "tid=? AND language=?", tid, language); err == nil {
|
|
return
|
|
}
|
|
if err = mysql.FindOne(comm.TableEchomeetTemplate, template, "tid=? AND language=?", tid, language); err == nil {
|
|
return
|
|
}
|
|
// 回退：同基础语言（裸 base 或其任一区域变体，如 es / es-ES 都能兜底 es-MX）
|
|
if base := baseLang(language); base != "" && base != language {
|
|
if err = postgres.FindOne(comm.TableEchomeetTemplate, template, "tid=? AND (language=? OR language LIKE ?)", tid, base, base+"-%"); err == nil {
|
|
return
|
|
}
|
|
err = mysql.FindOne(comm.TableEchomeetTemplate, template, "tid=? AND (language=? OR language LIKE ?)", tid, base, base+"-%")
|
|
}
|
|
return
|
|
}
|
|
|
|
func (this *modelComp) addtemplates(template *pb.DBEchoMeetTemplate) (err error) {
|
|
err = mysql.Insert(comm.TableEchomeetTemplate, template)
|
|
return
|
|
}
|
|
|
|
func (this *modelComp) updatetemplates(template *pb.DBEchoMeetTemplate) (err error) {
|
|
err = mysql.Save(comm.TableEchomeetTemplate, template)
|
|
return
|
|
}
|
|
|
|
func (this *modelComp) deltemplates(id []uint64) (err error) {
|
|
err = mysql.Delete(comm.TableEchomeetTemplate, "id IN ?", id)
|
|
return
|
|
}
|
|
|
|
func (this *modelComp) getrecordsforuid(uid string) (models []*pb.DBEchoMeetRecord, err error) {
|
|
models = make([]*pb.DBEchoMeetRecord, 0)
|
|
err = mysql.Find(comm.TableEchomeetRecord, &models, "uid=?", uid)
|
|
return
|
|
}
|
|
|
|
// 查询用户记录的精简信息（不含 original、translate、summary 大字段）
|
|
func (this *modelComp) getrecordbriefforuid(uid string) (models []*pb.DBEchoMeetRecord, err error) {
|
|
models = make([]*pb.DBEchoMeetRecord, 0)
|
|
err = mysql.Table(comm.TableEchomeetRecord).
|
|
Omit("original", "translate", "summary").
|
|
Where("uid = ?", uid).
|
|
Find(&models).Error
|
|
return
|
|
}
|
|
func (this *modelComp) getrecord(id uint64) (model *pb.DBEchoMeetRecord, err error) {
|
|
model = &pb.DBEchoMeetRecord{}
|
|
err = mysql.FindOne(comm.TableEchomeetRecord, model, "id=?", id)
|
|
return
|
|
}
|
|
func (this *modelComp) getrecordByTaskId(taskId string) (model *pb.DBEchoMeetRecord, err error) {
|
|
model = &pb.DBEchoMeetRecord{}
|
|
err = mysql.FindOne(comm.TableEchomeetRecord, model, "taskid=?", taskId)
|
|
return
|
|
}
|
|
func (this *modelComp) getrecords(ids []uint64) (models []*pb.DBEchoMeetRecord, err error) {
|
|
models = make([]*pb.DBEchoMeetRecord, 0)
|
|
err = mysql.Find(comm.TableEchomeetRecord, &models, "id IN ?", ids)
|
|
return
|
|
}
|
|
func (this *modelComp) addrecord(model *pb.DBEchoMeetRecord) (err error) {
|
|
err = mysql.Insert(comm.TableEchomeetRecord, model)
|
|
return
|
|
}
|
|
func (this *modelComp) saverecords(record []*pb.DBEchoMeetRecord) (err error) {
|
|
caller := saveCaller()
|
|
for _, r := range record {
|
|
this.module.Infof("saverecord(batch) [from %s] id:%d state:%s taskid:%q logid:%q",
|
|
caller, r.Id, r.State.String(), r.Taskid, r.Logid)
|
|
}
|
|
err = mysql.Save(comm.TableEchomeetRecord, record)
|
|
if err != nil {
|
|
this.module.Errorf("saverecord(batch) [from %s] 失败 err:%v", caller, err)
|
|
}
|
|
return
|
|
}
|
|
func (this *modelComp) saverecord(record *pb.DBEchoMeetRecord) (err error) {
|
|
caller := saveCaller()
|
|
// 重点观测 taskid：保存时打印调用方+本次写入的 taskid/logid/state。
|
|
// taskid 由「有」变「空」时，对照这行的调用方即可定位是哪段代码覆盖掉的。
|
|
this.module.Infof("saverecord [from %s] id:%d state:%s taskid:%q logid:%q seconds:%d size:%d",
|
|
caller, record.Id, record.State.String(), record.Taskid, record.Logid, record.Seconds, record.Size)
|
|
if record.Taskid == "" && (record.State == pb.DBEchoMeetRecordState_Transcribing ||
|
|
record.State == pb.DBEchoMeetRecordState_AwaitSummarizing ||
|
|
record.State == pb.DBEchoMeetRecordState_Summarizing) {
|
|
this.module.Warnf("saverecord [from %s] id:%d 异常：state=%s 却写入空 taskid（疑似被覆盖）",
|
|
caller, record.Id, record.State.String())
|
|
}
|
|
err = mysql.Save(comm.TableEchomeetRecord, record)
|
|
if err != nil {
|
|
this.module.Errorf("saverecord [from %s] id:%d 失败 err:%v", caller, record.Id, err)
|
|
}
|
|
return
|
|
}
|
|
|
|
// saveCaller 返回调用 saverecord/saverecords 的上一层代码位置（file:line），
|
|
// 用于在日志里标明本次写库是哪条业务路径触发的。
|
|
func saveCaller() string {
|
|
if _, file, line, ok := runtime.Caller(2); ok {
|
|
return fmt.Sprintf("%s:%d", filepath.Base(file), line)
|
|
}
|
|
return "?"
|
|
}
|
|
func (this *modelComp) delrecords(ids []uint64) (err error) {
|
|
err = mysql.Delete(comm.TableEchomeetRecord, "id IN ?", ids)
|
|
return
|
|
}
|
|
|
|
func (this *modelComp) getStatistics(uid string) (statistics *pb.DBUserStatistics, err error) {
|
|
statistics = &pb.DBUserStatistics{Uid: uid}
|
|
mysql.FindOne(comm.TableUserStatistics, statistics, "uid=?", uid)
|
|
return
|
|
}
|
|
|
|
func (this *modelComp) updateStatistics(statistics *pb.DBUserStatistics) (err error) {
|
|
err = mysql.Table(comm.TableUserStatistics).
|
|
Clauses(clause.OnConflict{UpdateAll: true}).
|
|
Create(statistics).Error
|
|
return
|
|
}
|
|
|
|
func (this *modelComp) addUseLog(log *pb.DBUserUseLog) (err error) {
|
|
err = mysql.Insert(comm.TableUserUseLog, log)
|
|
return
|
|
}
|
|
|