blob: d866e5d4f4f356e3edcc1d527c5e2ffd6d550262 [file] [log] [blame]
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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 global
type ConsumerConfig struct {
Filter string `yaml:"filter" json:"filter,omitempty" property:"filter"`
RegistryIDs []string `yaml:"registry-ids" json:"registry-ids,omitempty" property:"registry-ids"`
Protocol string `yaml:"protocol" json:"protocol,omitempty" property:"protocol"`
RequestTimeout string `default:"3s" yaml:"request-timeout" json:"request-timeout,omitempty" property:"request-timeout"`
ProxyFactory string `default:"default" yaml:"proxy" json:"proxy,omitempty" property:"proxy"`
Check bool `yaml:"check" json:"check,omitempty" property:"check"`
AdaptiveService bool `default:"false" yaml:"adaptive-service" json:"adaptive-service" property:"adaptive-service"`
// there is no need to configure References, it will be replaced by instance.NewClient
//References map[string]*client.ReferenceConfig `yaml:"references" json:"references,omitempty" property:"references"`
TracingKey string `yaml:"tracing-key" json:"tracing-key" property:"tracing-key"`
FilterConf interface{} `yaml:"filter-conf" json:"filter-conf,omitempty" property:"filter-conf"`
MaxWaitTimeForServiceDiscovery string `default:"3s" yaml:"max-wait-time-for-service-discovery" json:"max-wait-time-for-service-discovery,omitempty" property:"max-wait-time-for-service-discovery"`
MeshEnabled bool `yaml:"mesh-enabled" json:"mesh-enabled,omitempty" property:"mesh-enabled"`
}
func DefaultConsumerConfig() *ConsumerConfig {
return &ConsumerConfig{
RequestTimeout: "3s",
Check: true,
}
}
type ConsumerOption func(*ConsumerConfig)
func WithConsumer_Filter(filter string) ConsumerOption {
return func(cfg *ConsumerConfig) {
cfg.Filter = filter
}
}
func WithConsumer_RegistryIDs(ids []string) ConsumerOption {
return func(cfg *ConsumerConfig) {
cfg.RegistryIDs = ids
}
}
func WithConsumer_Protocol(protocol string) ConsumerOption {
return func(cfg *ConsumerConfig) {
cfg.Protocol = protocol
}
}
func WithConsumer_RequestTimeout(timeout string) ConsumerOption {
return func(cfg *ConsumerConfig) {
cfg.RequestTimeout = timeout
}
}
func WithConsumer_ProxyFactory(factory string) ConsumerOption {
return func(cfg *ConsumerConfig) {
cfg.ProxyFactory = factory
}
}
func WithConsumer_Check(flag bool) ConsumerOption {
return func(cfg *ConsumerConfig) {
cfg.Check = flag
}
}
func WithConsumer_AdaptiveService(flag bool) ConsumerOption {
return func(cfg *ConsumerConfig) {
cfg.AdaptiveService = flag
}
}
func WithConsumer_TracingKey(key string) ConsumerOption {
return func(cfg *ConsumerConfig) {
cfg.TracingKey = key
}
}
func WithConsumer_FilterConf(conf interface{}) ConsumerOption {
return func(cfg *ConsumerConfig) {
cfg.FilterConf = conf
}
}
func WithConsumer_MaxWaitTimeForServiceDiscovery(time string) ConsumerOption {
return func(cfg *ConsumerConfig) {
cfg.MaxWaitTimeForServiceDiscovery = time
}
}
func WithConsumer_MeshEnabled(flag bool) ConsumerOption {
return func(cfg *ConsumerConfig) {
cfg.MeshEnabled = flag
}
}