chore: milestones added in proposal db for migration

icon-community · Nov 10, 2023 · dc3af9a · dc3af9a
1 parent 89a60b1
commit dc3af9a
Showing 1 changed file with 12 additions and 9 deletions.
diff --git a/CPSCore/src/main/java/community/icon/cps/score/cpscore/CPSCore.java b/CPSCore/src/main/java/community/icon/cps/score/cpscore/CPSCore.java
@@ -1409,11 +1409,7 @@ private void updateProgressReportResult() {
     }
 
 
-    @External // TODO : remove in production
-    public void updateProposalPeriod(String ipfsHash){
-        String ipfsHashPrefix = proposalPrefix(ipfsHash);
-        proposalPeriod.at(ipfsHashPrefix).set(25);
-    }
+
 
     private void updateProposalStatus(String _ipfs_hash, Map<String, Object> _proposal_details) {
         String _proposal_status = (String) _proposal_details.get(STATUS);
@@ -2557,7 +2553,7 @@ public void updatePeriodCount(String newHash, int count) {
     }
 
     @External
-    public void submitProposalMock(ProposalAttributes newProposal, String oldHash, int[] milestones) {
+    public void submitProposalMock(ProposalAttributes newProposal, String oldHash, MilestonesAttributes[] milestones) {
         String newHash = newProposal.ipfs_hash;
         if (!containsInArrayDb(oldHash, activeProposals)) {
             Context.revert(TAG + ": Proposal cannot be migrated.");
@@ -2592,6 +2588,14 @@ public void submitProposalMock(ProposalAttributes newProposal, String oldHash, i
 
         addDataToProposalDB(newProposal, newIpfsHashPrefix);
 
+        for (MilestonesAttributes milestone: milestones) {
+            String milestonePrefix = mileStonePrefix(newIpfsHashPrefix,milestone.id);
+            addDataToMilestoneDb(milestone,milestonePrefix);
+            milestoneIds.at(newIpfsHashPrefix).add(milestone.id);
+        }
+
+
+
         BigInteger timestamp = ProposalDataDb.timestamp.at(oldIpfsHashPrefix).getOrDefault(BigInteger.ZERO);
         ProposalDataDb.timestamp.at(newIpfsHashPrefix).set(timestamp);
         proposalPeriod.at(newIpfsHashPrefix).set(31);
@@ -2658,10 +2662,9 @@ public void submitProposalMock(ProposalAttributes newProposal, String oldHash, i
             String progressReportStatus = ProgressReportDataDb.status.at(progressHashPrefix).getOrDefault("");
 
             if (progressReportStatus.equals(APPROVED)) {
-                milestoneSubmitted.at(progressHashPrefix).add(milestones[i]);
+                milestoneSubmitted.at(progressHashPrefix).add(milestones[i].id);
             }
-            String milestonePrefix = mileStonePrefix(newHash, milestones[i]);
-            MilestoneDb.id.at(milestonePrefix).set(milestones[i]);
+            String milestonePrefix = mileStonePrefix(newHash, milestones[i].id);
             MilestoneDb.status.at(milestonePrefix).set(MILESTONE_REPORT_APPROVED);
             MilestoneDb.progressReportHash.at(milestonePrefix).set(report);