GraffTech/pixel-streaming-coordinator
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

logger upgraded, session server status upgraded

Browse Source
This commit is contained in:
PC-ROGE\c
2023-04-19 21:47:11 +03:00
parent ffa66977c3
commit 4895ca0d09
11 changed files with 135 additions and 80 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/database/database.js
+75 -42
View File
@@ -1,7 +1,7 @@
const { MongoClient } = require("mongodb")
const { database_url, database_name } = require('../../config')
const crypto = require("crypto")
const time = require('../modules/time')
const time = require('../../lib/src/time')
let max_database_connection_timeout = 1000
const client = new MongoClient(database_url, {
@@ -22,50 +22,50 @@ const get_db = async () => {
}
const get_titles = async (start, count) => {
var db = await get_db()
let db = await get_db()
if (!db) {
return null
}
var titles = await db.collection('title').find().skip(parseInt(start)).limit(parseInt(count)).toArray()
let titles = await db.collection('title').find().skip(parseInt(start)).limit(parseInt(count)).toArray()
return (!titles.length) ? null : titles
}
const get_titles_for_language = async (start, count, language) => {
var db = await get_db()
let db = await get_db()
if (!db) {
return null
}
var titles = await db.collection('title').find({language:language}).skip(parseInt(start)).limit(parseInt(count)).toArray()
let titles = await db.collection('title').find({language:language}).skip(parseInt(start)).limit(parseInt(count)).toArray()
return (!titles.length) ? null : titles
}
const get_private_titles_with_key = async (key) => {
var db = await get_db()
let db = await get_db()
if (!db) {
return null
}
var titles = await db.collection('private_title').find({key:key}).toArray()
let titles = await db.collection('private_title').find({key:key}).toArray()
return (!titles.length) ? null : titles
}
const get_free_session_servers = async (title) => {
var db = await get_db()
let db = await get_db()
if (!db) {
return null
}
var servers = await db.collection('session_server').find({title:title}).toArray()
let servers = await db.collection('session_server').find({title:title}).toArray()
if (!servers.length) {
return null
}
var active_session = db.collection('active_session')
let active_session = db.collection('active_session')
var free_servers = []
let free_servers = []
for (const server of servers) {
var sessions = await active_session.find({server_url:server.url}).toArray()
let sessions = await active_session.find({server_url:server.url}).toArray()
if (server.limit > sessions.length) {
free_servers.push(server)
}
@@ -73,33 +73,35 @@ const get_free_session_servers = async (title) => {
return (free_servers.length) ? free_servers : null
}
const add_active_session = async (server_url, session_id, websocket_url) => {
var db = await get_db()
const add_active_session = async (title, server_url, session_id, websocket_url) => {
let db = await get_db()
if (!db) {
return false
}
var active_session = db.collection('active_session')
let active_session = db.collection('active_session')
await active_session.insertOne({
title: title,
server_url: server_url,
session_id: session_id,
websocket_url: websocket_url
})
return true
}
const generate_session_id = async () => {
var db = await get_db()
let db = await get_db()
if (!db) {
return null
}
var active_session = db.collection('active_session')
var session_id = {}
let active_session = db.collection('active_session')
let session_id = {}
var max_id_len = 6
var sessions_with_id = 0
let max_id_len = 6
let sessions_with_id = 0
while(true) {
session_id = crypto.randomBytes(max_id_len/2).toString('hex').toUpperCase()
sessions_with_id = await active_session.find({session_id:session_id}).toArray().length
@@ -109,8 +111,18 @@ const generate_session_id = async () => {
}
}
const get_active_session = async (session_id) => {
let db = await get_db()
if (!db) {
return null
}
let active_session = await db.collection('active_session').findOne({session_id:session_id})
return active_session
}
const remove_active_session = async (session_id) => {
var db = await get_db()
let db = await get_db()
if (!db) {
return false
}
@@ -120,12 +132,12 @@ const remove_active_session = async (session_id) => {
}
const get_session_websocket_url = async (session_id) => {
var db = await get_db()
let db = await get_db()
if (!db) {
return null
}
var active_session = db.collection('active_session')
var session = await active_session.findOne({session_id:session_id})
let active_session = db.collection('active_session')
let session = await active_session.findOne({session_id:session_id})
if (!session) {
return null
}
@@ -133,21 +145,21 @@ const get_session_websocket_url = async (session_id) => {
}
const get_statuses = async () => {
var db = await get_db()
let db = await get_db()
if (!db) {
return null
}
var statuses = await db.collection('status').find().toArray()
let statuses = await db.collection('status').find().toArray()
return (!statuses.length) ? null : statuses
}
const get_all_session_servers = async () => {
var db = await get_db()
let db = await get_db()
if (!db) {
return null
}
var servers = await db.collection('session_server').find().toArray()
let servers = await db.collection('session_server').find().toArray()
if (!servers.length) {
return null
@@ -156,33 +168,33 @@ const get_all_session_servers = async () => {
}
const add_error = async (error_message) => {
var db = await get_db()
let db = await get_db()
if (!db) {
return false
}
var error_collection = db.collection('error')
let error_collection = db.collection('error_history')
await error_collection.insertOne({error_message:error_message, time: time.get_date_time()})
await error_collection.insertOne({error_message:error_message, time: time.get_date_time(), unix_timestamp: time.get_unix_timestamp()})
return true
}
const get_errors = async () => {
var db = await get_db()
let db = await get_db()
if (!db) {
return null
}
var errors = await db.collection('error').find().toArray()
let errors = await db.collection('error_history').find().toArray()
return (!errors.length) ? null : errors
}
const get_all_active_sessions = async () => {
var db = await get_db()
let db = await get_db()
if (!db) {
return null
}
var active_sessions = await db.collection('active_session').find().toArray()
let active_sessions = await db.collection('active_session').find().toArray()
if (!active_sessions.length) {
return null
@@ -190,30 +202,50 @@ const get_all_active_sessions = async () => {
return active_sessions
}
const add_session_to_history = async (server_url, session_id, websocket_url) => {
var db = await get_db()
const add_session_to_history = async (session_id,
title,
server_url,
websocket_url,
max_uniq_users,
users_at_close,
time_start,
unix_timestamp_start,
time_end,
unix_timestamp_end
) => {
let db = await get_db()
if (!db) {
return false
}
var session_history = db.collection('session_history')
let session_history = db.collection('session_history')
let duration_in_seconds = unix_timestamp_end - unix_timestamp_start
await session_history.insertOne({
server_url: server_url,
session_id: session_id,
title: title,
server_url: server_url,
websocket_url: websocket_url,
time: time.get_date_time()
max_uniq_users: max_uniq_users,
users_at_close: users_at_close,
time_start: time_start,
unix_timestamp_start: unix_timestamp_start,
time_end: time_end,
unix_timestamp_end: unix_timestamp_end,
duration_in_seconds: duration_in_seconds,
duration_in_minutes: duration_in_seconds / 60
})
return true
}
const get_all_session_history = async () => {
var db = await get_db()
let db = await get_db()
if (!db) {
return null
}
var session_history = await db.collection('session_history').find().toArray()
let session_history = await db.collection('session_history').find().toArray()
if (!session_history.length) {
return null
@@ -227,6 +259,7 @@ module.exports = {
add_active_session,
generate_session_id,
remove_active_session,
get_active_session,
get_session_websocket_url,
get_titles_for_language,
get_private_titles_with_key,