restore: reset milestone deadline_unix and closed_date_unix (#5264)

1 files changed, 12 insertions, 4 deletions

diff --git a/models/models.go b/models/models.go
index f0f63e07..6407f413 100644
--- a/models/models.go
+++ b/models/models.go
@@ -360,16 +360,24 @@ func ImportDatabase(dirPath string, verbose bool) (err error) {
 				return fmt.Errorf("insert strcut: %v", err)
 			}
 
+			meta := make(map[string]interface{})
+			if err = jsoniter.Unmarshal(scanner.Bytes(), &meta); err != nil {
+				log.Error(2, "Failed to unmarshal to map: %v", err)
+			}
+
 			// Reset created_unix back to the date save in archive because Insert method updates its value
 			if isInsertProcessor && !skipInsertProcessors[rawTableName] {
-				meta := make(map[string]interface{})
-				if err = jsoniter.Unmarshal(scanner.Bytes(), &meta); err != nil {
-					log.Error(2, "Failed to unmarshal to map: %v", err)
-				}
 				if _, err = x.Exec("UPDATE "+rawTableName+" SET created_unix=? WHERE id=?", meta["CreatedUnix"], meta["ID"]); err != nil {
 					log.Error(2, "Failed to reset 'created_unix': %v", err)
 				}
 			}
+
+			switch rawTableName {
+			case "milestone":
+				if _, err = x.Exec("UPDATE "+rawTableName+" SET deadline_unix=?, closed_date_unix=? WHERE id=?", meta["DeadlineUnix"], meta["ClosedDateUnix"], meta["ID"]); err != nil {
+					log.Error(2, "Failed to reset 'milestone.deadline_unix', 'milestone.closed_date_unix': %v", err)
+				}
+			}
 		}
 
 		// PostgreSQL needs manually reset table sequence for auto increment keys