mirror of
https://github.com/postmannen/ctrl.git
synced 2025-03-31 01:24:31 +00:00
Implemented env variables for all flags, and removed config flag. Also added use of .env file.
This commit is contained in:
postmannen
parent
19fc8665a8
commit
5c921f0eb4
5 changed files with 108 additions and 647 deletions
107
Dockerfile
107
Dockerfile
|
|
@ -17,109 +17,4 @@ RUN apk update && apk add curl && apk add nmap
|
|||
WORKDIR /app
|
||||
COPY --from=build-env /build/cmd/ctrl/ctrl /app/
|
||||
|
||||
ENV CONFIG_FOLDER "./etc"
|
||||
ENV SOCKET_FOLDER "./tmp"
|
||||
ENV TCP_LISTENER ""
|
||||
ENV HTTP_LISTENER "localhost:8091"
|
||||
ENV DATABASE_FOLDER "./var/lib"
|
||||
ENV NODE_NAME ""
|
||||
ENV BROKER_ADDRESS "127.0.0.1:4222"
|
||||
ENV NATS_CONN_OPT_TIMEOUT "20"
|
||||
ENV NATS_CONNECT_RETRY_INTERVAL "10"
|
||||
ENV NATS_RECONNECT_JITTER "100"
|
||||
ENV NATS_RECONNECT_JITTER_TLS "1"
|
||||
ENV REQ_KEYS_REQUEST_UPDATE_INTERVAL "60"
|
||||
ENV REQ_ACL_REQUEST_UPDATE_INTERVAL "60"
|
||||
ENV PROFILING_PORT ""
|
||||
ENV PROM_HOST_AND_PORT "127.0.0.1:2111"
|
||||
ENV DEFAULT_MESSAGE_TIMEOUT 10
|
||||
ENV DEFAULT_MESSAGE_RETRIES 3
|
||||
ENV DEFAULT_METHOD_TIMEOUT 10
|
||||
ENV SUBSCRIBERS_DATA_FOLDER "./var"
|
||||
ENV CENTRAL_NODE_NAME "central"
|
||||
ENV ROOT_CA_PATH ""
|
||||
ENV NKEY_SEED_FILE ""
|
||||
ENV NKEY_SEED ""
|
||||
ENV EXPOSE_DATA_FOLDER "127.0.0.1:8090"
|
||||
ENV ERROR_MESSAGE_RETRIES 3
|
||||
ENV ERROR_MESSAGE_TIMEOUT 10
|
||||
ENV COMPRESSION ""
|
||||
ENV SERIALIZATION ""
|
||||
ENV SET_BLOCK_PROFILE_RATE "0"
|
||||
ENV ENABLE_SOCKET "1"
|
||||
ENV ENABLE_SIGNATURE_CHECK "0"
|
||||
ENV ENABLE_ACL_CHECK "0"
|
||||
ENV IS_CENTRAL_AUTH "0"
|
||||
ENV ENABLE_DEBUG "0"
|
||||
ENV KEEP_PUBLISHERS_ALIVE_FOR "10"
|
||||
|
||||
ENV START_PUB_REQ_HELLO 60
|
||||
|
||||
ENV ENABLE_KEY_UPDATES "1"
|
||||
ENV ENABLE_ACL_UPDATES "1"
|
||||
ENV IS_CENTRAL_ERROR_LOGGER "0"
|
||||
ENV START_SUB_REQ_HELLO "1"
|
||||
ENV START_SUB_REQ_TO_FILE_APPEND "1"
|
||||
ENV START_SUB_REQ_TO_FILE "1"
|
||||
ENV START_SUB_REQ_TO_FILE_NACK "1"
|
||||
ENV START_SUB_REQ_COPY_SRC "1"
|
||||
ENV START_SUB_REQ_COPY_DST "1"
|
||||
ENV START_SUB_REQ_CLI_COMMAND "1"
|
||||
ENV START_SUB_REQ_TO_CONSOLE "1"
|
||||
ENV START_SUB_REQ_HTTP_GET "1"
|
||||
ENV START_SUB_REQ_HTTP_GET_SCHEDULED "1"
|
||||
ENV START_SUB_REQ_TAIL_FILE "1"
|
||||
ENV START_SUB_REQ_CLI_COMMAND_CONT "1"
|
||||
|
||||
CMD ["ash","-c","env CONFIGFOLDER=./etc/ /app/ctrl\
|
||||
-socketFolder=${SOCKET_FOLDER}\
|
||||
-tcpListener=${TCP_LISTENER}\
|
||||
-httpListener=${HTTP_LISTENER}\
|
||||
-databaseFolder=${DATABASE_FOLDER}\
|
||||
-nodeName=${NODE_NAME}\
|
||||
-brokerAddress=${BROKER_ADDRESS}\
|
||||
-natsConnOptTimeout=${NATS_CONN_OPT_TIMEOUT}\
|
||||
-natsConnectRetryInterval=${NATS_CONNECT_RETRY_INTERVAL}\
|
||||
-natsReconnectJitter=${NATS_RECONNECT_JITTER}\
|
||||
-natsReconnectJitterTLS=${NATS_RECONNECT_JITTER_TLS}\
|
||||
-REQKeysRequestUpdateInterval=${REQ_KEYS_REQUEST_UPDATE_INTERVAL}\
|
||||
-REQAclRequestUpdateInterval=${REQ_ACL_REQUEST_UPDATE_INTERVAL}\
|
||||
-profilingPort=${PROFILING_PORT}\
|
||||
-promHostAndPort=${PROM_HOST_AND_PORT}\
|
||||
-defaultMessageTimeout=${DEFAULT_MESSAGE_TIMEOUT}\
|
||||
-defaultMessageRetries=${DEFAULT_MESSAGE_RETRIES}\
|
||||
-defaultMethodTimeout=${DEFAULT_METHOD_TIMEOUT}\
|
||||
-subscribersDataFolder=${SUBSCRIBERS_DATA_FOLDER}\
|
||||
-centralNodeName=${CENTRAL_NODE_NAME}\
|
||||
-rootCAPath=${ROOT_CA_PATH}\
|
||||
-nkeySeedFile=${NKEY_SEED_FILE}\
|
||||
-nkeySeed=${NKEY_SEED}\
|
||||
-exposeDataFolder=${EXPOSE_DATA_FOLDER}\
|
||||
-errorMessageRetries=${ERROR_MESSAGE_RETRIES}\
|
||||
-errorMessageTimeout=${ERROR_MESSAGE_TIMEOUT}\
|
||||
-compression=${COMPRESSION}\
|
||||
-serialization=${SERIALIZATION}\
|
||||
-setBlockProfileRate=${SET_BLOCK_PROFILE_RATE}\
|
||||
-enableSocket=${ENABLE_SOCKET}\
|
||||
-enableSignatureCheck=${ENABLE_SIGNATURE_CHECK}\
|
||||
-enableAclCheck=${ENABLE_ACL_CHECK}\
|
||||
-isCentralAuth=${IS_CENTRAL_AUTH}\
|
||||
-enableDebug=${ENABLE_DEBUG}\
|
||||
-keepPublishersAliveFor=${KEEP_PUBLISHERS_ALIVE_FOR}\
|
||||
-startPubREQHello=${START_PUB_REQ_HELLO}\
|
||||
-EnableKeyUpdates=${ENABLE_KEY_UPDATES}\
|
||||
-EnableAclUpdates=${ENABLE_ACL_UPDATES}\
|
||||
-isCentralErrorLogger=${IS_CENTRAL_ERROR_LOGGER}\
|
||||
-startSubREQHello=${START_SUB_REQ_HELLO}\
|
||||
-startSubREQToFileAppend=${START_SUB_REQ_TO_FILE_APPEND}\
|
||||
-startSubREQToFile=${START_SUB_REQ_TO_FILE}\
|
||||
-startSubREQCopySrc=${START_SUB_REQ_COPY_SRC}\
|
||||
-startSubREQCopyDst=${START_SUB_REQ_COPY_DST}\
|
||||
-startSubREQToFileNACK=${START_SUB_REQ_TO_FILE_NACK}\
|
||||
-startSubREQCliCommand=${START_SUB_REQ_CLI_COMMAND}\
|
||||
-startSubREQToConsole=${START_SUB_REQ_TO_CONSOLE}\
|
||||
-startSubREQHttpGet=${START_SUB_REQ_HTTP_GET}\
|
||||
-startSubREQHttpGetScheduled=${START_SUB_REQ_HTTP_GET_SCHEDULED}\
|
||||
-startSubREQTailFile=${START_SUB_REQ_TAIL_FILE}\
|
||||
-startSubREQCliCommandCont=${START_SUB_REQ_CLI_COMMAND_CONT}\
|
||||
"]
|
||||
CMD ["ash","-c","/app/ctrl"]
|
||||
|
|
|
|||
|
|
@ -27,11 +27,11 @@ func main() {
|
|||
//defer profile.Start(profile.MemProfile, profile.MemProfileRate(1)).Stop()
|
||||
|
||||
c := ctrl.NewConfiguration()
|
||||
err := c.CheckFlags(version)
|
||||
if err != nil {
|
||||
log.Printf("%v\n", err)
|
||||
return
|
||||
}
|
||||
// err := c.CheckFlags(version)
|
||||
// if err != nil {
|
||||
// log.Printf("%v\n", err)
|
||||
// return
|
||||
// }
|
||||
|
||||
// Start profiling if profiling port is specified
|
||||
if c.ProfilingPort != "" {
|
||||
|
|
|
|||
|
|
@ -2,12 +2,11 @@ package ctrl
|
|||
|
||||
import (
|
||||
"flag"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strconv"
|
||||
|
||||
toml "github.com/pelletier/go-toml/v2"
|
||||
"github.com/joho/godotenv"
|
||||
)
|
||||
|
||||
// Configuration are the structure that holds all the different
|
||||
|
|
@ -139,75 +138,87 @@ type Configuration struct {
|
|||
StartSubREQCliCommandCont bool `comment:"Start subscriber for continously delivery of output from cli commands."`
|
||||
}
|
||||
|
||||
// ConfigurationFromFile should have the same structure as
|
||||
// Configuration. This structure is used when parsing the
|
||||
// configuration values from file, so we are able to detect
|
||||
// if a value were given or not when parsing.
|
||||
type ConfigurationFromFile struct {
|
||||
ConfigFolder *string
|
||||
RingBufferPersistStore *bool
|
||||
RingBufferSize *int
|
||||
SocketFolder *string
|
||||
ReadFolder *string
|
||||
EnableReadFolder *bool
|
||||
TCPListener *string
|
||||
HTTPListener *string
|
||||
DatabaseFolder *string
|
||||
NodeName *string
|
||||
BrokerAddress *string
|
||||
NatsConnOptTimeout *int
|
||||
NatsConnectRetryInterval *int
|
||||
NatsReconnectJitter *int
|
||||
NatsReconnectJitterTLS *int
|
||||
REQKeysRequestUpdateInterval *int
|
||||
REQAclRequestUpdateInterval *int
|
||||
ProfilingPort *string
|
||||
PromHostAndPort *string
|
||||
DefaultMessageTimeout *int
|
||||
DefaultMessageRetries *int
|
||||
DefaultMethodTimeout *int
|
||||
SubscribersDataFolder *string
|
||||
CentralNodeName *string
|
||||
RootCAPath *string
|
||||
NkeySeedFile *string
|
||||
NkeyFromED25519SSHKeyFile *string
|
||||
NkeySeed *string
|
||||
ExposeDataFolder *string
|
||||
ErrorMessageTimeout *int
|
||||
ErrorMessageRetries *int
|
||||
Compression *string
|
||||
Serialization *string
|
||||
SetBlockProfileRate *int
|
||||
EnableSocket *bool
|
||||
EnableSignatureCheck *bool
|
||||
EnableAclCheck *bool
|
||||
IsCentralAuth *bool
|
||||
EnableDebug *bool
|
||||
LogLevel *string
|
||||
LogConsoleTimestamps *bool
|
||||
KeepPublishersAliveFor *int
|
||||
|
||||
StartPubREQHello *int
|
||||
EnableKeyUpdates *bool
|
||||
EnableAclUpdates *bool
|
||||
IsCentralErrorLogger *bool
|
||||
StartSubREQHello *bool
|
||||
StartSubREQToFileAppend *bool
|
||||
StartSubREQToFile *bool
|
||||
StartSubREQToFileNACK *bool
|
||||
StartSubREQCopySrc *bool
|
||||
StartSubREQCopyDst *bool
|
||||
StartSubREQCliCommand *bool
|
||||
StartSubREQToConsole *bool
|
||||
StartSubREQHttpGet *bool
|
||||
StartSubREQHttpGetScheduled *bool
|
||||
StartSubREQTailFile *bool
|
||||
StartSubREQCliCommandCont *bool
|
||||
}
|
||||
|
||||
// NewConfiguration will return a *Configuration.
|
||||
func NewConfiguration() *Configuration {
|
||||
c := Configuration{}
|
||||
c := newConfigurationDefaults()
|
||||
|
||||
err := godotenv.Load()
|
||||
if err != nil {
|
||||
log.Printf("Error loading .env file: %v\n", err)
|
||||
}
|
||||
|
||||
//flag.StringVar(&c.ConfigFolder, "configFolder", fc.ConfigFolder, "Defaults to ./usr/local/ctrl/etc/. *NB* This flag is not used, if your config file are located somwhere else than default set the location in an env variable named CONFIGFOLDER")
|
||||
flag.StringVar(&c.SocketFolder, "socketFolder", CheckEnv("SOCKET_FOLDER", c.SocketFolder).(string), "folder who contains the socket file. Defaults to ./tmp/. If other folder is used this flag must be specified at startup.")
|
||||
flag.StringVar(&c.ReadFolder, "readFolder", CheckEnv("READ_FOLDER", c.ReadFolder).(string), "folder who contains the readfolder. Defaults to ./readfolder/. If other folder is used this flag must be specified at startup.")
|
||||
flag.StringVar(&c.TCPListener, "tcpListener", CheckEnv("TCP_LISTENER", c.TCPListener).(string), "start up a TCP listener in addition to the Unix Socket, to give messages to the system. e.g. localhost:8888. No value means not to start the listener, which is default. NB: You probably don't want to start this on any other interface than localhost")
|
||||
flag.StringVar(&c.HTTPListener, "httpListener", CheckEnv("HTTP_LISTENER", c.HTTPListener).(string), "start up a HTTP listener in addition to the Unix Socket, to give messages to the system. e.g. localhost:8888. No value means not to start the listener, which is default. NB: You probably don't want to start this on any other interface than localhost")
|
||||
flag.StringVar(&c.DatabaseFolder, "databaseFolder", CheckEnv("DATABASE_FOLDER", c.DatabaseFolder).(string), "folder who contains the database file. Defaults to ./var/lib/. If other folder is used this flag must be specified at startup.")
|
||||
flag.StringVar(&c.NodeName, "nodeName", CheckEnv("NODE_NAME", c.NodeName).(string), "some unique string to identify this Edge unit")
|
||||
flag.StringVar(&c.BrokerAddress, "brokerAddress", CheckEnv("BROKER_ADDRESS", c.BrokerAddress).(string), "the address of the message broker")
|
||||
flag.IntVar(&c.NatsConnOptTimeout, "natsConnOptTimeout", CheckEnv("NATS_CONN_OPT_TIMEOUT", c.NatsConnOptTimeout).(int), "default nats client conn timeout in seconds")
|
||||
flag.IntVar(&c.NatsConnectRetryInterval, "natsConnectRetryInterval", CheckEnv("NATS_CONNECT_RETRY_INTERVAL", c.NatsConnectRetryInterval).(int), "default nats retry connect interval in seconds.")
|
||||
flag.IntVar(&c.NatsReconnectJitter, "natsReconnectJitter", CheckEnv("NATS_RECONNECT_JITTER", c.NatsReconnectJitter).(int), "default nats ReconnectJitter interval in milliseconds.")
|
||||
flag.IntVar(&c.NatsReconnectJitterTLS, "natsReconnectJitterTLS", CheckEnv("NATS_RECONNECT_JITTER_TLS", c.NatsReconnectJitterTLS).(int), "default nats ReconnectJitterTLS interval in seconds.")
|
||||
flag.IntVar(&c.REQKeysRequestUpdateInterval, "REQKeysRequestUpdateInterval", CheckEnv("REQ_KEYS_UPDATE_INTERVAL", c.REQKeysRequestUpdateInterval).(int), "default interval in seconds for asking the central for public keys")
|
||||
flag.IntVar(&c.REQAclRequestUpdateInterval, "REQAclRequestUpdateInterval", CheckEnv("REQ_ACL_REQUEST_UPDATE_INTERVAL", c.REQAclRequestUpdateInterval).(int), "default interval in seconds for asking the central for acl updates")
|
||||
flag.StringVar(&c.ProfilingPort, "profilingPort", CheckEnv("PROFILING_PORT", c.ProfilingPort).(string), "The number of the profiling port")
|
||||
flag.StringVar(&c.PromHostAndPort, "promHostAndPort", CheckEnv("PROM_HOST_AND_PORT", c.PromHostAndPort).(string), "host and port for prometheus listener, e.g. localhost:2112")
|
||||
flag.IntVar(&c.DefaultMessageTimeout, "defaultMessageTimeout", CheckEnv("DEFAULT_MESSAGE_TIMEOUT", c.DefaultMessageTimeout).(int), "default message timeout in seconds. This can be overridden on the message level")
|
||||
flag.IntVar(&c.DefaultMessageRetries, "defaultMessageRetries", CheckEnv("DEFAULT_MESSAGE_RETRIES", c.DefaultMessageRetries).(int), "default amount of retries that will be done before a message is thrown away, and out of the system")
|
||||
flag.IntVar(&c.DefaultMethodTimeout, "defaultMethodTimeout", CheckEnv("DEFAULT_METHOD_TIMEOUT", c.DefaultMethodTimeout).(int), "default amount of seconds a request method max will be allowed to run")
|
||||
flag.StringVar(&c.SubscribersDataFolder, "subscribersDataFolder", CheckEnv("SUBSCRIBER_DATA_FOLDER", c.SubscribersDataFolder).(string), "The data folder where subscribers are allowed to write their data if needed")
|
||||
flag.StringVar(&c.CentralNodeName, "centralNodeName", CheckEnv("CENTRAL_NODE_NAME", c.CentralNodeName).(string), "The name of the central node to receive messages published by this node")
|
||||
flag.StringVar(&c.RootCAPath, "rootCAPath", CheckEnv("ROOT_CA_PATH", c.RootCAPath).(string), "If TLS, enter the path for where to find the root CA certificate")
|
||||
flag.StringVar(&c.NkeyFromED25519SSHKeyFile, "nkeyFromED25519SSHKeyFile", CheckEnv("NKEY_FROM_ED25519_SSH_KEY_FILE", c.NkeyFromED25519SSHKeyFile).(string), "The full path of the nkeys seed file")
|
||||
flag.StringVar(&c.NkeySeedFile, "nkeySeedFile", CheckEnv("NKEY_SEED_FILE", c.NkeySeedFile).(string), "Full path to the ED25519 SSH private key. Will generate the NKEY Seed from an SSH ED25519 private key file. NB: This option will take precedence over NkeySeedFile if specified")
|
||||
flag.StringVar(&c.NkeySeed, "nkeySeed", CheckEnv("NKEY_SEED", c.NkeySeed).(string), "The actual nkey seed. To use if not stored in file")
|
||||
flag.StringVar(&c.ExposeDataFolder, "exposeDataFolder", CheckEnv("EXPOSE_DATA_FOLDER", c.ExposeDataFolder).(string), "If set the data folder will be exposed on the given host:port. Default value is not exposed at all")
|
||||
flag.IntVar(&c.ErrorMessageTimeout, "errorMessageTimeout", CheckEnv("ERROR_MESSAGE_TIMEOUT", c.ErrorMessageTimeout).(int), "The number of seconds to wait for an error message to time out")
|
||||
flag.IntVar(&c.ErrorMessageRetries, "errorMessageRetries", CheckEnv("ERROR_MESSAGE_RETRIES", c.ErrorMessageRetries).(int), "The number of if times to retry an error message before we drop it")
|
||||
flag.StringVar(&c.Compression, "compression", CheckEnv("COMPRESSION", c.Compression).(string), "compression method to use. defaults to no compression, z = zstd, g = gzip. Undefined value will default to no compression")
|
||||
flag.StringVar(&c.Serialization, "serialization", CheckEnv("SERIALIZATION", c.Serialization).(string), "Serialization method to use. defaults to gob, other values are = cbor. Undefined value will default to gob")
|
||||
flag.IntVar(&c.SetBlockProfileRate, "setBlockProfileRate", CheckEnv("BLOCK_PROFILE_RATE", c.SetBlockProfileRate).(int), "Enable block profiling by setting the value to f.ex. 1. 0 = disabled")
|
||||
flag.BoolVar(&c.EnableSocket, "enableSocket", CheckEnv("ENABLE_SOCKET", c.EnableSocket).(bool), "true/false, for enabling the creation of ctrl.sock file")
|
||||
flag.BoolVar(&c.EnableSignatureCheck, "enableSignatureCheck", CheckEnv("ENABLE_SIGNATURE_CHECK", c.EnableSignatureCheck).(bool), "true/false *TESTING* enable signature checking.")
|
||||
flag.BoolVar(&c.EnableAclCheck, "enableAclCheck", CheckEnv("ENABLE_ACL_CHECK", c.EnableAclCheck).(bool), "true/false *TESTING* enable Acl checking.")
|
||||
flag.BoolVar(&c.IsCentralAuth, "isCentralAuth", CheckEnv("IS_CENTRAL_AUTH", c.IsCentralAuth).(bool), "true/false, *TESTING* is this the central auth server")
|
||||
flag.BoolVar(&c.EnableDebug, "enableDebug", CheckEnv("ENABLE_DEBUG", c.EnableDebug).(bool), "true/false, will enable debug logging so all messages sent to the errorKernel will also be printed to STDERR")
|
||||
flag.StringVar(&c.LogLevel, "logLevel", CheckEnv("LOG_LEVEL", c.LogLevel).(string), "error/info/warning/debug/none")
|
||||
flag.BoolVar(&c.LogConsoleTimestamps, "LogConsoleTimestamps", CheckEnv("LOG_CONSOLE_TIMESTAMPS", c.LogConsoleTimestamps).(bool), "true/false for enabling or disabling timestamps when printing errors and information to stderr")
|
||||
flag.IntVar(&c.KeepPublishersAliveFor, "keepPublishersAliveFor", CheckEnv("KEEP_PUBLISHERS_ALIVE_FOR", c.KeepPublishersAliveFor).(int), "The amount of time we allow a publisher to stay alive without receiving any messages to publish")
|
||||
|
||||
// Start of Request publishers/subscribers
|
||||
|
||||
flag.IntVar(&c.StartPubREQHello, "startPubREQHello", CheckEnv("START_PUB_REQ_HELLO", c.StartPubREQHello).(int), "Make the current node send hello messages to central at given interval in seconds")
|
||||
|
||||
flag.BoolVar(&c.EnableKeyUpdates, "EnableKeyUpdates", CheckEnv("ENABLE_KEY_UPDATES", c.EnableKeyUpdates).(bool), "true/false")
|
||||
|
||||
flag.BoolVar(&c.EnableAclUpdates, "EnableAclUpdates", CheckEnv("ENABLE_ACL_UPDATES", c.EnableAclUpdates).(bool), "true/false")
|
||||
|
||||
flag.BoolVar(&c.IsCentralErrorLogger, "isCentralErrorLogger", CheckEnv("IS_CENTRAL_ERROR_LOGGER", c.IsCentralErrorLogger).(bool), "true/false")
|
||||
flag.BoolVar(&c.StartSubREQHello, "startSubREQHello", CheckEnv("START_SUB_REQ_HELLO", c.StartSubREQHello).(bool), "true/false")
|
||||
flag.BoolVar(&c.StartSubREQToFileAppend, "startSubREQToFileAppend", CheckEnv("START_SUB_REQ_TO_FILE_APPEND", c.StartSubREQToFileAppend).(bool), "true/false")
|
||||
flag.BoolVar(&c.StartSubREQToFile, "startSubREQToFile", CheckEnv("START_SUB_REQ_TO_FILE", c.StartSubREQToFile).(bool), "true/false")
|
||||
flag.BoolVar(&c.StartSubREQToFileNACK, "startSubREQToFileNACK", CheckEnv("START_SUB_REQ_TO_FILE_NACK", c.StartSubREQToFileNACK).(bool), "true/false")
|
||||
flag.BoolVar(&c.StartSubREQCopySrc, "startSubREQCopySrc", CheckEnv("START_SUB_REQ_COPY_SRC", c.StartSubREQCopySrc).(bool), "true/false")
|
||||
flag.BoolVar(&c.StartSubREQCopyDst, "startSubREQCopyDst", CheckEnv("START_SUB_REQ_COPY_DST", c.StartSubREQCopyDst).(bool), "true/false")
|
||||
flag.BoolVar(&c.StartSubREQCliCommand, "startSubREQCliCommand", CheckEnv("START_SUB_REQ_CLI_COMMAND", c.StartSubREQCliCommand).(bool), "true/false")
|
||||
flag.BoolVar(&c.StartSubREQToConsole, "startSubREQToConsole", CheckEnv("START_SUB_REQ_TO_CONSOLE", c.StartSubREQToConsole).(bool), "true/false")
|
||||
flag.BoolVar(&c.StartSubREQHttpGet, "startSubREQHttpGet", CheckEnv("START_SUB_REQ_HTTP_GET", c.StartSubREQHttpGet).(bool), "true/false")
|
||||
flag.BoolVar(&c.StartSubREQHttpGetScheduled, "startSubREQHttpGetScheduled", CheckEnv("START_SUB_REQ_HTTP_GET_SCHEDULED", c.StartSubREQHttpGetScheduled).(bool), "true/false")
|
||||
flag.BoolVar(&c.StartSubREQTailFile, "startSubREQTailFile", CheckEnv("START_SUB_REQ_TAIL_FILE", c.StartSubREQTailFile).(bool), "true/false")
|
||||
flag.BoolVar(&c.StartSubREQCliCommandCont, "startSubREQCliCommandCont", CheckEnv("START_SUB_REQ_CLI_COMMAND_CONT", c.StartSubREQCliCommandCont).(bool), "true/false")
|
||||
|
||||
// Check that mandatory flag values have been set.
|
||||
switch {
|
||||
case c.NodeName == "":
|
||||
log.Fatalf("error: the nodeName config option or flag cannot be empty, check -help\n")
|
||||
case c.CentralNodeName == "":
|
||||
log.Fatalf("error: the centralNodeName config option or flag cannot be empty, check -help\n")
|
||||
}
|
||||
|
||||
flag.Parse()
|
||||
|
||||
return &c
|
||||
}
|
||||
|
||||
|
|
@ -275,474 +286,30 @@ func newConfigurationDefaults() Configuration {
|
|||
return c
|
||||
}
|
||||
|
||||
// Check if all values are present in config file, and if not
|
||||
// found use the default value.
|
||||
func checkConfigValues(cf ConfigurationFromFile) Configuration {
|
||||
var conf Configuration
|
||||
cd := newConfigurationDefaults()
|
||||
|
||||
if cf.ConfigFolder == nil {
|
||||
conf.ConfigFolder = cd.ConfigFolder
|
||||
} else {
|
||||
conf.ConfigFolder = *cf.ConfigFolder
|
||||
}
|
||||
if cf.SocketFolder == nil {
|
||||
conf.SocketFolder = cd.SocketFolder
|
||||
} else {
|
||||
conf.SocketFolder = *cf.SocketFolder
|
||||
}
|
||||
if cf.ReadFolder == nil {
|
||||
conf.ReadFolder = cd.ReadFolder
|
||||
} else {
|
||||
conf.ReadFolder = *cf.ReadFolder
|
||||
}
|
||||
if cf.EnableReadFolder == nil {
|
||||
conf.EnableReadFolder = cd.EnableReadFolder
|
||||
} else {
|
||||
conf.EnableReadFolder = *cf.EnableReadFolder
|
||||
}
|
||||
if cf.TCPListener == nil {
|
||||
conf.TCPListener = cd.TCPListener
|
||||
} else {
|
||||
conf.TCPListener = *cf.TCPListener
|
||||
}
|
||||
if cf.HTTPListener == nil {
|
||||
conf.HTTPListener = cd.HTTPListener
|
||||
} else {
|
||||
conf.HTTPListener = *cf.HTTPListener
|
||||
}
|
||||
if cf.DatabaseFolder == nil {
|
||||
conf.DatabaseFolder = cd.DatabaseFolder
|
||||
} else {
|
||||
conf.DatabaseFolder = *cf.DatabaseFolder
|
||||
}
|
||||
if cf.NodeName == nil {
|
||||
conf.NodeName = cd.NodeName
|
||||
} else {
|
||||
conf.NodeName = *cf.NodeName
|
||||
}
|
||||
if cf.BrokerAddress == nil {
|
||||
conf.BrokerAddress = cd.BrokerAddress
|
||||
} else {
|
||||
conf.BrokerAddress = *cf.BrokerAddress
|
||||
}
|
||||
if cf.NatsConnOptTimeout == nil {
|
||||
conf.NatsConnOptTimeout = cd.NatsConnOptTimeout
|
||||
} else {
|
||||
conf.NatsConnOptTimeout = *cf.NatsConnOptTimeout
|
||||
}
|
||||
if cf.NatsConnectRetryInterval == nil {
|
||||
conf.NatsConnectRetryInterval = cd.NatsConnectRetryInterval
|
||||
} else {
|
||||
conf.NatsConnectRetryInterval = *cf.NatsConnectRetryInterval
|
||||
}
|
||||
if cf.NatsReconnectJitter == nil {
|
||||
conf.NatsReconnectJitter = cd.NatsReconnectJitter
|
||||
} else {
|
||||
conf.NatsReconnectJitter = *cf.NatsReconnectJitter
|
||||
}
|
||||
if cf.NatsReconnectJitterTLS == nil {
|
||||
conf.NatsReconnectJitterTLS = cd.NatsReconnectJitterTLS
|
||||
} else {
|
||||
conf.NatsReconnectJitterTLS = *cf.NatsReconnectJitterTLS
|
||||
}
|
||||
if cf.REQKeysRequestUpdateInterval == nil {
|
||||
conf.REQKeysRequestUpdateInterval = cd.REQKeysRequestUpdateInterval
|
||||
} else {
|
||||
conf.REQKeysRequestUpdateInterval = *cf.REQKeysRequestUpdateInterval
|
||||
}
|
||||
if cf.REQAclRequestUpdateInterval == nil {
|
||||
conf.REQAclRequestUpdateInterval = cd.REQAclRequestUpdateInterval
|
||||
} else {
|
||||
conf.REQAclRequestUpdateInterval = *cf.REQAclRequestUpdateInterval
|
||||
}
|
||||
if cf.ProfilingPort == nil {
|
||||
conf.ProfilingPort = cd.ProfilingPort
|
||||
} else {
|
||||
conf.ProfilingPort = *cf.ProfilingPort
|
||||
}
|
||||
if cf.PromHostAndPort == nil {
|
||||
conf.PromHostAndPort = cd.PromHostAndPort
|
||||
} else {
|
||||
conf.PromHostAndPort = *cf.PromHostAndPort
|
||||
}
|
||||
if cf.DefaultMessageTimeout == nil {
|
||||
conf.DefaultMessageTimeout = cd.DefaultMessageTimeout
|
||||
} else {
|
||||
conf.DefaultMessageTimeout = *cf.DefaultMessageTimeout
|
||||
}
|
||||
if cf.DefaultMessageRetries == nil {
|
||||
conf.DefaultMessageRetries = cd.DefaultMessageRetries
|
||||
} else {
|
||||
conf.DefaultMessageRetries = *cf.DefaultMessageRetries
|
||||
}
|
||||
if cf.DefaultMethodTimeout == nil {
|
||||
conf.DefaultMethodTimeout = cd.DefaultMethodTimeout
|
||||
} else {
|
||||
conf.DefaultMethodTimeout = *cf.DefaultMethodTimeout
|
||||
}
|
||||
if cf.SubscribersDataFolder == nil {
|
||||
conf.SubscribersDataFolder = cd.SubscribersDataFolder
|
||||
} else {
|
||||
conf.SubscribersDataFolder = *cf.SubscribersDataFolder
|
||||
}
|
||||
if cf.CentralNodeName == nil {
|
||||
conf.CentralNodeName = cd.CentralNodeName
|
||||
} else {
|
||||
conf.CentralNodeName = *cf.CentralNodeName
|
||||
}
|
||||
if cf.RootCAPath == nil {
|
||||
conf.RootCAPath = cd.RootCAPath
|
||||
} else {
|
||||
conf.RootCAPath = *cf.RootCAPath
|
||||
}
|
||||
if cf.NkeySeedFile == nil {
|
||||
conf.NkeySeedFile = cd.NkeySeedFile
|
||||
} else {
|
||||
conf.NkeySeedFile = *cf.NkeySeedFile
|
||||
}
|
||||
if cf.NkeyFromED25519SSHKeyFile == nil {
|
||||
conf.NkeyFromED25519SSHKeyFile = cd.NkeyFromED25519SSHKeyFile
|
||||
} else {
|
||||
conf.NkeyFromED25519SSHKeyFile = *cf.NkeyFromED25519SSHKeyFile
|
||||
}
|
||||
if cf.NkeySeed == nil {
|
||||
conf.NkeySeed = cd.NkeySeed
|
||||
} else {
|
||||
conf.NkeySeed = *cf.NkeySeed
|
||||
}
|
||||
if cf.ExposeDataFolder == nil {
|
||||
conf.ExposeDataFolder = cd.ExposeDataFolder
|
||||
} else {
|
||||
conf.ExposeDataFolder = *cf.ExposeDataFolder
|
||||
}
|
||||
if cf.ErrorMessageTimeout == nil {
|
||||
conf.ErrorMessageTimeout = cd.ErrorMessageTimeout
|
||||
} else {
|
||||
conf.ErrorMessageTimeout = *cf.ErrorMessageTimeout
|
||||
}
|
||||
if cf.ErrorMessageRetries == nil {
|
||||
conf.ErrorMessageRetries = cd.ErrorMessageRetries
|
||||
} else {
|
||||
conf.ErrorMessageRetries = *cf.ErrorMessageRetries
|
||||
}
|
||||
if cf.Compression == nil {
|
||||
conf.Compression = cd.Compression
|
||||
} else {
|
||||
conf.Compression = *cf.Compression
|
||||
}
|
||||
if cf.Serialization == nil {
|
||||
conf.Serialization = cd.Serialization
|
||||
} else {
|
||||
conf.Serialization = *cf.Serialization
|
||||
}
|
||||
if cf.SetBlockProfileRate == nil {
|
||||
conf.SetBlockProfileRate = cd.SetBlockProfileRate
|
||||
} else {
|
||||
conf.SetBlockProfileRate = *cf.SetBlockProfileRate
|
||||
}
|
||||
if cf.EnableSocket == nil {
|
||||
conf.EnableSocket = cd.EnableSocket
|
||||
} else {
|
||||
conf.EnableSocket = *cf.EnableSocket
|
||||
}
|
||||
if cf.EnableSignatureCheck == nil {
|
||||
conf.EnableSignatureCheck = cd.EnableSignatureCheck
|
||||
} else {
|
||||
conf.EnableSignatureCheck = *cf.EnableSignatureCheck
|
||||
}
|
||||
if cf.EnableAclCheck == nil {
|
||||
conf.EnableAclCheck = cd.EnableAclCheck
|
||||
} else {
|
||||
conf.EnableAclCheck = *cf.EnableAclCheck
|
||||
}
|
||||
if cf.IsCentralAuth == nil {
|
||||
conf.IsCentralAuth = cd.IsCentralAuth
|
||||
} else {
|
||||
conf.IsCentralAuth = *cf.IsCentralAuth
|
||||
}
|
||||
if cf.EnableDebug == nil {
|
||||
conf.EnableDebug = cd.EnableDebug
|
||||
} else {
|
||||
conf.EnableDebug = *cf.EnableDebug
|
||||
}
|
||||
if cf.LogLevel == nil {
|
||||
conf.LogLevel = cd.LogLevel
|
||||
} else {
|
||||
conf.LogLevel = *cf.LogLevel
|
||||
}
|
||||
if cf.LogConsoleTimestamps == nil {
|
||||
conf.LogConsoleTimestamps = cd.LogConsoleTimestamps
|
||||
} else {
|
||||
conf.LogConsoleTimestamps = *cf.LogConsoleTimestamps
|
||||
}
|
||||
if cf.KeepPublishersAliveFor == nil {
|
||||
conf.KeepPublishersAliveFor = cd.KeepPublishersAliveFor
|
||||
} else {
|
||||
conf.KeepPublishersAliveFor = *cf.KeepPublishersAliveFor
|
||||
func CheckEnv[T any](key string, v T) any {
|
||||
val, ok := os.LookupEnv(key)
|
||||
if !ok {
|
||||
return v
|
||||
}
|
||||
|
||||
// --- Start pub/sub
|
||||
|
||||
if cf.StartPubREQHello == nil {
|
||||
conf.StartPubREQHello = cd.StartPubREQHello
|
||||
} else {
|
||||
conf.StartPubREQHello = *cf.StartPubREQHello
|
||||
}
|
||||
if cf.EnableKeyUpdates == nil {
|
||||
conf.EnableKeyUpdates = cd.EnableKeyUpdates
|
||||
} else {
|
||||
conf.EnableKeyUpdates = *cf.EnableKeyUpdates
|
||||
}
|
||||
|
||||
if cf.EnableAclUpdates == nil {
|
||||
conf.EnableAclUpdates = cd.EnableAclUpdates
|
||||
} else {
|
||||
conf.EnableAclUpdates = *cf.EnableAclUpdates
|
||||
}
|
||||
|
||||
if cf.IsCentralErrorLogger == nil {
|
||||
conf.IsCentralErrorLogger = cd.IsCentralErrorLogger
|
||||
} else {
|
||||
conf.IsCentralErrorLogger = *cf.IsCentralErrorLogger
|
||||
}
|
||||
if cf.StartSubREQHello == nil {
|
||||
conf.StartSubREQHello = cd.StartSubREQHello
|
||||
} else {
|
||||
conf.StartSubREQHello = *cf.StartSubREQHello
|
||||
}
|
||||
if cf.StartSubREQToFileAppend == nil {
|
||||
conf.StartSubREQToFileAppend = cd.StartSubREQToFileAppend
|
||||
} else {
|
||||
conf.StartSubREQToFileAppend = *cf.StartSubREQToFileAppend
|
||||
}
|
||||
if cf.StartSubREQToFile == nil {
|
||||
conf.StartSubREQToFile = cd.StartSubREQToFile
|
||||
} else {
|
||||
conf.StartSubREQToFile = *cf.StartSubREQToFile
|
||||
}
|
||||
if cf.StartSubREQToFileNACK == nil {
|
||||
conf.StartSubREQToFileNACK = cd.StartSubREQToFileNACK
|
||||
} else {
|
||||
conf.StartSubREQToFileNACK = *cf.StartSubREQToFileNACK
|
||||
}
|
||||
if cf.StartSubREQCopySrc == nil {
|
||||
conf.StartSubREQCopySrc = cd.StartSubREQCopySrc
|
||||
} else {
|
||||
conf.StartSubREQCopySrc = *cf.StartSubREQCopySrc
|
||||
}
|
||||
if cf.StartSubREQCopyDst == nil {
|
||||
conf.StartSubREQCopyDst = cd.StartSubREQCopyDst
|
||||
} else {
|
||||
conf.StartSubREQCopyDst = *cf.StartSubREQCopyDst
|
||||
}
|
||||
if cf.StartSubREQCliCommand == nil {
|
||||
conf.StartSubREQCliCommand = cd.StartSubREQCliCommand
|
||||
} else {
|
||||
conf.StartSubREQCliCommand = *cf.StartSubREQCliCommand
|
||||
}
|
||||
if cf.StartSubREQToConsole == nil {
|
||||
conf.StartSubREQToConsole = cd.StartSubREQToConsole
|
||||
} else {
|
||||
conf.StartSubREQToConsole = *cf.StartSubREQToConsole
|
||||
}
|
||||
if cf.StartSubREQHttpGet == nil {
|
||||
conf.StartSubREQHttpGet = cd.StartSubREQHttpGet
|
||||
} else {
|
||||
conf.StartSubREQHttpGet = *cf.StartSubREQHttpGet
|
||||
}
|
||||
if cf.StartSubREQHttpGetScheduled == nil {
|
||||
conf.StartSubREQHttpGetScheduled = cd.StartSubREQHttpGetScheduled
|
||||
} else {
|
||||
conf.StartSubREQHttpGetScheduled = *cf.StartSubREQHttpGetScheduled
|
||||
}
|
||||
if cf.StartSubREQTailFile == nil {
|
||||
conf.StartSubREQTailFile = cd.StartSubREQTailFile
|
||||
} else {
|
||||
conf.StartSubREQTailFile = *cf.StartSubREQTailFile
|
||||
}
|
||||
if cf.StartSubREQCliCommandCont == nil {
|
||||
conf.StartSubREQCliCommandCont = cd.StartSubREQCliCommandCont
|
||||
} else {
|
||||
conf.StartSubREQCliCommandCont = *cf.StartSubREQCliCommandCont
|
||||
}
|
||||
|
||||
return conf
|
||||
}
|
||||
|
||||
// CheckFlags will parse all flags
|
||||
func (c *Configuration) CheckFlags(version string) error {
|
||||
|
||||
// Create an empty default config
|
||||
var fc Configuration
|
||||
|
||||
// Set default configfolder if no env was provided.
|
||||
configFolder := os.Getenv("CONFIG_FOLDER")
|
||||
|
||||
if configFolder == "" {
|
||||
configFolder = "./etc/"
|
||||
}
|
||||
|
||||
// Read file config. Set system default if it can't find config file.
|
||||
fc, err := c.ReadConfigFile(configFolder)
|
||||
if err != nil {
|
||||
log.Printf("%v\n", err)
|
||||
fc = newConfigurationDefaults()
|
||||
}
|
||||
|
||||
if configFolder == "" {
|
||||
fc.ConfigFolder = "./etc/"
|
||||
} else {
|
||||
fc.ConfigFolder = configFolder
|
||||
}
|
||||
|
||||
*c = fc
|
||||
|
||||
//flag.StringVar(&c.ConfigFolder, "configFolder", fc.ConfigFolder, "Defaults to ./usr/local/ctrl/etc/. *NB* This flag is not used, if your config file are located somwhere else than default set the location in an env variable named CONFIGFOLDER")
|
||||
flag.StringVar(&c.SocketFolder, "socketFolder", fc.SocketFolder, "folder who contains the socket file. Defaults to ./tmp/. If other folder is used this flag must be specified at startup.")
|
||||
flag.StringVar(&c.ReadFolder, "readFolder", fc.ReadFolder, "folder who contains the readfolder. Defaults to ./readfolder/. If other folder is used this flag must be specified at startup.")
|
||||
flag.StringVar(&c.TCPListener, "tcpListener", fc.TCPListener, "start up a TCP listener in addition to the Unix Socket, to give messages to the system. e.g. localhost:8888. No value means not to start the listener, which is default. NB: You probably don't want to start this on any other interface than localhost")
|
||||
flag.StringVar(&c.HTTPListener, "httpListener", fc.HTTPListener, "start up a HTTP listener in addition to the Unix Socket, to give messages to the system. e.g. localhost:8888. No value means not to start the listener, which is default. NB: You probably don't want to start this on any other interface than localhost")
|
||||
flag.StringVar(&c.DatabaseFolder, "databaseFolder", fc.DatabaseFolder, "folder who contains the database file. Defaults to ./var/lib/. If other folder is used this flag must be specified at startup.")
|
||||
flag.StringVar(&c.NodeName, "nodeName", fc.NodeName, "some unique string to identify this Edge unit")
|
||||
flag.StringVar(&c.BrokerAddress, "brokerAddress", fc.BrokerAddress, "the address of the message broker")
|
||||
flag.IntVar(&c.NatsConnOptTimeout, "natsConnOptTimeout", fc.NatsConnOptTimeout, "default nats client conn timeout in seconds")
|
||||
flag.IntVar(&c.NatsConnectRetryInterval, "natsConnectRetryInterval", fc.NatsConnectRetryInterval, "default nats retry connect interval in seconds.")
|
||||
flag.IntVar(&c.NatsReconnectJitter, "natsReconnectJitter", fc.NatsReconnectJitter, "default nats ReconnectJitter interval in milliseconds.")
|
||||
flag.IntVar(&c.NatsReconnectJitterTLS, "natsReconnectJitterTLS", fc.NatsReconnectJitterTLS, "default nats ReconnectJitterTLS interval in seconds.")
|
||||
flag.IntVar(&c.REQKeysRequestUpdateInterval, "REQKeysRequestUpdateInterval", fc.REQKeysRequestUpdateInterval, "default interval in seconds for asking the central for public keys")
|
||||
flag.IntVar(&c.REQAclRequestUpdateInterval, "REQAclRequestUpdateInterval", fc.REQAclRequestUpdateInterval, "default interval in seconds for asking the central for acl updates")
|
||||
flag.StringVar(&c.ProfilingPort, "profilingPort", fc.ProfilingPort, "The number of the profiling port")
|
||||
flag.StringVar(&c.PromHostAndPort, "promHostAndPort", fc.PromHostAndPort, "host and port for prometheus listener, e.g. localhost:2112")
|
||||
flag.IntVar(&c.DefaultMessageTimeout, "defaultMessageTimeout", fc.DefaultMessageTimeout, "default message timeout in seconds. This can be overridden on the message level")
|
||||
flag.IntVar(&c.DefaultMessageRetries, "defaultMessageRetries", fc.DefaultMessageRetries, "default amount of retries that will be done before a message is thrown away, and out of the system")
|
||||
flag.IntVar(&c.DefaultMethodTimeout, "defaultMethodTimeout", fc.DefaultMethodTimeout, "default amount of seconds a request method max will be allowed to run")
|
||||
flag.StringVar(&c.SubscribersDataFolder, "subscribersDataFolder", fc.SubscribersDataFolder, "The data folder where subscribers are allowed to write their data if needed")
|
||||
flag.StringVar(&c.CentralNodeName, "centralNodeName", fc.CentralNodeName, "The name of the central node to receive messages published by this node")
|
||||
flag.StringVar(&c.RootCAPath, "rootCAPath", fc.RootCAPath, "If TLS, enter the path for where to find the root CA certificate")
|
||||
flag.StringVar(&c.NkeyFromED25519SSHKeyFile, "nkeyFromED25519SSHKeyFile", fc.NkeyFromED25519SSHKeyFile, "The full path of the nkeys seed file")
|
||||
flag.StringVar(&c.NkeySeedFile, "nkeySeedFile", fc.NkeySeedFile, "Full path to the ED25519 SSH private key. Will generate the NKEY Seed from an SSH ED25519 private key file. NB: This option will take precedence over NkeySeedFile if specified")
|
||||
flag.StringVar(&c.NkeySeed, "nkeySeed", fc.NkeySeed, "The actual nkey seed. To use if not stored in file")
|
||||
flag.StringVar(&c.ExposeDataFolder, "exposeDataFolder", fc.ExposeDataFolder, "If set the data folder will be exposed on the given host:port. Default value is not exposed at all")
|
||||
flag.IntVar(&c.ErrorMessageTimeout, "errorMessageTimeout", fc.ErrorMessageTimeout, "The number of seconds to wait for an error message to time out")
|
||||
flag.IntVar(&c.ErrorMessageRetries, "errorMessageRetries", fc.ErrorMessageRetries, "The number of if times to retry an error message before we drop it")
|
||||
flag.StringVar(&c.Compression, "compression", fc.Compression, "compression method to use. defaults to no compression, z = zstd, g = gzip. Undefined value will default to no compression")
|
||||
flag.StringVar(&c.Serialization, "serialization", fc.Serialization, "Serialization method to use. defaults to gob, other values are = cbor. Undefined value will default to gob")
|
||||
flag.IntVar(&c.SetBlockProfileRate, "setBlockProfileRate", fc.SetBlockProfileRate, "Enable block profiling by setting the value to f.ex. 1. 0 = disabled")
|
||||
flag.BoolVar(&c.EnableSocket, "enableSocket", fc.EnableSocket, "true/false, for enabling the creation of ctrl.sock file")
|
||||
flag.BoolVar(&c.EnableSignatureCheck, "enableSignatureCheck", fc.EnableSignatureCheck, "true/false *TESTING* enable signature checking.")
|
||||
flag.BoolVar(&c.EnableAclCheck, "enableAclCheck", fc.EnableAclCheck, "true/false *TESTING* enable Acl checking.")
|
||||
flag.BoolVar(&c.IsCentralAuth, "isCentralAuth", fc.IsCentralAuth, "true/false, *TESTING* is this the central auth server")
|
||||
flag.BoolVar(&c.EnableDebug, "enableDebug", fc.EnableDebug, "true/false, will enable debug logging so all messages sent to the errorKernel will also be printed to STDERR")
|
||||
flag.StringVar(&c.LogLevel, "logLevel", fc.LogLevel, "error/info/warning/debug/none")
|
||||
flag.BoolVar(&c.LogConsoleTimestamps, "LogConsoleTimestamps", fc.LogConsoleTimestamps, "true/false for enabling or disabling timestamps when printing errors and information to stderr")
|
||||
flag.IntVar(&c.KeepPublishersAliveFor, "keepPublishersAliveFor", fc.KeepPublishersAliveFor, "The amount of time we allow a publisher to stay alive without receiving any messages to publish")
|
||||
|
||||
// Start of Request publishers/subscribers
|
||||
|
||||
flag.IntVar(&c.StartPubREQHello, "startPubREQHello", fc.StartPubREQHello, "Make the current node send hello messages to central at given interval in seconds")
|
||||
|
||||
flag.BoolVar(&c.EnableKeyUpdates, "EnableKeyUpdates", fc.EnableKeyUpdates, "true/false")
|
||||
|
||||
flag.BoolVar(&c.EnableAclUpdates, "EnableAclUpdates", fc.EnableAclUpdates, "true/false")
|
||||
|
||||
flag.BoolVar(&c.IsCentralErrorLogger, "isCentralErrorLogger", fc.IsCentralErrorLogger, "true/false")
|
||||
flag.BoolVar(&c.StartSubREQHello, "startSubREQHello", fc.StartSubREQHello, "true/false")
|
||||
flag.BoolVar(&c.StartSubREQToFileAppend, "startSubREQToFileAppend", fc.StartSubREQToFileAppend, "true/false")
|
||||
flag.BoolVar(&c.StartSubREQToFile, "startSubREQToFile", fc.StartSubREQToFile, "true/false")
|
||||
flag.BoolVar(&c.StartSubREQToFileNACK, "startSubREQToFileNACK", fc.StartSubREQToFileNACK, "true/false")
|
||||
flag.BoolVar(&c.StartSubREQCopySrc, "startSubREQCopySrc", fc.StartSubREQCopySrc, "true/false")
|
||||
flag.BoolVar(&c.StartSubREQCopyDst, "startSubREQCopyDst", fc.StartSubREQCopyDst, "true/false")
|
||||
flag.BoolVar(&c.StartSubREQCliCommand, "startSubREQCliCommand", fc.StartSubREQCliCommand, "true/false")
|
||||
flag.BoolVar(&c.StartSubREQToConsole, "startSubREQToConsole", fc.StartSubREQToConsole, "true/false")
|
||||
flag.BoolVar(&c.StartSubREQHttpGet, "startSubREQHttpGet", fc.StartSubREQHttpGet, "true/false")
|
||||
flag.BoolVar(&c.StartSubREQHttpGetScheduled, "startSubREQHttpGetScheduled", fc.StartSubREQHttpGetScheduled, "true/false")
|
||||
flag.BoolVar(&c.StartSubREQTailFile, "startSubREQTailFile", fc.StartSubREQTailFile, "true/false")
|
||||
flag.BoolVar(&c.StartSubREQCliCommandCont, "startSubREQCliCommandCont", fc.StartSubREQCliCommandCont, "true/false")
|
||||
|
||||
purgeBufferDB := flag.Bool("purgeBufferDB", false, "true/false, purge the incoming buffer db and all it's state")
|
||||
|
||||
ver := flag.Bool("version", false, "print version and exit")
|
||||
|
||||
flag.Parse()
|
||||
|
||||
if *ver {
|
||||
fmt.Printf("version %v\n", version)
|
||||
os.Exit(0)
|
||||
}
|
||||
|
||||
// Check that mandatory flag values have been set.
|
||||
switch {
|
||||
case c.NodeName == "":
|
||||
return fmt.Errorf("error: the nodeName config option or flag cannot be empty, check -help")
|
||||
case c.CentralNodeName == "":
|
||||
return fmt.Errorf("error: the centralNodeName config option or flag cannot be empty, check -help")
|
||||
}
|
||||
|
||||
if err := c.WriteConfigFile(); err != nil {
|
||||
log.Printf("error: checkFlags: failed writing config file: %v\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
if *purgeBufferDB {
|
||||
fp := filepath.Join(c.DatabaseFolder, "incomingBuffer.db")
|
||||
err := os.Remove(fp)
|
||||
switch any(v).(type) {
|
||||
case int:
|
||||
n, err := strconv.Atoi(val)
|
||||
if err != nil {
|
||||
log.Printf("error: failed to purge buffer state database: %v\n", err)
|
||||
log.Fatalf("error: failed to convert env to int: %v\n", n)
|
||||
}
|
||||
return n
|
||||
case string:
|
||||
return val
|
||||
case bool:
|
||||
switch {
|
||||
case val == "true" || val == "1":
|
||||
return true
|
||||
case val == "false" || val == "0" || val == "":
|
||||
return true
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// Reads the current config file from disk.
|
||||
func (c *Configuration) ReadConfigFile(configFolder string) (Configuration, error) {
|
||||
fPath := filepath.Join(configFolder, "config.toml")
|
||||
|
||||
if _, err := os.Stat(fPath); os.IsNotExist(err) {
|
||||
return Configuration{}, fmt.Errorf("error: no config file found %v: %v", fPath, err)
|
||||
}
|
||||
|
||||
f, err := os.OpenFile(fPath, os.O_RDONLY, 0660)
|
||||
if err != nil {
|
||||
return Configuration{}, fmt.Errorf("error: ReadConfigFile: failed to open file: %v", err)
|
||||
}
|
||||
defer f.Close()
|
||||
|
||||
var cFile ConfigurationFromFile
|
||||
dec := toml.NewDecoder(f)
|
||||
err = dec.Decode(&cFile)
|
||||
if err != nil {
|
||||
log.Printf("error: decoding config.toml file. The program will automatically try to correct the problem, and use sane default where it kind find a value to use, but beware of this error in case the program start to behave in not expected ways: path=%v: err=%v", fPath, err)
|
||||
}
|
||||
|
||||
// Check that all values read are ok.
|
||||
conf := checkConfigValues(cFile)
|
||||
|
||||
return conf, nil
|
||||
}
|
||||
|
||||
// WriteConfigFile will write the current config to file. If the file or the
|
||||
// directory for the config file does not exist it will be created.
|
||||
func (c *Configuration) WriteConfigFile() error {
|
||||
if _, err := os.Stat(c.ConfigFolder); os.IsNotExist(err) {
|
||||
err := os.MkdirAll(c.ConfigFolder, 0770)
|
||||
if err != nil {
|
||||
return fmt.Errorf("error: failed to create config directory %v: %v", c.ConfigFolder, err)
|
||||
}
|
||||
}
|
||||
|
||||
fp := filepath.Join(c.ConfigFolder, "config.toml")
|
||||
|
||||
f, err := os.OpenFile(fp, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0660)
|
||||
if err != nil {
|
||||
return fmt.Errorf("error: WriteConfigFile: failed to open file: %v", err)
|
||||
}
|
||||
defer f.Close()
|
||||
|
||||
enc := toml.NewEncoder(f)
|
||||
enc.Encode(c)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
|
|
|||
2
go.mod
2
go.mod
|
|
@ -9,11 +9,11 @@ require (
|
|||
github.com/google/uuid v1.3.0
|
||||
github.com/hpcloud/tail v1.0.0
|
||||
github.com/jinzhu/copier v0.4.0
|
||||
github.com/joho/godotenv v1.5.1
|
||||
github.com/klauspost/compress v1.17.0
|
||||
github.com/nats-io/nats-server/v2 v2.8.4
|
||||
github.com/nats-io/nats.go v1.25.0
|
||||
github.com/nats-io/nkeys v0.4.4
|
||||
github.com/pelletier/go-toml/v2 v2.0.7
|
||||
github.com/pkg/profile v1.7.0
|
||||
github.com/prometheus/client_golang v1.14.0
|
||||
go.etcd.io/bbolt v1.3.7
|
||||
|
|
|
|||
5
go.sum
5
go.sum
|
|
@ -41,6 +41,8 @@ github.com/hpcloud/tail v1.0.0/go.mod h1:ab1qPbhIpdTxEkNHXyeSf5vhxWSCs/tWer42PpO
|
|||
github.com/ianlancetaylor/demangle v0.0.0-20210905161508-09a460cdf81d/go.mod h1:aYm2/VgdVmcIU8iMfdMvDMsRAQjcfZSKFby6HOFvi/w=
|
||||
github.com/jinzhu/copier v0.4.0 h1:w3ciUoD19shMCRargcpm0cm91ytaBhDvuRpz1ODO/U8=
|
||||
github.com/jinzhu/copier v0.4.0/go.mod h1:DfbEm0FYsaqBcKcFuvmOZb218JkPGtvSHsKg8S8hyyg=
|
||||
github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0=
|
||||
github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4=
|
||||
github.com/klauspost/compress v1.17.0 h1:Rnbp4K9EjcDuVuHtd0dgA4qNuv9yKDYKK1ulpJwgrqM=
|
||||
github.com/klauspost/compress v1.17.0/go.mod h1:ntbaceVETuRiXiv4DpjP66DpAtAGkEQskQzEyD//IeE=
|
||||
github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
|
||||
|
|
@ -68,8 +70,6 @@ github.com/nats-io/nkeys v0.4.4 h1:xvBJ8d69TznjcQl9t6//Q5xXuVhyYiSos6RPtvQNTwA=
|
|||
github.com/nats-io/nkeys v0.4.4/go.mod h1:XUkxdLPTufzlihbamfzQ7mw/VGx6ObUs+0bN5sNvt64=
|
||||
github.com/nats-io/nuid v1.0.1 h1:5iA8DT8V7q8WK2EScv2padNa/rTESc1KdnPw4TC2paw=
|
||||
github.com/nats-io/nuid v1.0.1/go.mod h1:19wcPz3Ph3q0Jbyiqsd0kePYG7A95tJPxeL+1OSON2c=
|
||||
github.com/pelletier/go-toml/v2 v2.0.7 h1:muncTPStnKRos5dpVKULv2FVd4bMOhNePj9CjgDb8Us=
|
||||
github.com/pelletier/go-toml/v2 v2.0.7/go.mod h1:eumQOmlWiOPt5WriQQqoM5y18pDHwha2N+QD+EUNTek=
|
||||
github.com/pkg/diff v0.0.0-20210226163009-20ebb0f2a09e/go.mod h1:pJLUxLENpZxwdsKMEsNbx1VGcRFpLqf3715MtcvvzbA=
|
||||
github.com/pkg/profile v1.7.0 h1:hnbDkaNWPCLMO9wGLdBFTIZvzDrDfBM2072E1S9gJkA=
|
||||
github.com/pkg/profile v1.7.0/go.mod h1:8Uer0jas47ZQMJ7VD+OHknK4YDY07LPUC6dEvqDjvNo=
|
||||
|
|
@ -88,7 +88,6 @@ github.com/rogpeppe/go-internal v1.8.0 h1:FCbCCtXNOY3UtUuHUYaghJg4y7Fd14rXifAYUA
|
|||
github.com/rogpeppe/go-internal v1.8.0/go.mod h1:WmiCO8CzOY8rg0OYDC4/i/2WRWAB6poM+XZ2dLUbcbE=
|
||||
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||
github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
|
||||
github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
|
||||
github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||
github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue