/*
 * Copyright (c) 2023 Amazon.com, Inc. or its affiliates. All rights reserved.
 */

#include "config.h"

#include <assert.h>

#include "nccl_ofi.h"
#include "nccl_ofi_deque.h"

int nccl_ofi_deque_init(nccl_ofi_deque_t **deque_p)
{
	nccl_ofi_deque_t *deque = malloc(sizeof(nccl_ofi_deque_t));

	if (deque == NULL) {
		NCCL_OFI_WARN("Failed to allocate deque");
		return -ENOMEM;
	}

	deque->head.prev = &deque->head;
	deque->head.next = &deque->head;

	int ret = pthread_mutex_init(&deque->lock, NULL);
	if (ret != 0) {
		NCCL_OFI_WARN("Failed to initialize deque mutex.");
		free(deque);
		return -ret;
	}

	assert(deque_p);
	*deque_p = deque;

	return 0;
}

int nccl_ofi_deque_finalize(nccl_ofi_deque_t *deque)
{
	assert(deque);

	/* Since user allocates all memory used for deque elements, we don't need to
	   deallocate any entries here. :D */
	int ret = pthread_mutex_destroy(&deque->lock);
	if (ret != 0) {
		NCCL_OFI_WARN("Failed to destroy deque mutex.");
		return -ret;
	}

	free(deque);
	return 0;
}