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