Save messages in a database

1 files changed, 4 insertions, 2 deletions

diff --git a/server/server.go b/server/server.go
index ef2bb5f..74328a4 100644
--- a/server/server.go
+++ b/server/server.go
@@ -18,15 +18,17 @@ type Server struct {
 	usersMu    sync.RWMutex
 	serversMu  sync.RWMutex
 	channelsMu sync.RWMutex
+	storage    Storage
 }
 
-func NewServer(listenAddr string, name string) *Server {
+func NewServer(listenAddr string, name string, storage Storage) *Server {
 	return &Server{
 		listenAddr: listenAddr,
 		name:       name,
 		users:      make(map[string]User),
 		servers:    make(map[string]RemoteServer),
 		channels:   make(map[string]*Channel),
+		storage:    storage,
 	}
 }
 
@@ -50,7 +52,7 @@ func (s *Server) Start() error {
 
 func (s *Server) AddChannel(name string) {
 	s.channelsMu.Lock()
-	s.channelsMu.Unlock()
+	defer s.channelsMu.Unlock()
 	s.channels[name] = NewChannel(name)
 	log.Println("created channel", name)
 }