@@ -24,13 +24,13 @@ package v1
2424
2525import (
2626 "fmt"
27- "strings"
28- "strconv"
27+ "github.com/microsoft/frameworkcontroller/pkg/common"
2928 log "github.com/sirupsen/logrus"
3029 core "k8s.io/api/core/v1"
31- "k8s.io/apimachinery/pkg/types"
3230 meta "k8s.io/apimachinery/pkg/apis/meta/v1"
33- "github.com/microsoft/frameworkcontroller/pkg/common"
31+ "k8s.io/apimachinery/pkg/types"
32+ "strconv"
33+ "strings"
3434)
3535
3636///////////////////////////////////////////////////////////////////////////////////////
@@ -63,7 +63,7 @@ func GetFrameworkAttemptInstanceUID(frameworkAttemptID int32, configMapUID *type
6363}
6464
6565func SplitFrameworkAttemptInstanceUID (frameworkAttemptInstanceUID * types.UID ) (
66- frameworkAttemptID int32 , configMapUID * types.UID ) {
66+ frameworkAttemptID int32 , configMapUID * types.UID ) {
6767 parts := strings .Split (string (* frameworkAttemptInstanceUID ), "_" )
6868 i , err := strconv .ParseInt (parts [0 ], 10 , 32 )
6969 if err != nil {
@@ -79,7 +79,7 @@ func GetTaskAttemptInstanceUID(taskAttemptID int32, podUID *types.UID) *types.UI
7979}
8080
8181func SplitTaskAttemptInstanceUID (taskAttemptInstanceUID * types.UID ) (
82- taskAttemptID int32 , podUID * types.UID ) {
82+ taskAttemptID int32 , podUID * types.UID ) {
8383 parts := strings .Split (string (* taskAttemptInstanceUID ), "_" )
8484 i , err := strconv .ParseInt (parts [0 ], 10 , 32 )
8585 if err != nil {
@@ -193,14 +193,14 @@ func (ts *TaskStatus) IsRunning() bool {
193193
194194func (f * Framework ) IsCompleting () bool {
195195 return f .Status .State == FrameworkAttemptDeletionPending ||
196- f .Status .State == FrameworkAttemptDeletionRequested ||
197- f .Status .State == FrameworkAttemptDeleting
196+ f .Status .State == FrameworkAttemptDeletionRequested ||
197+ f .Status .State == FrameworkAttemptDeleting
198198}
199199
200200func (ts * TaskStatus ) IsCompleting () bool {
201201 return ts .State == TaskAttemptDeletionPending ||
202- ts .State == TaskAttemptDeletionRequested ||
203- ts .State == TaskAttemptDeleting
202+ ts .State == TaskAttemptDeletionRequested ||
203+ ts .State == TaskAttemptDeleting
204204}
205205
206206func (ct CompletionType ) IsSucceeded () bool {
@@ -419,7 +419,7 @@ func (f *Framework) NewFrameworkStatus() *FrameworkStatus {
419419}
420420
421421func (f * Framework ) NewFrameworkAttemptStatus (
422- frameworkAttemptID int32 ) FrameworkAttemptStatus {
422+ frameworkAttemptID int32 ) FrameworkAttemptStatus {
423423 return FrameworkAttemptStatus {
424424 ID : frameworkAttemptID ,
425425 StartTime : meta .Now (),
@@ -461,7 +461,7 @@ func (f *Framework) NewTaskStatus(taskRoleName string, taskIndex int32) *TaskSta
461461}
462462
463463func (f * Framework ) NewTaskAttemptStatus (
464- taskRoleName string , taskIndex int32 , taskAttemptID int32 ) TaskAttemptStatus {
464+ taskRoleName string , taskIndex int32 , taskAttemptID int32 ) TaskAttemptStatus {
465465 return TaskAttemptStatus {
466466 ID : taskAttemptID ,
467467 StartTime : meta .Now (),
@@ -510,16 +510,16 @@ func (rd RetryDecision) String() string {
510510}
511511
512512func (rp RetryPolicySpec ) ShouldRetry (
513- rps RetryPolicyStatus ,
514- cs * CompletionStatus ,
515- minDelaySecForTransientConflictFailed int64 ,
516- maxDelaySecForTransientConflictFailed int64 ) RetryDecision {
513+ rps RetryPolicyStatus ,
514+ cs * CompletionStatus ,
515+ minDelaySecForTransientConflictFailed int64 ,
516+ maxDelaySecForTransientConflictFailed int64 ) RetryDecision {
517517 ct := cs .Type
518518
519519 // 0. Built-in Always-on RetryPolicy
520520 if cs .Code == CompletionCodePodSpecInvalid ||
521- cs .Code == CompletionCodeStopFrameworkRequested ||
522- cs .Code == CompletionCodeFrameworkAttemptCompletion {
521+ cs .Code == CompletionCodeStopFrameworkRequested ||
522+ cs .Code == CompletionCodeFrameworkAttemptCompletion {
523523 return RetryDecision {false , true , 0 , cs .Diagnostics }
524524 }
525525
@@ -547,8 +547,8 @@ func (rp RetryPolicySpec) ShouldRetry(
547547
548548 // 2. NormalRetryPolicy
549549 if (rp .MaxRetryCount == ExtendedUnlimitedValue ) ||
550- (ct .IsFailed () && rp .MaxRetryCount == UnlimitedValue ) ||
551- (ct .IsFailed () && rps .AccountableRetriedCount < rp .MaxRetryCount ) {
550+ (ct .IsFailed () && rp .MaxRetryCount == UnlimitedValue ) ||
551+ (ct .IsFailed () && rps .AccountableRetriedCount < rp .MaxRetryCount ) {
552552 return RetryDecision {true , true , 0 , fmt .Sprintf (
553553 "AccountableRetriedCount %v has not reached MaxRetryCount %v" ,
554554 rps .AccountableRetriedCount , rp .MaxRetryCount )}
@@ -584,7 +584,7 @@ func (f *Framework) TransitionFrameworkState(dstState FrameworkState) {
584584
585585// This is the only interface to modify TaskState
586586func (f * Framework ) TransitionTaskState (
587- taskRoleName string , taskIndex int32 , dstState TaskState ) {
587+ taskRoleName string , taskIndex int32 , dstState TaskState ) {
588588 taskStatus := f .TaskStatus (taskRoleName , taskIndex )
589589 srcState := taskStatus .State
590590 if srcState == dstState {
0 commit comments