diff options
Diffstat (limited to 'server')
| -rw-r--r-- | server/message.go | 2 | ||||
| -rw-r--r-- | server/storage.go | 11 | ||||
| -rw-r--r-- | server/user.go | 2 |
3 files changed, 11 insertions, 4 deletions
diff --git a/server/message.go b/server/message.go index e814d9d..19bfadf 100644 --- a/server/message.go +++ b/server/message.go @@ -42,7 +42,7 @@ func (s *Server) handleMessage(sender net.Conn, connType core.ConnType, msg core return err } - if err := s.Storage.Write(msg); err != nil { + if err := s.Storage.AddMessage(msg); err != nil { log.Println("cannot write to database", err) } diff --git a/server/storage.go b/server/storage.go index 3473898..037c40f 100644 --- a/server/storage.go +++ b/server/storage.go @@ -7,6 +7,13 @@ import ( ) type Storage interface { - Write(msg core.Message) (err error) - Read(channel string, since time.Time, num, offset int) (history []core.Message, err error) + AddMessage(msg core.Message) (err error) + GetHistory(channel string, since time.Time, num, offset int) (history []core.Message, err error) + + SetUser(user core.UserData) error + DelUser(name string) error + GetUserPass(name string) (string, error) + + SetPermission(data core.PermissionData) error + GetPermission(user, channel string) (core.PermissionData, error) } diff --git a/server/user.go b/server/user.go index af3e1a6..5b8049f 100644 --- a/server/user.go +++ b/server/user.go @@ -175,7 +175,7 @@ func (s *Server) handleHistory(user *User, conn net.Conn, hist core.History) err return user.Send(conn, core.Error{core.ErrorNotFound}) } - messages, err := s.Storage.Read(hist.Channel, hist.Since, int(hist.Count), int(hist.Offset)) + messages, err := s.Storage.GetHistory(hist.Channel, hist.Since, int(hist.Count), int(hist.Offset)) if err != nil { fmt.Println("cannot get message history:", err) return user.Send(conn, core.Error{core.ErrorNotFound}) |