kirill-refactor

mine/lab_5/worker.c

@@ -0,0 +1,123 @@
+#define _GNU_SOURCE    // для определения расширенных возможностей glibc
+#include <errno.h>
+#include <mqueue.h>    // POSIX очереди сообщений: mq_open, mq_send, mq_receive
+#include <signal.h>
+#include <stdint.h>
+#include <stdio.h>     // fprintf, perror, dprintf
+#include <stdlib.h>    // atoi, rand_r
+#include <string.h>    // memset, strncpy, snprintf
+#include <sys/stat.h>  // права для mq_open (0666)
+#include <time.h>      // time(), nanosleep, struct timespec
+#include <unistd.h>    // getpid, STDOUT_FILENO
+
+#include "common.h"    // REQ_QUEUE, NAME_MAXLEN, req_msg_t, rep_msg_t
+
+// Пауза на заданное количество миллисекунд
+static void msleep(int ms) {
+    struct timespec ts = {
+        .tv_sec = ms / 1000,
+        .tv_nsec = (ms % 1000) * 1000000L
+    };
+    nanosleep(&ts, NULL);
+}
+
+int main(int argc, char **argv) {
+    // Ожидается один аргумент: сколько мёда пчела просит за раз
+    if (argc != 2) {
+        fprintf(stderr, "Usage: %s <bee_portion>\n", argv[0]);
+        return 2;
+    }
+    int portion = atoi(argv[1]);
+    if (portion <= 0) {
+        fprintf(stderr, "portion must be >0\n");
+        return 2;
+    }
+
+    // PID пчелы и имя её личной очереди ответов
+    pid_t me = getpid();
+    char replyq[NAME_MAXLEN];
+    snprintf(replyq, sizeof(replyq), "/bee_%d", (int) me);
+
+    // На всякий случай удаляем старую очередь с таким именем
+    mq_unlink(replyq);
+
+    // Атрибуты очереди ответов (куда сервер будет слать rep_msg_t)
+    struct mq_attr attr;
+    memset(&attr, 0, sizeof(attr));
+    attr.mq_maxmsg = 10;
+    attr.mq_msgsize = sizeof(rep_msg_t);
+
+    // Создаём очередь ответов пчелы, только для чтения
+    mqd_t qrep = mq_open(replyq, O_CREAT | O_RDONLY, 0666, &attr);
+    if (qrep == (mqd_t) -1) {
+        perror("mq_open reply");
+        return 1;
+    }
+
+    // Открываем очередь запросов к серверу (общая очередь REQ_QUEUE)
+    mqd_t qreq = -1;
+    for (int i = 0; i < 50; i++) {
+        qreq = mq_open(REQ_QUEUE, O_WRONLY);
+        if (qreq != -1) break; // удалось открыть — выходим из цикла
+        if (errno != ENOENT) {
+            // другая ошибка, не "очередь ещё не создана"
+            perror("mq_open req");
+            break;
+        }
+        // Если сервер ещё не создал очередь (ENOENT) — подождать и попробовать снова
+        msleep(100);
+    }
+    if (qreq == -1) {
+        // Не смогли открыть очередь запросов — выходим
+        perror("mq_open req");
+        mq_close(qrep);
+        mq_unlink(replyq);
+        return 1;
+    }
+
+    // Инициализация отдельного генератора случайных чисел для этой пчелы
+    unsigned seed = (unsigned) (time(NULL) ^ (uintptr_t) me);
+
+    // Основной рабочий цикл пчелы
+    while (1) {
+        // Ждём случайное время 100–699 мс перед очередным запросом
+        int ms = 100 + (rand_r(&seed) % 600);
+        msleep(ms);
+
+        // Формируем запрос к серверу
+        req_msg_t req;
+        memset(&req, 0, sizeof(req));
+        req.pid = me;
+        req.want = portion; // сколько мёда хотим получить
+        strncpy(req.replyq, replyq, sizeof(req.replyq) - 1); // куда слать ответ
+
+        // Отправляем запрос в очередь REQ_QUEUE
+        if (mq_send(qreq, (const char *) &req, sizeof(req), 0) == -1) {
+            perror("mq_send");
+            break;
+        }
+
+        // Ждём ответ от сервера в своей очереди
+        rep_msg_t rep;
+        ssize_t rd = mq_receive(qrep, (char *) &rep, sizeof(rep), NULL);
+        if (rd == -1) {
+            perror("mq_receive");
+            break;
+        }
+
+        // Если нам больше ничего не дают (granted <= 0) — выходим
+        if (rep.granted <= 0) {
+            break;
+        }
+
+        // Иначе логируем, сколько мёда получили и сколько осталось у сервера
+        dprintf(STDOUT_FILENO, "Bee %d got %d, remain %d\n",
+                (int) me, rep.granted, rep.remain);
+    }
+
+    // Очистка ресурсов: закрываем очереди и удаляем личную очередь ответов
+    mq_close(qreq);
+    mq_close(qrep);
+    mq_unlink(replyq);
+    return 0;
+}