2016-12-28 16:49:51 -06:00
|
|
|
import * as elasticsearch from '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: {
|
2018-09-05 02:48:59 -05:00
|
|
|
normalizer: {
|
|
|
|
lowercase_normalizer: {
|
|
|
|
type: 'custom',
|
|
|
|
filter: ['lowercase']
|
|
|
|
}
|
|
|
|
},
|
2018-07-04 06:36:06 -05: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 02:48:59 -05:00
|
|
|
analyzer: 'bigram',
|
|
|
|
normalizer: 'lowercase_normalizer'
|
2018-07-04 06:36:06 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
2016-12-28 16:49:51 -06:00
|
|
|
// Init ElasticSearch connection
|
2018-07-04 06:13:05 -05:00
|
|
|
const client = config.elasticsearch ? new elasticsearch.Client({
|
2016-12-28 16:49:51 -06:00
|
|
|
host: `${config.elasticsearch.host}:${config.elasticsearch.port}`
|
2018-07-04 06:13:05 -05:00
|
|
|
}) : null;
|
2016-12-28 16:49:51 -06:00
|
|
|
|
2018-07-04 06:13:05 -05:00
|
|
|
if (client) {
|
|
|
|
// Send a HEAD request
|
|
|
|
client.ping({
|
|
|
|
// Ping usually has a 3000ms timeout
|
|
|
|
requestTimeout: 30000
|
|
|
|
}, error => {
|
|
|
|
if (error) {
|
|
|
|
console.error('elasticsearch is down!');
|
|
|
|
} else {
|
|
|
|
console.log('elasticsearch is available!');
|
|
|
|
}
|
|
|
|
});
|
2018-07-04 06:02:45 -05:00
|
|
|
|
2018-07-04 06:36:06 -05:00
|
|
|
client.indices.exists({
|
|
|
|
index: 'misskey'
|
|
|
|
}).then(exist => {
|
|
|
|
if (exist) return;
|
|
|
|
|
|
|
|
client.indices.create({
|
|
|
|
index: 'misskey',
|
|
|
|
body: index
|
|
|
|
});
|
2018-07-04 06:13:05 -05:00
|
|
|
});
|
|
|
|
}
|
2016-12-28 16:49:51 -06:00
|
|
|
|
|
|
|
export default client;
|