feat: 交叉验证 AI vs LAMP(#15,一致确认/不一致升级会诊)

This commit is contained in:
weijuesen
2026-08-12 18:00:02 +08:00
parent 6f367d19ae
commit 1798b2909e
9 changed files with 199 additions and 0 deletions
+75
View File
@@ -2,6 +2,7 @@ package handler
import (
"bytes"
"encoding/json"
"io"
"net/http"
"strconv"
@@ -26,6 +27,7 @@ func RegisterLampRoutes(rg *gin.RouterGroup, db *gorm.DB, s3 *service.S3Service,
rg.GET("/lamp-tests/:id/steps", read, listLampTestSteps(db))
rg.PATCH("/lamp-tests/:id/steps/:stepNo", write, updateLampTestStep(db))
rg.POST("/lamp-tests/:id/result-image", write, uploadLampResultImage(db, s3, imageBucket))
rg.GET("/lamp-tests/:id/cross-validation", read, getLampCrossValidation(db))
}
// listLampTests 检测任务单列表(roomId/batchId/status 过滤)
@@ -97,6 +99,8 @@ func updateLampTest(db *gorm.DB) gin.HandlerFunc {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
resultSet := false
resultValue := ""
if r, ok := updates["result"]; ok {
result, _ := r.(string)
if result != "" && !model.ValidLampResult(result) {
@@ -109,15 +113,86 @@ func updateLampTest(db *gorm.DB) gin.HandlerFunc {
} else {
updates["status"] = "testing"
}
resultSet = true
resultValue = result
}
if len(updates) > 0 {
db.Model(&model.LampTest{}).Where("id = ?", id).Updates(updates)
}
// 交叉验证:结果落库后与同房间最近巡检比对
if resultSet && resultValue != "" {
_ = runCrossValidation(db, id, resultValue)
}
db.Where("id = ?", id).First(&t)
c.JSON(http.StatusOK, t)
}
}
// runCrossValidation 用同房间最近的巡检记录做交叉验证,并回写 cross_status/cross_reason
func runCrossValidation(db *gorm.DB, lampTestID, lampResult string) error {
var t model.LampTest
if err := db.Where("id = ?", lampTestID).First(&t).Error; err != nil {
return err
}
if t.RoomID == nil {
return nil
}
var inspection model.InspectionRecord
if err := db.Where("room_id = ?", *t.RoomID).Order("created_at DESC").First(&inspection).Error; err != nil {
return nil // 无关联巡检,保持 pending
}
var detections []service.AIDetection
if len(inspection.Detections) > 0 {
_ = json.Unmarshal(inspection.Detections, &detections)
}
aiClass := service.AIClassFromDetections(detections)
var diseases []string
if len(t.Diseases) > 0 {
_ = json.Unmarshal(t.Diseases, &diseases)
}
ok, reason := service.CrossValidate(aiClass, lampResult, diseases)
status := "inconsistent"
if ok {
status = "consistent"
}
return db.Model(&model.LampTest{}).Where("id = ?", lampTestID).
Updates(map[string]interface{}{"cross_status": status, "cross_reason": reason}).Error
}
// getLampCrossValidation 交叉验证详情(含关联巡检摘要)
func getLampCrossValidation(db *gorm.DB) gin.HandlerFunc {
return func(c *gin.Context) {
id := c.Param("id")
var t model.LampTest
if db.Where("id = ?", id).First(&t).Error != nil {
c.JSON(http.StatusNotFound, gin.H{"error": "lamp test not found"})
return
}
var related map[string]interface{}
if t.RoomID != nil {
var inspection model.InspectionRecord
if db.Where("room_id = ?", *t.RoomID).Order("created_at DESC").First(&inspection).Error == nil {
var detections []service.AIDetection
if len(inspection.Detections) > 0 {
_ = json.Unmarshal(inspection.Detections, &detections)
}
related = map[string]interface{}{
"id": inspection.ID,
"imageUrl": inspection.ImageURL,
"aiClass": service.AIClassFromDetections(detections),
"riskLevel": inspection.RiskLevel,
"createdAt": inspection.CreatedAt,
}
}
}
c.JSON(http.StatusOK, gin.H{
"crossStatus": t.CrossStatus,
"crossReason": t.CrossReason,
"relatedInspection": related,
})
}
}
// deleteLampTest 删除任务单(级联删除步骤)
func deleteLampTest(db *gorm.DB) gin.HandlerFunc {
return func(c *gin.Context) {
+2
View File
@@ -17,6 +17,8 @@ type LampTest struct {
ResultImageURL *string `gorm:"column:result_image_url;size:512" json:"resultImageUrl,omitempty"`
OperatorID *string `gorm:"column:operator_id;type:uuid" json:"operatorId,omitempty"`
ResultedAt *time.Time `gorm:"column:resulted_at;type:timestamptz" json:"resultedAt,omitempty"`
CrossStatus string `gorm:"column:cross_status;size:16;default:pending" json:"crossStatus"` // pending/consistent/inconsistent
CrossReason *string `gorm:"column:cross_reason;type:text" json:"crossReason,omitempty"`
Note *string `gorm:"type:text" json:"note,omitempty"`
CreatedAt time.Time `gorm:"type:timestamptz" json:"createdAt"`
UpdatedAt time.Time `gorm:"type:timestamptz" json:"updatedAt"`
@@ -0,0 +1,34 @@
package service
// CrossValidate 交叉验证:AI 检测结果 vs LAMP 检测结果(规格书:一致→确认诊断;不一致→升级专家会诊)
func CrossValidate(aiClass, lampResult string, lampDiseases []string) (bool, string) {
switch {
case lampResult == "invalid":
return false, "LAMP 判读无效,建议复检或专家会诊"
case aiClass == "":
return false, "未找到关联巡检记录,暂无法交叉验证"
case aiClass == "sick" && lampResult == "positive":
return true, "AI 检出异常与 LAMP 阳性一致,确认诊断"
case aiClass == "healthy" && lampResult == "negative":
return true, "AI 未见异常与 LAMP 阴性一致"
case aiClass == "sick" && lampResult == "negative":
return false, "AI 检出异常但 LAMP 阴性,建议专家会诊"
case aiClass == "healthy" && lampResult == "positive":
return false, "AI 未见异常但 LAMP 阳性,建议专家会诊"
default:
return false, "交叉验证结果待确认"
}
}
// AIClassFromDetections 从巡检检测结果归纳 AI 结论(任一非 healthy 视为 sick
func AIClassFromDetections(detections []AIDetection) string {
if len(detections) == 0 {
return ""
}
for _, d := range detections {
if d.ClassName != "healthy" {
return "sick"
}
}
return "healthy"
}
@@ -0,0 +1,40 @@
package service
import "testing"
func TestCrossValidate(t *testing.T) {
cases := []struct {
name string
aiClass string
lampResult string
wantOK bool
}{
{"AI异常+LAMP阳性 一致", "sick", "positive", true},
{"AI健康+LAMP阴性 一致", "healthy", "negative", true},
{"AI异常+LAMP阴性 不一致", "sick", "negative", false},
{"AI健康+LAMP阳性 不一致", "healthy", "positive", false},
{"LAMP无效 不一致", "sick", "invalid", false},
{"无AI记录 不一致", "", "positive", false},
}
for _, c := range cases {
ok, reason := CrossValidate(c.aiClass, c.lampResult, []string{"核型多角体病"})
if ok != c.wantOK {
t.Errorf("%s: ok=%v, want %v(原因 %s", c.name, ok, c.wantOK, reason)
}
if reason == "" {
t.Errorf("%s: 缺少原因说明", c.name)
}
}
}
func TestAIClassFromDetections(t *testing.T) {
if got := AIClassFromDetections(nil); got != "" {
t.Errorf("空检测应为空,实际 %s", got)
}
if got := AIClassFromDetections([]AIDetection{{ClassName: "healthy", Confidence: 0.9}}); got != "healthy" {
t.Errorf("全健康应为 healthy,实际 %s", got)
}
if got := AIClassFromDetections([]AIDetection{{ClassName: "healthy"}, {ClassName: "sick", Confidence: 0.6}}); got != "sick" {
t.Errorf("含 sick 应为 sick,实际 %s", got)
}
}