Node.js Rockset Example

gistfile1.js

"use strict";

const API_KEY = "****";
const COLLECTION_NAME = "myCollection";

const rockset = require("rockset").default(
  API_KEY,
  "https://api.rs2.usw2.rockset.com"
);

const sleep = ms => {
  return new Promise(resolve => setTimeout(resolve, ms));
};

const setup = async () => {
  console.log(`Creating collection: ${COLLECTION_NAME}`);
  try {
    await rockset.collections.createCollection("commons", {
      name: COLLECTION_NAME,
      field_mappings: [
        {
          name: "filter",
          input_fields: [
            {
              field_name: "country_code",
              if_missing: "SKIP",
              is_drop: true,
              param: "country_code"
            }
          ],
          output_field: {
            field_name: "country_code",
            on_error: "FAIL",
            value: {
              sql: "if(:country_code = 'US', :country_code, cast('raise exception' as int))"
            }
          }
        }
      ]
    });

    await sleep(1000 * 5);
    console.log(`Collection creation complete.`);
  } catch (e) {
    console.log(`setup error`, e);
  }
};

const tearDown = async () => {
  try {
    console.log(`Deleting collection: ${COLLECTION_NAME}`);
    await rockset.collections.deleteCollection("commons", COLLECTION_NAME);
    await sleep(1000 * 5);
    console.log(`Deleted collection: ${COLLECTION_NAME}`);
  } catch (e) {
    // silently fail if there is no collection to delete for convenience
    if (e.status && e.status === 404) {
      console.log(`Collection did not exist.`);
      return;
    }

    console.log(`Exception from tearDown(): ${e}`);
  }
};

const add = async (data = []) => {
  console.log(`Inserting data...`);
  const f = await rockset.documents.addDocuments("commons", COLLECTION_NAME, {
    data: data
  });

  await sleep(1000 * 5);
  console.log(`Insert complete`);
};

const get = async query => {
  console.log(`Getting data...`);

  const out = await rockset.queries.query({
    sql: {
      query
    }
  });
  return out;
};

(async () => {
  try {
    await tearDown();
    await setup();

    const records = [
      {
        _id: "1",
        country_code: "US"
      },
      {
        _id: "2",
        country_code: "FOO"
      },
      {
        _id: "3",
        country_code: "BAR"
      },
      {
        _id: "4",
        country_code: "US"
      }
    ];

    await add(records);

    const out = await get(`SELECT * FROM commons.${COLLECTION_NAME};`);

    console.log(out.results);

    await tearDown();

    console.log(`Complete.`);
  } catch (e) {
    console.log(e);
  }
})();