Added sqllite db to store messages

Author: TutorialsAndroid

requirements.txt

@@ -1,3 +1,4 @@
 fastapi
 uvicorn
-websockets
+websockets
+sqlite3

server.py

@@ -2,9 +2,24 @@
 from typing import List
 import uvicorn
 import json
+import sqlite3
 
 app = FastAPI()
 
+conn = sqlite3.connect("chat.db", check_same_thread=False)
+cursor = conn.cursor()
+
+cursor.execute("""
+CREATE TABLE IF NOT EXISTS messages (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    name TEXT,
+    text TEXT,
+    timestamp DATETIME DEFAULT CURRENT_TIMESTAMP
+)
+""")
+
+conn.commit()
+
 
 class ConnectionManager:
 
@@ -32,6 +47,18 @@ async def websocket_endpoint(websocket: WebSocket):
 
     await manager.connect(websocket)
 
+    cursor.execute(
+        "SELECT name, text FROM messages ORDER BY id DESC LIMIT 50"
+    )
+
+    rows = cursor.fetchall()
+
+    for name, text in reversed(rows):
+        await websocket.send_text(json.dumps({
+            "name": name,
+            "text": text
+        }))
+
     try:
 
         while True:
@@ -40,6 +67,16 @@ async def websocket_endpoint(websocket: WebSocket):
 
             message = json.loads(data)
 
+            name = message.get("name")
+            text = message.get("text")
+
+            cursor.execute(
+                "INSERT INTO messages (name, text) VALUES (?, ?)",
+                (name, text)
+            )
+
+            conn.commit()
+
             await manager.broadcast(json.dumps(message))
 
     except WebSocketDisconnect: