| 20 |
#include "log.h" |
#include "log.h" |
| 21 |
#include <stdlib.h> |
#include <stdlib.h> |
| 22 |
#include <string.h> |
#include <string.h> |
| 23 |
|
#include <sys/param.h> |
| 24 |
|
|
| 25 |
ARTICLE_FAVOR BBS_article_favor; |
ARTICLE_FAVOR BBS_article_favor; |
| 26 |
|
|
| 42 |
if (uid == 0) |
if (uid == 0) |
| 43 |
{ |
{ |
| 44 |
p_favor->aid_base_cnt = 0; |
p_favor->aid_base_cnt = 0; |
|
p_favor->aid_base = NULL; |
|
| 45 |
|
|
| 46 |
if (!keep_inc) |
if (!keep_inc) |
| 47 |
{ |
{ |
| 66 |
log_error("Query article_favorite error: %s\n", mysql_error(db)); |
log_error("Query article_favorite error: %s\n", mysql_error(db)); |
| 67 |
return -3; |
return -3; |
| 68 |
} |
} |
| 69 |
if ((rs = mysql_store_result(db)) == NULL) |
if ((rs = mysql_use_result(db)) == NULL) |
| 70 |
{ |
{ |
| 71 |
log_error("Get article_favorite data failed\n"); |
log_error("Get article_favorite data failed\n"); |
| 72 |
return -3; |
return -3; |
| 73 |
} |
} |
| 74 |
|
|
| 75 |
p_favor->aid_base_cnt = 0; |
p_favor->aid_base_cnt = 0; |
|
p_favor->aid_base = malloc(sizeof(int32_t) * mysql_num_rows(rs)); |
|
|
if (p_favor->aid_base == NULL) |
|
|
{ |
|
|
log_error("malloc(INT32 * %d) error: OOM\n", mysql_num_rows(rs)); |
|
|
mysql_free_result(rs); |
|
|
mysql_close(db); |
|
|
return -4; |
|
|
} |
|
| 76 |
|
|
| 77 |
while ((row = mysql_fetch_row(rs))) |
while ((row = mysql_fetch_row(rs))) |
| 78 |
{ |
{ |
| 79 |
p_favor->aid_base[(p_favor->aid_base_cnt)++] = atoi(row[0]); |
p_favor->aid_base[p_favor->aid_base_cnt] = atoi(row[0]); |
| 80 |
|
(p_favor->aid_base_cnt)++; |
| 81 |
|
if (p_favor->aid_base_cnt >= MAX_FAVOR_AID_BASE_CNT) |
| 82 |
|
{ |
| 83 |
|
log_error("Too many article_favorite records for uid=%d\n", |
| 84 |
|
uid); |
| 85 |
|
break; |
| 86 |
|
} |
| 87 |
} |
} |
| 88 |
mysql_free_result(rs); |
mysql_free_result(rs); |
| 89 |
|
|
| 107 |
return -1; |
return -1; |
| 108 |
} |
} |
| 109 |
|
|
| 110 |
if (p_favor->aid_base != NULL) |
p_favor->aid_base_cnt = 0; |
|
{ |
|
|
free(p_favor->aid_base); |
|
|
p_favor->aid_base = NULL; |
|
|
p_favor->aid_base_cnt = 0; |
|
|
} |
|
| 111 |
|
|
| 112 |
return 0; |
return 0; |
| 113 |
} |
} |
| 148 |
"DELETE FROM article_favorite WHERE UID = %d AND AID IN (", |
"DELETE FROM article_favorite WHERE UID = %d AND AID IN (", |
| 149 |
p_favor->uid); |
p_favor->uid); |
| 150 |
|
|
| 151 |
for (i = 0, j = 0; i < p_favor->aid_base_cnt && j < p_favor->aid_inc_cnt;) |
for (i = 0, j = 0; j < p_favor->aid_inc_cnt;) |
| 152 |
{ |
{ |
| 153 |
if (p_favor->aid_base[i] == p_favor->aid_inc[j]) // XOR - delete record |
if (i < p_favor->aid_base_cnt && p_favor->aid_base[i] == p_favor->aid_inc[j]) // XOR - delete record |
| 154 |
{ |
{ |
| 155 |
snprintf(tuple_tmp, sizeof(tuple_tmp), "%d, ", p_favor->aid_inc[j]); |
snprintf(tuple_tmp, sizeof(tuple_tmp), "%d, ", p_favor->aid_inc[j]); |
| 156 |
strncat(sql_del, tuple_tmp, sizeof(sql_del) - 1 - strnlen(sql_del, sizeof(sql_del))); |
strncat(sql_del, tuple_tmp, sizeof(sql_del) - 1 - strnlen(sql_del, sizeof(sql_del))); |
| 159 |
i++; |
i++; |
| 160 |
j++; |
j++; |
| 161 |
} |
} |
| 162 |
else if (p_favor->aid_base[i] < p_favor->aid_inc[j]) // skip existing record |
else if (i < p_favor->aid_base_cnt && p_favor->aid_base[i] < p_favor->aid_inc[j]) // skip existing record |
| 163 |
{ |
{ |
| 164 |
i++; |
i++; |
| 165 |
} |
} |
| 166 |
else // if (p_favor->aid_base[i] > p_favor->aid_inc[j]) |
else // if (i >= p_favor->aid_base_cnt || p_favor->aid_base[i] > p_favor->aid_inc[j]) |
| 167 |
{ |
{ |
| 168 |
snprintf(tuple_tmp, sizeof(tuple_tmp), |
snprintf(tuple_tmp, sizeof(tuple_tmp), |
| 169 |
"(%d, %d), ", |
"(%d, %d), ", |
| 226 |
|
|
| 227 |
int article_favor_merge_inc(ARTICLE_FAVOR *p_favor) |
int article_favor_merge_inc(ARTICLE_FAVOR *p_favor) |
| 228 |
{ |
{ |
| 229 |
int32_t *aid_new; |
int32_t aid_new[MAX_FAVOR_AID_BASE_CNT]; |
|
int aid_new_cnt; |
|
| 230 |
int i, j, k; |
int i, j, k; |
| 231 |
|
int len; |
| 232 |
|
|
| 233 |
if (p_favor == NULL) |
if (p_favor == NULL) |
| 234 |
{ |
{ |
| 241 |
return 0; |
return 0; |
| 242 |
} |
} |
| 243 |
|
|
| 244 |
aid_new_cnt = p_favor->aid_base_cnt + p_favor->aid_inc_cnt; |
for (i = 0, j = 0, k = 0; i < p_favor->aid_base_cnt && j < p_favor->aid_inc_cnt && k < MAX_FAVOR_AID_BASE_CNT;) |
|
|
|
|
aid_new = malloc(sizeof(int32_t) * (size_t)aid_new_cnt); |
|
|
if (aid_new == NULL) |
|
|
{ |
|
|
log_error("malloc(INT32 * %d) error: OOM\n", aid_new_cnt); |
|
|
return -2; |
|
|
} |
|
|
|
|
|
for (i = 0, j = 0, k = 0; i < p_favor->aid_base_cnt && j < p_favor->aid_inc_cnt;) |
|
| 245 |
{ |
{ |
| 246 |
if (p_favor->aid_base[i] == p_favor->aid_inc[j]) // XOR - discard duplicate pair |
if (p_favor->aid_base[i] == p_favor->aid_inc[j]) // XOR - discard duplicate pair |
| 247 |
{ |
{ |
| 258 |
} |
} |
| 259 |
} |
} |
| 260 |
|
|
| 261 |
memcpy(aid_new + k, p_favor->aid_base + i, sizeof(int32_t) * (size_t)(p_favor->aid_base_cnt - i)); |
len = MIN(p_favor->aid_base_cnt - i, MAX_FAVOR_AID_BASE_CNT - k); |
| 262 |
k += (p_favor->aid_base_cnt - i); |
if (len > 0) |
| 263 |
memcpy(aid_new + k, p_favor->aid_inc + j, sizeof(int32_t) * (size_t)(p_favor->aid_inc_cnt - j)); |
{ |
| 264 |
k += (p_favor->aid_inc_cnt - j); |
memcpy(aid_new + k, p_favor->aid_base + i, |
| 265 |
|
sizeof(int32_t) * (size_t)len); |
| 266 |
|
i += len; |
| 267 |
|
k += len; |
| 268 |
|
} |
| 269 |
|
if (i < p_favor->aid_base_cnt) |
| 270 |
|
{ |
| 271 |
|
log_error("Too many base aids, %d will be discarded\n", p_favor->aid_base_cnt - i); |
| 272 |
|
} |
| 273 |
|
|
| 274 |
free(p_favor->aid_base); |
len = MIN(p_favor->aid_inc_cnt - j, MAX_FAVOR_AID_BASE_CNT - k); |
| 275 |
p_favor->aid_base = aid_new; |
if (len > 0) |
| 276 |
p_favor->aid_base_cnt = k; |
{ |
| 277 |
|
memcpy(aid_new + k, p_favor->aid_inc + j, |
| 278 |
|
sizeof(int32_t) * (size_t)len); |
| 279 |
|
j += len; |
| 280 |
|
k += len; |
| 281 |
|
} |
| 282 |
|
if (j < p_favor->aid_inc_cnt) |
| 283 |
|
{ |
| 284 |
|
log_error("Too many inc aids, %d will be discarded\n", p_favor->aid_inc_cnt - j); |
| 285 |
|
} |
| 286 |
|
|
| 287 |
|
memcpy(p_favor->aid_base, aid_new, sizeof(int32_t) * (size_t)k); |
| 288 |
|
|
| 289 |
|
p_favor->aid_base_cnt = k; |
| 290 |
p_favor->aid_inc_cnt = 0; |
p_favor->aid_inc_cnt = 0; |
| 291 |
|
|
| 292 |
return 0; |
return 0; |
| 321 |
mid = (left + right) / 2; |
mid = (left + right) / 2; |
| 322 |
if (aid < (i == 0 ? p_favor->aid_base[mid] : p_favor->aid_inc[mid])) |
if (aid < (i == 0 ? p_favor->aid_base[mid] : p_favor->aid_inc[mid])) |
| 323 |
{ |
{ |
| 324 |
right = mid; |
right = mid - 1; |
| 325 |
} |
} |
| 326 |
else if (aid > (i == 0 ? p_favor->aid_base[mid] : p_favor->aid_inc[mid])) |
else if (aid > (i == 0 ? p_favor->aid_base[mid] : p_favor->aid_inc[mid])) |
| 327 |
{ |
{ |
| 372 |
mid = (left + right) / 2; |
mid = (left + right) / 2; |
| 373 |
if (aid < (i == 0 ? p_favor->aid_base[mid] : p_favor->aid_inc[mid])) |
if (aid < (i == 0 ? p_favor->aid_base[mid] : p_favor->aid_inc[mid])) |
| 374 |
{ |
{ |
| 375 |
right = mid; |
right = mid - 1; |
| 376 |
} |
} |
| 377 |
else if (aid > (i == 0 ? p_favor->aid_base[mid] : p_favor->aid_inc[mid])) |
else if (aid > (i == 0 ? p_favor->aid_base[mid] : p_favor->aid_inc[mid])) |
| 378 |
{ |
{ |
| 398 |
|
|
| 399 |
if (aid == p_favor->aid_inc[left] && p_favor->aid_inc_cnt > 0) // Unset |
if (aid == p_favor->aid_inc[left] && p_favor->aid_inc_cnt > 0) // Unset |
| 400 |
{ |
{ |
| 401 |
for (i = left; i < p_favor->aid_inc_cnt - 1; i++) |
if (p_favor->aid_inc_cnt > left + 1) |
| 402 |
{ |
{ |
| 403 |
p_favor->aid_inc[i] = p_favor->aid_inc[i + 1]; |
memmove(p_favor->aid_inc + left, |
| 404 |
|
p_favor->aid_inc + left + 1, |
| 405 |
|
sizeof(int32_t) * (size_t)(p_favor->aid_inc_cnt - left - 1)); |
| 406 |
} |
} |
| 407 |
|
|
| 408 |
(p_favor->aid_inc_cnt)--; |
(p_favor->aid_inc_cnt)--; |
| 436 |
right = left + 1; |
right = left + 1; |
| 437 |
} |
} |
| 438 |
|
|
| 439 |
for (i = p_favor->aid_inc_cnt - 1; i >= right; i--) |
if (p_favor->aid_inc_cnt > right) |
| 440 |
{ |
{ |
| 441 |
p_favor->aid_inc[i + 1] = p_favor->aid_inc[i]; |
memmove(p_favor->aid_inc + right + 1, |
| 442 |
|
p_favor->aid_inc + right, |
| 443 |
|
sizeof(int32_t) * (size_t)(p_favor->aid_inc_cnt - right)); |
| 444 |
} |
} |
| 445 |
|
|
| 446 |
p_favor->aid_inc[right] = aid; |
p_favor->aid_inc[right] = aid; |