github.com/m3db/m3@v1.5.0/src/msg/producer/config/buffer.go (about) 1 // Copyright (c) 2018 Uber Technologies, Inc. 2 // 3 // Permission is hereby granted, free of charge, to any person obtaining a copy 4 // of this software and associated documentation files (the "Software"), to deal 5 // in the Software without restriction, including without limitation the rights 6 // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 7 // copies of the Software, and to permit persons to whom the Software is 8 // furnished to do so, subject to the following conditions: 9 // 10 // The above copyright notice and this permission notice shall be included in 11 // all copies or substantial portions of the Software. 12 // 13 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 14 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 15 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 16 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 17 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 18 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN 19 // THE SOFTWARE. 20 21 package config 22 23 import ( 24 "time" 25 26 "github.com/m3db/m3/src/msg/producer/buffer" 27 "github.com/m3db/m3/src/x/instrument" 28 "github.com/m3db/m3/src/x/retry" 29 ) 30 31 // BufferConfiguration configs the buffer. 32 type BufferConfiguration struct { 33 OnFullStrategy *buffer.OnFullStrategy `yaml:"onFullStrategy"` 34 MaxBufferSize *int `yaml:"maxBufferSize"` 35 MaxMessageSize *int `yaml:"maxMessageSize"` 36 CloseCheckInterval *time.Duration `yaml:"closeCheckInterval"` 37 DropOldestInterval *time.Duration `yaml:"dropOldestInterval"` 38 ScanBatchSize *int `yaml:"scanBatchSize"` 39 AllowedSpilloverRatio *float64 `yaml:"allowedSpilloverRatio"` 40 CleanupRetry *retry.Configuration `yaml:"cleanupRetry"` 41 } 42 43 // NewOptions creates new buffer options. 44 func (c *BufferConfiguration) NewOptions(iOpts instrument.Options) buffer.Options { 45 opts := buffer.NewOptions().SetOnFullStrategy(buffer.DropOldest) 46 if c.MaxBufferSize != nil { 47 opts = opts.SetMaxBufferSize(*c.MaxBufferSize) 48 } 49 if c.MaxMessageSize != nil { 50 opts = opts.SetMaxMessageSize(*c.MaxMessageSize) 51 } 52 if c.CloseCheckInterval != nil { 53 opts = opts.SetCloseCheckInterval(*c.CloseCheckInterval) 54 } 55 if c.OnFullStrategy != nil { 56 opts = opts.SetOnFullStrategy(*c.OnFullStrategy) 57 } 58 if c.DropOldestInterval != nil { 59 opts = opts.SetDropOldestInterval(*c.DropOldestInterval) 60 } 61 if c.ScanBatchSize != nil { 62 opts = opts.SetScanBatchSize(*c.ScanBatchSize) 63 } 64 if c.AllowedSpilloverRatio != nil { 65 opts = opts.SetAllowedSpilloverRatio(*c.AllowedSpilloverRatio) 66 } 67 if c.CleanupRetry != nil { 68 opts = opts.SetCleanupRetryOptions(c.CleanupRetry.NewOptions(iOpts.MetricsScope())) 69 } 70 return opts.SetInstrumentOptions(iOpts) 71 }