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 return l, nil
} }
func (l *Logger) Debugf(format string, args ...interface{}) { func (l Logger) Debugf(format string, args ...interface{}) {
l.Logger.Debugf(format, args) 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) 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) 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) 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) l.Logger.Fatalf(format, args)
os.Exit(2) os.Exit(2)
} }

View File

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

View File

@ -7,16 +7,16 @@ import (
) )
type Storage struct { type Storage struct {
Events map[int64]event.Event Events map[event.ID]event.Event
lastID int64 lastID event.ID
Mu sync.RWMutex Mu sync.RWMutex
} }
func New() *Storage { 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.Mu.Lock()
s.lastID++ s.lastID++
s.Events[s.lastID] = event s.Events[s.lastID] = event
@ -24,25 +24,25 @@ func (s *Storage) Create(event event.Event) (int64, error) {
return s.lastID, nil 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.Mu.Lock()
s.Events[id] = event s.Events[id] = event
s.Mu.Unlock() s.Mu.Unlock()
return nil return nil
} }
func (s *Storage) Delete(id int64) error { func (s *Storage) Delete(id event.ID) error {
s.Mu.Lock() s.Mu.Lock()
delete(s.Events, id) delete(s.Events, id)
s.Mu.Unlock() s.Mu.Unlock()
return nil 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 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 == "" { if s.Events[id].Title == "" {
return event.Event{}, false return event.Event{}, false
} }

View File

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

View File

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