读写消息

Author: suyh

tmp-flink-queue/backup/opt_long.c

@@ -14,9 +14,9 @@
        示例：在struct option long_options[]数组中，如果有一个元素{"help", 0, 0, 'h'}，这里的"help"就是长选项的名称。这意味着在命令行中，用户可以使用--help来表示这个选项。长选项名称通常是具有明确语义的单词，方便用户记忆和使用，比如"version"、"input - file"等，这样用户通过--version或--input - file这样的命令行参数就能很直观地理解选项的用途。
    has_arg属性
        定义：has_arg用于表示长选项是否需要参数以及参数的性质，它是一个整数，有以下几种取值：
-       0：表示长选项不需要参数。例如，{"verbose", 0, 0, 'v'}中的verbose选项可能只是一个开关，用于开启或关闭详细输出模式，不需要额外的参数跟在后面。
-       1：表示长选项需要一个参数。比如{"input - file", 1, 0, 'i'}，当用户在命令行输入--input - file filename.txt时，filename.txt就是这个长选项input - file的参数，用于指定输入文件的名称。
-       2：表示长选项的参数是可选的。这种情况相对复杂一些，例如{"output - file", 2, 0, 'o'}，用户可以选择提供一个输出文件名称作为参数，如--output - file result.txt，也可以不提供参数，程序可能会使用默认的输出文件名或者有其他的处理方式。
+       0(no_argument)：表示长选项不需要参数。例如，{"verbose", 0, 0, 'v'}中的verbose选项可能只是一个开关，用于开启或关闭详细输出模式，不需要额外的参数跟在后面。
+       1(required_argument)：表示长选项需要一个参数。比如{"input - file", 1, 0, 'i'}，当用户在命令行输入--input - file filename.txt时，filename.txt就是这个长选项input - file的参数，用于指定输入文件的名称。
+       2(optional_argument)：表示长选项的参数是可选的。这种情况相对复杂一些，例如{"output - file", 2, 0, 'o'}，用户可以选择提供一个输出文件名称作为参数，如--output - file result.txt，也可以不提供参数，程序可能会使用默认的输出文件名或者有其他的处理方式。
    *flag属性
        定义：*flag是一个指针，通常用于和val属性配合来改变一个变量的值。如果flag为NULL，getopt_long函数返回val的值作为选项的返回值；如果flag不为NULL，当找到该选项时，*flag指向的变量会被设置为val的值。
        示例：假设我们有一个变量int option_enabled = 0;，并且有一个选项{"enable - option", 0, &option_enabled, 1}。当在命令行中发现--enable - option这个选项时，option_enabled的值会被设置为1，用于表示这个选项被启用。

tmp-flink-queue/v002/ipcmqs.c

@@ -20,18 +20,19 @@ struct msgbuf {
 
 // 打印使用帮助信息
 void usage() {
-    printf("Usage:./ipc_msg_queue [-r|-w] [--dates <date>] [--pns <pns>] [--channelList <channels>]\n");
+    printf("Usage:./ipc_msg_queue [-r|-w] [--dates <date>] [--pns <pns>] [--channelList <channels>] [--jobName <jobName>]\n");
     printf("Options:\n");
     printf("  -r, --read        Read from the message queue\n");
     printf("  -w, --write       Write to the message queue\n");
     printf("  -d, --dates       Specify date\n");
     printf("  -p, --pns         Specify pns\n");
     printf("  -c, --channelList Specify channel list\n");
+    printf("  -j, --jobName     Specify job name\n");
     exit(EXIT_FAILURE);
 }
 
 // 格式化消息并返回消息长度
-int formatMessage(char *buffer, const char *date, const char *pns, const char *channelList) {
+int formatMessage(char *buffer, const char *date, const char *pns, const char *channelList, const char *jobName) {
     int offset = 0;
     int dateLen = strlen(date);
     *((int*)(buffer + offset)) = dateLen;
@@ -51,11 +52,17 @@ int formatMessage(char *buffer, const char *date, const char *pns, const char *c
     strcpy(buffer + offset, channelList);
     offset += channelListLen;
 
+    int jobNameLen = strlen(jobName);
+    *((int*)(buffer + offset)) = jobNameLen;
+    offset += sizeof(int);
+    strcpy(buffer + offset, jobName);
+    offset += jobNameLen;
+
     return offset;
 }
 
 // 解析读取到的消息
-void parseReceivedMessage(const char *buffer, char *date, char *pns, char *channelList) {
+void parseReceivedMessage(const char *buffer, char *date, char *pns, char *channelList, char *jobName) {
     int offset = 0;
     int dateLen = *((int*)(buffer + offset));
     offset += sizeof(int);
@@ -73,21 +80,30 @@ void parseReceivedMessage(const char *buffer, char *date, char *pns, char *chann
     offset += sizeof(int);
     strncpy(channelList, buffer + offset, channelListLen);
     channelList[channelListLen] = '\0';
+    offset += channelListLen;
+
+    int jobNameLen = *((int*)(buffer + offset));
+    offset += sizeof(int);
+    strncpy(jobName, buffer + offset, jobNameLen);
+    jobName[jobNameLen] = '\0';
+    offset += channelListLen;
 }
 
-// ./ipcmqs -w --pns hy --date 20241102
+// ./ipcmqs -w --pns hy --date 20241102 --jobName cohort
 int main(int argc, char *argv[]) {
     int opt;
     int operation = OTHER_OPERATION;
-    char date[9] = "";
-    char pns[800] = "";
+    char date[20] = "";
+    char pns[1024] = "";
     char channelList[2048] = "";
+    char jobName[1024] = "";
     struct option long_options[] = {
         {"read", no_argument, 0, 'r'},
         {"write", no_argument, 0, 'w'},
         {"dates", required_argument, 0, 'd'},
         {"pns", required_argument, 0, 'p'},
         {"channelList", no_argument, 0, 'c'},
+        {"jobName", required_argument, 0, 'j'},
         {0, 0, 0, 0}
     };
 
@@ -108,6 +124,9 @@ int main(int argc, char *argv[]) {
             case 'c':
                 strcpy(channelList, optarg);
                 break;
+            case 'j':
+                strcpy(jobName, optarg);
+                break;
             default:
                 usage();
         }
@@ -144,18 +163,19 @@ int main(int argc, char *argv[]) {
             exit(EXIT_SUCCESS);
         }
 
-        parseReceivedMessage(buffer.mdata, date, pns, channelList);
+        parseReceivedMessage(buffer.mdata, date, pns, channelList, jobName);
         printf("dates: %s\n", date);
         printf("pns: %s\n", pns);
         printf("channelList: %s\n", channelList);
+        printf("jobName: %s\n", jobName);
     } else if (operation == WRITE_OPERATION) {
-        if (strlen(date) == 0 || strlen(pns) == 0) {
-            fprintf(stderr, "Dates and PNs cannot be empty when writing to the message queue.\n");
+        if (strlen(date) == 0 || strlen(pns) == 0 || strlen(jobName) == 0) {
+            fprintf(stderr, "Dates, pns jobName cannot be empty when writing to the message queue.\n");
             exit(EXIT_FAILURE);
         }
         struct msgbuf buffer;
         buffer.mtype = MSG_TYPE;
-        int msgLength = formatMessage(buffer.mdata, date, pns, channelList);
+        int msgLength = formatMessage(buffer.mdata, date, pns, channelList, jobName);
         if (msgsnd(msgid, &buffer, msgLength, 0) == -1) {
             perror("msgsnd");
             exit(EXIT_FAILURE);