tuple_extensions.c 4.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199
  1. #include "presentity.h"
  2. #include "pa_mod.h"
  3. #include "tuple_extensions.h"
  4. #include <cds/logger.h>
  5. /* DB manipulation */
  6. static int db_add_tuple_extension(presentity_t *p, presence_tuple_t *t, extension_element_t *n, int is_status_extension)
  7. {
  8. db_key_t cols[20];
  9. db_val_t vals[20];
  10. int n_updates = 0;
  11. if (!use_db) return 0;
  12. /* set data */
  13. cols[n_updates] = col_pres_id;
  14. vals[n_updates].type = DB_STR;
  15. vals[n_updates].nul = 0;
  16. vals[n_updates].val.str_val = p->pres_id;
  17. n_updates++;
  18. cols[n_updates] = col_tupleid;
  19. vals[n_updates].type = DB_STR;
  20. vals[n_updates].nul = 0;
  21. vals[n_updates].val.str_val = t->data.id;
  22. n_updates++;
  23. cols[n_updates] = col_element;
  24. vals[n_updates].type = DB_BLOB;
  25. vals[n_updates].nul = 0;
  26. vals[n_updates].val.blob_val = n->element;
  27. n_updates++;
  28. cols[n_updates] = col_status_extension;
  29. vals[n_updates].type = DB_INT;
  30. vals[n_updates].nul = 0;
  31. vals[n_updates].val.int_val = is_status_extension;
  32. n_updates++;
  33. /* run update */
  34. if (pa_dbf.use_table(pa_db, tuple_extensions_table) < 0) {
  35. LOG(L_ERR, "db_add_pres_note: Error in use_table\n");
  36. return -1;
  37. }
  38. if (pa_dbf.insert(pa_db, cols, vals, n_updates) < 0) {
  39. LOG(L_ERR, "db_add_pres_note: Can't insert record\n");
  40. return -1;
  41. }
  42. return 0;
  43. }
  44. int db_add_tuple_extensions(presentity_t *p, presence_tuple_t *t)
  45. {
  46. extension_element_t *n;
  47. n = t->data.first_unknown_element;
  48. while (n) {
  49. db_add_tuple_extension(p, t, n, 0);
  50. n = n->next;
  51. }
  52. n = t->data.status.first_unknown_element;
  53. while (n) {
  54. db_add_tuple_extension(p, t, n, 1);
  55. n = n->next;
  56. }
  57. return 0;
  58. }
  59. int db_remove_tuple_extensions(presentity_t *p, presence_tuple_t *t)
  60. {
  61. db_key_t keys[] = { col_pres_id, col_tupleid };
  62. db_op_t ops[] = { OP_EQ, OP_EQ };
  63. db_val_t k_vals[] = {
  64. { DB_STR, 0, { .str_val = p->pres_id } },
  65. { DB_STR, 0, { .str_val = t->data.id } }
  66. };
  67. if (!use_db) return 0;
  68. if (pa_dbf.use_table(pa_db, tuple_extensions_table) < 0) {
  69. LOG(L_ERR, "db_remove_tuple_extensions: Error in use_table\n");
  70. return -1;
  71. }
  72. if (pa_dbf.delete(pa_db, keys, ops, k_vals, 2) < 0) {
  73. LOG(L_ERR, "db_remove_tuple_extensions: Can't delete record\n");
  74. return -1;
  75. }
  76. return 0;
  77. }
  78. int db_update_tuple_extensions(presentity_t *p, presence_tuple_t *t)
  79. {
  80. db_remove_tuple_extensions(p, t);
  81. db_add_tuple_extensions(p, t);
  82. return 0;
  83. }
  84. int db_read_tuple_extensions(presentity_t *p, presence_tuple_t *t, db_con_t* db)
  85. {
  86. db_key_t keys[] = { col_pres_id, col_tupleid };
  87. db_op_t ops[] = { OP_EQ, OP_EQ };
  88. db_val_t k_vals[] = {
  89. { DB_STR, 0, { .str_val = p->pres_id } },
  90. { DB_STR, 0, { .str_val = t->data.id } }
  91. };
  92. int i;
  93. int r = 0;
  94. db_res_t *res = NULL;
  95. db_key_t result_cols[] = { col_element, col_status_extension };
  96. if (!use_db) return 0;
  97. if (pa_dbf.use_table(db, tuple_extensions_table) < 0) {
  98. LOG(L_ERR, "db_read_tuple_extensions: Error in use_table\n");
  99. return -1;
  100. }
  101. if (pa_dbf.query (db, keys, ops, k_vals,
  102. result_cols, 2, sizeof(result_cols) / sizeof(db_key_t),
  103. 0, &res) < 0) {
  104. LOG(L_ERR, "db_read_tuple_extensions(): Error while querying extensions\n");
  105. return -1;
  106. }
  107. if (!res) return 0; /* ? */
  108. for (i = 0; i < res->n; i++) {
  109. extension_element_t *n = NULL;
  110. db_row_t *row = &res->rows[i];
  111. db_val_t *row_vals = ROW_VALUES(row);
  112. str element = STR_NULL;
  113. int status_extension = 0;
  114. #define get_str_val(i,dst) do{if(!row_vals[i].nul){dst.s=(char*)row_vals[i].val.string_val;dst.len=strlen(dst.s);}}while(0)
  115. #define get_int_val(rvi,dst) do{if(!rvi.nul){dst=rvi.val.int_val;}else dst=0;}while(0)
  116. get_str_val(0, element);
  117. get_int_val(row_vals[1], status_extension);
  118. #undef get_str_val
  119. #undef get_int_val
  120. n = create_extension_element(&element);
  121. if (n) add_tuple_extension_no_wb(t, n, status_extension);
  122. }
  123. pa_dbf.free_result(db, res);
  124. return r;
  125. }
  126. /* in memory operations */
  127. void add_tuple_extension_no_wb(presence_tuple_t *t, extension_element_t *n, int is_status_extension)
  128. {
  129. if (is_status_extension)
  130. DOUBLE_LINKED_LIST_ADD(t->data.status.first_unknown_element,
  131. t->data.status.last_unknown_element, n);
  132. else DOUBLE_LINKED_LIST_ADD(t->data.first_unknown_element,
  133. t->data.last_unknown_element, n);
  134. }
  135. void free_tuple_extensions(presence_tuple_t *t)
  136. {
  137. extension_element_t *n, *nn;
  138. /* remove all extensions for this tuple */
  139. n = t->data.first_unknown_element;
  140. while (n) {
  141. nn = n->next;
  142. free_extension_element(n);
  143. n = nn;
  144. }
  145. /* remove all status extensions for this tuple */
  146. n = t->data.status.first_unknown_element;
  147. while (n) {
  148. nn = n->next;
  149. free_extension_element(n);
  150. n = nn;
  151. }
  152. t->data.first_unknown_element = NULL;
  153. t->data.last_unknown_element = NULL;
  154. t->data.status.first_unknown_element = NULL;
  155. t->data.status.last_unknown_element = NULL;
  156. }