@@ -2,10 +2,11 @@ package state
22
33import (
44 "fmt"
5- "github.com/jfrog/gofrog/safeconvert"
65 "path/filepath"
76 "time"
87
8+ "github.com/jfrog/gofrog/safeconvert"
9+
910 "github.com/jfrog/gofrog/datastructures"
1011 "github.com/jfrog/jfrog-cli-core/v2/artifactory/commands/transferfiles/api"
1112 "github.com/jfrog/jfrog-cli-core/v2/utils/coreutils"
@@ -49,7 +50,7 @@ func NewTransferStateManager(loadRunStatus bool) (*TransferStateManager, error)
4950 }
5051 stateManager .TransferRunStatus = transferRunStatus
5152 }
52- stateManager .TimeEstimationManager . stateManager = & stateManager
53+ stateManager .stateManager = & stateManager
5354 return & stateManager , nil
5455}
5556
@@ -128,7 +129,7 @@ func (ts *TransferStateManager) SetRepoState(repoKey string, totalSizeBytes, tot
128129func (ts * TransferStateManager ) SetRepoFullTransferStarted (startTime time.Time ) error {
129130 // We do not want to change the start time if it already exists, because it means we continue transferring from a snapshot.
130131 // Some dirs may not be searched again (if done exploring or completed), so handling their diffs from the original time is required.
131- return ts .TransferState . Action (func (state * TransferState ) error {
132+ return ts .Action (func (state * TransferState ) error {
132133 if state .CurrentRepo .FullTransfer .Started == "" {
133134 state .CurrentRepo .FullTransfer .Started = ConvertTimeToRFC3339 (startTime )
134135 }
@@ -137,23 +138,23 @@ func (ts *TransferStateManager) SetRepoFullTransferStarted(startTime time.Time)
137138}
138139
139140func (ts * TransferStateManager ) SetRepoFullTransferCompleted () error {
140- return ts .TransferState . Action (func (state * TransferState ) error {
141+ return ts .Action (func (state * TransferState ) error {
141142 state .CurrentRepo .FullTransfer .Ended = ConvertTimeToRFC3339 (time .Now ())
142143 return nil
143144 })
144145}
145146
146147// Increasing Transferred Diff files (modified files) and SizeByBytes value in suitable repository progress state
147148func (ts * TransferStateManager ) IncTransferredSizeAndFilesPhase1 (chunkTotalFiles , chunkTotalSizeInBytes int64 ) error {
148- err := ts .TransferState . Action (func (state * TransferState ) error {
149+ err := ts .Action (func (state * TransferState ) error {
149150 atomicallyAddInt64 (& state .CurrentRepo .Phase1Info .TransferredSizeBytes , chunkTotalSizeInBytes )
150151 atomicallyAddInt64 (& state .CurrentRepo .Phase1Info .TransferredUnits , chunkTotalFiles )
151152 return nil
152153 })
153154 if err != nil {
154155 return err
155156 }
156- return ts .TransferRunStatus . action (func (transferRunStatus * TransferRunStatus ) error {
157+ return ts .action (func (transferRunStatus * TransferRunStatus ) error {
157158 atomicallyAddInt64 (& transferRunStatus .OverallTransfer .TransferredSizeBytes , chunkTotalSizeInBytes )
158159 atomicallyAddInt64 (& transferRunStatus .OverallTransfer .TransferredUnits , chunkTotalFiles )
159160
@@ -165,15 +166,15 @@ func (ts *TransferStateManager) IncTransferredSizeAndFilesPhase1(chunkTotalFiles
165166}
166167
167168func (ts * TransferStateManager ) IncTransferredSizeAndFilesPhase2 (chunkTotalFiles , chunkTotalSizeInBytes int64 ) error {
168- return ts .TransferState . Action (func (state * TransferState ) error {
169+ return ts .Action (func (state * TransferState ) error {
169170 atomicallyAddInt64 (& state .CurrentRepo .Phase2Info .TransferredSizeBytes , chunkTotalSizeInBytes )
170171 atomicallyAddInt64 (& state .CurrentRepo .Phase2Info .TransferredUnits , chunkTotalFiles )
171172 return nil
172173 })
173174}
174175
175176func (ts * TransferStateManager ) IncTotalSizeAndFilesPhase2 (filesNumber , totalSize int64 ) error {
176- return ts .TransferState . Action (func (state * TransferState ) error {
177+ return ts .Action (func (state * TransferState ) error {
177178 atomicallyAddInt64 (& state .CurrentRepo .Phase2Info .TotalSizeBytes , totalSize )
178179 atomicallyAddInt64 (& state .CurrentRepo .Phase2Info .TotalUnits , filesNumber )
179180 return nil
@@ -182,7 +183,7 @@ func (ts *TransferStateManager) IncTotalSizeAndFilesPhase2(filesNumber, totalSiz
182183
183184// Set relevant information of files and storage we need to transfer in phase3
184185func (ts * TransferStateManager ) SetTotalSizeAndFilesPhase3 (filesNumber , totalSize int64 ) error {
185- return ts .TransferState . Action (func (state * TransferState ) error {
186+ return ts .Action (func (state * TransferState ) error {
186187 state .CurrentRepo .Phase3Info .TransferredUnits = 0
187188 state .CurrentRepo .Phase3Info .TransferredSizeBytes = 0
188189 atomicallyAddInt64 (& state .CurrentRepo .Phase3Info .TotalSizeBytes , totalSize )
@@ -193,7 +194,7 @@ func (ts *TransferStateManager) SetTotalSizeAndFilesPhase3(filesNumber, totalSiz
193194
194195// Increase transferred storage and files in phase 3
195196func (ts * TransferStateManager ) IncTransferredSizeAndFilesPhase3 (chunkTotalFiles , chunkTotalSizeInBytes int64 ) error {
196- return ts .TransferState . Action (func (state * TransferState ) error {
197+ return ts .Action (func (state * TransferState ) error {
197198 atomicallyAddInt64 (& state .CurrentRepo .Phase3Info .TransferredSizeBytes , chunkTotalSizeInBytes )
198199 atomicallyAddInt64 (& state .CurrentRepo .Phase3Info .TransferredUnits , chunkTotalFiles )
199200 return nil
@@ -202,7 +203,7 @@ func (ts *TransferStateManager) IncTransferredSizeAndFilesPhase3(chunkTotalFiles
202203
203204// Returns pointers to TotalStorage, TotalFiles, TransferredFiles and TransferredStorage from progressState of a specific Repository.
204205func (ts * TransferStateManager ) GetStorageAndFilesRepoPointers (phase int ) (totalFailedStorage , totalUploadedFailedStorage , totalFailedFiles , totalUploadedFailedFiles * int64 , err error ) {
205- err = ts .TransferState . Action (func (state * TransferState ) error {
206+ err = ts .Action (func (state * TransferState ) error {
206207 switch phase {
207208 case api .Phase1 :
208209 totalFailedStorage = & ts .CurrentRepo .Phase1Info .TotalSizeBytes
@@ -228,7 +229,7 @@ func (ts *TransferStateManager) GetStorageAndFilesRepoPointers(phase int) (total
228229// Adds new diff details to the repo's diff array in state.
229230// Marks files handling as started, and sets the handling range.
230231func (ts * TransferStateManager ) AddNewDiffToState (startTime time.Time ) error {
231- return ts .TransferState . Action (func (state * TransferState ) error {
232+ return ts .Action (func (state * TransferState ) error {
232233
233234 newDiff := DiffDetails {}
234235
@@ -254,7 +255,7 @@ func (ts *TransferStateManager) AddNewDiffToState(startTime time.Time) error {
254255}
255256
256257func (ts * TransferStateManager ) SetFilesDiffHandlingCompleted () error {
257- return ts .TransferState . Action (func (state * TransferState ) error {
258+ return ts .Action (func (state * TransferState ) error {
258259 state .CurrentRepo .Diffs [len (state .CurrentRepo .Diffs )- 1 ].FilesDiffRunTime .Ended = ConvertTimeToRFC3339 (time .Now ())
259260 state .CurrentRepo .Diffs [len (state .CurrentRepo .Diffs )- 1 ].Completed = true
260261 return nil
@@ -281,14 +282,14 @@ func (ts *TransferStateManager) GetReposTransferredSizeBytes(repoKeys ...string)
281282}
282283
283284func (ts * TransferStateManager ) GetTransferredSizeBytes () (transferredSizeBytes int64 , err error ) {
284- return transferredSizeBytes , ts .TransferRunStatus . action (func (transferRunStatus * TransferRunStatus ) error {
285+ return transferredSizeBytes , ts .action (func (transferRunStatus * TransferRunStatus ) error {
285286 transferredSizeBytes = transferRunStatus .OverallTransfer .TransferredSizeBytes
286287 return nil
287288 })
288289}
289290
290291func (ts * TransferStateManager ) GetDiffHandlingRange () (start , end time.Time , err error ) {
291- return start , end , ts .TransferState . Action (func (state * TransferState ) error {
292+ return start , end , ts .Action (func (state * TransferState ) error {
292293 var inErr error
293294 start , inErr = ConvertRFC3339ToTime (state .CurrentRepo .Diffs [len (state .CurrentRepo .Diffs )- 1 ].HandledRange .Started )
294295 if inErr != nil {
@@ -307,42 +308,42 @@ func (ts *TransferStateManager) IncVisitedFolders() error {
307308}
308309
309310func (ts * TransferStateManager ) ChangeDelayedFilesCountBy (count uint64 , increase bool ) error {
310- return ts .TransferRunStatus . action (func (transferRunStatus * TransferRunStatus ) error {
311+ return ts .action (func (transferRunStatus * TransferRunStatus ) error {
311312 atomicallyAddUint64 (& transferRunStatus .DelayedFiles , count , increase )
312313 return nil
313314 })
314315}
315316
316317func (ts * TransferStateManager ) ChangeTransferFailureCountBy (count uint64 , increase bool ) error {
317- return ts .TransferRunStatus . action (func (transferRunStatus * TransferRunStatus ) error {
318+ return ts .action (func (transferRunStatus * TransferRunStatus ) error {
318319 atomicallyAddUint64 (& transferRunStatus .TransferFailures , count , increase )
319320 return nil
320321 })
321322}
322323
323324func (ts * TransferStateManager ) IncRepositoriesTransferred () error {
324- return ts .TransferRunStatus . action (func (transferRunStatus * TransferRunStatus ) error {
325+ return ts .action (func (transferRunStatus * TransferRunStatus ) error {
325326 transferRunStatus .TotalRepositories .TransferredUnits ++
326327 return nil
327328 })
328329}
329330
330331func (ts * TransferStateManager ) SetRepoPhase (phaseId int ) error {
331- return ts .TransferRunStatus . action (func (transferRunStatus * TransferRunStatus ) error {
332+ return ts .action (func (transferRunStatus * TransferRunStatus ) error {
332333 transferRunStatus .CurrentRepoPhase = phaseId
333334 return nil
334335 })
335336}
336337
337338func (ts * TransferStateManager ) SetWorkingThreads (workingThreads int ) error {
338- return ts .TransferRunStatus . action (func (transferRunStatus * TransferRunStatus ) error {
339+ return ts .action (func (transferRunStatus * TransferRunStatus ) error {
339340 transferRunStatus .WorkingThreads = workingThreads
340341 return nil
341342 })
342343}
343344
344345func (ts * TransferStateManager ) GetWorkingThreads () (workingThreads int , err error ) {
345- return workingThreads , ts .TransferRunStatus . action (func (transferRunStatus * TransferRunStatus ) error {
346+ return workingThreads , ts .action (func (transferRunStatus * TransferRunStatus ) error {
346347 workingThreads = transferRunStatus .WorkingThreads
347348 return nil
348349 })
0 commit comments