node-media-server/src/api/handlers/sessions.js

A Node.js implementation of RTMP Server

// @ts-check
//
//  Created by Chen Mingliang on 24/11/27.
//  illuspas@msn.com
//  Copyright (c) 2024 NodeMedia. All rights reserved.
//

const logger = require("../../core/logger.js");
const Context = require("../../core/context.js");

class SessionsHandler {
  /**
   * Get all active sessions
   * @param {express.Request} req
   * @param {express.Response} res
   */
  static getSessions(req, res) {
    try {
      const sessions = [];

      Context.sessions.forEach((session, id) => {
        sessions.push({
          id,
          ip: session.ip,
          isPublisher: session.isPublisher,
          protocol: session.protocol,
          app: session.app,
          name: session.name,
          type: session.type,
          inBytes: session.inBytes,
          outBytes: session.outBytes,
          createTime: session.createTime
        });
      });

      res.json({
        success: true,
        data: {
          sessions,
          total: sessions.length
        },
        message: "Sessions retrieved successfully"
      });
    } catch (error) {
      logger.error("Error getting sessions:", error);
      res.status(500).json({
        success: false,
        data: {},
        message: "Internal server error"
      });
    }
  }

  /**
   * Delete a session by ID
   * @param {express.Request} req
   * @param {express.Response} res
   */
  static deleteSession(req, res) {
    try {
      const { id } = req.params;

      if (!id) {
        return res.status(400).json({
          success: false,
          data: {},
          message: "Session ID is required"
        });
      }

      const session = Context.sessions.get(id);

      if (!session) {
        return res.status(404).json({
          success: false,
          data: {},
          message: "Session not found"
        });
      }

      // Call the close method on the session
      session.close();

      res.json({
        success: true,
        data: {
          id
        },
        message: "Session deleted successfully"
      });
    } catch (error) {
      logger.error("Error deleting session:", error);
      res.status(500).json({
        success: false,
        data: {},
        message: "Internal server error"
      });
    }
  }
}

module.exports = SessionsHandler;