forked from proddata/nodeIngestBench
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
0 parents
commit 188eb8c
Showing
5 changed files
with
613 additions
and
0 deletions.
There are no files selected for viewing
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
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,104 @@ | ||
# Logs | ||
logs | ||
*.log | ||
npm-debug.log* | ||
yarn-debug.log* | ||
yarn-error.log* | ||
lerna-debug.log* | ||
|
||
# Diagnostic reports (https://nodejs.org/api/report.html) | ||
report.[0-9]*.[0-9]*.[0-9]*.[0-9]*.json | ||
|
||
# Runtime data | ||
pids | ||
*.pid | ||
*.seed | ||
*.pid.lock | ||
|
||
# Directory for instrumented libs generated by jscoverage/JSCover | ||
lib-cov | ||
|
||
# Coverage directory used by tools like istanbul | ||
coverage | ||
*.lcov | ||
|
||
# nyc test coverage | ||
.nyc_output | ||
|
||
# Grunt intermediate storage (https://gruntjs.com/creating-plugins#storing-task-files) | ||
.grunt | ||
|
||
# Bower dependency directory (https://bower.io/) | ||
bower_components | ||
|
||
# node-waf configuration | ||
.lock-wscript | ||
|
||
# Compiled binary addons (https://nodejs.org/api/addons.html) | ||
build/Release | ||
|
||
# Dependency directories | ||
node_modules/ | ||
jspm_packages/ | ||
|
||
# TypeScript v1 declaration files | ||
typings/ | ||
|
||
# TypeScript cache | ||
*.tsbuildinfo | ||
|
||
# Optional npm cache directory | ||
.npm | ||
|
||
# Optional eslint cache | ||
.eslintcache | ||
|
||
# Microbundle cache | ||
.rpt2_cache/ | ||
.rts2_cache_cjs/ | ||
.rts2_cache_es/ | ||
.rts2_cache_umd/ | ||
|
||
# Optional REPL history | ||
.node_repl_history | ||
|
||
# Output of 'npm pack' | ||
*.tgz | ||
|
||
# Yarn Integrity file | ||
.yarn-integrity | ||
|
||
# dotenv environment variables file | ||
.env | ||
.env.test | ||
|
||
# parcel-bundler cache (https://parceljs.org/) | ||
.cache | ||
|
||
# Next.js build output | ||
.next | ||
|
||
# Nuxt.js build / generate output | ||
.nuxt | ||
dist | ||
|
||
# Gatsby files | ||
.cache/ | ||
# Comment in the public line in if your project uses Gatsby and *not* Next.js | ||
# https://nextjs.org/blog/next-9-1#public-directory-support | ||
# public | ||
|
||
# vuepress build output | ||
.vuepress/dist | ||
|
||
# Serverless directories | ||
.serverless/ | ||
|
||
# FuseBox cache | ||
.fusebox/ | ||
|
||
# DynamoDB Local files | ||
.dynamodb/ | ||
|
||
# TernJS port file | ||
.tern-port |
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
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,17 @@ | ||
{ | ||
// Use IntelliSense to learn about possible attributes. | ||
// Hover to view descriptions of existing attributes. | ||
// For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387 | ||
"version": "0.2.0", | ||
"configurations": [ | ||
{ | ||
"type": "pwa-node", | ||
"request": "launch", | ||
"name": "Launch Program", | ||
"skipFiles": [ | ||
"<node_internals>/**" | ||
], | ||
"program": "${workspaceFolder}/main.js" | ||
} | ||
] | ||
} |
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
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,119 @@ | ||
require("dotenv").config(); | ||
const { Pool } = require("pg"); | ||
|
||
const args = process.argv.slice(2); | ||
|
||
const crateConfig = { | ||
user: process.env.CRATE_USER || "crate", | ||
host: process.env.CRATE_HOST || "localhost", | ||
database: process.env.CRATE_DB || "doc", | ||
password: process.env.CRATE_PASSWORD|| "", | ||
port: process.env.CRATE_PORT || 5432, | ||
ssl: process.env.CRATE_SSL === undefined ? false : process.env.CRATE_SSL.toLowerCase() == 'true', | ||
max: 24 | ||
}; | ||
|
||
const testOptions = { | ||
batchsize: Number(args[0]) || 10000, | ||
max_rows: Number(args[1]) || 1*1000*1000, | ||
table: "ge.time_series_data", | ||
columns: "(tag_name, ts, value_float,value_int, attributes, tenant_id)" | ||
}; | ||
|
||
console.log("-------- Options ---------"); | ||
console.log(testOptions); | ||
console.log("--------------------------"); | ||
|
||
const pool = new Pool(crateConfig) | ||
|
||
|
||
let ts_start; | ||
let ts_end; | ||
let inserts_started = 0; | ||
let inserts_done = 0; | ||
let max = Math.ceil(testOptions.max_rows / testOptions.batchsize); | ||
let done = false; | ||
let start = 0; | ||
let end = 0; | ||
|
||
// Get inital count(*) from table | ||
pool.query(`REFRESH TABLE ${testOptions.table}; SELECT COUNT(*) as count FROM ${testOptions.table}`).then( res => { | ||
|
||
start = Number(res[1].rows[0].count); | ||
ts_start = Date.now() / 1000; | ||
|
||
console.log("Start\t",start.toLocaleString(),"records"); | ||
|
||
query(); | ||
}).catch(err => console.log(err)); | ||
|
||
function query(){ | ||
if(inserts_started <= max){ | ||
inserts_started++; | ||
pool.query(getInsertQuery()).then(raise).catch(err => console.log(err)); | ||
|
||
let timeout = 20; | ||
setTimeout(query,timeout); | ||
} | ||
} | ||
|
||
function raise(){ | ||
inserts_done++; | ||
if(!done && inserts_done == max){ | ||
done = true; | ||
pool.query(`REFRESH TABLE ${testOptions.table}; SELECT COUNT(*) as count FROM ${testOptions.table}`).then( res => { | ||
|
||
end = Number(res[1].rows[0].count); | ||
let rows = end-start; | ||
ts_end = Date.now() / 1000; | ||
let rate = rows/(ts_end-ts_start); | ||
|
||
console.log("End\t",end.toLocaleString(),"records"); | ||
console.log("New\t",rows.toLocaleString(),"records"); | ||
console.log("Time\t",(ts_end-ts_start).toLocaleString(),'s'); | ||
console.log("Speed\t",rate.toLocaleString(),'rows per sec'); | ||
|
||
pool.end(); | ||
}).catch(err => console.log(err)); | ||
|
||
} | ||
} | ||
|
||
function getInsertQuery() { | ||
let query = `INSERT INTO ${testOptions.table} ${testOptions.columns} VALUES `; | ||
for (let i = 0; i < testOptions.batchsize; i++) { | ||
let s = getRandomObject(); | ||
if (i > 0) query += `,`; | ||
query += `('${s.tag_name}',${s.ts},${s.measurement},${s.value_int},'${s.attributes}',${s.tenant_id})`; | ||
} | ||
query += ';'; | ||
//console.log(query); | ||
return query | ||
} | ||
|
||
function getRandomObject(){ | ||
let obj = { | ||
tag_name : randomString("tag",1,200), | ||
ts: randomInt(1621814400000,1625443200000-1), | ||
value_int : randomInt(1,Number.MAX_SAFE_INTEGER-1), | ||
measurement : randomFloat(1.0,Number.MAX_VALUE-1), | ||
quality : randomInt(1,3), | ||
attributes : `{"${randomString("tag",1,20)}":"${randomString("value",100,10000)}","${randomString("tag",21,40)}":"${randomString("value",100,10000)}"}`, | ||
tenant_id : randomInt(10000,1000000) | ||
} | ||
|
||
return obj; | ||
} | ||
|
||
|
||
function randomInt(min, max) { | ||
return min + Math.floor(Math.random() * (max - min)); | ||
} | ||
|
||
function randomFloat(min, max) { | ||
return min + Math.random() * (max - min); | ||
} | ||
|
||
function randomString(base, min, max) { | ||
return base + min + Math.floor(Math.random() * (max - min)); | ||
} |
Oops, something went wrong.