Skip to content
Snippets Groups Projects
elasticsearch.ts 1.1 KiB
Newer Older
syuilo's avatar
syuilo committed
import * as elasticsearch from '@elastic/elasticsearch';
Akihiko Odaki's avatar
Akihiko Odaki committed
import config from '../config';
syuilo's avatar
syuilo committed
const index = {
	settings: {
		analysis: {
			analyzer: {
syuilo's avatar
syuilo committed
				ngram: {
					tokenizer: 'ngram'
syuilo's avatar
syuilo committed
				}
			}
		}
	},
	mappings: {
syuilo's avatar
syuilo committed
		properties: {
			text: {
				type: 'text',
				index: true,
				analyzer: 'ngram',
			},
			userId: {
				type: 'keyword',
				index: true,
			},
			userHost: {
				type: 'keyword',
				index: true,
syuilo's avatar
syuilo committed
			}
		}
	}
};

// Init ElasticSearch connection
const client = config.elasticsearch ? new elasticsearch.Client({
	node: `${config.elasticsearch.ssl ? 'https://' : 'http://'}${config.elasticsearch.host}:${config.elasticsearch.port}`,
	auth: (config.elasticsearch.user && config.elasticsearch.pass) ? {
		username: config.elasticsearch.user,
		password: config.elasticsearch.pass
	} : undefined,
syuilo's avatar
syuilo committed
	pingTimeout: 30000
syuilo's avatar
syuilo committed
if (client) {
syuilo's avatar
syuilo committed
	client.indices.exists({
		index: config.elasticsearch.index || 'misskey_note',
syuilo's avatar
syuilo committed
	}).then(exist => {
syuilo's avatar
syuilo committed
		if (!exist.body) {
			client.indices.create({
				index: config.elasticsearch.index || 'misskey_note',
syuilo's avatar
syuilo committed
				body: index
			});
		}
syuilo's avatar
syuilo committed
	});
}

export default client;