HW12 completed

pull/13/head
Andrey Ivanov 2020-09-25 12:24:06 +03:00 committed by Andrey Ivanov
parent 73a669ab90
commit 6fd369434a
5 changed files with 39 additions and 36 deletions

View File

@ -44,23 +44,23 @@ func New(conf config.Config) (Interface, error) {
return l, nil
}
func (l *Logger) Debugf(format string, args ...interface{}) {
func (l Logger) Debugf(format string, args ...interface{}) {
l.Logger.Debugf(format, args)
}
func (l *Logger) Infof(format string, args ...interface{}) {
func (l Logger) Infof(format string, args ...interface{}) {
l.Logger.Infof(format, args)
}
func (l *Logger) Warnf(format string, args ...interface{}) {
func (l Logger) Warnf(format string, args ...interface{}) {
l.Logger.Warnf(format, args)
}
func (l *Logger) Errorf(format string, args ...interface{}) {
func (l Logger) Errorf(format string, args ...interface{}) {
l.Logger.Errorf(format, args)
}
func (l *Logger) Fatalf(format string, args ...interface{}) {
func (l Logger) Fatalf(format string, args ...interface{}) {
l.Logger.Fatalf(format, args)
os.Exit(2)
}

View File

@ -4,6 +4,8 @@ import (
"time"
)
type ID int64
type Event struct {
Title string
Date time.Time

View File

@ -7,16 +7,16 @@ import (
)
type Storage struct {
Events map[int64]event.Event
lastID int64
Events map[event.ID]event.Event
lastID event.ID
Mu sync.RWMutex
}
func New() *Storage {
return &Storage{Events: make(map[int64]event.Event)}
return &Storage{Events: make(map[event.ID]event.Event)}
}
func (s *Storage) Create(event event.Event) (int64, error) {
func (s *Storage) Create(event event.Event) (event.ID, error) {
s.Mu.Lock()
s.lastID++
s.Events[s.lastID] = event
@ -24,25 +24,25 @@ func (s *Storage) Create(event event.Event) (int64, error) {
return s.lastID, nil
}
func (s *Storage) Update(id int64, event event.Event) error {
func (s *Storage) Update(id event.ID, event event.Event) error {
s.Mu.Lock()
s.Events[id] = event
s.Mu.Unlock()
return nil
}
func (s *Storage) Delete(id int64) error {
func (s *Storage) Delete(id event.ID) error {
s.Mu.Lock()
delete(s.Events, id)
s.Mu.Unlock()
return nil
}
func (s *Storage) List() (map[int64]event.Event, error) {
func (s *Storage) List() (map[event.ID]event.Event, error) {
return s.Events, nil
}
func (s *Storage) GetByID(id int64) (event.Event, bool) {
func (s *Storage) GetByID(id event.ID) (event.Event, bool) {
if s.Events[id].Title == "" {
return event.Event{}, false
}

View File

@ -38,25 +38,26 @@ func (s *Storage) Close() error {
return s.db.Close()
}
func (s *Storage) Create(event event.Event) (int64, error) {
func (s *Storage) Create(ev event.Event) (event.ID, error) {
res, err := s.db.Exec(
`INSERT INTO events
(title, date, latency, note, userID, notifyTime) VALUES
($1, $2, $3, $4, $5, $6)`,
event.Title,
event.Date.Format(dateTimeLayout),
event.Latency,
event.Note,
event.UserID,
event.NotifyTime,
ev.Title,
ev.Date.Format(dateTimeLayout),
ev.Latency,
ev.Note,
ev.UserID,
ev.NotifyTime,
)
if err != nil {
return -1, err
}
return res.LastInsertId()
idint64, err := res.LastInsertId()
return event.ID(idint64), err
}
func (s *Storage) Update(id int64, event event.Event) error {
func (s *Storage) Update(id event.ID, event event.Event) error {
_, err := s.db.Exec(
`UPDATE events set
title=$1, date=$2, latency=$3, note=$4, userID=$5, notifyTime=$6
@ -72,7 +73,7 @@ func (s *Storage) Update(id int64, event event.Event) error {
return err
}
func (s *Storage) Delete(id int64) error {
func (s *Storage) Delete(id event.ID) error {
_, err := s.db.Exec(
`DELETE from events where id=$1`,
id,
@ -80,35 +81,35 @@ func (s *Storage) Delete(id int64) error {
return err
}
func (s *Storage) List() (map[int64]event.Event, error) {
res := make(map[int64]event.Event)
func (s *Storage) List() (map[event.ID]event.Event, error) {
res := make(map[event.ID]event.Event)
results, err := s.db.Query(
`SELECT (id,title,date,latency,note,userID,notifyTime) from events ORDER BY id`)
if err != nil {
return map[int64]event.Event{}, err
return map[event.ID]event.Event{}, err
}
defer results.Close()
for results.Next() {
var id int64
var id event.ID
var evt event.Event
var dateRaw string
err = results.Scan(&id, &evt.Title, &dateRaw, &evt.Latency, &evt.Note, &evt.UserID, &evt.NotifyTime)
if err != nil {
return map[int64]event.Event{}, err
return map[event.ID]event.Event{}, err
}
evt.Date, err = time.Parse(dateTimeLayout, dateRaw)
if err != nil {
return map[int64]event.Event{}, err
return map[event.ID]event.Event{}, err
}
res[id] = evt
}
if results.Err() != nil {
return map[int64]event.Event{}, results.Err()
return map[event.ID]event.Event{}, results.Err()
}
return res, nil
}
func (s *Storage) GetByID(id int64) (event.Event, bool) {
func (s *Storage) GetByID(id event.ID) (event.Event, bool) {
var res event.Event
var dateRaw string
err := s.db.QueryRow(

View File

@ -16,11 +16,11 @@ type Config struct {
}
type StorageInterface interface {
Create(event event.Event) (int64, error)
Update(id int64, event event.Event) error
Delete(id int64) error
List() (map[int64]event.Event, error)
GetByID(id int64) (event.Event, bool)
Create(event event.Event) (event.ID, error)
Update(id event.ID, event event.Event) error
Delete(id event.ID) error
List() (map[event.ID]event.Event, error)
GetByID(id event.ID) (event.Event, bool)
}
func NewStore(conf Config) StorageInterface {