Fix logging

This commit is contained in:
David Newhall II 2019-06-21 04:56:17 -07:00
parent 38dfab9d58
commit bf8c465560
3 changed files with 8 additions and 15 deletions

View File

@ -38,7 +38,6 @@ after_success:
- ls -l release/
- |
declare -r SSH_FILE="$(mktemp -u $HOME/.ssh/XXXXX)"
echo -e "Host github.com\n\tStrictHostKeyChecking no\n" >> ~/.ssh/config
# Get deploy key for golift/homebrew-mugs.
openssl aes-256-cbc -K $encrypted_9f3147001275_key -iv $encrypted_9f3147001275_iv -in ".travis/github_deploy_key.enc" -out "$SSH_FILE" -d

View File

@ -9,6 +9,7 @@ import (
)
func main() {
log.SetFlags(log.LstdFlags)
unifi := &unifipoller.UnifiPoller{}
if unifi.ParseFlags(os.Args[1:]); unifi.ShowVer {
fmt.Printf("unifi-poller v%s\n", unifipoller.Version)

View File

@ -44,29 +44,22 @@ func (u *UnifiPoller) GetConfig() (err error) {
UnifiBase: defaultUnifURL,
Interval: Dur{defaultInterval},
Sites: []string{"default"},
Quiet: u.DumpJSON != "",
}
u.Logf("Loading Configuration File: %s", u.ConfigFile)
var buf []byte
switch buf, err = ioutil.ReadFile(u.ConfigFile); {
case err != nil:
return err
case strings.HasSuffix(u.ConfigFile, ".json"):
err = json.Unmarshal(buf, u.Config)
return json.Unmarshal(buf, u.Config)
case strings.HasSuffix(u.ConfigFile, ".xml"):
err = xml.Unmarshal(buf, u.Config)
return xml.Unmarshal(buf, u.Config)
case strings.HasSuffix(u.ConfigFile, ".yaml"):
err = yaml.Unmarshal(buf, u.Config)
return yaml.Unmarshal(buf, u.Config)
default:
err = toml.Unmarshal(buf, u.Config)
return toml.Unmarshal(buf, u.Config)
}
if err != nil {
return err
}
if u.DumpJSON != "" {
u.Quiet = true
}
u.Logf("Loaded Configuration: %s", u.ConfigFile)
return nil
}
// Run invokes all the application logic and routines.
@ -74,7 +67,7 @@ func (u *UnifiPoller) Run() (err error) {
if u.DumpJSON != "" {
return u.DumpJSONPayload()
}
if log.SetFlags(0); u.Debug {
if u.Debug {
log.SetFlags(log.Lshortfile | log.Lmicroseconds | log.Ldate)
u.LogDebugf("Debug Logging Enabled")
}