mirror of
https://github.com/linux-odyssey/linux-odyssey.git
synced 2026-11-06 03:20:13 +08:00
feat: seperate container creation and get
This commit is contained in:
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()
|
||||||
|
|||||||
@@ -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) => {
|
||||||
|
|||||||
@@ -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
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in new issue
Block a user