1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
/*
* Copyright (c) 2012-2016 Cisco Systems, Inc. All rights reserved.
* Copyright (c) 2012 Los Alamos National Security, LLC. All rights reserved
* Copyright (c) 2015 Intel, Inc. All rights reserved
* Copyright (c) 2020 Amazon.com, Inc. or its affiliates.
* All Rights reserved.
* $COPYRIGHT$
*
* Additional copyrights may follow
*
* $HEADER$
*/
#include "opal_config.h"
#include <stddef.h>
#include <stdio.h>
#include "opal/class/opal_hotel.h"
#include "opal/util/event.h"
static void local_eviction_callback(int fd, short flags, void *arg)
{
opal_hotel_room_eviction_callback_arg_t *eargs = (opal_hotel_room_eviction_callback_arg_t *)
arg;
void *occupant = eargs->hotel->rooms[eargs->room_num].occupant;
/* Remove the occurpant from the room.
Do not change this logic without also changing the same logic
in opal_hotel_checkout() and
opal_hotel_checkout_and_return_occupant(). */
opal_hotel_t *hotel = eargs->hotel;
opal_hotel_room_t *room = &(hotel->rooms[eargs->room_num]);
room->occupant = NULL;
hotel->last_unoccupied_room++;
assert(hotel->last_unoccupied_room < hotel->num_rooms);
hotel->unoccupied_rooms[hotel->last_unoccupied_room] = eargs->room_num;
/* Invoke the user callback to tell them that they were evicted */
hotel->evict_callback_fn(hotel, eargs->room_num, occupant);
}
int opal_hotel_init(opal_hotel_t *h, int num_rooms, opal_event_base_t *evbase,
uint32_t eviction_timeout, int eviction_event_priority,
opal_hotel_eviction_callback_fn_t evict_callback_fn)
{
int i;
/* Bozo check */
if (num_rooms <= 0 || NULL == evict_callback_fn) {
return OPAL_ERR_BAD_PARAM;
}
h->num_rooms = num_rooms;
h->evbase = evbase;
h->eviction_timeout.tv_usec = eviction_timeout % 1000000;
h->eviction_timeout.tv_sec = eviction_timeout / 1000000;
h->evict_callback_fn = evict_callback_fn;
h->rooms = (opal_hotel_room_t *) malloc(num_rooms * sizeof(opal_hotel_room_t));
if (NULL != evict_callback_fn) {
h->eviction_args = (opal_hotel_room_eviction_callback_arg_t *) malloc(
num_rooms * sizeof(opal_hotel_room_eviction_callback_arg_t));
}
h->unoccupied_rooms = (int *) malloc(num_rooms * sizeof(int));
h->last_unoccupied_room = num_rooms - 1;
for (i = 0; i < num_rooms; ++i) {
/* Mark this room as unoccupied */
h->rooms[i].occupant = NULL;
/* Setup this room in the unoccupied index array */
h->unoccupied_rooms[i] = i;
/* Setup the eviction callback args */
h->eviction_args[i].hotel = h;
h->eviction_args[i].room_num = i;
/* Create this room's event (but don't add it) */
if (NULL != h->evbase) {
opal_event_set(h->evbase, &(h->rooms[i].eviction_timer_event), -1, 0,
local_eviction_callback, &(h->eviction_args[i]));
/* Set the priority so it gets serviced properly */
opal_event_set_priority(&(h->rooms[i].eviction_timer_event), eviction_event_priority);
}
}
return OPAL_SUCCESS;
}
static void constructor(opal_hotel_t *h)
{
h->num_rooms = 0;
h->evbase = NULL;
h->eviction_timeout.tv_sec = 0;
h->eviction_timeout.tv_usec = 0;
h->evict_callback_fn = NULL;
h->rooms = NULL;
h->eviction_args = NULL;
h->unoccupied_rooms = NULL;
h->last_unoccupied_room = -1;
}
static void destructor(opal_hotel_t *h)
{
int i;
/* Go through all occupied rooms and destroy their events */
if (NULL != h->evbase) {
for (i = 0; i < h->num_rooms; ++i) {
if (NULL != h->rooms[i].occupant) {
opal_event_del(&(h->rooms[i].eviction_timer_event));
}
}
}
if (NULL != h->rooms) {
free(h->rooms);
}
if (NULL != h->eviction_args) {
free(h->eviction_args);
}
if (NULL != h->unoccupied_rooms) {
free(h->unoccupied_rooms);
}
}
OBJ_CLASS_INSTANCE(opal_hotel_t, opal_object_t, constructor, destructor);
|
