/[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.13 by sysadm, Wed Oct 22 14:42:08 2025 UTC Revision 1.32 by sysadm, Tue Nov 11 00:28:05 2025 UTC
# Line 1  Line 1 
1  /***************************************************************************  /* SPDX-License-Identifier: GPL-3.0-or-later */
2                                                   user_list.c  -  description  /*
3                                                           -------------------   * user_list
4          Copyright            : (C) 2004-2025 by Leaflet   *   - data model and basic operations of (online) user list
5          Email                : leaflet@leafok.com   *
6   ***************************************************************************/   * Copyright (C) 2004-2025  Leaflet <leaflet@leafok.com>
7     */
8  /***************************************************************************  
9   *                                                                         *  #ifdef HAVE_CONFIG_H
10   *   This program is free software; you can redistribute it and/or modify  *  #include "config.h"
11   *   it under the terms of the GNU General Public License as published by  *  #endif
  *   the Free Software Foundation; either version 3 of the License, or     *  
  *   (at your option) any later version.                                   *  
  *                                                                         *  
  ***************************************************************************/  
12    
13  #include "common.h"  #include "common.h"
14  #include "database.h"  #include "database.h"
15  #include "log.h"  #include "log.h"
16  #include "trie_dict.h"  #include "trie_dict.h"
17  #include "user_list.h"  #include "user_list.h"
18    #include "user_stat.h"
19  #include <errno.h>  #include <errno.h>
20  #include <stdlib.h>  #include <stdlib.h>
21  #include <string.h>  #include <string.h>
# Line 40  union semun Line 37  union semun
37  };  };
38  #endif // #ifdef _SEM_SEMUN_UNDEFINED  #endif // #ifdef _SEM_SEMUN_UNDEFINED
39    
40  #define USER_LIST_TRY_LOCK_WAIT_TIME 1 // second  enum _user_list_constant_t
41  #define USER_LIST_TRY_LOCK_TIMES 10  {
42            USER_LIST_TRY_LOCK_WAIT_TIME = 1, // second
43            USER_LIST_TRY_LOCK_TIMES = 10,
44    };
45    
46  struct user_list_pool_t  struct user_list_pool_t
47  {  {
# Line 53  struct user_list_pool_t Line 53  struct user_list_pool_t
53          USER_ONLINE_LIST user_online_list[2];          USER_ONLINE_LIST user_online_list[2];
54          USER_ONLINE_LIST *p_online_current;          USER_ONLINE_LIST *p_online_current;
55          USER_ONLINE_LIST *p_online_new;          USER_ONLINE_LIST *p_online_new;
56            USER_STAT_MAP user_stat_map;
57            int user_login_count;
58  };  };
59  typedef struct user_list_pool_t USER_LIST_POOL;  typedef struct user_list_pool_t USER_LIST_POOL;
60    
# Line 91  static int user_list_rd_lock(int semid); Line 93  static int user_list_rd_lock(int semid);
93  static int user_list_rw_lock(int semid);  static int user_list_rw_lock(int semid);
94    
95  static int user_list_load(MYSQL *db, USER_LIST *p_list);  static int user_list_load(MYSQL *db, USER_LIST *p_list);
96  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);
97    static int user_login_count_load(MYSQL *db);
98    
99  static int user_info_index_uid_comp(const void *ptr1, const void *ptr2)  static int user_info_index_uid_comp(const void *ptr1, const void *ptr2)
100  {  {
# Line 119  static int user_info_index_uid_comp(cons Line 122  static int user_info_index_uid_comp(cons
122    
123  int user_list_load(MYSQL *db, USER_LIST *p_list)  int user_list_load(MYSQL *db, USER_LIST *p_list)
124  {  {
         USER_INFO_INDEX_UID index_uid[BBS_max_user_count];  
125          MYSQL_RES *rs = NULL;          MYSQL_RES *rs = NULL;
126          MYSQL_ROW row;          MYSQL_ROW row;
127          char sql[SQL_BUFFER_LEN];          char sql[SQL_BUFFER_LEN];
128          int ret = 0;          int ret = 0;
129          int i;          int i;
130          int32_t last_uid = -1;          int j;
131            int32_t last_uid;
132            size_t intro_buf_offset;
133            size_t intro_len;
134    
135          if (db == NULL || p_list == NULL)          if (db == NULL || p_list == NULL)
136          {          {
# Line 137  int user_list_load(MYSQL *db, USER_LIST Line 142  int user_list_load(MYSQL *db, USER_LIST
142          {          {
143                  last_uid = p_list->users[p_list->user_count - 1].uid;                  last_uid = p_list->users[p_list->user_count - 1].uid;
144          }          }
145            else
146            {
147                    last_uid = -1;
148            }
149    
150          snprintf(sql, sizeof(sql),          snprintf(sql, sizeof(sql),
151                           "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, "
152                           "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), "
153                             "UNIX_TIMESTAMP(birthday), `introduction` "
154                           "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 "
155                           "INNER JOIN user_reginfo ON user_list.UID = user_reginfo.UID "                           "INNER JOIN user_reginfo ON user_list.UID = user_reginfo.UID "
156                           "WHERE enable ORDER BY username");                           "WHERE enable ORDER BY username");
# Line 159  int user_list_load(MYSQL *db, USER_LIST Line 169  int user_list_load(MYSQL *db, USER_LIST
169                  goto cleanup;                  goto cleanup;
170          }          }
171    
172            intro_buf_offset = 0;
173          i = 0;          i = 0;
174          while ((row = mysql_fetch_row(rs)))          while ((row = mysql_fetch_row(rs)))
175          {          {
# Line 173  int user_list_load(MYSQL *db, USER_LIST Line 184  int user_list_load(MYSQL *db, USER_LIST
184                  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]));
185                  p_list->users[i].life = (row[5] == NULL ? 0 : atoi(row[5]));                  p_list->users[i].life = (row[5] == NULL ? 0 : atoi(row[5]));
186                  p_list->users[i].exp = (row[6] == NULL ? 0 : atoi(row[6]));                  p_list->users[i].exp = (row[6] == NULL ? 0 : atoi(row[6]));
187                  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]));
188                  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]));
189                  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]));
190                    p_list->users[i].last_logout_dt = (row[10] == NULL ? 0 : atol(row[10]));
191                  // index                  p_list->users[i].birthday = (row[11] == NULL ? 0 : atol(row[11]));
192                  index_uid[i].uid = p_list->users[i].uid;                  intro_len = strlen((row[12] == NULL ? "" : row[12]));
193                  index_uid[i].id = i;                  if (intro_len >= sizeof(p_list->user_intro_buf) - 1 - intro_buf_offset)
194                    {
195                            log_error("OOM for user introduction: len=%d, i=%d\n", intro_len, i);
196                            break;
197                    }
198                    memcpy(p_list->user_intro_buf + intro_buf_offset,
199                               (row[12] == NULL ? "" : row[12]),
200                               intro_len + 1);
201                    p_list->users[i].intro = p_list->user_intro_buf + intro_buf_offset;
202                    intro_buf_offset += (intro_len + 1);
203    
204                  i++;                  i++;
205                  if (i >= BBS_max_user_count)                  if (i >= BBS_max_user_count)
# Line 191  int user_list_load(MYSQL *db, USER_LIST Line 211  int user_list_load(MYSQL *db, USER_LIST
211          mysql_free_result(rs);          mysql_free_result(rs);
212          rs = NULL;          rs = NULL;
213    
         // Sort index  
214          if (i != p_list->user_count || p_list->users[i - 1].uid != last_uid) // Count of users changed          if (i != p_list->user_count || p_list->users[i - 1].uid != last_uid) // Count of users changed
215          {          {
216                  qsort(index_uid, (size_t)i, sizeof(USER_INFO_INDEX_UID), user_info_index_uid_comp);                  // Rebuild index
217                  memcpy(p_list->index_uid, index_uid, sizeof(USER_INFO_INDEX_UID) * (size_t)i);                  for (j = 0; j < i; j++)
218                    {
219                            p_list->index_uid[j].uid = p_list->users[j].uid;
220                            p_list->index_uid[j].id = j;
221                    }
222    
223                    qsort(p_list->index_uid, (size_t)i, sizeof(USER_INFO_INDEX_UID), user_info_index_uid_comp);
224    
225  #ifdef _DEBUG  #ifdef _DEBUG
226                  log_error("Rebuild index of %d users, last_uid=%d\n", i, p_list->users[i - 1].uid);                  log_error("Rebuild index of %d users, last_uid=%d\n", i, p_list->users[i - 1].uid);
# Line 214  cleanup: Line 239  cleanup:
239          return ret;          return ret;
240  }  }
241    
242  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)
243  {  {
244          MYSQL_RES *rs = NULL;          MYSQL_RES *rs = NULL;
245          MYSQL_ROW row;          MYSQL_ROW row;
246          char sql[SQL_BUFFER_LEN];          char sql[SQL_BUFFER_LEN];
247          int ret = 0;          int ret = 0;
248          int i;          int i;
249            int j;
250            int user_cnt;
251            int guest_cnt;
252    
253          if (db == NULL || p_list == NULL)          if (db == NULL || p_online_list == NULL)
254          {          {
255                  log_error("NULL pointer error\n");                  log_error("NULL pointer error\n");
256                  return -1;                  return -1;
# Line 231  int user_online_list_load(MYSQL *db, USE Line 259  int user_online_list_load(MYSQL *db, USE
259          snprintf(sql, sizeof(sql),          snprintf(sql, sizeof(sql),
260                           "SELECT SID, UID, ip, current_action, UNIX_TIMESTAMP(login_tm), "                           "SELECT SID, UID, ip, current_action, UNIX_TIMESTAMP(login_tm), "
261                           "UNIX_TIMESTAMP(last_tm) FROM user_online "                           "UNIX_TIMESTAMP(last_tm) FROM user_online "
262                           "WHERE last_tm >= SUBDATE(NOW(), INTERVAL %d SECOND) AND UID <> 0 "                           "WHERE last_tm >= SUBDATE(NOW(), INTERVAL %d SECOND) "
263                           "ORDER BY last_tm DESC",                           "ORDER BY last_tm DESC",
264                           BBS_user_off_line);                           BBS_user_off_line);
265    
# Line 250  int user_online_list_load(MYSQL *db, USE Line 278  int user_online_list_load(MYSQL *db, USE
278          }          }
279    
280          i = 0;          i = 0;
281            user_cnt = 0;
282            guest_cnt = 0;
283          while ((row = mysql_fetch_row(rs)))          while ((row = mysql_fetch_row(rs)))
284          {          {
285                  p_list->users[i].id = i;                  if (atoi(row[1]) == 0) // guest
286                  strncpy(p_list->users[i].session_id, row[0], sizeof(p_list->users[i].session_id) - 1);                  {
287                  p_list->users[i].session_id[sizeof(p_list->users[i].session_id) - 1] = '\0';                          guest_cnt++;
288                            continue;
289                    }
290                    else
291                    {
292                            user_cnt++;
293                    }
294    
295                  if ((ret = query_user_info_by_uid(atoi(row[1]), &(p_list->users[i].user_info))) <= 0)                  p_online_list->users[i].id = i;
296                    strncpy(p_online_list->users[i].session_id, row[0], sizeof(p_online_list->users[i].session_id) - 1);
297                    p_online_list->users[i].session_id[sizeof(p_online_list->users[i].session_id) - 1] = '\0';
298    
299                    if ((ret = query_user_info_by_uid(atoi(row[1]), &(p_online_list->users[i].user_info), NULL, 0)) <= 0)
300                  {                  {
301                          log_error("query_user_info_by_uid(%d) error\n", atoi(row[1]));                          log_error("query_user_info_by_uid(%d) error\n", atoi(row[1]));
302                          continue;                          continue;
303                  }                  }
304    
305                  strncpy(p_list->users[i].ip, row[2], sizeof(p_list->users[i].ip) - 1);                  strncpy(p_online_list->users[i].ip, row[2], sizeof(p_online_list->users[i].ip) - 1);
306                  p_list->users[i].ip[sizeof(p_list->users[i].ip) - 1] = '\0';                  p_online_list->users[i].ip[sizeof(p_online_list->users[i].ip) - 1] = '\0';
307    
308                  strncpy(p_list->users[i].current_action, row[3], sizeof(p_list->users[i].current_action) - 1);                  strncpy(p_online_list->users[i].current_action, row[3], sizeof(p_online_list->users[i].current_action) - 1);
309                  p_list->users[i].current_action[sizeof(p_list->users[i].current_action) - 1] = '\0';                  p_online_list->users[i].current_action[sizeof(p_online_list->users[i].current_action) - 1] = '\0';
310                  p_list->users[i].current_action_title = NULL;                  p_online_list->users[i].current_action_title = NULL;
311                  if (p_list->users[i].current_action[0] == '\0')                  if (p_online_list->users[i].current_action[0] == '\0')
312                  {                  {
313                          p_list->users[i].current_action_title = "";                          p_online_list->users[i].current_action_title = "";
314                  }                  }
315                  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)                  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)
316                  {                  {
317                          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",
318                                            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);
319                          continue;                          continue;
320                  }                  }
321    
322                  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]));
323                  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]));
   
                 // index  
                 p_list->index_uid[i].uid = p_list->users[i].user_info.uid;  
                 p_list->index_uid[i].id = i;  
324    
325                  i++;                  i++;
326                  if (i >= BBS_max_user_online_count)                  if (i >= BBS_max_user_online_count)
# Line 296  int user_online_list_load(MYSQL *db, USE Line 332  int user_online_list_load(MYSQL *db, USE
332          mysql_free_result(rs);          mysql_free_result(rs);
333          rs = NULL;          rs = NULL;
334    
335          // Sort index          if (user_cnt > 0)
         if (i > 0)  
336          {          {
337                  qsort(p_list->index_uid, (size_t)i, sizeof(USER_INFO_INDEX_UID), user_info_index_uid_comp);                  // Rebuild index
338  #ifdef _DEBUG                  for (j = 0; j < user_cnt; j++)
339                  log_error("Rebuild index of %d online users\n", i);                  {
340  #endif                          p_online_list->index_uid[j].uid = p_online_list->users[j].user_info.uid;
341          }                          p_online_list->index_uid[j].id = j;
342                    }
343    
344          p_list->user_count = i;                  qsort(p_online_list->index_uid, (size_t)user_cnt, sizeof(USER_INFO_INDEX_UID), user_info_index_uid_comp);
345            }
346    
347  #ifdef _DEBUG          p_online_list->user_count = user_cnt;
348          log_error("Loaded %d online users\n", p_list->user_count);          p_online_list->guest_count = guest_cnt;
 #endif  
349    
350  cleanup:  cleanup:
351          mysql_free_result(rs);          mysql_free_result(rs);
# Line 317  cleanup: Line 353  cleanup:
353          return ret;          return ret;
354  }  }
355    
356  int user_list_pool_init(void)  int user_login_count_load(MYSQL *db)
357    {
358            MYSQL_RES *rs = NULL;
359            MYSQL_ROW row;
360            char sql[SQL_BUFFER_LEN];
361    
362            if (db == NULL)
363            {
364                    log_error("NULL pointer error\n");
365                    return -1;
366            }
367    
368            snprintf(sql, sizeof(sql),
369                             "SELECT ID FROM user_login_log ORDER BY ID DESC LIMIT 1");
370            if (mysql_query(db, sql) != 0)
371            {
372                    log_error("Query user_login_log error: %s\n", mysql_error(db));
373                    return -2;
374            }
375            if ((rs = mysql_store_result(db)) == NULL)
376            {
377                    log_error("Get user_login_log data failed\n");
378                    return -2;
379            }
380            if ((row = mysql_fetch_row(rs)))
381            {
382                    p_user_list_pool->user_login_count = atoi(row[0]);
383            }
384            mysql_free_result(rs);
385    
386            return 0;
387    }
388    
389    int user_list_pool_init(const char *filename)
390  {  {
391          int shmid;          int shmid;
392          int semid;          int semid;
# Line 351  int user_list_pool_init(void) Line 420  int user_list_pool_init(void)
420    
421          // Allocate shared memory          // Allocate shared memory
422          proj_id = (int)(time(NULL) % getpid());          proj_id = (int)(time(NULL) % getpid());
423          key = ftok(VAR_USER_LIST_SHM, proj_id);          key = ftok(filename, proj_id);
424          if (key == -1)          if (key == -1)
425          {          {
426                  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);
427                  return -2;                  return -2;
428          }          }
429    
# Line 407  int user_list_pool_init(void) Line 476  int user_list_pool_init(void)
476          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]);
477          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]);
478    
479            user_stat_map_init(&(p_user_list_pool->user_stat_map));
480    
481          return 0;          return 0;
482  }  }
483    
# Line 431  void user_list_pool_cleanup(void) Line 502  void user_list_pool_cleanup(void)
502                  log_error("shmdt(shmid = %d) error (%d)\n", shmid, errno);                  log_error("shmdt(shmid = %d) error (%d)\n", shmid, errno);
503          }          }
504    
505          if (shmctl(shmid, IPC_RMID, NULL) == -1)          if (shmid != 0 && shmctl(shmid, IPC_RMID, NULL) == -1)
506          {          {
507                  log_error("shmctl(shmid = %d, IPC_RMID) error (%d)\n", shmid, errno);                  log_error("shmctl(shmid = %d, IPC_RMID) error (%d)\n", shmid, errno);
508          }          }
# Line 513  int user_list_pool_reload(int online_use Line 584  int user_list_pool_reload(int online_use
584                          ret = -2;                          ret = -2;
585                          goto cleanup;                          goto cleanup;
586                  }                  }
587    
588                    if (user_login_count_load(db) < 0)
589                    {
590                            log_error("user_login_count_load() error\n");
591                            ret = -2;
592                            goto cleanup;
593                    }
594          }          }
595          else          else
596          {          {
# Line 826  cleanup: Line 904  cleanup:
904          return ret;          return ret;
905  }  }
906    
907    int get_user_list_count(int *p_user_cnt)
908    {
909            if (p_user_cnt == NULL)
910            {
911                    log_error("NULL pointer error\n");
912                    return -1;
913            }
914    
915            // acquire lock of user list
916            if (user_list_rd_lock(p_user_list_pool->semid) < 0)
917            {
918                    log_error("user_list_rd_lock() error\n");
919                    return -2;
920            }
921    
922            *p_user_cnt = p_user_list_pool->p_current->user_count;
923    
924            // release lock of user list
925            if (user_list_rd_unlock(p_user_list_pool->semid) < 0)
926            {
927                    log_error("user_list_rd_unlock() error\n");
928                    return -2;
929            }
930    
931            return 0;
932    }
933    
934    int get_user_online_list_count(int *p_user_cnt, int *p_guest_cnt)
935    {
936            if (p_user_cnt == NULL || p_guest_cnt == NULL)
937            {
938                    log_error("NULL pointer error\n");
939                    return -1;
940            }
941    
942            // acquire lock of user list
943            if (user_list_rd_lock(p_user_list_pool->semid) < 0)
944            {
945                    log_error("user_list_rd_lock() error\n");
946                    return -2;
947            }
948    
949            *p_user_cnt = p_user_list_pool->p_online_current->user_count;
950            *p_guest_cnt = p_user_list_pool->p_online_current->guest_count;
951    
952            // release lock of user list
953            if (user_list_rd_unlock(p_user_list_pool->semid) < 0)
954            {
955                    log_error("user_list_rd_unlock() error\n");
956                    return -2;
957            }
958    
959            return 0;
960    }
961    
962    int get_user_login_count(int *p_login_cnt)
963    {
964            if (p_login_cnt == NULL)
965            {
966                    log_error("NULL pointer error\n");
967                    return -1;
968            }
969    
970            *p_login_cnt = p_user_list_pool->user_login_count;
971    
972            return 0;
973    }
974    
975  int query_user_info(int32_t id, USER_INFO *p_user)  int query_user_info(int32_t id, USER_INFO *p_user)
976  {  {
977          int ret = 0;          int ret = 0;
# Line 859  int query_user_info(int32_t id, USER_INF Line 1005  int query_user_info(int32_t id, USER_INF
1005          return ret;          return ret;
1006  }  }
1007    
1008  int query_user_info_by_uid(int32_t uid, USER_INFO *p_user)  int query_user_info_by_uid(int32_t uid, USER_INFO *p_user, char *p_intro_buf, size_t intro_buf_len)
1009  {  {
1010          int left;          int left;
1011          int right;          int right;
# Line 888  int query_user_info_by_uid(int32_t uid, Line 1034  int query_user_info_by_uid(int32_t uid,
1034                  mid = (left + right) / 2;                  mid = (left + right) / 2;
1035                  if (uid < p_user_list_pool->p_current->index_uid[mid].uid)                  if (uid < p_user_list_pool->p_current->index_uid[mid].uid)
1036                  {                  {
1037                          right = mid;                          right = mid - 1;
1038                  }                  }
1039                  else if (uid > p_user_list_pool->p_current->index_uid[mid].uid)                  else if (uid > p_user_list_pool->p_current->index_uid[mid].uid)
1040                  {                  {
# Line 906  int query_user_info_by_uid(int32_t uid, Line 1052  int query_user_info_by_uid(int32_t uid,
1052                  id = p_user_list_pool->p_current->index_uid[left].id;                  id = p_user_list_pool->p_current->index_uid[left].id;
1053                  *p_user = p_user_list_pool->p_current->users[id];                  *p_user = p_user_list_pool->p_current->users[id];
1054                  ret = 1;                  ret = 1;
1055    
1056                    if (p_intro_buf != NULL)
1057                    {
1058                            strncpy(p_intro_buf, p_user_list_pool->p_current->users[id].intro, intro_buf_len - 1);
1059                            p_intro_buf[intro_buf_len - 1] = '\0';
1060                            p_user->intro = p_intro_buf;
1061                    }
1062            }
1063    
1064            // release lock of user list
1065            if (user_list_rd_unlock(p_user_list_pool->semid) < 0)
1066            {
1067                    log_error("user_list_rd_unlock() error\n");
1068                    ret = -1;
1069            }
1070    
1071            return ret;
1072    }
1073    
1074    int query_user_info_by_username(const char *username_prefix, int max_user_cnt,
1075                                                                    int32_t uid_list[], char username_list[][BBS_username_max_len + 1])
1076    {
1077            int left;
1078            int right;
1079            int mid;
1080            int left_save;
1081            int ret = 0;
1082            size_t prefix_len;
1083            int comp;
1084            int i;
1085    
1086            if (username_prefix == NULL || uid_list == NULL || username_list == NULL)
1087            {
1088                    log_error("NULL pointer error\n");
1089                    return -1;
1090            }
1091    
1092            prefix_len = strlen(username_prefix);
1093    
1094            // acquire lock of user list
1095            if (user_list_rd_lock(p_user_list_pool->semid) < 0)
1096            {
1097                    log_error("user_list_rd_lock() error\n");
1098                    return -2;
1099            }
1100    
1101            left = 0;
1102            right = p_user_list_pool->p_current->user_count - 1;
1103    
1104            while (left < right)
1105            {
1106                    mid = (left + right) / 2;
1107                    comp = strncasecmp(username_prefix, p_user_list_pool->p_current->users[mid].username, prefix_len);
1108                    if (comp < 0)
1109                    {
1110                            right = mid - 1;
1111                    }
1112                    else if (comp > 0)
1113                    {
1114                            left = mid + 1;
1115                    }
1116                    else // if (comp == 0)
1117                    {
1118                            left = mid;
1119                            break;
1120                    }
1121            }
1122    
1123            if (strncasecmp(username_prefix, p_user_list_pool->p_current->users[left].username, prefix_len) == 0) // Found
1124            {
1125    #ifdef _DEBUG
1126                    log_error("Debug: match found, pos=%d\n", left);
1127    #endif
1128    
1129                    left_save = left;
1130                    right = left;
1131                    left = 0;
1132    
1133                    while (left < right)
1134                    {
1135                            mid = (left + right) / 2;
1136                            comp = strncasecmp(username_prefix, p_user_list_pool->p_current->users[mid].username, prefix_len);
1137                            if (comp > 0)
1138                            {
1139                                    left = mid + 1;
1140                            }
1141                            else if (comp == 0)
1142                            {
1143                                    right = mid;
1144                            }
1145                            else // if (comp < 0)
1146                            {
1147                                    log_error("Bug: left=%d right=%d mid=%d");
1148                                    ret = -2;
1149                                    goto cleanup;
1150                            }
1151                    }
1152    
1153    #ifdef _DEBUG
1154                    log_error("Debug: first match found, pos=%d\n", right);
1155    #endif
1156    
1157                    left = left_save;
1158                    left_save = right;
1159                    right = p_user_list_pool->p_current->user_count - 1;
1160    
1161                    while (left < right)
1162                    {
1163                            mid = (left + right) / 2 + (left + right) % 2;
1164                            comp = strncasecmp(username_prefix, p_user_list_pool->p_current->users[mid].username, prefix_len);
1165                            if (comp < 0)
1166                            {
1167                                    right = mid - 1;
1168                            }
1169                            else if (comp == 0)
1170                            {
1171                                    left = mid;
1172                            }
1173                            else // if (comp > 0)
1174                            {
1175                                    log_error("Bug: left=%d right=%d mid=%d");
1176                                    ret = -2;
1177                                    goto cleanup;
1178                            }
1179                    }
1180    
1181    #ifdef _DEBUG
1182                    log_error("Debug: last match found, pos=%d\n", left);
1183    #endif
1184    
1185                    right = left;
1186                    left = left_save;
1187    
1188                    for (i = 0; i < max_user_cnt && left + i <= right; i++)
1189                    {
1190                            uid_list[i] = p_user_list_pool->p_current->users[left + i].uid;
1191                            strncpy(username_list[i], p_user_list_pool->p_current->users[left + i].username, sizeof(username_list[i]) - 1);
1192                            username_list[i][sizeof(username_list[i]) - 1] = '\0';
1193                    }
1194                    ret = i;
1195          }          }
1196    
1197    cleanup:
1198          // release lock of user list          // release lock of user list
1199          if (user_list_rd_unlock(p_user_list_pool->semid) < 0)          if (user_list_rd_unlock(p_user_list_pool->semid) < 0)
1200          {          {
# Line 959  int query_user_online_info_by_uid(int32_ Line 1246  int query_user_online_info_by_uid(int32_
1246          int32_t id;          int32_t id;
1247          int ret = 0;          int ret = 0;
1248          int i;          int i;
1249            int user_cnt;
1250    
1251          if (p_users == NULL || p_user_cnt == NULL)          if (p_users == NULL || p_user_cnt == NULL)
1252          {          {
# Line 966  int query_user_online_info_by_uid(int32_ Line 1254  int query_user_online_info_by_uid(int32_
1254                  return -1;                  return -1;
1255          }          }
1256    
1257            user_cnt = *p_user_cnt;
1258            *p_user_cnt = 0;
1259    
1260          // acquire lock of user list          // acquire lock of user list
1261          if (user_list_rd_lock(p_user_list_pool->semid) < 0)          if (user_list_rd_lock(p_user_list_pool->semid) < 0)
1262          {          {
# Line 981  int query_user_online_info_by_uid(int32_ Line 1272  int query_user_online_info_by_uid(int32_
1272                  mid = (left + right) / 2;                  mid = (left + right) / 2;
1273                  if (uid < p_user_list_pool->p_online_current->index_uid[mid].uid)                  if (uid < p_user_list_pool->p_online_current->index_uid[mid].uid)
1274                  {                  {
1275                          right = mid;                          right = mid - 1;
1276                  }                  }
1277                  else if (uid > p_user_list_pool->p_online_current->index_uid[mid].uid)                  else if (uid > p_user_list_pool->p_online_current->index_uid[mid].uid)
1278                  {                  {
# Line 1013  int query_user_online_info_by_uid(int32_ Line 1304  int query_user_online_info_by_uid(int32_
1304                  }                  }
1305    
1306                  for (i = 0;                  for (i = 0;
1307                           left < p_user_list_pool->p_online_current->user_count && i < *p_user_cnt &&                           left < p_user_list_pool->p_online_current->user_count && i < user_cnt &&
1308                           uid == p_user_list_pool->p_online_current->index_uid[left].uid;                           uid == p_user_list_pool->p_online_current->index_uid[left].uid;
1309                           left++, i++)                           left++, i++)
1310                  {                  {
# Line 1037  int query_user_online_info_by_uid(int32_ Line 1328  int query_user_online_info_by_uid(int32_
1328    
1329          return ret;          return ret;
1330  }  }
1331    
1332    int get_user_id_list(int32_t *p_uid_list, int *p_user_cnt, int start_uid)
1333    {
1334            int left;
1335            int right;
1336            int mid;
1337            int ret = 0;
1338            int i;
1339    
1340            if (p_uid_list == NULL || p_user_cnt == NULL)
1341            {
1342                    log_error("NULL pointer error\n");
1343                    return -1;
1344            }
1345    
1346            // acquire lock of user list
1347            if (user_list_rd_lock(p_user_list_pool->semid) < 0)
1348            {
1349                    log_error("user_list_rd_lock() error\n");
1350                    return -2;
1351            }
1352    
1353            left = 0;
1354            right = p_user_list_pool->p_current->user_count - 1;
1355    
1356            while (left < right)
1357            {
1358                    mid = (left + right) / 2;
1359                    if (start_uid < p_user_list_pool->p_current->index_uid[mid].uid)
1360                    {
1361                            right = mid - 1;
1362                    }
1363                    else if (start_uid > p_user_list_pool->p_current->index_uid[mid].uid)
1364                    {
1365                            left = mid + 1;
1366                    }
1367                    else // if (start_uid == p_user_list_pool->p_current->index_uid[mid].uid)
1368                    {
1369                            left = mid;
1370                            break;
1371                    }
1372            }
1373    
1374            for (i = 0; i < *p_user_cnt && left + i < p_user_list_pool->p_current->user_count; i++)
1375            {
1376                    p_uid_list[i] = p_user_list_pool->p_current->index_uid[left + i].uid;
1377            }
1378            *p_user_cnt = i;
1379    
1380            // release lock of user list
1381            if (user_list_rd_unlock(p_user_list_pool->semid) < 0)
1382            {
1383                    log_error("user_list_rd_unlock() error\n");
1384                    ret = -1;
1385            }
1386    
1387            return ret;
1388    }
1389    
1390    int user_stat_update(void)
1391    {
1392            return user_stat_map_update(&(p_user_list_pool->user_stat_map));
1393    }
1394    
1395    int user_article_cnt_inc(int32_t uid, int n)
1396    {
1397            return user_stat_article_cnt_inc(&(p_user_list_pool->user_stat_map), uid, n);
1398    }
1399    
1400    int get_user_article_cnt(int32_t uid)
1401    {
1402            const USER_STAT *p_stat;
1403            int ret;
1404    
1405            ret = user_stat_get(&(p_user_list_pool->user_stat_map), uid, &p_stat);
1406            if (ret < 0)
1407            {
1408                    log_error("user_stat_get(uid=%d) error: %d\n", uid, ret);
1409                    return -1;
1410            }
1411            else if (ret == 0) // user not found
1412            {
1413                    return -1;
1414            }
1415    
1416            return p_stat->article_count;
1417    }

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