grim/convey
Clone
Summary
Browse
Changes
Graph
closing merged branch
expand-list
2017-10-03, Gary Kramlich
345a52ef04c6
closing merged branch
/*
* Convey
* Copyright 2016-2017 Gary Kramlich <grim@reaperworld.com>
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package
logging
import
(
"github.com/aphistic/gomol"
"github.com/aphistic/gomol-console"
)
var
(
logTemplate
=
`{{.Timestamp.Format "2006-01-02 15:04:05.000"}} {{index .Attrs "id"}}: {{.Message}}`
logColorTemplate
=
`{{.Timestamp.Format "2006-01-02 15:04:05.000"}} {{index .Attrs "idColor"}}{{index .Attrs "id"}}{{reset}}: {{color}}{{.Message}}{{reset}}`
)
func
setupConsole
(
color
bool
)
(
gomol
.
Logger
,
error
)
{
cfg
:=
gomolconsole
.
NewConsoleLoggerConfig
()
logger
,
err
:=
gomolconsole
.
NewConsoleLogger
(
cfg
)
if
err
!=
nil
{
return
nil
,
err
}
template
:=
logTemplate
if
color
{
template
=
logColorTemplate
}
tpl
,
err
:=
gomol
.
NewTemplate
(
template
)
if
err
!=
nil
{
return
nil
,
err
}
err
=
logger
.
SetTemplate
(
tpl
)
if
err
!=
nil
{
return
nil
,
err
}
return
logger
,
nil
}
// Setup sets up the logging system based on the input parameters
func
Setup
(
color
,
verbose
bool
)
error
{
consoleLogger
,
err
:=
setupConsole
(
color
)
if
err
!=
nil
{
return
err
}
gomol
.
AddLogger
(
consoleLogger
)
if
verbose
{
gomol
.
SetLogLevel
(
gomol
.
LevelDebug
)
}
else
{
gomol
.
SetLogLevel
(
gomol
.
LevelInfo
)
}
err
=
gomol
.
InitLoggers
()
return
err
}