/[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.17 by sysadm, Thu Oct 23 01:58:05 2025 UTC Revision 1.24 by sysadm, Sun Nov 2 04:30:29 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 92  static int user_list_rw_lock(int semid); Line 95  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_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)  static int user_info_index_uid_comp(const void *ptr1, const void *ptr2)
101  {  {
# Line 146  int user_list_load(MYSQL *db, USER_LIST Line 150  int user_list_load(MYSQL *db, USER_LIST
150    
151          snprintf(sql, sizeof(sql),          snprintf(sql, sizeof(sql),
152                           "SELECT user_list.UID AS UID, username, nickname, gender, gender_pub, life, exp, visit_count, "                           "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), `introduction` "                           "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 username");                           "WHERE enable ORDER BY username");
# Line 183  int user_list_load(MYSQL *db, USER_LIST Line 188  int user_list_load(MYSQL *db, USER_LIST
188                  p_list->users[i].visit_count = (row[7] == NULL ? 0 : atoi(row[7]));                  p_list->users[i].visit_count = (row[7] == NULL ? 0 : atoi(row[7]));
189                  p_list->users[i].signup_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].last_login_dt = (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].birthday = (row[10] == NULL ? 0 : atol(row[10]));                  p_list->users[i].last_logout_dt = (row[10] == NULL ? 0 : atol(row[10]));
192                  intro_len = strlen((row[11] == NULL ? "" : row[11]));                  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)                  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);                          log_error("OOM for user introduction: len=%d, i=%d\n", intro_len, i);
197                          break;                          break;
198                  }                  }
199                  memcpy(p_list->user_intro_buf + intro_buf_offset,                  memcpy(p_list->user_intro_buf + intro_buf_offset,
200                             (row[11] == NULL ? "" : row[11]),                             (row[12] == NULL ? "" : row[12]),
201                             intro_len + 1);                             intro_len + 1);
202                  p_list->users[i].intro = p_list->user_intro_buf + intro_buf_offset;                  p_list->users[i].intro = p_list->user_intro_buf + intro_buf_offset;
203                  intro_buf_offset += (intro_len + 1);                  intro_buf_offset += (intro_len + 1);
# Line 242  int user_online_list_load(MYSQL *db, USE Line 248  int user_online_list_load(MYSQL *db, USE
248          int ret = 0;          int ret = 0;
249          int i;          int i;
250          int j;          int j;
251            int user_cnt;
252            int guest_cnt;
253    
254          if (db == NULL || p_list == NULL)          if (db == NULL || p_list == NULL)
255          {          {
# Line 252  int user_online_list_load(MYSQL *db, USE Line 260  int user_online_list_load(MYSQL *db, USE
260          snprintf(sql, sizeof(sql),          snprintf(sql, sizeof(sql),
261                           "SELECT SID, UID, ip, current_action, UNIX_TIMESTAMP(login_tm), "                           "SELECT SID, UID, ip, current_action, UNIX_TIMESTAMP(login_tm), "
262                           "UNIX_TIMESTAMP(last_tm) FROM user_online "                           "UNIX_TIMESTAMP(last_tm) FROM user_online "
263                           "WHERE last_tm >= SUBDATE(NOW(), INTERVAL %d SECOND) AND UID <> 0 "                           "WHERE last_tm >= SUBDATE(NOW(), INTERVAL %d SECOND) "
264                           "ORDER BY last_tm DESC",                           "ORDER BY last_tm DESC",
265                           BBS_user_off_line);                           BBS_user_off_line);
266    
# Line 271  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                    if (atoi(row[1]) == 0) // guest
287                    {
288                            guest_cnt++;
289                            continue;
290                    }
291                    else
292                    {
293                            user_cnt++;
294                    }
295    
296                  p_list->users[i].id = i;                  p_list->users[i].id = i;
297                  strncpy(p_list->users[i].session_id, row[0], sizeof(p_list->users[i].session_id) - 1);                  strncpy(p_list->users[i].session_id, row[0], sizeof(p_list->users[i].session_id) - 1);
298                  p_list->users[i].session_id[sizeof(p_list->users[i].session_id) - 1] = '\0';                  p_list->users[i].session_id[sizeof(p_list->users[i].session_id) - 1] = '\0';
# Line 313  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          if (i > 0)          if (user_cnt > 0)
337          {          {
338                  // Rebuild index                  // Rebuild index
339                  for (j = 0; j < i; j++)                  for (j = 0; j < user_cnt; j++)
340                  {                  {
341                          p_list->index_uid[j].uid = p_list->users[j].user_info.uid;                          p_list->index_uid[j].uid = p_list->users[j].user_info.uid;
342                          p_list->index_uid[j].id = j;                          p_list->index_uid[j].id = j;
343                  }                  }
344    
345                  qsort(p_list->index_uid, (size_t)i, sizeof(USER_INFO_INDEX_UID), user_info_index_uid_comp);                  qsort(p_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("Rebuild index of %d online users\n", i);                  log_error("Rebuild index of %d online users\n", user_cnt);
349  #endif  #endif
350          }          }
351    
352          p_list->user_count = i;          p_list->user_count = user_cnt;
353            p_list->guest_count = guest_cnt;
354    
355  #ifdef _DEBUG  #ifdef _DEBUG
356          log_error("Loaded %d online users\n", p_list->user_count);          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 341  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 375  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 431  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 537  int user_list_pool_reload(int online_use Line 593  int user_list_pool_reload(int online_use
593                          ret = -2;                          ret = -2;
594                          goto cleanup;                          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
605          {          {
# Line 850  cleanup: Line 913  cleanup:
913          return ret;          return ret;
914  }  }
915    
916    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)  int query_user_info(int32_t id, USER_INFO *p_user)
985  {  {
986          int ret = 0;          int ret = 0;
# Line 912  int query_user_info_by_uid(int32_t uid, Line 1043  int query_user_info_by_uid(int32_t uid,
1043                  mid = (left + right) / 2;                  mid = (left + right) / 2;
1044                  if (uid < p_user_list_pool->p_current->index_uid[mid].uid)                  if (uid < p_user_list_pool->p_current->index_uid[mid].uid)
1045                  {                  {
1046                          right = mid;                          right = mid - 1;
1047                  }                  }
1048                  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)
1049                  {                  {
# Line 983  int query_user_online_info_by_uid(int32_ Line 1114  int query_user_online_info_by_uid(int32_
1114          int32_t id;          int32_t id;
1115          int ret = 0;          int ret = 0;
1116          int i;          int i;
1117            int user_cnt;
1118    
1119          if (p_users == NULL || p_user_cnt == NULL)          if (p_users == NULL || p_user_cnt == NULL)
1120          {          {
# Line 990  int query_user_online_info_by_uid(int32_ Line 1122  int query_user_online_info_by_uid(int32_
1122                  return -1;                  return -1;
1123          }          }
1124    
1125            user_cnt = *p_user_cnt;
1126            *p_user_cnt = 0;
1127    
1128          // acquire lock of user list          // acquire lock of user list
1129          if (user_list_rd_lock(p_user_list_pool->semid) < 0)          if (user_list_rd_lock(p_user_list_pool->semid) < 0)
1130          {          {
# Line 1005  int query_user_online_info_by_uid(int32_ Line 1140  int query_user_online_info_by_uid(int32_
1140                  mid = (left + right) / 2;                  mid = (left + right) / 2;
1141                  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)
1142                  {                  {
1143                          right = mid;                          right = mid - 1;
1144                  }                  }
1145                  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)
1146                  {                  {
# Line 1037  int query_user_online_info_by_uid(int32_ Line 1172  int query_user_online_info_by_uid(int32_
1172                  }                  }
1173    
1174                  for (i = 0;                  for (i = 0;
1175                           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 &&
1176                           uid == p_user_list_pool->p_online_current->index_uid[left].uid;                           uid == p_user_list_pool->p_online_current->index_uid[left].uid;
1177                           left++, i++)                           left++, i++)
1178                  {                  {
# Line 1061  int query_user_online_info_by_uid(int32_ Line 1196  int query_user_online_info_by_uid(int32_
1196    
1197          return ret;          return ret;
1198  }  }
1199    
1200    int get_user_id_list(int32_t *p_uid_list, int *p_user_cnt, int start_uid)
1201    {
1202            int left;
1203            int right;
1204            int mid;
1205            int ret = 0;
1206            int i;
1207    
1208            if (p_uid_list == NULL || p_user_cnt == NULL)
1209            {
1210                    log_error("NULL pointer error\n");
1211                    return -1;
1212            }
1213    
1214            // acquire lock of user list
1215            if (user_list_rd_lock(p_user_list_pool->semid) < 0)
1216            {
1217                    log_error("user_list_rd_lock() error\n");
1218                    return -2;
1219            }
1220    
1221            left = 0;
1222            right = p_user_list_pool->p_current->user_count - 1;
1223    
1224            while (left < right)
1225            {
1226                    mid = (left + right) / 2;
1227                    if (start_uid < p_user_list_pool->p_current->index_uid[mid].uid)
1228                    {
1229                            right = mid - 1;
1230                    }
1231                    else if (start_uid > p_user_list_pool->p_current->index_uid[mid].uid)
1232                    {
1233                            left = mid + 1;
1234                    }
1235                    else // if (start_uid == p_user_list_pool->p_current->index_uid[mid].uid)
1236                    {
1237                            left = mid;
1238                            break;
1239                    }
1240            }
1241    
1242            for (i = 0; i < *p_user_cnt && left + i < p_user_list_pool->p_current->user_count; i++)
1243            {
1244                    p_uid_list[i] = p_user_list_pool->p_current->index_uid[left + i].uid;
1245            }
1246            *p_user_cnt = i;
1247    
1248            // release lock of user list
1249            if (user_list_rd_unlock(p_user_list_pool->semid) < 0)
1250            {
1251                    log_error("user_list_rd_unlock() error\n");
1252                    ret = -1;
1253            }
1254    
1255            return ret;
1256    }
1257    
1258    int user_stat_update(void)
1259    {
1260            return user_stat_map_update(&(p_user_list_pool->user_stat_map));
1261    }
1262    
1263    int user_article_cnt_inc(int32_t uid, int n)
1264    {
1265            return user_stat_article_cnt_inc(&(p_user_list_pool->user_stat_map), uid, n);
1266    }
1267    
1268    int get_user_article_cnt(int32_t uid)
1269    {
1270            const USER_STAT *p_stat;
1271            int ret;
1272    
1273            ret = user_stat_get(&(p_user_list_pool->user_stat_map), uid, &p_stat);
1274            if (ret < 0)
1275            {
1276                    log_error("user_stat_get(uid=%d) error: %d\n", uid, ret);
1277                    return -1;
1278            }
1279            else if (ret == 0) // user not found
1280            {
1281                    return -1;
1282            }
1283    
1284            return p_stat->article_count;
1285    }

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