hamok/examples/src/queue-push-pop-example.ts

Lightweight Distributed Object Storage on RAFT consensus algorithm

import { Hamok, setHamokLogLevel } from 'hamok';
import * as pino from 'pino';
import { HamokMessageHub } from './utils/HamokMessageHub';

const logger = pino.pino({
	name: 'queue-push-pop-example',
	level: 'debug',
});

export async function run() {

	const server_1 = new Hamok();
	const server_2 = new Hamok();
	const messageHub = new HamokMessageHub();
	
	messageHub.add(server_1, server_2);

	await Promise.all([
		server_1.join(),
		server_2.join(),
	]);

	const queue_1 = server_1.createQueue<number>({
		queueId: 'my-distributed-queue',
	});
	const queue_2 = server_2.createQueue<number>({
		queueId: 'my-distributed-queue',
	});
	// const storage_3 = server_3.createReplicatedStorage<string, number>({
	// 	storageId: 'my-replicated-storage',
	// });

	const value_1 = Math.random();
	const value_2 = Math.random();

	logger.debug(`Pushing values into distributed queue. Candidates to push from server1: ${value_1}, server2: ${value_2}`);

	await Promise.all([
		queue_1.push(value_1),
		queue_2.push(value_2),
	]).catch(err => {
		logger.error('Error pushing values into distributed queue: %s', `${err}`);
		throw err;
	});

	const [ popppedValue_1, popppedValue_2 ] = await Promise.all([
		queue_1.pop(),
		queue_2.pop(),
	]).catch(err => {
		logger.error('Error popping values into distributed queue: %s', `${err}`);
		throw err;
	});

	logger.debug('First round of popped values from distributed queue by server_1: %s, server_2: %s', popppedValue_1 ?? 'empty', popppedValue_2 ?? 'empty');

	const [ popppedValue_3, popppedValue_4 ] = await Promise.all([
		queue_1.pop(),
		queue_2.pop(),
	]).catch(err => {
		logger.error('Error popping values into distributed queue: %s', `${err}`);
		throw err;
	});

	logger.debug('Second round of popped values from distributed queue by server_1: %s, server_2: %s', popppedValue_3 ?? 'empty', popppedValue_4 ?? 'empty');

	server_1.close();
	server_2.close();
}

if (require.main === module) {
	logger.info('Running from module file');
	setHamokLogLevel('info');
	run();
}