Last active
June 21, 2021 07:15
-
-
Save mdvsh/4038a2ae6c9d41423aa771b7b0da39b2 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
const http = require("http"); | |
const fs = require("fs"); | |
const PORT = 4000; | |
const origin = `http://localhost:${PORT}`; | |
class Database { | |
constructor(fname) { | |
this.fn = `${fname}.json`; | |
let state = new Map(); | |
try { | |
let joe = fs.readFileSync(this.fn, { | |
encoding: "utf-8", | |
flag: "r+", | |
}); | |
state = new Map(Object.entries(JSON.parse(joe))); | |
console.log("\nInitializing from existing DB.\n"); | |
} catch (err) { | |
console.log("\nNo DB found. Initializing new.\n"); | |
fs.writeFileSync(this.fn, JSON.stringify(Object.fromEntries(state))); | |
} | |
this.state = state; | |
} | |
setState(key, value) { | |
this.state.set(key, value); | |
return this.updateDb(this.state); | |
} | |
getValue(key) { | |
return this.state.get(key); | |
} | |
hasValue(key) { | |
return this.state.has(key); | |
} | |
getState(len = false) { | |
return len ? this.state.size : this.state; | |
} | |
updateDb(state) { | |
return fs.writeFile( | |
this.fn, | |
JSON.stringify(Object.fromEntries(state), undefined, 2), | |
(err) => { | |
if (err) { | |
console.log("couldnt update db"); | |
} | |
return true; | |
} | |
); | |
} | |
} | |
const db2 = new Database("storage"); | |
const server = http.createServer((req, res) => { | |
const url = new URL(req.url, origin); | |
const query = url.search.substr(1).split("="); | |
// route: /set?key=value | |
if (req.method === "GET" && url.pathname === "/set" && query.length === 2) { | |
if (query[0].length >= 1) { | |
let [key, value] = query; | |
if (db2.hasValue(key)) { | |
// console.log(db); | |
res | |
.writeHead(200, { "Content-Type": "text/plain" }) | |
.end(`Value with key [${key}] already exists in the database.`); | |
} else { | |
let record = {}; | |
record[key] = value; | |
db2.setState(key, value); | |
// console.log(db); | |
res | |
.writeHead(200, { "Content-Type": "text/plain" }) | |
.end( | |
`New Entry: ${JSON.stringify( | |
record, | |
undefined, | |
2 | |
)}\n\nCurrent Database Size: ${db2.getState((len = true))}` | |
); | |
} | |
} else { | |
res.end("key passed should be at least 1 character"); | |
} | |
} | |
// route: /get?key=somekey | |
else if ( | |
req.method === "GET" && | |
url.pathname === "/get" && | |
query[0] === "key" && | |
query.length === 2 | |
) { | |
if (query[1].length >= 1) { | |
let [, key] = query; | |
let fVal = db2.getValue(key); | |
if (!fVal) { | |
res.end(`Value with key [${key}] not found in database.`); | |
} else { | |
let record = {}; | |
record[key] = fVal; | |
res | |
.writeHead(200, { "Content-Type": "text/plain" }) | |
.end(`Found Entry:\n${JSON.stringify(record, undefined, 2)}`); | |
} | |
} else { | |
res.end("key passed should be at least 1 character"); | |
} | |
} | |
// route: /db (test route to display current state of db) | |
else if (req.method === "GET" && url.pathname === "/db") { | |
res | |
.writeHead(200, { "Content-Type": "text/plain" }) | |
.end( | |
`Current DB State: ${JSON.stringify( | |
Object.fromEntries(db2.getState()), | |
undefined, | |
2 | |
)}\n\nCurrent Database Size: ${db2.getState((len = true))}` | |
); | |
} | |
// route: /* (404 Not Found) | |
else { | |
res.statusCode = 404; | |
res.end("unexpected route"); | |
} | |
}); | |
server.listen(PORT, () => { | |
console.log(`Server ready at: ${origin}`); | |
}); |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment