github.com/matrixorigin/matrixone@v1.2.0/pkg/vm/engine/tae/common/retry.go (about) 1 // Copyright 2022 Matrix Origin 2 // 3 // Licensed under the Apache License, Version 2.0 (the "License"); 4 // you may not use this file except in compliance with the License. 5 // You may obtain a copy of the License at 6 // 7 // http://www.apache.org/licenses/LICENSE-2.0 8 // 9 // Unless required by applicable law or agreed to in writing, software 10 // distributed under the License is distributed on an "AS IS" BASIS, 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 // See the License for the specific language governing permissions and 13 // limitations under the License. 14 15 package common 16 17 import ( 18 "context" 19 "time" 20 21 "github.com/matrixorigin/matrixone/pkg/common/moerr" 22 ) 23 24 type RetryOp = func() error 25 type WaitOp = func() (ok bool, err error) 26 27 func RetryWithIntervalAndTimeout( 28 op WaitOp, 29 timeout time.Duration, 30 interval time.Duration, suppressTimout bool) (err error) { 31 32 ctx, cancel := context.WithTimeout(context.Background(), timeout) 33 defer cancel() 34 35 ticker := time.NewTicker(interval) 36 defer ticker.Stop() 37 38 var ok bool 39 ok, err = op() 40 if ok { 41 return 42 } 43 44 for { 45 select { 46 case <-ctx.Done(): 47 if suppressTimout { 48 return moerr.GetOkExpectedEOB() 49 } 50 return moerr.NewInternalError(ctx, "timeout") 51 case <-ticker.C: 52 ok, err = op() 53 if ok { 54 return 55 } 56 } 57 } 58 }