This commit is contained in:
Andrey Ivanov 2020-09-27 09:56:35 +03:00 committed by Andrey Ivanov
parent fc19ac2e89
commit 16a9c8f9bc
2 changed files with 8 additions and 12 deletions

View File

@ -22,8 +22,10 @@ type Logger struct {
Logger amitralog.Logger Logger amitralog.Logger
} }
var validLevel = map[string]bool{"debug": true, "info":true, "warn": true, "error": true, "fatal": true}
func New(conf config.Config) (Interface, error) { func New(conf config.Config) (Interface, error) {
if conf.Logger.File == "" || !validLevel(conf.Logger.Level) { if conf.Logger.File == "" || !validLevel[strings.ToLower(conf.Logger.Level)] {
return nil, errors.New("invalid logger config") return nil, errors.New("invalid logger config")
} }
@ -64,8 +66,3 @@ func (l *Logger) Fatalf(format string, args ...interface{}) {
l.Logger.Fatalf(format, args) l.Logger.Fatalf(format, args)
os.Exit(2) os.Exit(2)
} }
func validLevel(level string) bool {
var l = map[string]int{"debug": 1, "info": 1, "warn": 1, "error": 1, "fatal": 1}
return l[strings.ToLower(level)] == 1
}

View File

@ -83,10 +83,9 @@ func (s *Storage) Delete(id event.ID) error {
func (s *Storage) List() (map[event.ID]event.Event, error) { func (s *Storage) List() (map[event.ID]event.Event, error) {
res := make(map[event.ID]event.Event) res := make(map[event.ID]event.Event)
results, err := s.db.Query( results, err := s.db.Query( `SELECT (id,title,date,latency,note,userID,notifyTime) from events ORDER BY id`)
`SELECT (id,title,date,latency,note,userID,notifyTime) from events ORDER BY id`)
if err != nil { if err != nil {
return map[event.ID]event.Event{}, err return nil, err
} }
defer results.Close() defer results.Close()
for results.Next() { for results.Next() {
@ -95,16 +94,16 @@ func (s *Storage) List() (map[event.ID]event.Event, error) {
var dateRaw string var dateRaw string
err = results.Scan(&id, &evt.Title, &dateRaw, &evt.Latency, &evt.Note, &evt.UserID, &evt.NotifyTime) err = results.Scan(&id, &evt.Title, &dateRaw, &evt.Latency, &evt.Note, &evt.UserID, &evt.NotifyTime)
if err != nil { if err != nil {
return map[event.ID]event.Event{}, err return nil, err
} }
evt.Date, err = time.Parse(dateTimeLayout, dateRaw) evt.Date, err = time.Parse(dateTimeLayout, dateRaw)
if err != nil { if err != nil {
return map[event.ID]event.Event{}, err return nil, err
} }
res[id] = evt res[id] = evt
} }
if results.Err() != nil { if results.Err() != nil {
return map[event.ID]event.Event{}, results.Err() return nil, results.Err()
} }
return res, nil return res, nil
} }