blob: 17d270ba880d71d88d05912edb351a8c16e84a09 [file] [log] [blame]
/*
* Copyright 2018-present Open Networking Foundation
* 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 config provides the Log, kvstore, Kafka configuration
package config
import (
"flag"
"fmt"
"github.com/opencord/voltha-go/common/log"
"os"
)
// Open OLT default constants
const (
EtcdStoreName = "etcd"
defaultInstanceid = "openOlt001"
defaultKafkaadapterhost = "127.0.0.1"
defaultKafkaadapterport = 9092
defaultKafkaclusterhost = "127.0.0.1"
defaultKafkaclusterport = 9094
defaultKvstoretype = EtcdStoreName
defaultKvstoretimeout = 5 //in seconds
defaultKvstorehost = "127.0.0.1"
defaultKvstoreport = 2379 // Consul = 8500; Etcd = 2379
defaultLoglevel = 0
defaultBanner = false
defaultDisplayVersionOnly = false
defaultTopic = "openolt"
defaultCoretopic = "rwcore"
defaultOnunumber = 1
)
// AdapterFlags represents the set of configurations used by the read-write adaptercore service
type AdapterFlags struct {
// Command line parameters
InstanceID string
KafkaAdapterHost string
KafkaAdapterPort int
KafkaClusterHost string
KafkaClusterPort int
KVStoreType string
KVStoreTimeout int // in seconds
KVStoreHost string
KVStorePort int
Topic string
CoreTopic string
LogLevel int
OnuNumber int
Banner bool
DisplayVersionOnly bool
}
func init() {
_, _ = log.AddPackage(log.JSON, log.WarnLevel, nil)
}
// NewAdapterFlags returns a new RWCore config
func NewAdapterFlags() *AdapterFlags {
var adapterFlags = AdapterFlags{ // Default values
InstanceID: defaultInstanceid,
KafkaAdapterHost: defaultKafkaadapterhost,
KafkaAdapterPort: defaultKafkaadapterport,
KafkaClusterHost: defaultKafkaclusterhost,
KafkaClusterPort: defaultKafkaclusterport,
KVStoreType: defaultKvstoretype,
KVStoreTimeout: defaultKvstoretimeout,
KVStoreHost: defaultKvstorehost,
KVStorePort: defaultKvstoreport,
Topic: defaultTopic,
CoreTopic: defaultCoretopic,
LogLevel: defaultLoglevel,
OnuNumber: defaultOnunumber,
Banner: defaultBanner,
DisplayVersionOnly: defaultDisplayVersionOnly,
}
return &adapterFlags
}
// ParseCommandArguments parses the arguments when running read-write adaptercore service
func (so *AdapterFlags) ParseCommandArguments() {
help := fmt.Sprintf("Kafka - Adapter messaging host")
flag.StringVar(&(so.KafkaAdapterHost), "kafka_adapter_host", defaultKafkaadapterhost, help)
help = fmt.Sprintf("Kafka - Adapter messaging port")
flag.IntVar(&(so.KafkaAdapterPort), "kafka_adapter_port", defaultKafkaadapterport, help)
help = fmt.Sprintf("Kafka - Cluster messaging host")
flag.StringVar(&(so.KafkaClusterHost), "kafka_cluster_host", defaultKafkaclusterhost, help)
help = fmt.Sprintf("Kafka - Cluster messaging port")
flag.IntVar(&(so.KafkaClusterPort), "kafka_cluster_port", defaultKafkaclusterport, help)
help = fmt.Sprintf("Open OLT topic")
flag.StringVar(&(so.Topic), "adapter_topic", defaultTopic, help)
help = fmt.Sprintf("Core topic")
flag.StringVar(&(so.CoreTopic), "core_topic", defaultCoretopic, help)
help = fmt.Sprintf("KV store type")
flag.StringVar(&(so.KVStoreType), "kv_store_type", defaultKvstoretype, help)
help = fmt.Sprintf("The default timeout when making a kv store request")
flag.IntVar(&(so.KVStoreTimeout), "kv_store_request_timeout", defaultKvstoretimeout, help)
help = fmt.Sprintf("KV store host")
flag.StringVar(&(so.KVStoreHost), "kv_store_host", defaultKvstorehost, help)
help = fmt.Sprintf("KV store port")
flag.IntVar(&(so.KVStorePort), "kv_store_port", defaultKvstoreport, help)
help = fmt.Sprintf("Log level")
flag.IntVar(&(so.LogLevel), "log_level", defaultLoglevel, help)
help = fmt.Sprintf("Number of ONUs")
flag.IntVar(&(so.OnuNumber), "onu_number", defaultOnunumber, help)
help = fmt.Sprintf("Show startup banner log lines")
flag.BoolVar(&(so.Banner), "banner", defaultBanner, help)
help = fmt.Sprintf("Show version information and exit")
flag.BoolVar(&(so.DisplayVersionOnly), "version", defaultDisplayVersionOnly, help)
flag.Parse()
containerName := getContainerInfo()
if len(containerName) > 0 {
so.InstanceID = containerName
}
}
func getContainerInfo() string {
return os.Getenv("HOSTNAME")
}