feat: seperate container creation and get

This commit is contained in:
wancat committed 2023-08-23 11:51:16 +08:00
1 parent 2c866d2e83
commit 8e63284211
3 files changed
+43 -19

No files matched your search

@@ -1,5 +1,6 @@
import Session from '../../models/session.js' import Session from '../../models/session.js'
import Quest from '../../models/quest.js' import Quest from '../../models/quest.js'
import { createContainer } from '../../containers/docker.js'
export async function getSessionList(req, res) { export async function getSessionList(req, res) {
try { try {
@@ -7,8 +8,31 @@ export async function getSessionList(req, res) {
if (req.query.quest_id) { if (req.query.quest_id) {
query.quest = req.query.quest_id query.quest = req.query.quest_id
} }
if (req.query.status) {
query.status = req.query.status
}
const sessions = await Session.find(query) const sessions = await Session.find(query)
res.json(sessions) res.json(
sessions.map(
({
_id,
user,
quest,
status,
progress,
createdAt,
lastActivityAt,
}) => ({
_id,
user,
quest,
status,
progress,
createdAt,
lastActivityAt,
})
)
)
} catch (err) { } catch (err) {
console.error(err) console.error(err)
res.status(500).json({ message: err.message }) res.status(500).json({ message: err.message })
@@ -24,11 +48,14 @@ export async function createSession(req, res) {
try { try {
const progress = quest.stages[0].id const progress = quest.stages[0].id
const container = await createContainer(
`quest-${quest.id}-${req.user.username}-${Date.now()}`
)
const newSession = new Session({ const newSession = new Session({
user: req.user, user: req.user,
quest, quest,
progress, progress,
containerId: `quest-${quest.id}-${req.user.username}-${Date.now()}`, containerId: container.id,
}) })
const session = await newSession.save() const session = await newSession.save()
+3 -2
View File
@@ -1,5 +1,5 @@
import { Server } from 'socket.io' import { Server } from 'socket.io'
import { getOrCreateContainer, attachContainer } from '../containers/docker.js' import { getAndStartContainer, attachContainer } from '../containers/docker.js'
import Session from '../models/session.js' import Session from '../models/session.js'
import { defaultUser, genSessionJWT } from '../utils/auth.js' import { defaultUser, genSessionJWT } from '../utils/auth.js'
@@ -47,6 +47,7 @@ export default (server) => {
session = await Session.findOne({ session = await Session.findOne({
_id: sessionId, _id: sessionId,
user: socket.user, user: socket.user,
status: 'active',
}) })
if (!session) { if (!session) {
socket.send('Session not found.') socket.send('Session not found.')
@@ -61,7 +62,7 @@ export default (server) => {
const token = await genSessionJWT(session) const token = await genSessionJWT(session)
const container = await getOrCreateContainer(session.containerId) const container = await getAndStartContainer(session.containerId)
const stream = await attachContainer(container, { token }) const stream = await attachContainer(container, { token })
stream.on('data', (chunk) => { stream.on('data', (chunk) => {
+11 -15
View File
@@ -24,7 +24,16 @@ const containerOptions = {
const network = engine.getNetwork(config.dockerNetwork) const network = engine.getNetwork(config.dockerNetwork)
export async function getOrCreateContainer(id) { export async function createContainer(name) {
const container = await engine.createContainer({
...containerOptions,
name,
})
await network.connect({ Container: container.id })
return container
}
export async function getAndStartContainer(id) {
console.log(`Getting container: ${id}`) console.log(`Getting container: ${id}`)
let container let container
try { try {
@@ -34,20 +43,7 @@ export async function getOrCreateContainer(id) {
} }
return container return container
} catch (error) { } catch (error) {
try { throw new Error(`Container not found: ${id}`)
container = await engine.createContainer({
...containerOptions,
name: id,
})
} catch (err) {
console.log(err)
throw err
}
await network.connect({ Container: container.id })
await container.start()
console.log(container.id)
return container
} }
} }