fix inserting member story overwriting existing data

munchtoast · Aug 20, 2024 · d19b2c1 · d19b2c1
1 parent 6734bbb
commit d19b2c1
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/subsystem/user_story_member/insert_member_story.go b/subsystem/user_story_member/insert_member_story.go
@@ -3,13 +3,19 @@ package user_story_member
 import (
 	"elichika/client"
 	"elichika/userdata"
+	"elichika/utils"
 )
 
 func InsertMemberStory(session *userdata.Session, storyMemberMasterId int32) {
-	userStoryMember := client.UserStoryMember{
-		StoryMemberMasterId: storyMemberMasterId,
-		IsNew:               true,
-		AcquiredAt:          session.Time.Unix(),
+	userStoryMember := client.UserStoryMember{}
+	exist, err := session.Db.Table("u_story_member").
+		Where("user_id = ? AND story_member_master_id = ?", session.UserId, storyMemberMasterId).Get(&userStoryMember)
+	utils.CheckErr(err)
+	if !exist {
+		session.UserModel.UserStoryMemberById.Set(storyMemberMasterId, client.UserStoryMember{
+			StoryMemberMasterId: storyMemberMasterId,
+			IsNew:               true,
+			AcquiredAt:          session.Time.Unix(),
+		})
 	}
-	session.UserModel.UserStoryMemberById.Set(storyMemberMasterId, userStoryMember)
 }