-
Notifications
You must be signed in to change notification settings - Fork 2
/
loadRequests.js
124 lines (113 loc) · 3.87 KB
/
loadRequests.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
const fs = require('fs');
const superagent = require('superagent');
const winston = require('winston');
const readline = require('readline');
const path = require('path');
const { getAuthToken } = require('./lib/login');
let inFile = process.argv[2];
let ep = 'circulation/requests';
let debug = process.env.DEBUG;
let dolog = process.env.LOG;
(async () => {
try {
const start = new Date().valueOf();
if (!inFile) {
throw 'Usage: node loadRequests.js <jsonl_file>';
} else if (!fs.existsSync(inFile)) {
throw new Error('Can\'t find input file');
}
const config = (fs.existsSync('./config.js')) ? require('./config.js') : require('./config.default.js');
const workingDir = path.dirname(inFile);
const baseName = path.basename(inFile, '.jsonl');
const errPath = `${workingDir}/${baseName}Err.jsonl`;
const outPath = `${workingDir}/${baseName}Out.jsonl`;
const logPath = `${workingDir}/${baseName}.log`;
if (fs.existsSync(errPath)) {
fs.unlinkSync(errPath);
}
if (fs.existsSync(outPath)) {
fs.unlinkSync(outPath);
}
var logger;
if (config.logpath || dolog) {
const lpath = config.logpath;
const lname = inFile.replace(/.+\//, '');
const logFileName = (dolog) ? logPath : `${lpath}/${lname}.log`;
if (fs.existsSync(logFileName)) {
fs.unlinkSync(logFileName);
}
logger = winston.createLogger({
level: 'info',
format: winston.format.json(),
defaultMeta: { service: 'user-service' },
transports: [
new winston.transports.File({ filename: logFileName })
]
});
} else {
logger = console;
}
const postReq = async (actionUrl, rec, x, rt) => {
if (!rt) rt = {};
let lDate = new Date();
logger.info(`[${x}] ${lDate} POST ${rec.id} to ${actionUrl}`);
try {
let res = await superagent
.post(actionUrl)
.send(rec)
.set('x-okapi-token', authToken)
.set('content-type', 'application/json')
.set('accept', 'application/json');
logger.info(` Successfully added record id ${rec.id}`);
} catch (e) {
let errMsg = (e.response && e.response.text && !debug) ? e.response.text : e;
if (errMsg.match(/Hold|Recall/) && !rt.Page) {
logger.warn(` WARN Hold/Recall type request failed, retrying as Page...`)
rec.requestType = 'Page';
rt[rec.requestType] = 1;
await postReq(actionUrl, rec, x, rt);
} else if (errMsg.match(/Page/) && !rt.Hold) {
logger.warn(` WARN Page type request failed, retrying as Hold...`)
rec.requestType = 'Hold';
rt[rec.requestType] = 1;
await postReq(actionUrl, rec, x, rt);
} else {
throw new Error(errMsg);
}
}
}
const authToken = await getAuthToken(superagent, config.okapi, config.tenant, config.authpath, config.username, config.password);
const actionUrl = `${config.okapi}/${ep}`;
let success = 0;
let fail = 0;
const fileStream = fs.createReadStream(inFile);
const rl = readline.createInterface({
input: fileStream,
crlfDelay: Infinity
});
let x = 0;
for await (const line of rl) {
x++;
let rec = JSON.parse(line);
delete rec.errorMessage;
try {
await postReq(actionUrl, rec, x);
success++;
} catch (e) {
logger.error(`${e}`);
let l = JSON.parse(line);
l.errorMessage = `${e}`;
fs.writeFileSync(errPath, JSON.stringify(l) + '\n', { flag: 'a'});
fail++;
}
}
const end = new Date().valueOf();
const ms = end - start;
const time = Math.floor(ms / 1000);
logger.info(`Time: ${time} sec`);
logger.info(`Records added: ${success}`);
logger.info(`Failures: ${fail}`);
} catch (e) {
console.error(e);
}
})();