2019-04-25 00:46:39 +02:00
|
|
|
import * as elasticsearch from '@elastic/elasticsearch';
|
2018-04-02 06:15:53 +02:00
|
|
|
import config from '../config';
|
2016-12-28 23:49:51 +01:00
|
|
|
|
2018-07-04 13:36:06 +02:00
|
|
|
const index = {
|
|
|
|
settings: {
|
|
|
|
analysis: {
|
|
|
|
analyzer: {
|
2019-04-25 00:46:39 +02:00
|
|
|
ngram: {
|
|
|
|
tokenizer: 'ngram'
|
2018-07-04 13:36:06 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
},
|
|
|
|
mappings: {
|
2019-04-25 00:46:39 +02:00
|
|
|
properties: {
|
|
|
|
text: {
|
|
|
|
type: 'text',
|
|
|
|
index: true,
|
|
|
|
analyzer: 'ngram',
|
|
|
|
},
|
|
|
|
userId: {
|
|
|
|
type: 'keyword',
|
|
|
|
index: true,
|
|
|
|
},
|
|
|
|
userHost: {
|
|
|
|
type: 'keyword',
|
|
|
|
index: true,
|
2018-07-04 13:36:06 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
2016-12-28 23:49:51 +01:00
|
|
|
// Init ElasticSearch connection
|
2019-02-07 13:02:33 +01:00
|
|
|
const client = config.elasticsearch ? new elasticsearch.Client({
|
2019-10-20 17:41:12 +02:00
|
|
|
node: `${config.elasticsearch.ssl ? 'https://' : 'http://'}${config.elasticsearch.host}:${config.elasticsearch.port}`,
|
2020-03-20 06:00:34 +01:00
|
|
|
auth: (config.elasticsearch.user && config.elasticsearch.pass) ? {
|
|
|
|
username: config.elasticsearch.user,
|
|
|
|
password: config.elasticsearch.pass
|
|
|
|
} : undefined,
|
2019-04-25 00:46:39 +02:00
|
|
|
pingTimeout: 30000
|
2019-02-07 13:02:33 +01:00
|
|
|
}) : null;
|
2016-12-28 23:49:51 +01:00
|
|
|
|
2018-07-04 13:13:05 +02:00
|
|
|
if (client) {
|
2018-07-04 13:36:06 +02:00
|
|
|
client.indices.exists({
|
2019-08-09 06:04:35 +02:00
|
|
|
index: config.elasticsearch.index || 'misskey_note',
|
2018-07-04 13:36:06 +02:00
|
|
|
}).then(exist => {
|
2019-04-25 00:46:39 +02:00
|
|
|
if (!exist.body) {
|
|
|
|
client.indices.create({
|
2019-08-09 06:04:35 +02:00
|
|
|
index: config.elasticsearch.index || 'misskey_note',
|
2019-04-25 00:46:39 +02:00
|
|
|
body: index
|
|
|
|
});
|
|
|
|
}
|
2018-07-04 13:13:05 +02:00
|
|
|
});
|
|
|
|
}
|
2016-12-28 23:49:51 +01:00
|
|
|
|
|
|
|
export default client;
|