Compare commits

...

2 Commits

4 changed files with 43 additions and 5 deletions

View File

@ -1,7 +1,7 @@
# gopackagebase
```sh
go get git.wisellama.rocks/Wisellama/gopackagebase@v0.0.1
go get git.wisellama.rocks/Wisellama/gopackagebase@v0.0.4
```
This modules contains basic setup and initialization functions that

View File

@ -30,10 +30,11 @@ func Initialize(configFile string, defaultConfig gosimpleconf.ConfigMap) (*BaseC
log.Fatalf("error during configure: %v\n", err)
}
utcTime := gosimpleconf.Bool(configMap["log.utcTime"])
writeToFile := gosimpleconf.Bool(configMap["log.writeToFile"])
logFilename := configMap["log.file"]
logWriter, err := SetupLogging(writeToFile, logFilename)
logWriter, err := SetupLogging(writeToFile, utcTime, logFilename)
if err != nil {
return nil, err
}

31
gopackagebase_test.go Normal file
View File

@ -0,0 +1,31 @@
package gopackagebase_test
import (
"testing"
"git.wisellama.rocks/Wisellama/gopackagebase"
"git.wisellama.rocks/Wisellama/gosimpleconf"
)
func TestGoPackageBase(t *testing.T) {
defaultConf := gosimpleconf.ConfigMap{
"log.writeToFile": "false",
}
baseConfig, err := gopackagebase.Initialize("idontexist.conf", defaultConf)
if err != nil {
t.Errorf("unexpected error: %v", err)
}
if baseConfig == nil {
t.Fatalf("baseConfig was nil")
}
value := baseConfig.ConfigMap["log.writeToFile"]
expected := defaultConf["log.writeToFile"]
if value != expected {
t.Errorf("incorrect value for log.writeToFile: expected %v, got %v", expected, value)
}
baseConfig.Cancel()
}

View File

@ -8,20 +8,25 @@ import (
)
type logWriter struct {
utcTime bool
writeToFile bool
logFile *os.File
}
func (w *logWriter) Write(bytes []byte) (int, error) {
t := time.Now().UTC().Format(time.RFC3339)
return fmt.Fprintf(w.logFile, "%v %v", t, string(bytes))
t := time.Now()
if w.utcTime {
t = t.UTC()
}
format := t.Format(time.RFC3339)
return fmt.Fprintf(w.logFile, "%v %v", format, string(bytes))
}
func (w *logWriter) Cleanup() {
defer w.logFile.Close()
}
func SetupLogging(writeToFile bool, logFilename string) (*logWriter, error) {
func SetupLogging(writeToFile bool, utcTime bool, logFilename string) (*logWriter, error) {
var err error
log.SetFlags(0)
@ -34,6 +39,7 @@ func SetupLogging(writeToFile bool, logFilename string) (*logWriter, error) {
}
writer := &logWriter{
utcTime: utcTime,
writeToFile: writeToFile,
logFile: logFile,
}