/[LeafOK_CVS]/lbbs/src/user_list.c
ViewVC logotype

Diff of /lbbs/src/user_list.c

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

Revision 1.7 by sysadm, Wed Oct 22 05:50:13 2025 UTC Revision 1.25 by sysadm, Sun Nov 2 14:38:53 2025 UTC
# Line 19  Line 19 
19  #include "log.h"  #include "log.h"
20  #include "trie_dict.h"  #include "trie_dict.h"
21  #include "user_list.h"  #include "user_list.h"
22    #include "user_stat.h"
23  #include <errno.h>  #include <errno.h>
24  #include <stdlib.h>  #include <stdlib.h>
25  #include <string.h>  #include <string.h>
# Line 53  struct user_list_pool_t Line 54  struct user_list_pool_t
54          USER_ONLINE_LIST user_online_list[2];          USER_ONLINE_LIST user_online_list[2];
55          USER_ONLINE_LIST *p_online_current;          USER_ONLINE_LIST *p_online_current;
56          USER_ONLINE_LIST *p_online_new;          USER_ONLINE_LIST *p_online_new;
57            USER_STAT_MAP user_stat_map;
58            int user_login_count;
59  };  };
60  typedef struct user_list_pool_t USER_LIST_POOL;  typedef struct user_list_pool_t USER_LIST_POOL;
61    
# Line 91  static int user_list_rd_lock(int semid); Line 94  static int user_list_rd_lock(int semid);
94  static int user_list_rw_lock(int semid);  static int user_list_rw_lock(int semid);
95    
96  static int user_list_load(MYSQL *db, USER_LIST *p_list);  static int user_list_load(MYSQL *db, USER_LIST *p_list);
97  static int user_online_list_load(MYSQL *db, USER_ONLINE_LIST *p_list);  static int user_online_list_load(MYSQL *db, USER_ONLINE_LIST *p_online_list);
98    static int user_login_count_load(MYSQL *db);
99    
100    static int user_info_index_uid_comp(const void *ptr1, const void *ptr2)
101    {
102            const USER_INFO_INDEX_UID *p1 = ptr1;
103            const USER_INFO_INDEX_UID *p2 = ptr2;
104    
105            if (p1->uid < p2->uid)
106            {
107                    return -1;
108            }
109            else if (p1->uid > p2->uid)
110            {
111                    return 1;
112            }
113            else if (p1->id < p2->id)
114            {
115                    return -1;
116            }
117            else if (p1->id > p2->id)
118            {
119                    return 1;
120            }
121            return 0;
122    }
123    
124  int user_list_load(MYSQL *db, USER_LIST *p_list)  int user_list_load(MYSQL *db, USER_LIST *p_list)
125  {  {
# Line 100  int user_list_load(MYSQL *db, USER_LIST Line 128  int user_list_load(MYSQL *db, USER_LIST
128          char sql[SQL_BUFFER_LEN];          char sql[SQL_BUFFER_LEN];
129          int ret = 0;          int ret = 0;
130          int i;          int i;
131            int j;
132            int32_t last_uid;
133            size_t intro_buf_offset;
134            size_t intro_len;
135    
136          if (db == NULL || p_list == NULL)          if (db == NULL || p_list == NULL)
137          {          {
# Line 107  int user_list_load(MYSQL *db, USER_LIST Line 139  int user_list_load(MYSQL *db, USER_LIST
139                  return -1;                  return -1;
140          }          }
141    
142            if (p_list->user_count > 0)
143            {
144                    last_uid = p_list->users[p_list->user_count - 1].uid;
145            }
146            else
147            {
148                    last_uid = -1;
149            }
150    
151          snprintf(sql, sizeof(sql),          snprintf(sql, sizeof(sql),
152                           "SELECT user_list.UID AS UID, username, nickname, gender, gender_pub, life, exp, "                           "SELECT user_list.UID AS UID, username, nickname, gender, gender_pub, life, exp, visit_count, "
153                           "UNIX_TIMESTAMP(signup_dt), UNIX_TIMESTAMP(last_login_dt), UNIX_TIMESTAMP(birthday) "                           "UNIX_TIMESTAMP(signup_dt), UNIX_TIMESTAMP(last_login_dt), UNIX_TIMESTAMP(last_logout_dt), "
154                             "UNIX_TIMESTAMP(birthday), `introduction` "
155                           "FROM user_list INNER JOIN user_pubinfo ON user_list.UID = user_pubinfo.UID "                           "FROM user_list INNER JOIN user_pubinfo ON user_list.UID = user_pubinfo.UID "
156                           "INNER JOIN user_reginfo ON user_list.UID = user_reginfo.UID "                           "INNER JOIN user_reginfo ON user_list.UID = user_reginfo.UID "
157                           "WHERE enable ORDER BY UID");                           "WHERE enable ORDER BY username");
158    
159          if (mysql_query(db, sql) != 0)          if (mysql_query(db, sql) != 0)
160          {          {
# Line 128  int user_list_load(MYSQL *db, USER_LIST Line 170  int user_list_load(MYSQL *db, USER_LIST
170                  goto cleanup;                  goto cleanup;
171          }          }
172    
173            intro_buf_offset = 0;
174          i = 0;          i = 0;
175          while ((row = mysql_fetch_row(rs)))          while ((row = mysql_fetch_row(rs)))
176          {          {
177                    // record
178                    p_list->users[i].id = i;
179                  p_list->users[i].uid = atoi(row[0]);                  p_list->users[i].uid = atoi(row[0]);
180                  strncpy(p_list->users[i].username, row[1], sizeof(p_list->users[i].username) - 1);                  strncpy(p_list->users[i].username, row[1], sizeof(p_list->users[i].username) - 1);
181                  p_list->users[i].username[sizeof(p_list->users[i].username) - 1] = '\0';                  p_list->users[i].username[sizeof(p_list->users[i].username) - 1] = '\0';
# Line 140  int user_list_load(MYSQL *db, USER_LIST Line 185  int user_list_load(MYSQL *db, USER_LIST
185                  p_list->users[i].gender_pub = (int8_t)(row[4] == NULL ? 0 : atoi(row[4]));                  p_list->users[i].gender_pub = (int8_t)(row[4] == NULL ? 0 : atoi(row[4]));
186                  p_list->users[i].life = (row[5] == NULL ? 0 : atoi(row[5]));                  p_list->users[i].life = (row[5] == NULL ? 0 : atoi(row[5]));
187                  p_list->users[i].exp = (row[6] == NULL ? 0 : atoi(row[6]));                  p_list->users[i].exp = (row[6] == NULL ? 0 : atoi(row[6]));
188                  p_list->users[i].signup_dt = (row[7] == NULL ? 0 : atol(row[7]));                  p_list->users[i].visit_count = (row[7] == NULL ? 0 : atoi(row[7]));
189                  p_list->users[i].last_login_dt = (row[8] == NULL ? 0 : atol(row[8]));                  p_list->users[i].signup_dt = (row[8] == NULL ? 0 : atol(row[8]));
190                  p_list->users[i].birthday = (row[9] == NULL ? 0 : atol(row[9]));                  p_list->users[i].last_login_dt = (row[9] == NULL ? 0 : atol(row[9]));
191                    p_list->users[i].last_logout_dt = (row[10] == NULL ? 0 : atol(row[10]));
192                    p_list->users[i].birthday = (row[10] == NULL ? 0 : atol(row[11]));
193                    intro_len = strlen((row[12] == NULL ? "" : row[12]));
194                    if (intro_len >= sizeof(p_list->user_intro_buf) - 1 - intro_buf_offset)
195                    {
196                            log_error("OOM for user introduction: len=%d, i=%d\n", intro_len, i);
197                            break;
198                    }
199                    memcpy(p_list->user_intro_buf + intro_buf_offset,
200                               (row[12] == NULL ? "" : row[12]),
201                               intro_len + 1);
202                    p_list->users[i].intro = p_list->user_intro_buf + intro_buf_offset;
203                    intro_buf_offset += (intro_len + 1);
204    
205                  i++;                  i++;
206                  if (i >= BBS_max_user_count)                  if (i >= BBS_max_user_count)
# Line 154  int user_list_load(MYSQL *db, USER_LIST Line 212  int user_list_load(MYSQL *db, USER_LIST
212          mysql_free_result(rs);          mysql_free_result(rs);
213          rs = NULL;          rs = NULL;
214    
215            if (i != p_list->user_count || p_list->users[i - 1].uid != last_uid) // Count of users changed
216            {
217                    // Rebuild index
218                    for (j = 0; j < i; j++)
219                    {
220                            p_list->index_uid[j].uid = p_list->users[j].uid;
221                            p_list->index_uid[j].id = j;
222                    }
223    
224                    qsort(p_list->index_uid, (size_t)i, sizeof(USER_INFO_INDEX_UID), user_info_index_uid_comp);
225    
226    #ifdef _DEBUG
227                    log_error("Rebuild index of %d users, last_uid=%d\n", i, p_list->users[i - 1].uid);
228    #endif
229            }
230    
231          p_list->user_count = i;          p_list->user_count = i;
232    
233  #ifdef _DEBUG  #ifdef _DEBUG
# Line 166  cleanup: Line 240  cleanup:
240          return ret;          return ret;
241  }  }
242    
243  int user_online_list_load(MYSQL *db, USER_ONLINE_LIST *p_list)  int user_online_list_load(MYSQL *db, USER_ONLINE_LIST *p_online_list)
244  {  {
245          MYSQL_RES *rs = NULL;          MYSQL_RES *rs = NULL;
246          MYSQL_ROW row;          MYSQL_ROW row;
247          char sql[SQL_BUFFER_LEN];          char sql[SQL_BUFFER_LEN];
248          int ret = 0;          int ret = 0;
249          int i;          int i;
250            int j;
251            int user_cnt;
252            int guest_cnt;
253    
254          if (db == NULL || p_list == NULL)          if (db == NULL || p_online_list == NULL)
255          {          {
256                  log_error("NULL pointer error\n");                  log_error("NULL pointer error\n");
257                  return -1;                  return -1;
# Line 202  int user_online_list_load(MYSQL *db, USE Line 279  int user_online_list_load(MYSQL *db, USE
279          }          }
280    
281          i = 0;          i = 0;
282            user_cnt = 0;
283            guest_cnt = 0;
284          while ((row = mysql_fetch_row(rs)))          while ((row = mysql_fetch_row(rs)))
285          {          {
286                  strncpy(p_list->users[i].session_id, row[0], sizeof(p_list->users[i].session_id) - 1);                  if (atoi(row[1]) == 0) // guest
   
                 p_list->users[i].session_id[sizeof(p_list->users[i].session_id) - 1] = '\0';  
                 if ((ret = query_user_info(atoi(row[1]), &(p_list->users[i].user_info))) < 0)  
287                  {                  {
288                          log_error("query_user_info(%d) error\n", atoi(row[1]));                          guest_cnt++;
289                          continue;                          continue;
290                  }                  }
291                  else if (ret == 0) // Guest                  else
292                  {                  {
293                          p_list->users[i].user_info.uid = 0;                          user_cnt++;
294                          strncpy(p_list->users[i].user_info.username, "guest", sizeof(p_list->users[i].user_info.username) - 1);                  }
295                          p_list->users[i].user_info.username[sizeof(p_list->users[i].user_info.username) - 1] = '\0';  
296                          strncpy(p_list->users[i].user_info.nickname, "Guest", sizeof(p_list->users[i].user_info.nickname) - 1);                  p_online_list->users[i].id = i;
297                          p_list->users[i].user_info.nickname[sizeof(p_list->users[i].user_info.nickname) - 1] = '\0';                  strncpy(p_online_list->users[i].session_id, row[0], sizeof(p_online_list->users[i].session_id) - 1);
298                          p_list->users[i].user_info.gender = 'M';                  p_online_list->users[i].session_id[sizeof(p_online_list->users[i].session_id) - 1] = '\0';
299                          p_list->users[i].user_info.gender_pub = 0;  
300                          p_list->users[i].user_info.life = 150;                  if ((ret = query_user_info_by_uid(atoi(row[1]), &(p_online_list->users[i].user_info), NULL, 0)) <= 0)
                         p_list->users[i].user_info.exp = 0;  
                         p_list->users[i].user_info.signup_dt = 0;  
                         p_list->users[i].user_info.last_login_dt = 0;  
                         p_list->users[i].user_info.birthday = 0;  
                 }  
   
                 strncpy(p_list->users[i].ip, row[2], sizeof(p_list->users[i].ip) - 1);  
                 p_list->users[i].ip[sizeof(p_list->users[i].ip) - 1] = '\0';  
   
                 strncpy(p_list->users[i].current_action, row[3], sizeof(p_list->users[i].current_action) - 1);  
                 p_list->users[i].current_action[sizeof(p_list->users[i].current_action) - 1] = '\0';  
                 p_list->users[i].current_action_title = NULL;  
                 if (p_list->users[i].current_action[0] == '\0')  
301                  {                  {
302                          p_list->users[i].current_action_title = "";                          log_error("query_user_info_by_uid(%d) error\n", atoi(row[1]));
303                            continue;
304                  }                  }
305                  else if (trie_dict_get(p_trie_action_dict, p_list->users[i].current_action, (int64_t *)(&(p_list->users[i].current_action_title))) < 0)  
306                    strncpy(p_online_list->users[i].ip, row[2], sizeof(p_online_list->users[i].ip) - 1);
307                    p_online_list->users[i].ip[sizeof(p_online_list->users[i].ip) - 1] = '\0';
308    
309                    strncpy(p_online_list->users[i].current_action, row[3], sizeof(p_online_list->users[i].current_action) - 1);
310                    p_online_list->users[i].current_action[sizeof(p_online_list->users[i].current_action) - 1] = '\0';
311                    p_online_list->users[i].current_action_title = NULL;
312                    if (p_online_list->users[i].current_action[0] == '\0')
313                    {
314                            p_online_list->users[i].current_action_title = "";
315                    }
316                    else if (trie_dict_get(p_trie_action_dict, p_online_list->users[i].current_action, (int64_t *)(&(p_online_list->users[i].current_action_title))) < 0)
317                  {                  {
318                          log_error("trie_dict_get(p_trie_action_dict, %s) error on session_id=%s\n",                          log_error("trie_dict_get(p_trie_action_dict, %s) error on session_id=%s\n",
319                                            p_list->users[i].current_action, p_list->users[i].session_id);                                            p_online_list->users[i].current_action, p_online_list->users[i].session_id);
320                          continue;                          continue;
321                  }                  }
322    
323                  p_list->users[i].login_tm = (row[4] == NULL ? 0 : atol(row[4]));                  p_online_list->users[i].login_tm = (row[4] == NULL ? 0 : atol(row[4]));
324                  p_list->users[i].last_tm = (row[5] == NULL ? 0 : atol(row[5]));                  p_online_list->users[i].last_tm = (row[5] == NULL ? 0 : atol(row[5]));
325    
326                  i++;                  i++;
327                  if (i >= BBS_max_user_online_count)                  if (i >= BBS_max_user_online_count)
# Line 258  int user_online_list_load(MYSQL *db, USE Line 333  int user_online_list_load(MYSQL *db, USE
333          mysql_free_result(rs);          mysql_free_result(rs);
334          rs = NULL;          rs = NULL;
335    
336          p_list->user_count = i;          if (user_cnt > 0)
337            {
338                    // Rebuild index
339                    for (j = 0; j < user_cnt; j++)
340                    {
341                            p_online_list->index_uid[j].uid = p_online_list->users[j].user_info.uid;
342                            p_online_list->index_uid[j].id = j;
343                    }
344    
345                    qsort(p_online_list->index_uid, (size_t)user_cnt, sizeof(USER_INFO_INDEX_UID), user_info_index_uid_comp);
346    
347  #ifdef _DEBUG  #ifdef _DEBUG
348          log_error("Loaded %d users\n", p_list->user_count);                  log_error("Rebuild index of %d online users\n", user_cnt);
349    #endif
350            }
351    
352            p_online_list->user_count = user_cnt;
353            p_online_list->guest_count = guest_cnt;
354    
355    #ifdef _DEBUG
356            log_error("Loaded %d online users and %d guest users\n", p_list->user_count, p_list->guest_count);
357  #endif  #endif
358    
359  cleanup:  cleanup:
# Line 270  cleanup: Line 362  cleanup:
362          return ret;          return ret;
363  }  }
364    
365  int user_list_pool_init(void)  int user_login_count_load(MYSQL *db)
366    {
367            MYSQL_RES *rs = NULL;
368            MYSQL_ROW row;
369            char sql[SQL_BUFFER_LEN];
370    
371            if (db == NULL)
372            {
373                    log_error("NULL pointer error\n");
374                    return -1;
375            }
376    
377            snprintf(sql, sizeof(sql),
378                             "SELECT ID FROM user_login_log ORDER BY ID DESC LIMIT 1");
379            if (mysql_query(db, sql) != 0)
380            {
381                    log_error("Query user_login_log error: %s\n", mysql_error(db));
382                    return -2;
383            }
384            if ((rs = mysql_store_result(db)) == NULL)
385            {
386                    log_error("Get user_login_log data failed\n");
387                    return -2;
388            }
389            if ((row = mysql_fetch_row(rs)))
390            {
391                    p_user_list_pool->user_login_count = atoi(row[0]);
392            }
393            mysql_free_result(rs);
394    
395            return 0;
396    }
397    
398    int user_list_pool_init(const char *filename)
399  {  {
400          int shmid;          int shmid;
401          int semid;          int semid;
# Line 304  int user_list_pool_init(void) Line 429  int user_list_pool_init(void)
429    
430          // Allocate shared memory          // Allocate shared memory
431          proj_id = (int)(time(NULL) % getpid());          proj_id = (int)(time(NULL) % getpid());
432          key = ftok(VAR_USER_LIST_SHM, proj_id);          key = ftok(filename, proj_id);
433          if (key == -1)          if (key == -1)
434          {          {
435                  log_error("ftok(%s %d) error (%d)\n", VAR_USER_LIST_SHM, proj_id, errno);                  log_error("ftok(%s %d) error (%d)\n", filename, proj_id, errno);
436                  return -2;                  return -2;
437          }          }
438    
# Line 360  int user_list_pool_init(void) Line 485  int user_list_pool_init(void)
485          p_user_list_pool->p_online_current = &(p_user_list_pool->user_online_list[0]);          p_user_list_pool->p_online_current = &(p_user_list_pool->user_online_list[0]);
486          p_user_list_pool->p_online_new = &(p_user_list_pool->user_online_list[1]);          p_user_list_pool->p_online_new = &(p_user_list_pool->user_online_list[1]);
487    
488            user_stat_map_init(&(p_user_list_pool->user_stat_map));
489    
490          return 0;          return 0;
491  }  }
492    
# Line 443  int user_list_pool_reload(int online_use Line 570  int user_list_pool_reload(int online_use
570          MYSQL *db = NULL;          MYSQL *db = NULL;
571          USER_LIST *p_tmp;          USER_LIST *p_tmp;
572          USER_ONLINE_LIST *p_online_tmp;          USER_ONLINE_LIST *p_online_tmp;
573            int ret = 0;
574    
575          if (p_user_list_pool == NULL)          if (p_user_list_pool == NULL)
576          {          {
# Line 462  int user_list_pool_reload(int online_use Line 590  int user_list_pool_reload(int online_use
590                  if (user_online_list_load(db, p_user_list_pool->p_online_new) < 0)                  if (user_online_list_load(db, p_user_list_pool->p_online_new) < 0)
591                  {                  {
592                          log_error("user_online_list_load() error\n");                          log_error("user_online_list_load() error\n");
593                          return -2;                          ret = -2;
594                            goto cleanup;
595                    }
596    
597                    if (user_login_count_load(db) < 0)
598                    {
599                            log_error("user_login_count_load() error\n");
600                            ret = -2;
601                            goto cleanup;
602                  }                  }
603          }          }
604          else          else
# Line 470  int user_list_pool_reload(int online_use Line 606  int user_list_pool_reload(int online_use
606                  if (user_list_load(db, p_user_list_pool->p_new) < 0)                  if (user_list_load(db, p_user_list_pool->p_new) < 0)
607                  {                  {
608                          log_error("user_list_load() error\n");                          log_error("user_list_load() error\n");
609                          return -2;                          ret = -2;
610                            goto cleanup;
611                  }                  }
612          }          }
613    
614          mysql_close(db);          mysql_close(db);
615            db = NULL;
616    
617          if (user_list_rw_lock(p_user_list_pool->semid) < 0)          if (user_list_rw_lock(p_user_list_pool->semid) < 0)
618          {          {
619                  log_error("user_list_rw_lock() error\n");                  log_error("user_list_rw_lock() error\n");
620                  return -3;                  ret = -3;
621                    goto cleanup;
622          }          }
623    
624          if (online_user)          if (online_user)
# Line 500  int user_list_pool_reload(int online_use Line 639  int user_list_pool_reload(int online_use
639          if (user_list_rw_unlock(p_user_list_pool->semid) < 0)          if (user_list_rw_unlock(p_user_list_pool->semid) < 0)
640          {          {
641                  log_error("user_list_rw_unlock() error\n");                  log_error("user_list_rw_unlock() error\n");
642                  return -3;                  ret = -3;
643                    goto cleanup;
644          }          }
645    
646          return 0;  cleanup:
647            mysql_close(db);
648    
649            return ret;
650  }  }
651    
652  int user_list_try_rd_lock(int semid, int wait_sec)  int user_list_try_rd_lock(int semid, int wait_sec)
# Line 668  int query_user_list(int page_id, USER_IN Line 811  int query_user_list(int page_id, USER_IN
811                  return -1;                  return -1;
812          }          }
813    
814            *p_user_count = 0;
815            *p_page_count = 0;
816    
817          // acquire lock of user list          // acquire lock of user list
818          if (user_list_rd_lock(p_user_list_pool->semid) < 0)          if (user_list_rd_lock(p_user_list_pool->semid) < 0)
819          {          {
# Line 721  int query_user_online_list(int page_id, Line 867  int query_user_online_list(int page_id,
867                  return -1;                  return -1;
868          }          }
869    
870            *p_user_count = 0;
871            *p_page_count = 0;
872    
873          // acquire lock of user list          // acquire lock of user list
874          if (user_list_rd_lock(p_user_list_pool->semid) < 0)          if (user_list_rd_lock(p_user_list_pool->semid) < 0)
875          {          {
# Line 764  cleanup: Line 913  cleanup:
913          return ret;          return ret;
914  }  }
915    
916  int query_user_info(int32_t uid, USER_INFO *p_user)  int get_user_list_count(int *p_user_cnt)
917    {
918            if (p_user_cnt == NULL)
919            {
920                    log_error("NULL pointer error\n");
921                    return -1;
922            }
923    
924            // acquire lock of user list
925            if (user_list_rd_lock(p_user_list_pool->semid) < 0)
926            {
927                    log_error("user_list_rd_lock() error\n");
928                    return -2;
929            }
930    
931            *p_user_cnt = p_user_list_pool->p_current->user_count;
932    
933            // release lock of user list
934            if (user_list_rd_unlock(p_user_list_pool->semid) < 0)
935            {
936                    log_error("user_list_rd_unlock() error\n");
937                    return -2;
938            }
939    
940            return 0;
941    }
942    
943    int get_user_online_list_count(int *p_user_cnt, int *p_guest_cnt)
944    {
945            if (p_user_cnt == NULL || p_guest_cnt == NULL)
946            {
947                    log_error("NULL pointer error\n");
948                    return -1;
949            }
950    
951            // acquire lock of user list
952            if (user_list_rd_lock(p_user_list_pool->semid) < 0)
953            {
954                    log_error("user_list_rd_lock() error\n");
955                    return -2;
956            }
957    
958            *p_user_cnt = p_user_list_pool->p_online_current->user_count;
959            *p_guest_cnt = p_user_list_pool->p_online_current->guest_count;
960    
961            // release lock of user list
962            if (user_list_rd_unlock(p_user_list_pool->semid) < 0)
963            {
964                    log_error("user_list_rd_unlock() error\n");
965                    return -2;
966            }
967    
968            return 0;
969    }
970    
971    int get_user_login_count(int *p_login_cnt)
972    {
973            if (p_login_cnt == NULL)
974            {
975                    log_error("NULL pointer error\n");
976                    return -1;
977            }
978    
979            *p_login_cnt = p_user_list_pool->user_login_count;
980    
981            return 0;
982    }
983    
984    int query_user_info(int32_t id, USER_INFO *p_user)
985    {
986            int ret = 0;
987    
988            if (p_user == NULL)
989            {
990                    log_error("NULL pointer error\n");
991                    return -1;
992            }
993    
994            // acquire lock of user list
995            if (user_list_rd_lock(p_user_list_pool->semid) < 0)
996            {
997                    log_error("user_list_rd_lock() error\n");
998                    return -2;
999            }
1000    
1001            if (id >= 0 && id < p_user_list_pool->p_current->user_count) // Found
1002            {
1003                    *p_user = p_user_list_pool->p_current->users[id];
1004                    ret = 1;
1005            }
1006    
1007            // release lock of user list
1008            if (user_list_rd_unlock(p_user_list_pool->semid) < 0)
1009            {
1010                    log_error("user_list_rd_unlock() error\n");
1011                    ret = -1;
1012            }
1013    
1014            return ret;
1015    }
1016    
1017    int query_user_info_by_uid(int32_t uid, USER_INFO *p_user, char *p_intro_buf, size_t intro_buf_len)
1018  {  {
1019          int left;          int left;
1020          int right;          int right;
1021          int mid;          int mid;
1022            int32_t id;
1023          int ret = 0;          int ret = 0;
1024    
1025          if (p_user == NULL)          if (p_user == NULL)
# Line 790  int query_user_info(int32_t uid, USER_IN Line 1041  int query_user_info(int32_t uid, USER_IN
1041          while (left < right)          while (left < right)
1042          {          {
1043                  mid = (left + right) / 2;                  mid = (left + right) / 2;
1044                  if (uid < p_user_list_pool->p_current->users[mid].uid)                  if (uid < p_user_list_pool->p_current->index_uid[mid].uid)
1045                    {
1046                            right = mid - 1;
1047                    }
1048                    else if (uid > p_user_list_pool->p_current->index_uid[mid].uid)
1049                    {
1050                            left = mid + 1;
1051                    }
1052                    else // if (uid == p_user_list_pool->p_current->index_uid[mid].uid)
1053                    {
1054                            left = mid;
1055                            break;
1056                    }
1057            }
1058    
1059            if (uid == p_user_list_pool->p_current->index_uid[left].uid) // Found
1060            {
1061                    id = p_user_list_pool->p_current->index_uid[left].id;
1062                    *p_user = p_user_list_pool->p_current->users[id];
1063                    ret = 1;
1064    
1065                    if (p_intro_buf != NULL)
1066                    {
1067                            strncpy(p_intro_buf, p_user_list_pool->p_current->users[id].intro, intro_buf_len - 1);
1068                            p_intro_buf[intro_buf_len - 1] = '\0';
1069                            p_user->intro = p_intro_buf;
1070                    }
1071            }
1072    
1073            // release lock of user list
1074            if (user_list_rd_unlock(p_user_list_pool->semid) < 0)
1075            {
1076                    log_error("user_list_rd_unlock() error\n");
1077                    ret = -1;
1078            }
1079    
1080            return ret;
1081    }
1082    
1083    int query_user_info_by_username(const char *username_prefix, int max_user_cnt,
1084                                                                    int32_t uid_list[], char username_list[][BBS_username_max_len + 1])
1085    {
1086            int left;
1087            int right;
1088            int mid;
1089            int left_save;
1090            int ret = 0;
1091            size_t prefix_len;
1092            int comp;
1093            int i;
1094    
1095            if (username_prefix == NULL || uid_list == NULL || username_list == NULL)
1096            {
1097                    log_error("NULL pointer error\n");
1098                    return -1;
1099            }
1100    
1101            prefix_len = strlen(username_prefix);
1102    
1103            // acquire lock of user list
1104            if (user_list_rd_lock(p_user_list_pool->semid) < 0)
1105            {
1106                    log_error("user_list_rd_lock() error\n");
1107                    return -2;
1108            }
1109    
1110            left = 0;
1111            right = p_user_list_pool->p_current->user_count - 1;
1112    
1113            while (left < right)
1114            {
1115                    mid = (left + right) / 2;
1116                    comp = strncasecmp(username_prefix, p_user_list_pool->p_current->users[mid].username, prefix_len);
1117                    if (comp < 0)
1118                  {                  {
1119                          right = mid;                          right = mid - 1;
1120                  }                  }
1121                  else if (uid > p_user_list_pool->p_current->users[mid].uid)                  else if (comp > 0)
1122                  {                  {
1123                          left = mid + 1;                          left = mid + 1;
1124                  }                  }
1125                  else // if (uid == p_user_list_pool->p_current->users[mid].uid)                  else // if (comp == 0)
1126                  {                  {
1127                          left = mid;                          left = mid;
1128                          break;                          break;
1129                  }                  }
1130          }          }
1131    
1132          if (uid == p_user_list_pool->p_current->users[left].uid) // Found          if (strncasecmp(username_prefix, p_user_list_pool->p_current->users[left].username, prefix_len) == 0) // Found
1133            {
1134    #ifdef _DEBUG
1135                    log_error("Debug: match found, pos=%d\n", left);
1136    #endif
1137    
1138                    left_save = left;
1139                    right = left;
1140                    left = 0;
1141    
1142                    while (left < right)
1143                    {
1144                            mid = (left + right) / 2;
1145                            comp = strncasecmp(username_prefix, p_user_list_pool->p_current->users[mid].username, prefix_len);
1146                            if (comp > 0)
1147                            {
1148                                    left = mid + 1;
1149                            }
1150                            else if (comp == 0)
1151                            {
1152                                    right = mid;
1153                            }
1154                            else // if (comp < 0)
1155                            {
1156                                    log_error("Bug: left=%d right=%d mid=%d");
1157                                    ret = -2;
1158                                    goto cleanup;
1159                            }
1160                    }
1161    
1162    #ifdef _DEBUG
1163                    log_error("Debug: first match found, pos=%d\n", right);
1164    #endif
1165    
1166                    left = left_save;
1167                    left_save = right;
1168                    right = p_user_list_pool->p_current->user_count - 1;
1169    
1170                    while (left < right)
1171                    {
1172                            mid = (left + right) / 2 + (left + right) % 2;
1173                            comp = strncasecmp(username_prefix, p_user_list_pool->p_current->users[mid].username, prefix_len);
1174                            if (comp < 0)
1175                            {
1176                                    right = mid - 1;
1177                            }
1178                            else if (comp == 0)
1179                            {
1180                                    left = mid;
1181                            }
1182                            else // if (comp > 0)
1183                            {
1184                                    log_error("Bug: left=%d right=%d mid=%d");
1185                                    ret = -2;
1186                                    goto cleanup;
1187                            }
1188                    }
1189    
1190    #ifdef _DEBUG
1191                    log_error("Debug: last match found, pos=%d\n", left);
1192    #endif
1193    
1194                    right = left;
1195                    left = left_save;
1196    
1197                    for (i = 0; i < max_user_cnt && left + i <= right; i++)
1198                    {
1199                            uid_list[i] = p_user_list_pool->p_current->users[left + i].uid;
1200                            strncpy(username_list[i], p_user_list_pool->p_current->users[left + i].username, sizeof(username_list[i]) - 1);
1201                            username_list[i][sizeof(username_list[i]) - 1] = '\0';
1202                    }
1203                    ret = i;
1204            }
1205    
1206    cleanup:
1207            // release lock of user list
1208            if (user_list_rd_unlock(p_user_list_pool->semid) < 0)
1209            {
1210                    log_error("user_list_rd_unlock() error\n");
1211                    ret = -1;
1212            }
1213    
1214            return ret;
1215    }
1216    
1217    int query_user_online_info(int32_t id, USER_ONLINE_INFO *p_user)
1218    {
1219            int ret = 0;
1220    
1221            if (p_user == NULL)
1222            {
1223                    log_error("NULL pointer error\n");
1224                    return -1;
1225            }
1226    
1227            // acquire lock of user list
1228            if (user_list_rd_lock(p_user_list_pool->semid) < 0)
1229            {
1230                    log_error("user_list_rd_lock() error\n");
1231                    return -2;
1232            }
1233    
1234            if (id >= 0 && id < p_user_list_pool->p_online_current->user_count) // Found
1235          {          {
1236                  *p_user = p_user_list_pool->p_current->users[left];                  *p_user = p_user_list_pool->p_online_current->users[id];
1237                  ret = 1;                  ret = 1;
1238          }          }
1239    
# Line 820  int query_user_info(int32_t uid, USER_IN Line 1246  int query_user_info(int32_t uid, USER_IN
1246    
1247          return ret;          return ret;
1248  }  }
1249    
1250    int query_user_online_info_by_uid(int32_t uid, USER_ONLINE_INFO *p_users, int *p_user_cnt, int start_id)
1251    {
1252            int left;
1253            int right;
1254            int mid;
1255            int32_t id;
1256            int ret = 0;
1257            int i;
1258            int user_cnt;
1259    
1260            if (p_users == NULL || p_user_cnt == NULL)
1261            {
1262                    log_error("NULL pointer error\n");
1263                    return -1;
1264            }
1265    
1266            user_cnt = *p_user_cnt;
1267            *p_user_cnt = 0;
1268    
1269            // acquire lock of user list
1270            if (user_list_rd_lock(p_user_list_pool->semid) < 0)
1271            {
1272                    log_error("user_list_rd_lock() error\n");
1273                    return -2;
1274            }
1275    
1276            left = start_id;
1277            right = p_user_list_pool->p_online_current->user_count - 1;
1278    
1279            while (left < right)
1280            {
1281                    mid = (left + right) / 2;
1282                    if (uid < p_user_list_pool->p_online_current->index_uid[mid].uid)
1283                    {
1284                            right = mid - 1;
1285                    }
1286                    else if (uid > p_user_list_pool->p_online_current->index_uid[mid].uid)
1287                    {
1288                            left = mid + 1;
1289                    }
1290                    else // if (uid == p_user_list_pool->p_online_current->index_uid[mid].uid)
1291                    {
1292                            left = mid;
1293                            break;
1294                    }
1295            }
1296    
1297            if (uid == p_user_list_pool->p_online_current->index_uid[left].uid)
1298            {
1299                    right = left;
1300                    left = start_id;
1301    
1302                    while (left < right)
1303                    {
1304                            mid = (left + right) / 2;
1305                            if (uid - 1 < p_user_list_pool->p_online_current->index_uid[mid].uid)
1306                            {
1307                                    right = mid;
1308                            }
1309                            else // if (uid - 1 >= p_user_list_pool->p_online_current->index_uid[mid].uid)
1310                            {
1311                                    left = mid + 1;
1312                            }
1313                    }
1314    
1315                    for (i = 0;
1316                             left < p_user_list_pool->p_online_current->user_count && i < user_cnt &&
1317                             uid == p_user_list_pool->p_online_current->index_uid[left].uid;
1318                             left++, i++)
1319                    {
1320                            id = p_user_list_pool->p_online_current->index_uid[left].id;
1321                            p_users[i] = p_user_list_pool->p_online_current->users[id];
1322                    }
1323    
1324                    if (i > 0)
1325                    {
1326                            *p_user_cnt = i;
1327                            ret = 1;
1328                    }
1329            }
1330    
1331            // release lock of user list
1332            if (user_list_rd_unlock(p_user_list_pool->semid) < 0)
1333            {
1334                    log_error("user_list_rd_unlock() error\n");
1335                    ret = -1;
1336            }
1337    
1338            return ret;
1339    }
1340    
1341    int get_user_id_list(int32_t *p_uid_list, int *p_user_cnt, int start_uid)
1342    {
1343            int left;
1344            int right;
1345            int mid;
1346            int ret = 0;
1347            int i;
1348    
1349            if (p_uid_list == NULL || p_user_cnt == NULL)
1350            {
1351                    log_error("NULL pointer error\n");
1352                    return -1;
1353            }
1354    
1355            // acquire lock of user list
1356            if (user_list_rd_lock(p_user_list_pool->semid) < 0)
1357            {
1358                    log_error("user_list_rd_lock() error\n");
1359                    return -2;
1360            }
1361    
1362            left = 0;
1363            right = p_user_list_pool->p_current->user_count - 1;
1364    
1365            while (left < right)
1366            {
1367                    mid = (left + right) / 2;
1368                    if (start_uid < p_user_list_pool->p_current->index_uid[mid].uid)
1369                    {
1370                            right = mid - 1;
1371                    }
1372                    else if (start_uid > p_user_list_pool->p_current->index_uid[mid].uid)
1373                    {
1374                            left = mid + 1;
1375                    }
1376                    else // if (start_uid == p_user_list_pool->p_current->index_uid[mid].uid)
1377                    {
1378                            left = mid;
1379                            break;
1380                    }
1381            }
1382    
1383            for (i = 0; i < *p_user_cnt && left + i < p_user_list_pool->p_current->user_count; i++)
1384            {
1385                    p_uid_list[i] = p_user_list_pool->p_current->index_uid[left + i].uid;
1386            }
1387            *p_user_cnt = i;
1388    
1389            // release lock of user list
1390            if (user_list_rd_unlock(p_user_list_pool->semid) < 0)
1391            {
1392                    log_error("user_list_rd_unlock() error\n");
1393                    ret = -1;
1394            }
1395    
1396            return ret;
1397    }
1398    
1399    int user_stat_update(void)
1400    {
1401            return user_stat_map_update(&(p_user_list_pool->user_stat_map));
1402    }
1403    
1404    int user_article_cnt_inc(int32_t uid, int n)
1405    {
1406            return user_stat_article_cnt_inc(&(p_user_list_pool->user_stat_map), uid, n);
1407    }
1408    
1409    int get_user_article_cnt(int32_t uid)
1410    {
1411            const USER_STAT *p_stat;
1412            int ret;
1413    
1414            ret = user_stat_get(&(p_user_list_pool->user_stat_map), uid, &p_stat);
1415            if (ret < 0)
1416            {
1417                    log_error("user_stat_get(uid=%d) error: %d\n", uid, ret);
1418                    return -1;
1419            }
1420            else if (ret == 0) // user not found
1421            {
1422                    return -1;
1423            }
1424    
1425            return p_stat->article_count;
1426    }

Legend:
Removed lines/characters  
Changed lines/characters
  Added lines/characters
webmaster@leafok.com
 ViewVC Help
Powered by ViewVC 1.3.0-beta1