import { LocalClient } from '../client/LocalClient';
import { Context } from '../client/Context';
import { serialize } from '../common/SerializeFunction';

// Every calculate API is async so we need async/await:
async function main() {
  // Create a local instance of dcf:
  const client = new LocalClient();
  // Wait creation process completed:
  await client.init();
  // Create API context:
  const dcc = new Context(client);

  // Create a new rdd.
  const rdd = dcc.parallelize([1, 2, 3, 4, 5, 6, 7, 8, 9, 10]).cache();

  console.log(await rdd.sort().collect());
  console.log(
    await rdd
      .map(v => v % 3)
      .sort()
      .collect(),
  );

  const arr = await dcc
    .range(0, 1000000)
    .sort(false)
    .collect();

  for (let i = 0; i < arr.length - 1; i++) {
    if (arr[i] < arr[i + 1]) {
      console.log('Error!');
    }
  }

  console.log(
    await dcc
      .range(0, 1000000)
      .sort(false)
      .glom()
      .map(v => v.length)
      .collect(),
  );

  console.log(
    await dcc
      .range(0, 1000000)
      .sortBy(v => {
        for (let j = 2; j < 1000; j++) {
          if (v % j === 0) {
            return j;
          }
        }
        return v;
      }, false)
      .take(10),
  );

  await rdd.unpersist();
  // Shutdown
  client.dispose();
}
main().catch(e => console.error(e.stack));