| 15 |
***************************************************************************/ |
***************************************************************************/ |
| 16 |
|
|
| 17 |
#include "section_list.h" |
#include "section_list.h" |
| 18 |
|
#include "trie_dict.h" |
| 19 |
#include "bbs.h" |
#include "bbs.h" |
| 20 |
#include "log.h" |
#include "log.h" |
| 21 |
#include <stdio.h> |
#include <stdio.h> |
| 24 |
|
|
| 25 |
#define ARTICLE_BLOCK_SHM_FILE "~article_block_shm.dat" |
#define ARTICLE_BLOCK_SHM_FILE "~article_block_shm.dat" |
| 26 |
#define SECTION_LIST_SHM_FILE "~section_list_shm.dat" |
#define SECTION_LIST_SHM_FILE "~section_list_shm.dat" |
| 27 |
|
#define TRIE_DICT_SHM_FILE "~trie_dict_shm.dat" |
| 28 |
|
|
| 29 |
const char *sname[] = { |
const char *sname[] = { |
| 30 |
"Test", |
"Test", |
| 90 |
} |
} |
| 91 |
fclose(fp); |
fclose(fp); |
| 92 |
|
|
| 93 |
|
if ((fp = fopen(TRIE_DICT_SHM_FILE, "w")) == NULL) |
| 94 |
|
{ |
| 95 |
|
log_error("fopen(%s) error\n", TRIE_DICT_SHM_FILE); |
| 96 |
|
return -1; |
| 97 |
|
} |
| 98 |
|
fclose(fp); |
| 99 |
|
|
| 100 |
|
if (trie_dict_init(TRIE_DICT_SHM_FILE, TRIE_NODE_PER_POOL) < 0) |
| 101 |
|
{ |
| 102 |
|
printf("trie_dict_init failed\n"); |
| 103 |
|
return -1; |
| 104 |
|
} |
| 105 |
|
|
| 106 |
if (article_block_init(ARTICLE_BLOCK_SHM_FILE, block_count) < 0) |
if (article_block_init(ARTICLE_BLOCK_SHM_FILE, block_count) < 0) |
| 107 |
{ |
{ |
| 108 |
log_error("article_block_init(%s, %d) error\n", ARTICLE_BLOCK_SHM_FILE, block_count); |
log_error("article_block_init(%s, %d) error\n", ARTICLE_BLOCK_SHM_FILE, block_count); |
| 109 |
return -2; |
return -2; |
| 110 |
} |
} |
| 111 |
|
|
| 112 |
if (section_list_pool_init(SECTION_LIST_SHM_FILE) < 0) |
if (section_list_init(SECTION_LIST_SHM_FILE) < 0) |
| 113 |
{ |
{ |
| 114 |
log_error("section_list_pool_init(%s) error\n", SECTION_LIST_SHM_FILE); |
log_error("section_list_pool_init(%s) error\n", SECTION_LIST_SHM_FILE); |
| 115 |
return -2; |
return -2; |
| 119 |
|
|
| 120 |
last_aid = 0; |
last_aid = 0; |
| 121 |
|
|
| 122 |
if (section_list_try_rw_lock(NULL, 1) < 0) |
if (section_list_rw_lock(NULL) < 0) |
| 123 |
{ |
{ |
| 124 |
printf("section_list_try_rw_lock(sid = %d) error\n", 0); |
printf("section_list_rw_lock(sid = %d) error\n", 0); |
| 125 |
} |
} |
| 126 |
|
|
| 127 |
for (i = 0; i < section_count; i++) |
for (i = 0; i < section_count; i++) |
| 185 |
article.ontop = 0; |
article.ontop = 0; |
| 186 |
article.lock = 0; |
article.lock = 0; |
| 187 |
|
|
| 188 |
if (section_list_try_rw_lock(p_section[i], 1) < 0) |
if (section_list_rw_lock(p_section[i]) < 0) |
| 189 |
{ |
{ |
| 190 |
printf("section_list_try_rw_lock(sid = %d) error\n", p_section[i]->sid); |
printf("section_list_rw_lock(sid = %d) error\n", p_section[i]->sid); |
| 191 |
break; |
break; |
| 192 |
} |
} |
| 193 |
|
|
| 210 |
// printf("Loaded %d articles into section %d\n", p_section[i]->article_count, i); |
// printf("Loaded %d articles into section %d\n", p_section[i]->article_count, i); |
| 211 |
} |
} |
| 212 |
|
|
| 213 |
|
if (last_aid != article_block_last_aid()) |
| 214 |
|
{ |
| 215 |
|
printf("last_aid != %d\n", article_block_last_aid()); |
| 216 |
|
} |
| 217 |
|
|
| 218 |
|
if (article_block_article_count() != section_count * BBS_article_limit_per_section) |
| 219 |
|
{ |
| 220 |
|
printf("article_block_article_count() error %d != %d * %d\n", |
| 221 |
|
article_block_article_count(), section_count, BBS_article_limit_per_section); |
| 222 |
|
} |
| 223 |
|
|
| 224 |
last_aid = 0; |
last_aid = 0; |
| 225 |
|
|
| 226 |
for (j = 0; j < BBS_article_limit_per_section; j++) |
for (j = 0; j < BBS_article_limit_per_section; j++) |
| 235 |
printf("article_block_find_by_aid() at section %d index %d, %d != %d\n", i, j, p_article->aid, last_aid); |
printf("article_block_find_by_aid() at section %d index %d, %d != %d\n", i, j, p_article->aid, last_aid); |
| 236 |
} |
} |
| 237 |
|
|
| 238 |
if (section_list_try_rw_lock(p_section[i], 1) < 0) |
if (section_list_rw_lock(p_section[i]) < 0) |
| 239 |
{ |
{ |
| 240 |
printf("section_list_try_rw_lock(sid = %d) error\n", p_section[i]->sid); |
printf("section_list_rw_lock(sid = %d) error\n", p_section[i]->sid); |
| 241 |
break; |
break; |
| 242 |
} |
} |
| 243 |
|
|
| 258 |
|
|
| 259 |
printf("Testing #2 ...\n"); |
printf("Testing #2 ...\n"); |
| 260 |
|
|
| 261 |
if (section_list_try_rw_lock(NULL, 1) < 0) |
if (section_list_rw_lock(NULL) < 0) |
| 262 |
{ |
{ |
| 263 |
printf("section_list_try_rw_lock(sid = %d) error\n", 0); |
printf("section_list_rw_lock(sid = %d) error\n", 0); |
| 264 |
} |
} |
| 265 |
|
|
| 266 |
if (article_block_reset() != 0) |
if (article_block_reset() != 0) |
| 285 |
{ |
{ |
| 286 |
section_first_aid = last_aid + 1; |
section_first_aid = last_aid + 1; |
| 287 |
|
|
| 288 |
if (section_list_try_rw_lock(p_section[i], 1) < 0) |
if (section_list_rw_lock(p_section[i]) < 0) |
| 289 |
{ |
{ |
| 290 |
printf("section_list_try_rw_lock(sid = %d) error\n", p_section[i]->sid); |
printf("section_list_rw_lock(sid = %d) error\n", p_section[i]->sid); |
| 291 |
break; |
break; |
| 292 |
} |
} |
| 293 |
|
|
| 333 |
article_count = 0; |
article_count = 0; |
| 334 |
last_aid = 0; |
last_aid = 0; |
| 335 |
|
|
| 336 |
if (section_list_try_rd_lock(p_section[i], 1) < 0) |
if (section_list_rd_lock(p_section[i]) < 0) |
| 337 |
{ |
{ |
| 338 |
printf("section_list_try_rd_lock(sid = %d) error\n", p_section[i]->sid); |
printf("section_list_rd_lock(sid = %d) error\n", p_section[i]->sid); |
| 339 |
break; |
break; |
| 340 |
} |
} |
| 341 |
|
|
| 377 |
continue; |
continue; |
| 378 |
} |
} |
| 379 |
|
|
| 380 |
if (section_list_try_rd_lock(p_section[i], 1) < 0) |
if (section_list_rd_lock(p_section[i]) < 0) |
| 381 |
{ |
{ |
| 382 |
printf("section_list_try_rd_lock(sid = %d) error\n", p_section[i]->sid); |
printf("section_list_rd_lock(sid = %d) error\n", p_section[i]->sid); |
| 383 |
break; |
break; |
| 384 |
} |
} |
| 385 |
|
|
| 459 |
{ |
{ |
| 460 |
last_aid = 0; |
last_aid = 0; |
| 461 |
|
|
| 462 |
if (section_list_try_rd_lock(p_section[i], 1) < 0) |
if (section_list_rd_lock(p_section[i]) < 0) |
| 463 |
{ |
{ |
| 464 |
printf("section_list_try_rd_lock(sid = %d) error\n", p_section[i]->sid); |
printf("section_list_rd_lock(sid = %d) error\n", p_section[i]->sid); |
| 465 |
break; |
break; |
| 466 |
} |
} |
| 467 |
|
|
| 501 |
|
|
| 502 |
for (i = 0; i < section_count; i++) |
for (i = 0; i < section_count; i++) |
| 503 |
{ |
{ |
| 504 |
if (section_list_try_rw_lock(p_section[i], 1) < 0) |
if (section_list_rw_lock(p_section[i]) < 0) |
| 505 |
{ |
{ |
| 506 |
printf("section_list_try_rw_lock(sid = %d) error\n", p_section[i]->sid); |
printf("section_list_rw_lock(sid = %d) error\n", p_section[i]->sid); |
| 507 |
break; |
break; |
| 508 |
} |
} |
| 509 |
|
|
| 646 |
|
|
| 647 |
for (i = 0; i < BBS_max_section; i++) |
for (i = 0; i < BBS_max_section; i++) |
| 648 |
{ |
{ |
| 649 |
if (section_list_try_rw_lock(p_section[i], 1) < 0) |
if (section_list_rw_lock(p_section[i]) < 0) |
| 650 |
{ |
{ |
| 651 |
printf("section_list_try_rw_lock(sid = %d) error\n", p_section[i]->sid); |
printf("section_list_rw_lock(sid = %d) error\n", p_section[i]->sid); |
| 652 |
break; |
break; |
| 653 |
} |
} |
| 654 |
|
|
| 715 |
|
|
| 716 |
printf("Testing #5 ...\n"); |
printf("Testing #5 ...\n"); |
| 717 |
|
|
| 718 |
if (section_list_try_rw_lock(NULL, 1) < 0) |
if (section_list_rw_lock(NULL) < 0) |
| 719 |
{ |
{ |
| 720 |
printf("section_list_try_rw_lock(sid = %d) error\n", 0); |
printf("section_list_rw_lock(sid = %d) error\n", 0); |
| 721 |
} |
} |
| 722 |
|
|
| 723 |
if (article_block_reset() != 0) |
if (article_block_reset() != 0) |
| 740 |
|
|
| 741 |
for (i = 0; i < section_count / 2; i++) |
for (i = 0; i < section_count / 2; i++) |
| 742 |
{ |
{ |
| 743 |
if (section_list_try_rw_lock(p_section[i], 1) < 0) |
if (section_list_rw_lock(p_section[i]) < 0) |
| 744 |
{ |
{ |
| 745 |
printf("section_list_try_rw_lock(sid = %d) error\n", p_section[i]->sid); |
printf("section_list_rw_lock(sid = %d) error\n", p_section[i]->sid); |
| 746 |
break; |
break; |
| 747 |
} |
} |
| 748 |
|
|
| 782 |
|
|
| 783 |
for (i = 0; i < section_count / 2; i++) |
for (i = 0; i < section_count / 2; i++) |
| 784 |
{ |
{ |
| 785 |
if (section_list_try_rw_lock(p_section[i], 1) < 0) |
if (section_list_rw_lock(p_section[i]) < 0) |
| 786 |
{ |
{ |
| 787 |
printf("section_list_try_rw_lock(sid = %d) error\n", p_section[i]->sid); |
printf("section_list_rw_lock(sid = %d) error\n", p_section[i]->sid); |
| 788 |
break; |
break; |
| 789 |
} |
} |
| 790 |
|
|
| 815 |
|
|
| 816 |
for (i = 0; i < section_count / 2; i++) |
for (i = 0; i < section_count / 2; i++) |
| 817 |
{ |
{ |
| 818 |
if (section_list_try_rw_lock(p_section[i], 1) < 0) |
if (section_list_rw_lock(p_section[i]) < 0) |
| 819 |
{ |
{ |
| 820 |
printf("section_list_try_rw_lock(sid = %d) error\n", p_section[i]->sid); |
printf("section_list_rw_lock(sid = %d) error\n", p_section[i]->sid); |
| 821 |
break; |
break; |
| 822 |
} |
} |
| 823 |
|
|
| 824 |
if (section_list_try_rw_lock(p_section[section_count / 2 + i], 1) < 0) |
if (section_list_rw_lock(p_section[section_count / 2 + i]) < 0) |
| 825 |
{ |
{ |
| 826 |
printf("section_list_try_rw_lock(sid = %d) error\n", p_section[section_count / 2 + i]->sid); |
printf("section_list_rw_lock(sid = %d) error\n", p_section[section_count / 2 + i]->sid); |
| 827 |
|
|
| 828 |
if (section_list_rw_unlock(p_section[i]) < 0) |
if (section_list_rw_unlock(p_section[i]) < 0) |
| 829 |
{ |
{ |
| 873 |
|
|
| 874 |
for (i = 0; i < section_count; i++) |
for (i = 0; i < section_count; i++) |
| 875 |
{ |
{ |
| 876 |
if (section_list_try_rd_lock(p_section[i], 1) < 0) |
if (section_list_rd_lock(p_section[i]) < 0) |
| 877 |
{ |
{ |
| 878 |
printf("section_list_try_rd_lock(sid = %d) error\n", p_section[i]->sid); |
printf("section_list_rd_lock(sid = %d) error\n", p_section[i]->sid); |
| 879 |
break; |
break; |
| 880 |
} |
} |
| 881 |
|
|
| 921 |
} |
} |
| 922 |
} |
} |
| 923 |
|
|
| 924 |
|
printf("Testing #6 ...\n"); |
| 925 |
|
|
| 926 |
|
for (i = 0; i < section_count; i++) |
| 927 |
|
{ |
| 928 |
|
if (section_list_rd_lock(p_section[i]) < 0) |
| 929 |
|
{ |
| 930 |
|
printf("section_list_rd_lock(sid = %d) error\n", p_section[i]->sid); |
| 931 |
|
break; |
| 932 |
|
} |
| 933 |
|
} |
| 934 |
|
|
| 935 |
|
printf("Try rw_lock for 5 sec...\n"); |
| 936 |
|
if (section_list_try_rw_lock(NULL, 5) == 0) |
| 937 |
|
{ |
| 938 |
|
printf("section_list_try_rw_lock(sid = %d) error, expectation is timeout\n", p_section[i]->sid); |
| 939 |
|
} |
| 940 |
|
|
| 941 |
|
for (i = 0; i < section_count; i++) |
| 942 |
|
{ |
| 943 |
|
if (section_list_rd_unlock(p_section[i]) < 0) |
| 944 |
|
{ |
| 945 |
|
printf("section_list_rd_unlock(sid = %d) error\n", p_section[i]->sid); |
| 946 |
|
break; |
| 947 |
|
} |
| 948 |
|
} |
| 949 |
|
|
| 950 |
|
if (section_list_try_rw_lock(NULL, 5) < 0) |
| 951 |
|
{ |
| 952 |
|
printf("section_list_rd_lock(sid = %d) error\n", p_section[i]->sid); |
| 953 |
|
} |
| 954 |
|
|
| 955 |
|
for (i = 0; i < section_count; i++) |
| 956 |
|
{ |
| 957 |
|
if (section_list_try_rd_lock(p_section[i], 0) == 0) |
| 958 |
|
{ |
| 959 |
|
printf("section_list_try_rd_lock(sid = %d) error, expectation is timeout\n", p_section[i]->sid); |
| 960 |
|
break; |
| 961 |
|
} |
| 962 |
|
} |
| 963 |
|
|
| 964 |
|
if (section_list_rw_unlock(NULL) < 0) |
| 965 |
|
{ |
| 966 |
|
printf("section_list_rw_unlock(sid = %d) error\n", p_section[i]->sid); |
| 967 |
|
} |
| 968 |
|
|
| 969 |
printf("Press ENTER to exit..."); |
printf("Press ENTER to exit..."); |
| 970 |
getchar(); |
getchar(); |
| 971 |
|
|
| 972 |
|
section_list_cleanup(); |
| 973 |
article_block_cleanup(); |
article_block_cleanup(); |
| 974 |
section_list_pool_cleanup(); |
trie_dict_cleanup(); |
| 975 |
|
|
| 976 |
|
if (unlink(TRIE_DICT_SHM_FILE) < 0) |
| 977 |
|
{ |
| 978 |
|
log_error("unlink(%s) error\n", TRIE_DICT_SHM_FILE); |
| 979 |
|
return -1; |
| 980 |
|
} |
| 981 |
|
|
| 982 |
if (unlink(ARTICLE_BLOCK_SHM_FILE) < 0) |
if (unlink(ARTICLE_BLOCK_SHM_FILE) < 0) |
| 983 |
{ |
{ |