rudeshark.net/src/db/elasticsearch.ts

75 lines
1.3 KiB
TypeScript
Raw Normal View History

2016-12-28 23:49:51 +01:00
import * as elasticsearch from 'elasticsearch';
2018-04-02 06:15:53 +02:00
import config from '../config';
import Logger from '../services/logger';
const esLogger = new Logger('es');
2016-12-28 23:49:51 +01:00
2018-07-04 13:36:06 +02:00
const index = {
settings: {
analysis: {
2018-09-05 09:48:59 +02:00
normalizer: {
lowercase_normalizer: {
type: 'custom',
filter: ['lowercase']
}
},
2018-07-04 13:36:06 +02:00
analyzer: {
bigram: {
tokenizer: 'bigram_tokenizer'
}
},
tokenizer: {
bigram_tokenizer: {
type: 'nGram',
min_gram: 2,
max_gram: 2
}
}
}
},
mappings: {
note: {
properties: {
text: {
type: 'text',
index: true,
2018-09-05 09:48:59 +02:00
analyzer: 'bigram',
normalizer: 'lowercase_normalizer'
2018-07-04 13:36:06 +02:00
}
}
}
}
};
2016-12-28 23:49:51 +01:00
// Init ElasticSearch connection
const client = config.elasticsearch ? new elasticsearch.Client({
host: `${config.elasticsearch.host}:${config.elasticsearch.port}`
}) : null;
2016-12-28 23:49:51 +01:00
2018-07-04 13:13:05 +02:00
if (client) {
// Send a HEAD request
client.ping({
// Ping usually has a 3000ms timeout
requestTimeout: 30000
}, error => {
if (error) {
esLogger.error('elasticsearch is down!');
2018-07-04 13:13:05 +02:00
} else {
esLogger.succ('elasticsearch is available!');
2018-07-04 13:13:05 +02:00
}
});
2018-07-04 13:02:45 +02:00
2018-07-04 13:36:06 +02:00
client.indices.exists({
index: 'misskey'
}).then(exist => {
if (exist) return;
client.indices.create({
index: 'misskey',
body: index
});
2018-07-04 13:13:05 +02:00
});
}
2016-12-28 23:49:51 +01:00
export default client;