]> rtime.felk.cvut.cz Git - frescor/frsh.git/blobdiff - fres/cbroker/fcb.c
Progress on transaction negotiation
[frescor/frsh.git] / fres / cbroker / fcb.c
index 2e825bcec366f63f9eab02555bad886580cfe848..7fc217aa4bc07e867713931634c803876804f851 100644 (file)
@@ -53,6 +53,7 @@
  * 
  * 
  */
+#define WITH_C99               /* For ul_gsa.h iterators */
 #include <semaphore.h>
 #include <getopt.h>
 #include <forb.h>
@@ -63,6 +64,7 @@
 #include <errno.h>
 #include <stdio.h>
 #include <ul_gavlcust.h>
+#include <ul_gsacust.h>
 #include <string.h>
 #include <ul_log.h>
 #include <ul_logreg.h>
 #include <forb/proto_inet.h>
 #endif
 
+#define COMPILE_TIME_ASSERT(cond, msg) \
+       typedef char msg[(cond) ? 1 : -1]
+
+
 UL_LOG_CUST(ulogd_fcb);
 ul_log_domain_t ulogd_fcb = {UL_LOGL_MSG, "main"};
 UL_LOGREG_SINGLE_DOMAIN_INIT_FUNCTION(init_ulogd_fcb, ulogd_fcb);
@@ -140,6 +146,7 @@ struct fcb_contract {
  * Contract broker data
  */
 struct fcb {
+       fres_contract_id_t contract_counter;
        gavl_cust_root_field_t resources; /**< Registered resources */
        gavl_cust_root_field_t contracts; /**< Contracts negotiated by this FCB */
 };
@@ -265,6 +272,28 @@ GAVL_CUST_NODE_INT_IMP(fcb_contract         /* cust_prefix */,             \
 #include "fcb_contract_gavl.inc"
 #endif
 
+struct res_array {
+       gsa_array_field_t array;
+};
+
+GSA_CUST_DEC(res_array         /* cust_prefix */,
+            struct res_array   /* cust_array_t */,
+            struct resource    /* cust_item_t */,
+            struct res_key     /* cust_key_t */,
+            array              /* cust_array_field */,
+            key                /* cust_item_key */,
+            res_key_cmp        /* cust_cmp_fnc */);
+
+GSA_CUST_IMP(res_array         /* cust_prefix */,
+            struct res_array   /* cust_array_t */,
+            struct resource    /* cust_item_t */,
+            struct res_key     /* cust_key_t */,
+            array              /* cust_array_field */,
+            key                /* cust_item_key */,
+            res_key_cmp        /* cust_cmp_fnc */,
+            true               /* cust_ins_fl */);
+
+
 
 #define o2fcb(o) (struct fcb*)forb_instance_data(o)
 
@@ -319,7 +348,9 @@ prepare_fcb_contracts(struct fcb *fcb, struct fcb_contract *fcb_contracts[],
 
                if (fres_contract_id_is_empty(&c->id)) {
                        /* Normal negotiation request */
-                       forb_uuid_generate((forb_uuid_t *)&c->id);
+                       COMPILE_TIME_ASSERT(sizeof(c->id) == sizeof(fcb->contract_counter),
+                                           wrong_size_of_contract_id);
+                       c->id = ++fcb->contract_counter;
                        fc = fcb_contract_new(&c->id);
                        if (!fc)
                                return errno;
@@ -439,6 +470,57 @@ prepare_reservation_list(struct resource *resource,
        }
 }
 
+static int
+reserve_resource(struct resource *resource)
+{
+       int ret, i;
+       fres_contract_ptr_seq contracts;
+       struct fres_contract *c;
+       struct fcb_contract *fc;
+       CORBA_Environment ev;
+       
+       if (!forb_sequence_alloc_buf(&contracts, resource->rl.length)) {
+               ret = errno;
+               goto err;
+       }
+       forb_sequence_length(&contracts) = resource->rl.length;
+
+       i=0;
+       /* Prepare FORB sequence  */
+       ul_list_for_each(reservation_list, &resource->rl, fc) {
+               c = NULL;
+               if (fc->to_be_reserved_contract)
+                       /* Contract without spare capacity */
+                       c = fc->to_be_reserved_contract;
+               else if (fc->requested_contract)
+                       c = fc->requested_contract;
+               assert(c);
+               forb_sequence_elem(&contracts, i++) = c;
+       }
+       
+       /* Reserve contract */
+       ret = fres_resource_manager_reserve_contracts(resource->mng, &contracts, &ev);
+       if (forb_exception_occurred(&ev)) {
+               ret = fres_forbex2err(&ev);
+               ul_logerr("FORB exception when reserving contracts\n");
+               goto err_free;
+       }
+       if (ret < 0) {
+               ul_logerr("Contract reservation error %d\n", ret);
+               ret = FRES_ERR_ADMISSION_TEST;
+               goto err_free;
+       } else if (ret == 0)
+               ret = FRSH_NO_ERROR;
+       else if (ret == 1)
+               ret = FRSH_ERR_CONTRACT_REJECTED;
+       else
+               assert(false);
+err_free:
+       forb_sequence_free_buf(&contracts, forb_no_destructor);
+err:
+       return ret;
+}
+
 /** 
  * 
  * 
@@ -451,17 +533,10 @@ static int
 rebalance_spare_capacity_and_reserve(struct resource *resource)
 {
        int ret;
-       unsigned i;
        struct reservation_list *rl = &resource->rl;
        struct fcb_contract *fc;
        fres_block_spare_capacity *s;
 
-       fres_contract_ptr_seq contracts;
-       if (!forb_sequence_alloc_buf(&contracts, rl->length)) {
-               return errno;
-       }
-       contracts._length = rl->length;
-       i=0;
        /* Initialize optimization */
        ul_list_for_each(reservation_list, rl, fc) {
                fc->to_be_reserved_contract =
@@ -470,9 +545,6 @@ rebalance_spare_capacity_and_reserve(struct resource *resource)
                        NULL;
                assert(fc->to_be_reserved_contract != NULL);
 
-               forb_sequence_elem(&contracts, i) = fc->to_be_reserved_contract;
-               i++;
-               
                s = fres_contract_get_spare_capacity(fc->to_be_reserved_contract);
                if (s && s->granularity == FRSH_GR_DISCRETE) {
                        fc->sc_variant.initial = s->variants._length - 1;
@@ -516,21 +588,15 @@ rebalance_spare_capacity_and_reserve(struct resource *resource)
                }
 
                if (criterion > best_criterion) {
-                       CORBA_Environment ev;
-                       /* Reserve contract */
-                       ret = fres_resource_manager_reserve_contracts(resource->mng, &contracts, &ev);
-                       if (forb_exception_occurred(&ev)) {
-                               ret = fres_forbex2err(&ev);
-                               ul_logerr("FORB exception when reserving contracts\n");
-                               goto err;
-                       }
-                       if (ret < 0) {
-                               ul_logerr("Contract reservation error %d\n", ret);
-                               ret = FRES_ERR_ADMISSION_TEST;
-                               goto err;
-                       }
-                       if (ret == 0) { /* negotiation succeeded */
+                       ret = reserve_resource(resource);
+                       switch (ret) {
+                       case FRSH_NO_ERROR:
                                best_criterion = criterion;
+                               break;
+                       case FRSH_ERR_CONTRACT_REJECTED:
+                               break;
+                       default:
+                               goto err;
                        }
                }
        } while (!all_combinations_tried);
@@ -553,7 +619,6 @@ err:
                fres_contract_destroy(fc->to_be_reserved_contract);
                fc->to_be_reserved_contract = NULL;
        }
-       forb_sequence_free_buf(&contracts, forb_no_destructor);
        return ret;
 }
 
@@ -1002,40 +1067,42 @@ transaction_has_spare_capacity(const fres_transaction_t* transaction)
 
 static int
 transaction_get_resources(struct fcb *fcb, struct fcb_contract *fc[],
-                         int num, struct resource **resources[])
+                         int num, struct res_array *res_array)
 {
-       int i, j, r, alloc;
-       struct resource **res;
+       int i;
        struct resource *resource;
        struct res_key key;
+       int ret;
 
-       alloc = 10;
-       res = malloc(alloc*sizeof(*res));
-       if (!res) 
-               return errno;
-       res[0] = NULL;
-       for (r = 0, i = 0; i < num; i++) {
+       res_array_init_array_field(res_array);
+       for (i = 0; i < num; i++) {
                get_fc_res_key(fc[i], &key);
                resource = fcb_resource_find(fcb, &key);
-               /* TODO: Check whether to use GSA for resources. */
+               ret = res_array_insert(res_array, resource);
+               if (ret == -1) {
+                       ret = FRSH_ERR_INTERNAL_ERROR;
+                       goto err;
+               }
        }
-       res[r] = NULL;
-       return 0;
+       ret = 0;
+err:
+       return ret;
 }
                                 
-
-
 CORBA_long
 negotiate_transaction(fres_contract_broker _obj,
                      const fres_transaction_t* transaction,
                      CORBA_Environment *ev)
 {
        struct fcb *fcb = o2fcb(_obj);
-       struct fcb_contract **fcb_contracts;
-       const fres_contract_ptr_seq* contracts = &transaction->contracts;
-       int num = contracts->_length;
-       int ret;
+       struct fcb_contract **fcb_contracts, *fc;
+       const fres_contract_ptr_seq* user_contracts = &transaction->contracts;
+       int i, ret, num = user_contracts->_length;
+       struct res_array res_array;
+       struct res_key key;
+       struct resource *resource;
 
+       ul_logmsg("Negotiating transaction of %d contracts\n", num);
        if (transaction_has_spare_capacity(transaction)) {
                ret =  FRES_ERR_SPARE_CAPACITY_NOT_SUPPORTED;
                goto err;
@@ -1049,13 +1116,48 @@ negotiate_transaction(fres_contract_broker _obj,
        memset(fcb_contracts, 0, sizeof(*fcb_contracts)*num);
 
        ret = prepare_fcb_contracts(fcb, fcb_contracts,
-                                   contracts->_buffer, num);
+                                   user_contracts->_buffer, num);
        if (ret)
                goto err_free_fcb_contracts;
 
-       //ret = transaction_get_resources(fcb_contracts, num, &resources);
+       ret = transaction_get_resources(fcb, fcb_contracts, num, &res_array);
+
+       gsa_cust_for_each(res_array, &res_array, resource) {
+               reservation_list_init_head(&resource->rl);
+               resource->rl.length = 0;
+               for (i = 0; i < num; i++) {
+                       fc = fcb_contracts[i];
+                       get_fc_res_key(fc, &key);
+                       if (res_key_cmp(&key, &resource->key) == 0) {
+                               reservation_list_insert(&resource->rl, fc);
+                               resource->rl.length++;
+                       }
+               }
+       }
+       gsa_cust_for_each(res_array, &res_array, resource) {
+               ret = reserve_resource(resource);
+               if (ret) {
+                       ul_logerr("Reservation failed\n");
+                       goto err_cancel_reservation;
+               }
+       }
+       gsa_cust_for_each(res_array, &res_array, resource) {
+               ret = commit_resource(resource);
+               if (ret) {
+                       ul_logerr("Commit failed\n");
+                       goto err_cancel_reservation;
+               }
+       }
        
-       return FRSH_ERR_NOT_IMPLEMENTED;
+       return 0;
+err_cancel_reservation:
+       gsa_cust_for_each(res_array, &res_array, resource) {
+               cancel_reservations(resource);
+       }
+err_free_res_contracts:
+       forb_sequence_free_buf(&res_contracts, forb_no_destructor);
+err_free_resources:
+       res_array_delete_all(&res_array);
 err_free_fcb_contracts:
        free_fcb_contracts(fcb_contracts, num);
 err:
@@ -1202,6 +1304,7 @@ int main(int argc, char *argv[])
                }
        }
 
+       memset(&fcb_data, 0, sizeof(fcb_data));
        fosa_clock_get_time(CLOCK_REALTIME, &start_time);
 
        if (opt_daemon)