javascript 节点基本操作中的弹性搜索

Posted 2021-05-03

const elasticsearch = require('elasticsearch');
const LineByLineReader = require('line-by-line');

const client = new elasticsearch.Client({
  host: 'localhost:9200'
});

const index = 'dota';
const type = 'heroes';
const doc = {
  'hero': 'drow ranger', 
  'id': 1,
  'type': 'agility',
  'role': 'carry'
}

const docs = [{
  'hero': 'dazzle',
  'id': 2,
  'type': 'int',
  'role': 'support'
}, {
  'hero': 'warlock',
  'id': 3,
  'type': 'int',
  'role': 'support'
}];

client.ping({
  // ping usually has a 3000ms timeout
  requestTimeout: 1000
}, function (error) {
  if (error) {
    console.trace('elasticsearch cluster is down!');
  } else {
    console.log('All is well');
    insert(index, type, doc).then(console.log).catch(console.log);
    del(index, type, doc).then(console.log).catch(console.log)
    search(index, type, 'drow' ).then(console.log).catch(console.log);
    bulkInsert(index, type, docs).then(console.log).catch(console.log);
    bulkInsert(index, type, docs).then(console.log).catch(console.log);
  }
});

function delay(timer) {
  return new Promise((resolve) => {
    setTimeout(() => {
      resolve();
    }, timer);
  });
}

async function insert(index, type, doc) {
  return await client.index({
    index, 
    type, 
    id: doc.id, 
    body: doc,
  });
}

async function del(index, type, doc) {
  return await client.delete({
    index,
    type,
    id: doc.id,
  });
}

async function search(index, type, searchTerm) {
  return await client.search({
    index, 
    type,
    body: {
      query: {
        match: {
          body: searchTerm,
        },
      },
    },
  });
}

async function bulkInsert(index, type, docs) {
  const bulk = [];
  docs.forEach(doc => {
    bulk.push(
      { index: { _index: index, _type: type, _id: doc.id || doc.quizId } },
      doc,
    );
  });
  console.log(bulk);
  return await client.bulk({ body: bulk});
}

async function bulkDelete(index, type, docs) {
  const bulk = [];
  docs.forEach(doc => {
    bulk.push(
      { delete: { _index: index, _type: type, _id: doc.id || doc.quizId } }    
    );
  });
  return await client.bulk({ body: bulk});
}

async function bulkUpdate(index, type, docs) {
  const bulk = [];
  docs.forEach(doc => {
    bulk.push(
      { update: { _index: index, _type: type, _id: doc.id || doc.quizId } }, 
      {doc}
    );
  });
  return await client.bulk({ body: bulk});
}