EP.37 การสร้างระบบจัดการห้องแชทใน WebSocket และ GraphQL

ทำไมต้องมีระบบห้องแชท?

ระบบห้องแชทช่วยให้ผู้ใช้สามารถแบ่งการสนทนาออกเป็นกลุ่มย่อยได้ โดยแต่ละห้องจะมีผู้ใช้ที่เกี่ยวข้องเท่านั้น ซึ่งเหมาะสำหรับ:

• แชทกลุ่ม (Group Chat) ที่สมาชิกสามารถเข้าร่วมหรือออกจากห้องได้
• ห้องสนทนาเฉพาะเรื่อง (Topic-Based Chat Rooms) เช่น ห้องสนทนาของทีมในบริษัท
• ระบบสนับสนุนลูกค้า (Customer Support Chat) ที่ตัวแทนสามารถเข้าห้องเพื่อตอบคำถามของลูกค้าได้

โครงสร้างของระบบจัดการห้องแชท

1. WebSocket Server - จัดการการเชื่อมต่อของผู้ใช้และส่งข้อความไปยังห้องที่เกี่ยวข้อง
2. GraphQL API - รองรับคำสั่งสร้างห้อง, เข้าร่วมห้อง และรับส่งข้อความ
3. Database (PostgreSQL / MongoDB) - ใช้เก็บข้อมูลห้องและข้อความ
4. GraphQL Subscription - ส่งข้อความไปยังสมาชิกในห้องแบบเรียลไทม์

ติดตั้งไลบรารีที่จำเป็น

go get github.com/99designs/gqlgen
go get github.com/gorilla/websocket
go get github.com/jmoiron/sqlx

การตั้งค่าฐานข้อมูลสำหรับระบบห้องแชท

ไฟล์ schema.sql

CREATE TABLE chat_rooms (
    id SERIAL PRIMARY KEY,
    name TEXT NOT NULL UNIQUE
);

CREATE TABLE messages (
    id SERIAL PRIMARY KEY,
    room_id INTEGER REFERENCES chat_rooms(id),
    content TEXT NOT NULL,
    sender TEXT NOT NULL,
    timestamp TIMESTAMP DEFAULT CURRENT_TIMESTAMP
);

การสร้าง GraphQL Schema สำหรับระบบห้องแชท

ไฟล์ schema.graphql

type Query {
  rooms: [ChatRoom!]!
  messages(roomID: ID!): [Message!]!
}

type Mutation {
  createRoom(name: String!): ChatRoom!
  joinRoom(roomID: ID!): String!
  sendMessage(roomID: ID!, content: String!, sender: String!): Message!
}

type Subscription {
  messageAdded(roomID: ID!): Message!
}

type ChatRoom {
  id: ID!
  name: String!
}

type Message {
  id: ID!
  roomID: ID!
  content: String!
  sender: String!
  timestamp: String!
}

การสร้าง Resolver สำหรับระบบห้องแชท

ไฟล์ resolver.go

package main

import (
    "context"
    "database/sql"
    "fmt"
    "sync"
    "time"

    _ "github.com/lib/pq"
)

type ChatRoom struct {
    ID   int    `json:"id"`
    Name string `json:"name"`
}

type Message struct {
    ID        int       `json:"id"`
    RoomID    int       `json:"roomID"`
    Content   string    `json:"content"`
    Sender    string    `json:"sender"`
    Timestamp time.Time `json:"timestamp"`
}

type Resolver struct {
    mu          sync.Mutex
    db          *sql.DB
    subscribers map[int]map[string]chan Message
}

func NewResolver(db *sql.DB) *Resolver {
    return &Resolver{
        db:          db,
        subscribers: make(map[int]map[string]chan Message),
    }
}

func (r *Resolver) Query_rooms(ctx context.Context) ([]ChatRoom, error) {
    rows, err := r.db.Query("SELECT id, name FROM chat_rooms")
    if err != nil {
        return nil, err
    }
    defer rows.Close()

    var rooms []ChatRoom
    for rows.Next() {
        var room ChatRoom
        if err := rows.Scan(&room.ID, &room.Name); err != nil {
            return nil, err
        }
        rooms = append(rooms, room)
    }
    return rooms, nil
}

func (r *Resolver) Mutation_createRoom(ctx context.Context, name string) (ChatRoom, error) {
    var room ChatRoom
    err := r.db.QueryRow("INSERT INTO chat_rooms (name) VALUES ($1) RETURNING id, name", name).
        Scan(&room.ID, &room.Name)
    return room, err
}

func (r *Resolver) Mutation_sendMessage(ctx context.Context, roomID int, content string, sender string) (Message, error) {
    var message Message
    err := r.db.QueryRow("INSERT INTO messages (room_id, content, sender) VALUES ($1, $2, $3) RETURNING id, room_id, content, sender, timestamp", roomID, content, sender).
        Scan(&message.ID, &message.RoomID, &message.Content, &message.Sender, &message.Timestamp)
    
    r.mu.Lock()
    for _, ch := range r.subscribers[roomID] {
        ch <- message
    }
    r.mu.Unlock()
    return message, err
}

ท้าให้ลอง!

ลองเพิ่ม ระบบแอดมินห้องแชท ที่สามารถกำหนดสิทธิ์ผู้ใช้ เช่น ลบข้อความ หรือ จำกัดการเข้าถึง

EP ถัดไป

ใน EP.38, เราจะเพิ่ม การแจ้งเตือนแบบ Push Notification ให้กับ WebSocket Chat 🚀