put sessions in db

api/db.go

@@ -6,6 +6,7 @@ import (
 
 	"git.dubyatp.xyz/chat-api-server/db"
 	"github.com/gocql/gocql"
+	"github.com/google/uuid"
 )
 
 func dbGetUser(id string) (*User, error) {
@@ -107,6 +108,42 @@ func dbGetAllMessages() ([]*Message, error) {
 	return messages, nil
 }
 
+func dbAddSession(session *Session) error {
+	query := `INSERT INTO sessions (session_token, username) VALUES (?, ?)`
+	err := db.Session.Query(query, session.Token, session.Username).Exec()
+	if err != nil {
+		return fmt.Errorf("failed to add session: %v", err)
+	}
+	return nil
+}
+
+func dbGetSession(id uuid.UUID) (*Session, error) {
+	query := `SELECT session_token, username FROM sessions WHERE session_token = ?`
+	var session Session
+	err := db.Session.Query(query, id).Scan(
+		&session.Token,
+		&session.Username)
+	if err == gocql.ErrNotFound {
+		return nil, errors.New("Session not found")
+	} else if err != nil {
+		return nil, fmt.Errorf("failed to query session: %v", err)
+	}
+
+	return &session, nil
+}
+
+func dbDeleteSession(id uuid.UUID) error {
+	query := `DELETE FROM sessions WHERE session_token = ?`
+
+	err := db.Session.Query(query, id).Exec()
+
+	if err != nil {
+		return fmt.Errorf("failed to delete session: %v", err)
+	}
+
+	return nil
+}
+
 func dbAddUser(user *User) error {
 	query := `INSERT INTO users (id, name, password) VALUES (?, ?, ?)`
 	err := db.Session.Query(query, user.ID, user.Name, user.Password).Exec()