paricafe/src/db/elasticsearch.ts

53 lines
916 B
TypeScript
Raw Normal View History

2019-04-24 17:46:39 -05:00
import * as elasticsearch from '@elastic/elasticsearch';
2018-04-01 23:15:53 -05:00
import config from '../config';
2016-12-28 16:49:51 -06:00
2018-07-04 06:36:06 -05:00
const index = {
settings: {
analysis: {
analyzer: {
2019-04-24 17:46:39 -05:00
ngram: {
tokenizer: 'ngram'
2018-07-04 06:36:06 -05:00
}
}
}
},
mappings: {
2019-04-24 17:46:39 -05:00
properties: {
text: {
type: 'text',
index: true,
analyzer: 'ngram',
},
userId: {
type: 'keyword',
index: true,
},
userHost: {
type: 'keyword',
index: true,
2018-07-04 06:36:06 -05:00
}
}
}
};
2016-12-28 16:49:51 -06:00
// Init ElasticSearch connection
const client = config.elasticsearch ? new elasticsearch.Client({
2019-04-24 17:46:39 -05:00
node: `http://${config.elasticsearch.host}:${config.elasticsearch.port}`,
pingTimeout: 30000
}) : null;
2016-12-28 16:49:51 -06:00
2018-07-04 06:13:05 -05:00
if (client) {
2018-07-04 06:36:06 -05:00
client.indices.exists({
index: config.elasticsearch.index || 'misskey_note',
2018-07-04 06:36:06 -05:00
}).then(exist => {
2019-04-24 17:46:39 -05:00
if (!exist.body) {
client.indices.create({
index: config.elasticsearch.index || 'misskey_note',
2019-04-24 17:46:39 -05:00
body: index
});
}
2018-07-04 06:13:05 -05:00
});
}
2016-12-28 16:49:51 -06:00
export default client;