2020-03-16 03:20:17 +01:00
|
|
|
// Copyright Project Harbor Authors
|
|
|
|
//
|
|
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
// you may not use this file except in compliance with the License.
|
|
|
|
// You may obtain a copy of the License at
|
|
|
|
//
|
|
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
//
|
|
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
// See the License for the specific language governing permissions and
|
|
|
|
// limitations under the License.
|
|
|
|
|
|
|
|
package quota
|
|
|
|
|
2022-08-04 04:08:02 +02:00
|
|
|
import "github.com/goharbor/harbor/src/lib/retry"
|
|
|
|
|
2020-03-16 03:20:17 +01:00
|
|
|
// Option option for `Refresh` method of `Controller`
|
|
|
|
type Option func(*Options)
|
|
|
|
|
2020-07-22 05:18:05 +02:00
|
|
|
// Options options used by `Refresh`, `Get`, `List` methods of `Controller`
|
2020-03-16 03:20:17 +01:00
|
|
|
type Options struct {
|
2020-07-22 05:18:05 +02:00
|
|
|
IgnoreLimitation bool
|
|
|
|
WithReferenceObject bool
|
2022-08-04 04:08:02 +02:00
|
|
|
// RetryOptions is the sets of options but for retry function.
|
|
|
|
RetryOptions []retry.Option
|
2020-03-16 03:20:17 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
// IgnoreLimitation set IgnoreLimitation for the Options
|
|
|
|
func IgnoreLimitation(ignoreLimitation bool) func(*Options) {
|
|
|
|
return func(opts *Options) {
|
|
|
|
opts.IgnoreLimitation = ignoreLimitation
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-07-22 05:18:05 +02:00
|
|
|
// WithReferenceObject set WithReferenceObject to true for the Options
|
|
|
|
func WithReferenceObject() func(*Options) {
|
|
|
|
return func(opts *Options) {
|
|
|
|
opts.WithReferenceObject = true
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-08-04 04:08:02 +02:00
|
|
|
// WithRetryOptions set RetryOptions to Options
|
|
|
|
func WithRetryOptions(retryOpts []retry.Option) func(*Options) {
|
|
|
|
return func(opts *Options) {
|
|
|
|
opts.RetryOptions = retryOpts
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-03-16 03:20:17 +01:00
|
|
|
func newOptions(options ...Option) *Options {
|
|
|
|
opts := &Options{}
|
|
|
|
for _, f := range options {
|
|
|
|
f(opts)
|
|
|
|
}
|
|
|
|
return opts
|
|
|
|
}
|