e2net/iceshrimp-legacy
2
0
Fork 0
Code Issues Pull requests Packages Projects Releases Activity

wip

Browse source
This commit is contained in:
syuilo 2018-07-04 20:13:05 +09:00
parent 9d49636cd1
commit 7293baa1f9
2 changed files with 42 additions and 41 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
src/db/elasticsearch.ts
View file

@ -2,23 +2,24 @@ import * as elasticsearch from 'elasticsearch';
import config from '../config'; import config from '../config';
// Init ElasticSearch connection // Init ElasticSearch connection
const client = new elasticsearch.Client({ const client = config.elasticsearch ? new elasticsearch.Client({
host: `${config.elasticsearch.host}:${config.elasticsearch.port}` host: `${config.elasticsearch.host}:${config.elasticsearch.port}`
}); }) : null;
// Send a HEAD request if (client) {
client.ping({ // Send a HEAD request
client.ping({
// Ping usually has a 3000ms timeout // Ping usually has a 3000ms timeout
requestTimeout: 30000 requestTimeout: 30000
}, error => { }, error => {
if (error) { if (error) {
console.error('elasticsearch is down!'); console.error('elasticsearch is down!');
} else { } else {
console.log('elasticsearch is available!'); console.log('elasticsearch is available!');
} }
}); });
client.indices.create({ client.indices.create({
index: 'misskey', index: 'misskey',
body: { body: {
settings: { settings: {
@ -49,6 +50,7 @@ client.indices.create({
} }
} }
} }
}); });
}
export default client; export default client;

3
src/services/note/create.ts
View file

@ -1,3 +1,4 @@
import es from '../../db/elasticsearch';
import Note, { pack, INote } from '../../models/note'; import Note, { pack, INote } from '../../models/note';
import User, { isLocalUser, IUser, isRemoteUser, IRemoteUser, ILocalUser } from '../../models/user'; import User, { isLocalUser, IUser, isRemoteUser, IRemoteUser, ILocalUser } from '../../models/user';
import stream, { publishLocalTimelineStream, publishGlobalTimelineStream, publishUserListStream } from '../../publishers/stream'; import stream, { publishLocalTimelineStream, publishGlobalTimelineStream, publishUserListStream } from '../../publishers/stream';
@ -431,8 +432,6 @@ export default async (user: IUser, data: {
// Register to search database // Register to search database
if (note.text && config.elasticsearch) { if (note.text && config.elasticsearch) {
const es = require('../../../db/elasticsearch');
es.index({ es.index({
index: 'misskey', index: 'misskey',
type: 'note', type: 'note',