From 915b3a6d94a294906628be64e33a7b95d732d67e Mon Sep 17 00:00:00 2001
From: ychao_1983 <ychao_1983@sina.com>
Date: Thu, 10 Mar 2022 17:12:38 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 models/dataset_star.go  | 9 ++++++---
 modules/context/repo.go | 2 ++
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/models/dataset_star.go b/models/dataset_star.go
index fac08caee..5dedb2ba4 100644
--- a/models/dataset_star.go
+++ b/models/dataset_star.go
@@ -51,9 +51,12 @@ func StarDataset(userID, datasetID int64, star bool) error {
 	return sess.Commit()
 }
 
-func IsDatasetStaring(userID, datasetID int64) bool {
-
-	return isDatasetStaring(x, userID, datasetID)
+func IsDatasetStaringByRepoId(userID, repoID int64) bool {
+	dataset, _ := GetDatasetByRepo(&Repository{ID: repoID})
+	if dataset == nil {
+		return false
+	}
+	return isDatasetStaring(x, userID, dataset.ID)
 }
 
 func isDatasetStaring(e Engine, userID, datasetID int64) bool {
diff --git a/modules/context/repo.go b/modules/context/repo.go
index 64f02c921..7c425c8c0 100755
--- a/modules/context/repo.go
+++ b/modules/context/repo.go
@@ -475,6 +475,8 @@ func RepoAssignment() macaron.Handler {
 		if ctx.IsSigned {
 			ctx.Data["IsWatchingRepo"] = models.IsWatching(ctx.User.ID, repo.ID)
 			ctx.Data["IsStaringRepo"] = models.IsStaring(ctx.User.ID, repo.ID)
+
+			ctx.Data["IsStaringDataset"] = models.IsDatasetStaringByRepoId(ctx.User.ID, repo.ID)
 		}
 
 		if repo.IsFork {