!fixup: Support custom live deck

Signed-off-by: Yuan Si <do4suki@gmail.com>

handler/unit.go

@@ -110,127 +110,39 @@ func SetDeckHandler(ctx *gin.Context) {
 		panic(err)
 	}
 
-	UserEng.ShowSQL(true)
+	// 开始事务
+	// UserEng.ShowSQL(true)
 	session := UserEng.NewSession()
 	defer session.Close()
-
-	for _, deck := range deckReq.UnitDeckList {
-		exists, err := UserEng.Table("user_deck_m").Where("user_id = ? AND deck_id = ?", uId, deck.UnitDeckID).Exist()
-		CheckErr(err)
-
-		if exists { // 原队伍
-			// 原队伍信息
-			var userDeckId int
-			_, err = UserEng.Table("user_deck_m").Cols("id").Where("user_id = ? AND deck_id = ?", uId, deck.UnitDeckID).Get(&userDeckId)
-			CheckErr(err)
-
-			// fmt.Println(userDeckId)
-
-			// 事务
 	if err = session.Begin(); err != nil {
 		session.Rollback()
 		panic(err)
 	}
-			// 更新队伍信息
-			userDeck := tools.UserDeckData{
-				MainFlag: deck.MainFlag,
-				DeckName: deck.DeckName,
-			}
-			_, err := session.Table("user_deck_m").Update(&userDeck, &tools.UserDeckData{
-				UserID: uId,
-				DeckID: deck.UnitDeckID,
-			})
-			if err != nil {
-				session.Rollback()
-				panic(err)
-			}
-			// fmt.Println("Aff:", r)
 
-			// 更新队伍成员信息
-			for _, unit := range deck.UnitDeckDetail {
-				// 原队伍成员位置信息
-				unitDeckData := tools.UnitDeckData{}
-				_, err = session.Table("deck_unit_m").Where("user_deck_id = ? AND position = ?", userDeckId, unit.Position).Get(&unitDeckData)
+	// 原有队伍信息
+	var userDeckId []int
+	err = session.Table("user_deck_m").Cols("id").Where("user_id = ?", uId).Find(&userDeckId)
 	if err != nil {
 		session.Rollback()
 		panic(err)
 	}
 
-				// 该位置成员信息不变
-				if unitDeckData.UnitOwningUserID == unit.UnitOwningUserID {
-					continue
-				}
-
-				// 新成员信息
-				// fmt.Printf("位置: %d 成员发生改变, %d -> %d\n", unit.Position, unitDeckData.UnitOwningUserID, unit.UnitOwningUserID)
-				newUnitData := tools.UnitData{}
-				exists, err := session.Table("user_unit_m").Where("unit_owning_user_id = ?", unit.UnitOwningUserID).Exist()
+	// 删除全部原有队伍成员
+	_, err = session.Table("deck_unit_m").In("user_deck_id", userDeckId).Delete()
 	if err != nil {
 		session.Rollback()
 		panic(err)
 	}
-				if exists {
-					// fmt.Println("新成员为用户增加成员")
-					_, err = session.Table("user_unit_m").Where("unit_owning_user_id = ?", unit.UnitOwningUserID).Get(&newUnitData)
-					if err != nil {
-						session.Rollback()
-						panic(err)
-					}
-				} else {
-					exists, err := MainEng.Table("common_unit_m").Where("unit_owning_user_id = ?", unit.UnitOwningUserID).Exist()
-					if err != nil {
-						session.Rollback()
-						panic(err)
-					}
-					if exists {
-						// fmt.Println("新成员为公共成员")
-						_, err = MainEng.Table("common_unit_m").Where("unit_owning_user_id = ?", unit.UnitOwningUserID).Get(&newUnitData)
-						if err != nil {
-							session.Rollback()
-							panic(err)
-						}
-					} else {
-						// fmt.Println("新成员不存在")
-						session.Rollback()
-						panic("wtf?")
-					}
-				}
-				// fmt.Println("新的成员信息:", newUnitData)
-
-				// 更新新成员信息
-				newUnitDeckData := tools.UnitDeckData{}
-				b, err := json.Marshal(newUnitData)
-				if err != nil {
-					session.Rollback()
-					panic(err)
-				}
-				if err = json.Unmarshal(b, &newUnitDeckData); err != nil {
-					session.Rollback()
-					panic(err)
-				}
-				newUnitDeckData.BeforeLove = newUnitDeckData.MaxLove
-				newUnitDeckData.Position = unit.Position
-				newUnitDeckData.UserDeckID = userDeckId
-				newUnitDeckData.InsertData = time.Now().Unix()
-
-				_, err = session.Table("deck_unit_m").Update(&newUnitDeckData, &tools.UnitDeckData{
-					ID: unitDeckData.ID,
-				})
-				if err != nil {
-					session.Rollback()
-					panic(err)
-				}
-			}
-			if err = session.Commit(); err != nil {
-				session.Rollback()
-				panic(err)
-			}
-		} else { // 新队伍
-			if err := session.Begin(); err != nil {
-				session.Rollback()
-				panic(err)
-			}
 
+	// 删除全部原有队伍
+	_, err = session.Table("user_deck_m").In("id", userDeckId).Delete()
+	if err != nil {
+		session.Rollback()
+		panic(err)
+	}
+
+	// 遍历新队伍
+	for _, deck := range deckReq.UnitDeckList {
 		// 新队伍信息
 		userDeck := tools.UserDeckData{
 			DeckID:     deck.UnitDeckID,
@@ -279,7 +191,7 @@ func SetDeckHandler(ctx *gin.Context) {
 				} else {
 					// fmt.Println("新成员不存在")
 					session.Rollback()
-						panic("wtf?")
+					panic("unexpected operation")
 				}
 			}
 			// fmt.Println("新的成员信息:", newUnitData)
@@ -306,12 +218,13 @@ func SetDeckHandler(ctx *gin.Context) {
 				panic(err)
 			}
 		}
+	}
+
+	// 结束事务
 	if err = session.Commit(); err != nil {
 		session.Rollback()
 		panic(err)
 	}
-		}
-	}
 
 	dispResp := SetDisplayRankResp{
 		ResponseData: []interface{}{},