From d0ab407117bbe369382e9b0330bd330bee489421 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BC=A0=E6=96=8C?= <259278618@qq.com>
Date: Mon, 31 Aug 2026 15:59:40 +0800
Subject: [PATCH] =?UTF-8?q?feat(server):=20dao=20=E5=A2=9E=E5=8A=A0=20GetB?=
 =?UTF-8?q?yNamePrefix=EF=BC=88=E5=B0=81=E9=9D=A2=E5=85=AC=E5=BC=80?=
 =?UTF-8?q?=E8=B7=AF=E7=94=B1=E5=AE=9A=E4=BD=8D=E7=94=A8=EF=BC=89?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 server/biz/dao/dataset.go | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/server/biz/dao/dataset.go b/server/biz/dao/dataset.go
index 639ae7c..f564f10 100644
--- a/server/biz/dao/dataset.go
+++ b/server/biz/dao/dataset.go
@@ -108,6 +108,19 @@ func (d *datasetDao) GetByName(ctx context.Context, name string) (*entity.Datase
 	return &e, nil
 }
 
+// GetByNamePrefix 按训练文件名前缀查询（模型发布命名，前缀唯一），不存在返回 nil
+func (d *datasetDao) GetByNamePrefix(ctx context.Context, prefix string) (*entity.Dataset, error) {
+	var e entity.Dataset
+	err := g.DB().Model(consts.TableDataset).Ctx(ctx).Where("name_prefix", prefix).Scan(&e)
+	if err != nil {
+		if common.IsNoRows(err) {
+			return nil, nil
+		}
+		return nil, err
+	}
+	return &e, nil
+}
+
 // UpdateCounters 更新图片数/已标注数/状态（imgDelta 增量；labeledCount ≥0 时覆盖写）
 func (d *datasetDao) UpdateCounters(ctx context.Context, id int64, imgDelta, labeledCount int64, status string) error {
 	data := g.Map{"updated_at": gtime.Now()}