mirror of
https://github.com/linux-odyssey/linux-odyssey.git
synced 2026-11-06 03:20:13 +08:00
14 files changed
+1581
-11
No files matched your search
@@ -0,0 +1,73 @@
|
||||
export default class DAG {
|
||||
#nodes = new Map()
|
||||
|
||||
#edges = new Set()
|
||||
|
||||
#layers = []
|
||||
|
||||
constructor(nodes) {
|
||||
this.#nodes = new Map(nodes.map((node) => [node._id, node]))
|
||||
this.#edges = new Set()
|
||||
this.#nodes.forEach((_, id) => {
|
||||
this.#setLayer(id)
|
||||
this.#addEdges(id)
|
||||
})
|
||||
}
|
||||
|
||||
getNode(id) {
|
||||
return this.#nodes.get(id)
|
||||
}
|
||||
|
||||
getNodes() {
|
||||
return Array.from(this.#nodes.values())
|
||||
}
|
||||
|
||||
getEdges() {
|
||||
return this.#edges
|
||||
}
|
||||
|
||||
getEdgesArray() {
|
||||
return Array.from(this.#edges.values())
|
||||
}
|
||||
|
||||
getLayers() {
|
||||
return this.#layers
|
||||
}
|
||||
|
||||
getLayer(id) {
|
||||
const node = this.getNode(id)
|
||||
if (!node) {
|
||||
throw new Error(`Node ${id} not found`)
|
||||
}
|
||||
return this.#layers[node.layer - 1]
|
||||
}
|
||||
|
||||
#setLayer(id) {
|
||||
const node = this.getNode(id)
|
||||
if (!node) {
|
||||
throw new Error(`Node ${id} not found`)
|
||||
}
|
||||
if (node.layer !== undefined) {
|
||||
return node.layer
|
||||
}
|
||||
const previousLayer = node.requirements.map((rId) => this.#setLayer(rId))
|
||||
node.layer = Math.max(...previousLayer, 0) + 1
|
||||
if (this.#layers.length < node.layer) {
|
||||
this.#layers.push(1)
|
||||
} else {
|
||||
this.#layers[node.layer - 1] += 1
|
||||
}
|
||||
node.index = this.#layers[node.layer - 1]
|
||||
return node.layer
|
||||
}
|
||||
|
||||
#addEdges(id) {
|
||||
const node = this.getNode(id)
|
||||
if (!node) {
|
||||
throw new Error(`Node ${id} not found`)
|
||||
}
|
||||
node.requirements.forEach((rId) => {
|
||||
this.#edges.add([rId, id])
|
||||
})
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,165 @@
|
||||
import DAG from './dag.js'
|
||||
|
||||
describe('dag layers', () => {
|
||||
it('two layer', () => {
|
||||
const data = [
|
||||
{
|
||||
_id: 'helloworld',
|
||||
requirements: [],
|
||||
},
|
||||
{
|
||||
_id: 'spell',
|
||||
requirements: ['helloworld'],
|
||||
},
|
||||
]
|
||||
const dag = new DAG(data)
|
||||
expect(dag.getNode('helloworld').layer).toBe(1)
|
||||
expect(dag.getNode('spell').layer).toBe(2)
|
||||
})
|
||||
it('branch', () => {
|
||||
const data = [
|
||||
{
|
||||
_id: 'helloworld',
|
||||
requirements: [],
|
||||
},
|
||||
{
|
||||
_id: 'spell',
|
||||
requirements: ['helloworld'],
|
||||
},
|
||||
{
|
||||
_id: 'discover',
|
||||
requirements: ['helloworld'],
|
||||
},
|
||||
]
|
||||
const dag = new DAG(data)
|
||||
expect(dag.getNode('helloworld').layer).toBe(1)
|
||||
expect(dag.getNode('spell').layer).toBe(2)
|
||||
expect(dag.getNode('discover').layer).toBe(2)
|
||||
})
|
||||
it('merge', () => {
|
||||
const data = [
|
||||
{
|
||||
_id: 'A',
|
||||
requirements: [],
|
||||
},
|
||||
{
|
||||
_id: 'B',
|
||||
requirements: [],
|
||||
},
|
||||
{
|
||||
_id: 'C',
|
||||
requirements: ['A', 'B'],
|
||||
},
|
||||
]
|
||||
const dag = new DAG(data)
|
||||
expect(dag.getNode('A').layer).toBe(1)
|
||||
expect(dag.getNode('B').layer).toBe(1)
|
||||
expect(dag.getNode('C').layer).toBe(2)
|
||||
})
|
||||
it('skip', () => {
|
||||
const data = [
|
||||
{
|
||||
_id: 'A',
|
||||
requirements: [],
|
||||
},
|
||||
{
|
||||
_id: 'B',
|
||||
requirements: [],
|
||||
},
|
||||
{
|
||||
_id: 'C',
|
||||
requirements: ['A'],
|
||||
},
|
||||
{
|
||||
_id: 'D',
|
||||
requirements: ['B', 'C'],
|
||||
},
|
||||
]
|
||||
const dag = new DAG(data)
|
||||
expect(dag.getNode('A').layer).toBe(1)
|
||||
expect(dag.getNode('B').layer).toBe(1)
|
||||
expect(dag.getNode('C').layer).toBe(2)
|
||||
expect(dag.getNode('D').layer).toBe(3)
|
||||
})
|
||||
})
|
||||
|
||||
describe('dag edges', () => {
|
||||
it('one edge', () => {
|
||||
const data = [
|
||||
{
|
||||
_id: 'helloworld',
|
||||
requirements: [],
|
||||
},
|
||||
{
|
||||
_id: 'spell',
|
||||
requirements: ['helloworld'],
|
||||
},
|
||||
]
|
||||
|
||||
const dag = new DAG(data)
|
||||
expect(dag.getEdges()).toEqual(new Set([['helloworld', 'spell']]))
|
||||
})
|
||||
|
||||
it('branch edges', () => {
|
||||
const data = [
|
||||
{
|
||||
_id: 'helloworld',
|
||||
requirements: [],
|
||||
},
|
||||
{
|
||||
_id: 'spell',
|
||||
requirements: ['helloworld'],
|
||||
},
|
||||
{
|
||||
_id: 'discover',
|
||||
requirements: ['helloworld'],
|
||||
},
|
||||
]
|
||||
|
||||
const dag = new DAG(data)
|
||||
expect(dag.getEdges()).toEqual(
|
||||
new Set([
|
||||
['helloworld', 'discover'],
|
||||
['helloworld', 'spell'],
|
||||
])
|
||||
)
|
||||
})
|
||||
})
|
||||
|
||||
describe('dag layers', () => {
|
||||
it('two layer', () => {
|
||||
const data = [
|
||||
{
|
||||
_id: 'helloworld',
|
||||
requirements: [],
|
||||
},
|
||||
{
|
||||
_id: 'spell',
|
||||
requirements: ['helloworld'],
|
||||
},
|
||||
]
|
||||
|
||||
const dag = new DAG(data)
|
||||
expect(dag.getLayers()).toEqual([1, 1])
|
||||
})
|
||||
|
||||
it('branch layers', () => {
|
||||
const data = [
|
||||
{
|
||||
_id: 'helloworld',
|
||||
requirements: [],
|
||||
},
|
||||
{
|
||||
_id: 'spell',
|
||||
requirements: ['helloworld'],
|
||||
},
|
||||
{
|
||||
_id: 'discover',
|
||||
requirements: ['helloworld'],
|
||||
},
|
||||
]
|
||||
|
||||
const dag = new DAG(data)
|
||||
expect(dag.getLayers()).toEqual([1, 2])
|
||||
})
|
||||
})
|
||||
@@ -1,2 +1,3 @@
|
||||
export { isValidEmail, isValidUsername } from './auth.js'
|
||||
export { buildFileCheckCmd } from './commands.js'
|
||||
export { default as DAG } from './dag.js'
|
||||
Reference in new issue
Block a user