elasticsearch/ts/elasticsearch.classes.elasticsearch.ts

106 lines
2.9 KiB
TypeScript
Raw Normal View History

2018-01-24 23:38:07 +00:00
// interfaces
2018-08-12 14:42:09 +00:00
import { Client as ElasticClient } from 'elasticsearch';
2018-11-07 10:38:53 +00:00
import { ILogContext, ILogPackage, ILogDestination } from '@pushrocks/smartlog-interfaces';
2018-01-27 18:16:05 +00:00
// other classes
2018-11-09 22:15:11 +00:00
import { ElasticScheduler } from './elasticsearch.classes.elasticscheduler';
import { ElasticIndex } from './elasticsearch.classes.elasticindex';
2018-01-24 23:38:07 +00:00
export interface IStandardLogParams {
2018-01-27 18:16:05 +00:00
message: string;
severity: string;
2018-01-24 23:38:07 +00:00
}
2018-11-10 00:48:44 +00:00
export interface IElasticSearchConstructorOptions {
indexPrefix: string;
indexRetention: number;
2018-01-27 18:16:05 +00:00
port: number;
domain: string;
ssl: boolean;
user?: string;
pass?: string;
2018-01-24 23:38:07 +00:00
}
2018-11-07 10:38:53 +00:00
export class ElasticSearch<T> {
2018-11-09 22:15:11 +00:00
public client: ElasticClient;
public elasticScheduler = new ElasticScheduler(this);
public elasticIndex: ElasticIndex = new ElasticIndex(this);
2018-01-24 23:38:07 +00:00
2018-11-10 00:48:44 +00:00
public indexPrefix: string;
public indexRetention: number;
2018-01-24 23:38:07 +00:00
/**
* sets up an instance of Elastic log
* @param optionsArg
*/
2018-11-10 00:48:44 +00:00
constructor(optionsArg: IElasticSearchConstructorOptions) {
2018-01-24 23:38:07 +00:00
this.client = new ElasticClient({
2019-11-02 00:42:03 +00:00
host: this.computeHostString(optionsArg),
2018-11-11 01:00:40 +00:00
// log: 'trace'
2018-01-24 23:38:07 +00:00
});
2018-11-10 00:48:44 +00:00
this.indexPrefix = optionsArg.indexPrefix;
this.indexRetention = optionsArg.indexRetention;
2018-01-27 18:16:05 +00:00
}
2018-01-24 23:38:07 +00:00
/**
* computes the host string from the constructor options
* @param optionsArg
*/
2018-11-10 00:48:44 +00:00
private computeHostString(optionsArg: IElasticSearchConstructorOptions): string {
2018-01-24 23:38:07 +00:00
let hostString = `${optionsArg.domain}:${optionsArg.port}`;
2018-01-27 18:16:05 +00:00
if (optionsArg.user && optionsArg.pass) {
hostString = `${optionsArg.user}:${optionsArg.pass}@${hostString}`;
2018-01-24 23:38:07 +00:00
}
2018-01-27 18:16:05 +00:00
if (optionsArg.ssl) {
hostString = `https://${hostString}`;
2018-01-24 23:38:07 +00:00
} else {
2018-01-27 18:16:05 +00:00
hostString = `http://${hostString}`;
2018-01-24 23:38:07 +00:00
}
2019-11-02 00:42:03 +00:00
console.log(hostString);
2018-01-24 23:38:07 +00:00
return hostString;
}
2018-11-03 22:45:21 +00:00
public async log(logPackageArg: ILogPackage, scheduleOverwrite = false) {
2018-01-27 18:16:05 +00:00
const now = new Date();
2018-11-25 21:06:25 +00:00
const indexToUse = `${this.indexPrefix}-${now.getFullYear()}.${(
'0' +
(now.getMonth() + 1)
).slice(-2)}.${('0' + now.getDate()).slice(-2)}`;
2018-11-09 22:15:11 +00:00
if (this.elasticScheduler.docsScheduled && !scheduleOverwrite) {
this.elasticScheduler.scheduleDoc(logPackageArg);
2018-01-27 18:16:05 +00:00
return;
}
2018-11-09 22:15:11 +00:00
await this.elasticIndex.ensureIndex(indexToUse);
2018-01-27 18:16:05 +00:00
this.client.index(
{
2018-11-09 22:15:11 +00:00
index: indexToUse,
2018-08-12 14:42:09 +00:00
type: 'log',
2018-01-27 18:16:05 +00:00
body: {
2018-11-07 10:38:53 +00:00
'@timestamp': new Date(logPackageArg.timestamp).toISOString(),
...logPackageArg
2018-01-27 18:16:05 +00:00
}
},
(error, response) => {
if (error) {
2018-08-12 14:42:09 +00:00
console.log('ElasticLog encountered an error:');
2018-01-27 18:16:05 +00:00
console.log(error);
2018-11-09 22:15:11 +00:00
this.elasticScheduler.addFailedDoc(logPackageArg);
2018-01-27 18:16:05 +00:00
} else {
2018-11-11 01:00:40 +00:00
// console.log(`ElasticLog: ${logPackageArg.message}`);
2018-01-27 18:16:05 +00:00
}
2018-01-24 23:38:07 +00:00
}
2018-01-27 18:16:05 +00:00
);
2018-01-24 23:38:07 +00:00
}
2018-11-07 10:38:53 +00:00
2018-11-25 21:06:25 +00:00
get logDestination(): ILogDestination {
2018-11-07 10:38:53 +00:00
return {
handleLog: (smartlogPackageArg: ILogPackage) => {
this.log(smartlogPackageArg);
}
2018-11-25 21:06:25 +00:00
};
2018-11-07 10:38:53 +00:00
}
2018-01-27 18:16:05 +00:00
}