Русский
Русский
English
Статистика
Реклама

Создаем с нуля высоконагруженное приложение на Tarantool

image


В 2013 я пришел в Mail.ru Group, и я решал задачу, в которой мне нужна была очередь. Есть много разных инструментов для построения очередей, но я решил для начала узнать, что уже имеется в компании. Услышал, что есть такой продукт Tarantool. Узнал, как он устроен, и мне показалось, что в него отлично может быть встроен брокер очередей.


Я пошёл к главному по Tarantool Косте Осипову и постарался объяснить, что я хочу получить. Предполагалось, что код очереди будет написан на C, как и остальной код Tarantool, но На следующий день Костя дал мне скрипт на 250 строк, который реализовывал почти всё, что я хотел.


С того момента я влюбился в Tarantool. Оказалось, что можно написать совсем немного кода на очень простом скриптовом языке и получить совершенно новую для этой СУБД функциональность.


Прошло много времени, Tarantool развивался, в том числе и под влиянием наших запросов, но основные идеи и подходы сохранились. Я расскажу, как реализовать собственную очередь на современном Tarantool, например версии 2.2.


На тот момент я был знаком с несколькими реализациями очередей, и мне нравился простой и быстрый Beanstalkd. У него довольно удобный интерфейс, отслеживание состояния задачи по соединению (обрыв клиента возвращал задачу в очередь), а также удобные возможности по работе с отложенными задачами. При реализации очереди мне хотелось получить что-то подобное.


Сам сервис можно представить следующим образом: у нас есть процесс брокера очереди, который принимает и хранит задачи; есть клиенты: продюсеры, которые приносят задачи (метод put); и консюмеры, которые берут задачи в работу (метод take).



Жизненный цикл одной задачи можно описать следующей схемой. Задача появляется при помощи метода put и переходит в состояние ready. Операция take переводит задачу в taken. Из taken задача может быть обработана (ack) и удалена, или возвращена в ready (release).



Также мы можем расширить эту диаграмму и ввести дополнительно отложенную обработку задач:



Подготовка окружения


Tarantool сегодня это, в том числе, LuaJIT-интерпретатор. Чтобы начать с ним работать, нужно создать стартовый файл init.lua, точку входа, и прописать там вызов box.cfg(), который запускает внутренности СУБД.


Для локальной разработки осталось только подключить и запустить консоль. Затем создайте вот такой файл и запустите его:


require'strict'.on()box.cfg{}require'console'.start()os.exit()

Консоль интерактивная, в ней сразу можно что-то делать. Не нужно долго и много устанавливать и настраивать инструменты, разбираться в них. Просто пишете 10-15 строчек на любой локальной машине


Также я рекомендую сразу включить strict. Язык Lua довольно свободен в объявлении переменных, и этот режим должен немного помочь вам при ошибках. Кстати, если собрать Tarantool самому в режиме DEBUG, то strict будет включён по умолчанию.


Далее остаётся только запустить наш файл при помощи tarantool:


tarantool init.lua

Вы должны увидеть что-то похожее:


2020-07-09 20:00:11.344 [30043] main/102/init.lua C> Tarantool 2.2.3-1-g98ecc909a2020-07-09 20:00:11.345 [30043] main/102/init.lua C> log level 52020-07-09 20:00:11.346 [30043] main/102/init.lua I> mapping 268435456 bytes for memtx tuple arena...2020-07-09 20:00:11.347 [30043] main/102/init.lua I> mapping 134217728 bytes for vinyl tuple arena...2020-07-09 20:00:11.370 [30043] main/102/init.lua I> instance uuid 38c59892-263e-42de-875c-8f67539191a32020-07-09 20:00:11.371 [30043] main/102/init.lua I> initializing an empty data directory2020-07-09 20:00:11.408 [30043] main/102/init.lua I> assigned id 1 to replica 38c59892-263e-42de-875c-8f67539191a32020-07-09 20:00:11.408 [30043] main/102/init.lua I> cluster uuid 7723bdf4-24e8-4957-bd6c-6ab502a1911c2020-07-09 20:00:11.425 [30043] snapshot/101/main I> saving snapshot `./00000000000000000000.snap.inprogress'2020-07-09 20:00:11.437 [30043] snapshot/101/main I> done2020-07-09 20:00:11.439 [30043] main/102/init.lua I> ready to accept requests2020-07-09 20:00:11.439 [30043] main/104/checkpoint_daemon I> scheduled next checkpoint for Thu Jul  9 21:11:59 2020tarantool> 

Пишем очередь


Создадим отдельный файл queue.lua для написания нашего приложения. Конечно, можно было бы писать всё прямо в init.lua, но работать с отдельным файлом будет удобнее.


Подключим queue в виде модуля из файла init.lua:


require'strict'.on()box.cfg{}queue = require 'queue'require'console'.start()os.exit()

Все дальнейшие модификации мы будем делать в queue.lua.


Поскольку мы делаем очередь, нам понадобится где-то хранить информацию о задачах. Создадим спейс (space) таблицу для данных. Можно создавать его без опций, но мы сразу кое-что добавим. Чтобы нормально перезапускаться, мы укажем, что спейс нужно создавать только в том случае, если он не существует (if_not_exists). Также в современном Tarantool можно указывать формат полей с описанием содержимого (лучше так и делать). Так мы и поступим.


Под очередь возьмём совсем простую структуру. Мне понадобятся только id задач, их статусы и какие-то произвольные данные. Мне не важно, что там будет лежать. Работать с данными без первичного индекса нельзя, поэтому сразу создадим индекс по id. Проверяйте, чтобы тип поля совпадал и в формате, и в индексе.


box.schema.create_space('queue',{ if_not_exists = true; })box.space.queue:format( {    { name = 'id';     type = 'number' },    { name = 'status'; type = 'string' },    { name = 'data';   type = '*'      },} );box.space.queue:create_index('primary', {   parts = { 1,'number' };   if_not_exists = true;})

Объявим глобальную таблицу queue, которая будет нести в себе наши функции, атрибуты и методы. И для начала объявим две функции: положить задачу (put) и взять задачу (take).


У задач в очереди будут состояния. Для обозначения статуса заведём отдельную таблицу со статусом. В качестве значения можно использовать числа или строки, но я люблю использовать однобуквенные значения: их можно выбрать семантически значимыми и они занимают минимум места при хранении. Для начала сделаем два статуса: R=READY и T=TAKEN.


local queue = {}local STATUS = {}STATUS.READY = 'R'STATUS.TAKEN = 'T'function queue.put(...)endfunction queue.take(...)endreturn queue

Как сделать put? Очень просто. Нам нужно сгенерировать id и вставить данные в спейс со статусом READY. Есть много разных способов генерирования идентификатора, мы возьмём clock.realtime. Для очереди он хорош тем, что автоматически определяется порядок сообщений (но учтите, что часы могут жёстко переводиться, и в этом случае порядок задач может быть нарушен). Также, теоретически, может возникнуть ситуация, когда в очереди уже будет задача с таким же значением. Поэтому можно посмотреть, нет ли задачи с таким id, и в случае коллизии добавить единицу. На это уйдут микросекунды, и это крайне маловероятная ситуация, поэтому производительность не пострадает.


Все аргументы функции мы просто вставляем в качестве начинки в нашу задачу:


local clock = require 'clock'function gen_id()    local new_id    repeat        new_id = clock.realtime64()    until not box.space.queue:get(new_id)    return new_idendfunction queue.put(...)    local id = gen_id()    return box.space.queue:insert{ id, STATUS.READY, { ... } }end

Как только мы написали функцию put, можем перезапустить Tarantool и сразу вызвать эту функцию. Видим, что задача положена в очередь, она выглядит как тапл (кортеж). В него можно класть произвольные данные и даже вложенные структуры. Таплы, в которых Tarantool хранит данные, упаковываются в MessagePack, что позволяет сохранять такие структуры.


tarantool> queue.put("hello")---- [1594325382148311477, 'R', ['hello']]...tarantool> queue.put("my","data",1,2,3)---- [1594325394527830491, 'R', ['my', 'data', 1, 2, 3]]...tarantool> queue.put({ complex = { struct = "data" }})---- [1594325413166109943, 'R', [{'complex': {'struct': 'data'}}]]...

Всё, что мы кладём, находится в спейсе. Можно взять команды спейса и посмотреть, что там лежит.


tarantool> box.space.queue:select()---- - [1594325382148311477, 'R', ['hello']]  - [1594325394527830491, 'R', ['my', 'data', 1, 2, 3]]  - [1594325413166109943, 'R', [{'complex': {'struct': 'data'}}]]...

Теперь нужно научиться брать задачи сделаем функцию take. Для этого поработаем со статусом. Мы берем те задачи, которые готовы к обработке, то есть находятся в статусе READY. Можно было бы, конечно, пройтись по первичному ключу и найти первую готовую задачу, но в условиях нагрузки и большого количества обрабатываемых задач этот сценарий нам не подойдёт. Нужен отдельный индекс по полю статуса. Одна из основных черт Tarantool, которые отличают его от key-value баз, это возможность создавать различные индексы, почти как в реляционных базах: на разные поля, композитные, разного типа.


Создадим второй индекс, в котором укажем, что первое поле это статус. По нему и будем искать. А второе поле это id. Он упорядочит по возрастанию задачи в рамках одного статуса.


box.space.queue:create_index('status', {    parts = { 2, 'string', 1, 'number' };    if_not_exists = true;})

Возьмём встроенные функции для выборки. Есть специальный итератор, который применяется к спейсу как pairs. В него мы передаем часть ключа. Здесь мы сталкиваемся с составным индексом, который состоит из двух полей: ищем по первому, а упорядочиваем по второму. Говорим системе найти нам таплы, которые по первой части индекса равняются статусу READY. И будем их получать уже упорядоченными по второй части индекса. Если мы что-то нашли, то берём задачу, обновляем и возвращаем. Обновляем для того, чтобы никто другой, кто придет с таким же вызовом take, не взял её. Если задач нет, то возвращаем ничего.


function queue.take()    local found = box.space.queue.index.status        :pairs({STATUS.READY},{ iterator = 'EQ' }):nth(1)    if found then        return box.space.queue            :update( {found.id}, {{'=', 2, STATUS.TAKEN }})    end    returnend

Также стоит обратить внимание на то, что первый уровень тапла в Tarantool представлен в виде массива. Он не имеет имен, он нумерованный, поэтому при таких операциях, как update, до совсем недавнего времени нужно было указывать номер поля. В качестве вспомогательного элемента сделаем табличку, в которой сопоставим имя поля и номер. Для формирования этой таблицы мы можем воспользоваться уже описанным нами форматом:


local F = {}for no,def in pairs(box.space.queue:format()) do    F[no] = def.name    F[def.name] = noend

Для большей наглядности можем поправить описание индексов:


box.space.queue:format( {    { name = 'id';     type = 'number' },    { name = 'status'; type = 'string' },    { name = 'data';   type = '*'      },} );local F = {}for no,def in pairs(box.space.queue:format()) do    F[no] = def.name    F[def.name] = noendbox.space.queue:create_index('primary', {   parts = { F.id, 'number' };   if_not_exists = true;})box.space.queue:create_index('status', {    parts = { F.status, 'string', F.id, 'number' };    if_not_exists = true;})

Теперь можно реализовать take целиком:


function queue.take(...)    for _,t in        box.space.queue.index.status        :pairs({ STATUS.READY },{ iterator='EQ' })    do        return box.space.queue:update({t.id},{            { '=', F.status, STATUS.TAKEN }        })    end    returnend

Проверим, как это работает. Положим одну задачу и вызовем take дважды. Если к этому моменту у нас есть данные в спейсе, можем его очистить командой box.space.queue:truncate():


tarantool> queue.put("my","data",1,2,3)---- [1594325927025602515, 'R', ['my', 'data', 1, 2, 3]]...tarantool> queue.take()---- [1594325927025602515, 'T', ['my', 'data', 1, 2, 3]]...tarantool> queue.take()---...

Первый take возвращает нам ту самую задачу, которую мы положили. А когда вызовем take повторно, то больше ничего не вернется, потому что ready-задач (в статусе R) больше нет. Можем убедиться в этом, выполнив select из спейса:


tarantool> box.space.queue:select()---- - [1594325927025602515, 'T', ['my', 'data', 1, 2, 3]]...

Потребитель, который берет задачу, должен либо подтвердить её обработку, либо вернуть без обработки, если по какой-либо причине не справляется. Тогда задачу сможет взять кто-то другой. Реализуем для этого две функции: ack и release. Они принимают id задачи и ищут её. Если у задачи статус взятой, то мы обрабатываем её. Эти функции очень похожи. Одна удаляет обработанные задачи, другая возвращает их в статус ready.


function queue.ack(id)    local t = assert(box.space.queue:get{id},"Task not exists")    if t and t.status == STATUS.TAKEN then        return box.space.queue:delete{t.id}    else        error("Task not taken")    endendfunction queue.release(id)    local t = assert(box.space.queue:get{id},"Task not exists")    if t and t.status == STATUS.TAKEN then        return box.space.queue:update({t.id},{{'=', F.status, STATUS.READY }})    else        error("Task not taken")    endend

Посмотрим, как это работает со всеми четырьмя функциями. Кладём две задачи и берём первую, затем освобождаем её. Она возвращается обратно в статус R. Второй вызов take берёт ту же задачу. Если мы её обработаем, она удалится. Третий вызов take возьмёт уже вторую задачу. Порядок соблюдается. Если задача взята, то она не выдается ещё кому-нибудь.


tarantool> queue.put("task 1")---- [1594326185712343931, 'R', ['task 1']]...tarantool> queue.put("task 2")---- [1594326187061434882, 'R', ['task 2']]...tarantool> task = queue.take() return task---- [1594326185712343931, 'T', ['task 1']]...tarantool> queue.release(task.id)---- [1594326185712343931, 'R', ['task 1']]...tarantool> task = queue.take() return task---- [1594326185712343931, 'T', ['task 1']]...tarantool> queue.ack(task.id)---- [1594326185712343931, 'T', ['task 1']]...tarantool> task = queue.take() return task---- [1594326187061434882, 'T', ['task 2']]...tarantool> queue.ack(task.id)---- [1594326187061434882, 'T', ['task 2']]...tarantool> task = queue.take() return task---- null...

Получилась корректно работающая очередь. Мы уже можем написать потребителя, который будет обрабатывать задачи. Но у него есть, как минимум, одна проблема. Когда мы вызываем take, функция сразу возвращает либо задачу, либо пустую строку. Если написать цикл обработки задач и запустить его, то работать он будет, но вхолостую, ничего не делая, просто потребляя CPU.


while true do    local task = queue.take()    if task then        -- ...    endend

Чтобы это исправить, нам понадобится примитив канал (или channel). Он позволяет передавать сообщения. По сути, это FIFO-очередь для общения между файберами. У нас есть файбер, который кладет задачи, когда мы приходим в базу данных по сети или работаем с ней из консоли. В файбере исполняется наш Lua-код, ему нужно через какой-то примитив сообщить другому файберу, который ждёт задачи, что появилась новая.


Канал работает так: в нем может быть буфер на N слотов, в которые можно положить сообщение, даже если никто не читает из канала. Также можно создать канал без буферной ёмкости, тогда положить можно будет только в те слоты, которые кто-то ждет. Например, мы создаем канал на два буферных элемента. В нем два слота под put. Если на канале будет ожидать один потребитель, он создаст третий слот под put. Если мы будем класть сообщения в этот канал, то три операции put выполнятся без блокировки, а четвертый put заблокирует нам тот файбер, который кладет в этот канал. Это позволяет организовать межфайберное взаимодействие. Если вдруг вы знакомы с каналами в Go, то там они фактически такие же:



Немного переделаем нашу функцию take. Сначала добавим новый аргумент таймаут: мы готовы ждать задачу в течение определённого времени. Сделаем цикл, который будет искать готовую задачу. Если не найдёт, то будет вычислять, сколько времени ему осталось ждать.


Создадим канал, который будет ждать с этим таймаутом. Ели файбер спит в ожидании на канале, то его можно разбудить извне, передав сообщение в этот канал.


local fiber = require 'fiber'queue._wait = fiber.channel()function queue.take(timeout)    if not timeout then timeout = 0 end    local now = fiber.time()    local found    while not found do        found = box.space.queue.index.status            :pairs({STATUS.READY},{ iterator = 'EQ' }):nth(1)        if not found then            local left = (now + timeout) - fiber.time()            if left <= 0 then return end            queue._wait:get(left)        end    end    return box.space.queue        :update( {found.id}, {{'=', F.status, STATUS.TAKEN }})end

Итого: take пытается взять задачу, если получилось, то возвращает её. Но если задачи не нашлось, то можно подождать в течение остатка таймаута. Причем другая сторона, которая будет производить задачу, может этот файбер разбудить.


Чтобы удобно было проводить различные тесты, можем в файле init.lua подключить модуль fiber глобально:


fiber = require 'fiber'

Давайте посмотрим, как это будет работать без пробуждения файбера. В отдельном файбере положим задачу через 0,1 с. То есть сначала очередь пустая, а через 0,1 с. после запуска появляется задача. При этом вызов take сделаем с таймаутом 3. После запуска take попытается найти задачу. Не найдя её, он уснёт на 3 с. Затем проснётся, снова поищет и найдёт задачу.


tarantool> do    box.space.queue:truncate()    fiber.create(function()        fiber.sleep(0.1)        queue.put("task 3")    end)    local start = fiber.time()    return queue.take(3), { wait = fiber.time() - start }end---- [1594326905489650533, 'T', ['task 3']]- wait: 3.0017817020416...

Теперь сделаем так, чтобы take просыпался при появлении задачи. Для этого возьмем старую функцию put и добавим в нее отправку сообщения в канал. В качестве сообщения можно отправить что угодно, пусть в этом случае будет true.


Ранее я показывал, что put может заблокироваться, если в канале недостаточно места. При этом производителю задач не важно, есть с той стороны потребители или нет. Он не должен блокироваться в ожидании потребителя. Поэтому логично поставить здесь нулевой таймаут на блокировку. Если там есть потребители, то есть те, кому нужно сообщить о новой задаче, мы его разбудим. Иначе у нас сообщение в этот канал не положится. Или, в качестве альтернативного варианта, можно проверить, есть ли у канала активные читатели.


function queue.put(...)    local id = gen_id()    if queue._wait:has_readers() then        queue._wait:put(true,0)    end    return box.space.queue:insert{ id, STATUS.READY, { ... } }end

После этого тот же самый код take начнет работать совершенно иначе. Мы создаём задачу через 0,1 с. и take сразу же просыпается и получает её. Мы избавились от горячего цикла, который непрерывно висел в ожидании задачи. Если мы не положим задачу, то файбер будет ждать три секунды.


tarantool> do    box.space.queue:truncate()    fiber.create(function()        fiber.sleep(0.1)        queue.put("task 4")    end)    local start = fiber.time()    return queue.take(3), { wait = fiber.time() - start }end---- [1594327004302379957, 'T', ['task 4']]- wait: 0.10164666175842...

На текущий момент мы протестировали работу внутри экземпляра, теперь поработаем по сети. В первую очередь нам нужно наш сервер сделать сервером. Добавим в файле init.lua в box.cfg опцию listen порт, на котором он будет слушать. Вместе с этим нам понадобится сделать разрешения. Не будем сейчас детально рассматривать настройку привилегий, сделаем так, чтобы любое подключение имело привилегии на исполнение. Про права вы можете почитать отдельно.


require'strict'.on()fiber = require 'fiber'box.cfg{    listen = '127.0.0.1:3301'}box.schema.user.grant('guest', 'super', nil, nil, { if_not_exists = true })queue = require 'queue'require'console'.start()os.exit()

Создадим клиент-producer для генерирования задач. В поставке Tarantool уже есть модуль, который позволяет подключаться к другому Tarantool.


#!/usr/bin/env tarantoolif #arg < 1 then    error("Need arguments",0)endlocal netbox = require 'net.box'local conn = netbox.connect('127.0.0.1:3301')local yaml = require 'yaml'local res = conn:call('queue.put',{unpack(arg)})print(yaml.encode(res))conn:close()

$ tarantool producer.lua "hi"--- [1594327270675788959, 'R', ['hi']]...

Потребитель (consumer) будет подключаться, вызывать take с таймаутом и обрабатывать результат. Если он получил задачу, то будем печатать её и освобождать. Мы сейчас пока не будем обрабатывать. Допустим, задача получена.


#!/usr/bin/env tarantoollocal netbox = require 'net.box'local conn = netbox.connect('127.0.0.1:3301')local yaml = require 'yaml'while true do    local task = conn:call('queue.take', { 1 })    if task then        print("Got task: ", yaml.encode(task))        conn:call('queue.release', { task.id })    else        print "No more tasks"    endend

Но при попытке освободить задачу у нас произойдёт какая-то фигня.


$ tarantool consumer.lua Got task:        --- [1594327270675788959, 'T', ['hi']]...ER_EXACT_MATCH: Invalid key part count in an exact match (expected 1, got 0)

Давайте разберёмся. При повторной попытке исполнения потребителя мы обнаружим, что при предыдущем запуске он задачу взял, но не смог вернуть: у него произошла ошибка и задача застряла. Такие задачи больше никто не сможет взять, но и некому их вернуть, потому что код, который их брал, завершился.


$ tarantool consumer.lua No more tasksNo more tasks

С помощью select можно увидеть, что задачи взяты.


tarantool> box.space.queue:select()---- - [1594327004302379957, 'T', ['task 3']]  - [1594327270675788959, 'T', ['hi']]...

Здесь есть сразу несколько проблем, поэтому давайте начнём с автоматического освобождения задач при отключении клиента.


В Tarantool есть триггеры на подключение и отключение клиентов. Если мы их добавим, то сможем о фактах подключения и отключения.


local log = require 'log'box.session.on_connect(function()    log.info( "connected %s from %s", box.session.id(), box.session.peer() )end)box.session.on_disconnect(function()    log.info( "disconnected %s from %s", box.session.id(), box.session.peer() )end)

2020-07-09 20:52:09.107 [32604] main/115/main I> connected 2 from 127.0.0.1:366522020-07-09 20:52:10.260 [32604] main/116/main I> disconnected 2 from nil2020-07-09 20:52:10.823 [32604] main/116/main I> connected 3 from 127.0.0.1:366542020-07-09 20:52:11.541 [32604] main/115/main I> disconnected 3 from nil

Есть понятие session id, и можно узнать, с какого IP было подключение и время отключения. Правда, есть один нюанс. Вызов session.peer() по сути вызывает getpeername(2) непосредственно на сокете. Поэтому при отключении мы уже не видим, кто отключается (getpeername вызывается на закрытом сокете). Cделаем небольшой хак. В Tarantool есть box.session.storage временная таблица, в которую можно сохранять всё, что хочется, на время существования сессии. Во время подключения запомним, кто к нам подключился, чтобы знать, кто отключится. Это облегчает отладку.


box.session.on_connect(function()    box.session.storage.peer = box.session.peer()    log.info( "connected %s from %s", box.session.id(), box.session.storage.peer )end)box.session.on_disconnect(function()    log.info( "disconnected %s from %s", box.session.id(), box.session.storage.peer )end)

Теперь у нас есть событие отключения клиента. Нам нужно как-то освободить взятые им задачи. Введем понятие владения задачей. Та сессия, которая взяла задачу, должна за неё отвечать. Заведем две таблички, в которые будем сохранять эти данные, и модифицируем функцию take.


queue.taken = {}; -- список взятых задачqueue.bysid = {}; -- список задач для конкретной сессии

function queue.take(timeout)    if not timeout then timeout = 0 end    local now = fiber.time()    local found    while not found do        found = box.space.queue.index.status            :pairs({STATUS.READY},{ iterator = 'EQ' }):nth(1)        if not found then            local left = (now + timeout) - fiber.time()            if left <= 0 then return end            queue._wait:get(left)        end    end    local sid = box.session.id()    log.info("Register %s by %s", found.id, sid)    queue.taken[ found.id ] = sid    queue.bysid[ sid ] = queue.bysid[ sid ] or {}    queue.bysid[ sid ][ found.id ] = true    return box.space.queue        :update( {found.id}, {{'=', F.status, STATUS.TAKEN }})end

Мы в ней запомним, что конкретная задача взята конкретной сессией. Также нам понадобится модифицировать код возврата задач, ack и release. Сделаем одну общую функцию. Будем проверять, что задача есть и взята, причём взята конкретной сессией. Таким образом нельзя будет из одного соединения взять задачу, а из другого прийти и сказать: удалите её, я обработал.


local function get_task( id )    if not id then error("Task id required", 2) end    local t = box.space.queue:get{id}    if not t then        error(string.format( "Task {%s} was not found", id ), 2)    end    if not queue.taken[id] then        error(string.format( "Task %s not taken by anybody", id ), 2)    end    if queue.taken[id] ~= box.session.id() then        error(string.format( "Task %s taken by %d. Not you (%d)",            id, queue.taken[id], box.session.id() ), 2)    end    return tend

Теперь функции ack и release становятся очень простыми. Мы в них вызываем get_task, который проверяет, что задача принадлежит нам и взята. И дальше уже с ней работаем.


function queue.ack(id)    local t = get_task(id)    queue.taken[ t.id ] = nil    queue.bysid[ box.session.id() ][ t.id ] = nil    return box.space.queue:delete{t.id}endfunction queue.release(id)    local t = get_task(id)    if queue._wait:has_readers() then queue._wait:put(true,0) end    queue.taken[ t.id ] = nil    queue.bysid[ box.session.id() ][ t.id ] = nil    return box.space.queue        :update({t.id},{{'=', F.status, STATUS.READY }})end

Для сброса состояния всех задач в R можно воспользоваться SQL или Lua-cниппетом:


box.execute[[ update "queue" set "status" = 'R' where "status" = 'T' ]]box.space.queue.index.status:pairs({'T'}):each(function(t) box.space.queue:update({t.id},{{'=',2,'R'}}) end)

Когда мы вызовем consumer повторно, он ответит task ID required.


$ tarantool consumer.lua Got task:        --- [1594327004302379957, 'T', ['task 3']]...ER_PROC_LUA: queue.lua:113: Task id required

Так мы находим первую проблему в нашем коде. Когда мы работаем внутри Tarantool, кортеж всегда ассоциирован со спейсом. У того есть формат, а у формата есть имена полей. Поэтому в тапле можно пользоваться именами полей. А когда мы выносим это за пределы базы, тапл становится просто массивом с набором полей. Доработаем формат возврата из функций и будем возвращать не таплы, а объекты с именами. Для этого воспользуемся методом :tomap{ names_only = true }:


function queue.put(...)    --- ...    return box.space.queue        :insert{ id, STATUS.READY, { ... } }        :tomap{ names_only = true }endfunction queue.take(timeout)    --- ...    return box.space.queue        :update( {found.id}, {{'=', F.status, STATUS.TAKEN }})        :tomap{ names_only = true }endfunction queue.ack(id)    --- ...    return box.space.queue:delete{t.id}:tomap{ names_only = true }endfunction queue.release(id)    --- ...    return box.space.queue        :update({t.id},{{'=', F.status, STATUS.READY }})        :tomap{ names_only = true }endreturn queue

Поменяв это, мы столкнёмся с новой проблемой.


$ tarantool consumer.lua Got task:        --- {'status': 'T', 'data': ['hi'], 'id': 1594327270675788959}...ER_PROC_LUA: queue.lua:117: Task 1594327270675788959ULL not taken by anybody

При попытке освободить задачу система ответит, мы её не брали. При этом визуально мы увидим, что ID один и тот же. Только есть еще какая-то суффикс ULL.


Здесь мы сталкиваемся с одной особенностью расширения LuaJIT: FFI (Foreign Function Interface). Давайте рассмотрим детальнее. Положим в таблицу пять значений, используя в качестве ключей различные варианты записи числа 1.


tarantool> t = {}tarantool> t[1] = 1tarantool> t["1"] = 2tarantool> t[1LL] = 3tarantool> t[1ULL] = 4tarantool> t[1ULL] = 5tarantool> t---- 1: 1  1: 5  1: 4  '1': 2  1: 3...

Можно было бы предположить, что они положатся как 2 (строка + число). Максимум как 3 (строка, число, LL). Но при выводе на экран окажется, что все ключи лежат в таблице раздельно: мы видим все значения 1, 2, 3, 4, 5. Более того, при сериализации мы не видим разницы между обычными, знаковыми и беззнаковыми числами.


tarantool> return t[1], t['1'], t[1LL], t[1ULL]---- 1- 2- null- null...

Но самое веселье наступает, если попытаться достать данные из таблицы. С обычными Lua-типами всё хорошо (number и string), а вот с LL (long long) и ULL (unsigned long long) нет. Эти типы являются отдельным типом cdata. Он предназначен для работы с типами из языка C. И при сохранении в Lua-таблицу cdata хэшируется по адресу, а не по значению. У двух, пусть и одинаковых по значению, чисел просто два разных адреса. И когда мы складываем ULL в таблицу, то потом не можем по такому же значению достать его из таблицы.


Поэтому нам придется немного переделать нашу очередь и владение ключами. Шаг вынужденный, но он позволяет нам в дальнейшем модифицировать наши ключи произвольным образом. Нам нужно каким-либо образом превратить наш ключ в строку или число. Возьмем MessagePack. В Tarantool он используется для хранения таплов и будет упаковывать наши значения так же, как это делает сам Tarantool. С его помощью мы превратим произвольный ключ в строку, которая будет являться ключом в нашей таблице.


local msgpack = require 'msgpack'local function keypack( key )    return msgpack.encode( key )endlocal function keyunpack( data )    return msgpack.decode( data )end

Добавляем в take упаковку ключа и сохраняем его в таблице. В функции get_task проверим, что ключ прошел в правильном формате, и если это не так, то превратим его в int64. После этого воспользуемся тем же самым keypack, который упакует ключ в MessagePack. Поскольку этот упакованный ключ будет требоваться всем функциям, которые с ним работают, мы будем возвращать его из get_task, чтобы ack и release могли им пользоваться и вычищать его из сессий.


function queue.take(timeout)    if not timeout then timeout = 0 end    local now = fiber.time()    local found    while not found do        found = box.space.queue.index.status            :pairs({STATUS.READY},{ iterator = 'EQ' }):nth(1)        if not found then            local left = (now + timeout) - fiber.time()            if left <= 0 then return end            queue._wait:get(left)        end    end    local sid = box.session.id()    log.info("Register %s by %s", found.id, sid)    local key = keypack( found.id )    queue.taken[ key ] = sid    queue.bysid[ sid ] = queue.bysid[ sid ] or {}    queue.bysid[ sid ][ key ] = true    return box.space.queue        :update( {found.id}, {{'=', F.status, STATUS.TAKEN }})        :tomap{ names_only = true }endlocal function get_task( id )    if not id then error("Task id required", 2) end    id = tonumber64(id)    local key = keypack(id)    local t = box.space.queue:get{id}    if not t then        error(string.format( "Task {%s} was not found", id ), 2)    end    if not queue.taken[key] then        error(string.format( "Task %s not taken by anybody", id ), 2)    end    if queue.taken[key] ~= box.session.id() then        error(string.format( "Task %s taken by %d. Not you (%d)",            id, queue.taken[key], box.session.id() ), 2)    end    return t, keyendfunction queue.ack(id)    local t, key = get_task(id)    queue.taken[ key ] = nil    queue.bysid[ box.session.id() ][ key ] = nil    return box.space.queue:delete{t.id}:tomap{ names_only = true }endfunction queue.release(id)    local t, key = get_task(id)    queue.taken[ key ] = nil    queue.bysid[ box.session.id() ][ key ] = nil    if queue._wait:has_readers() then queue._wait:put(true,0) end    return box.space.queue        :update({t.id},{{'=', F.status, STATUS.READY }})        :tomap{ names_only = true }end

Поскольку у нас есть есть триггер на отключение, мы теперь знаем, что отключилась конкретная сессия, которая владеет какими-то ключами. Можно взять все ключи этой сессии и автоматически вернуть их в исходное состояние ready. Также, внутри этой сессии могут висеть ожидающие take. Оставим для них маркер в session.storage, что задачи брать не нужно.


box.session.on_disconnect(function()    log.info( "disconnected %s from %s", box.session.id(), box.session.storage.peer )    box.session.storage.destroyed = true    local sid = box.session.id()    local bysid = queue.bysid[ sid ]    if bysid then        while next(bysid) do            for key, id in pairs(bysid) do                log.info("Autorelease %s by disconnect", id);                queue.taken[key] = nil                bysid[key] = nil                local t = box.space.queue:get(id)                if t then                    if queue._wait:has_readers() then queue._wait:put(true,0) end                    box.space.queue:update({t.id},{{'=', F.status, STATUS.READY }})                end            end        end        queue.bysid[ sid ] = nil    endend)function queue.take(timeout)    if not timeout then timeout = 0 end    local now = fiber.time()    local found    while not found do        found = box.space.queue.index.status            :pairs({STATUS.READY},{ iterator = 'EQ' }):nth(1)        if not found then            local left = (now + timeout) - fiber.time()            if left <= 0 then return end            queue._wait:get(left)        end    end    if box.session.storage.destroyed then return end    local sid = box.session.id()    log.info("Register %s by %s", found.id, sid)    local key = keypack( found.id )    queue.taken[ key ] = sid    queue.bysid[ sid ] = queue.bysid[ sid ] or {}    queue.bysid[ sid ][ key ] = found.id    return box.space.queue        :update( {found.id}, {{'=', F.status, STATUS.TAKEN }})        :tomap{ names_only = true }end

Для теста можно брать задачи командой:


tarantoolctl connect 127.0.0.1:3301 <<< 'queue.take()'

Пока это всё отлаживалось, можно было столкнуться с тем, что вы взяли задачи, потушили очередь, запустили заново задачи никому не принадлежат (потому что соединения порвались при выключении), но при этом они в статусе taken. Поэтому добавим в код модификацию статусов при старте: база запускается и освобождает все взятые задачи.


while true do    local t = box.space.queue.index.status:pairs({STATUS.TAKEN}):nth(1)    if not t then break end    box.space.queue:update({ t.id }, {{'=', F.status, STATUS.READY }})    log.info("Autoreleased %s at start", t.id)end

Получилась очередь, готовая к эксплуатации.


Добавим отложенную обработку


Осталось добавить отложенные задачи. Для этого добавим новое поле и индекс по нему. В этом поле мы будем хранить время, когда определённую задачу нужно перевести в другое состояние. Модифицируем функцию put и добавим новый статус:W=WAITING.


box.space.queue:format( {    { name = 'id';     type = 'number' },    { name = 'status'; type = 'string' },    { name = 'runat';  type = 'number' },    { name = 'data';   type = '*'      },} )box.space.queue:create_index('runat', {    parts = { F.runat, 'number', F.id, 'number' };    if_not_exists = true;})STATUS.WAITING = 'W'

Поскольку мы кардинально меняем схему и это режим разработки, очистим предыдущую схему (выполняем в консоли):


box.space.queue.drop()box.snapshot()

Перезапустим очередь.


В put и release добавим поддержку delay. Если delay передан, то присваиваем задаче состояние WAITING и определяем, в какой момент времени она должна быть обработана. Также нам понадобится обработчик. Для этого мы можем воспользоваться фоновыми файберами. В любой момент можно создать файбер, не ассоциированный ни с каким соединением и который будет работать в фоне. Создадим файбер, который будет крутиться бесконечно и ждать ближайшей задачи.


function queue.put(data, opts)    local id = gen_id()    local runat = 0    local status = STATUS.READY    if opts and opts.delay then        runat = clock.realtime() + tonumber(opts.delay)        status = STATUS.WAITING    else        if queue._wait:has_readers() then            queue._wait:put(true,0)        end    end    return box.space.queue        :insert{ id, status, runat, data }        :tomap{ names_only=true }endfunction queue.release(id, opts)    local t, key = get_task(id)    queue.taken[ key ] = nil    queue.bysid[ box.session.id() ][ key ] = nil    local runat = 0    local status = STATUS.READY    if opts and opts.delay then        runat = clock.realtime() + tonumber(opts.delay)        status = STATUS.WAITING    else        if queue._wait:has_readers() then queue._wait:put(true,0) end    end    return box.space.queue        :update({t.id},{{ '=', F.status, status },{ '=', F.runat, runat }})        :tomap{ names_only = true }end

Если приходит время какой-то задачи, мы её модифицируем. Переводим из статуса ожидания в статус готовности, также нотифицируя тех клиентов, которые могут ждать задачу.


Теперь кладём задачу с задержкой. Вызываем take, готовой задачи нет. Вызываем повторно, уже с таймаутом, который укладывается в появление задачи. Как только она появляется, мы видим, что это заслуга файбера queue.runat.


queue._runat = fiber.create(function()    fiber.name('queue.runat')    while true do        local remaining        local now = clock.realtime()        for _,t in box.space.queue.index.runat            :pairs( { 0 }, { iterator = 'GT' })        do            if t.runat > now then                remaining = t.runat - now                break            else                if t.status == STATUS.WAITING then                    log.info("Runat: W->R %s",t.id)                    if queue._wait:has_readers() then queue._wait:put(true,0) end                    box.space.queue:update({ t.id }, {                        {'=', F.status, STATUS.READY },                        {'=', F.runat, 0 },                    })                else                    log.error("Runat: bad status %s for %s", t.status, t.id)                    box.space.queue:update({ t.id },{{ '=', F.runat, 0 }})                end            end        end        if not remaining or remaining > 1 then remaining = 1 end        fiber.sleep(remaining)    endend)

Мониторинг


Нельзя забывать про мониторинг очереди, потому что самое печальное, что может случиться с вашей очередью: она станет очень большой. Или вообще закончится.


Мы довольно просто можем наивно и в лоб собрать статистику с нашей очереди: посчитать количество по всем статусам, которые есть, и начать отправлять данные в мониторинг.


function queue.stats()    return {        total   = box.space.queue:len(),        ready   = box.space.queue.index.status:count({STATUS.READY}),        waiting = box.space.queue.index.status:count({STATUS.WAITING}),        taken   = box.space.queue.index.status:count({STATUS.TAKEN}),    }    end

tarantool> queue.stats()---- ready: 10  taken: 2  waiting: 5  total: 17...tarantool> local clock = require 'clock' local s = clock.time() local r = queue.stats() return r, clock.time() - s---- ready: 10  taken: 2  waiting: 5  total: 17- 0.00057339668273926...

Такой мониторинг будет работать довольно быстро. До того момента, пока задач не станет очень много. Нормальное состояние очереди пустое. Но предположим, что-то случается и прилетает, например, миллион задач. Наша функция stats продолжает показывать корректное значение. Правда, она начинает работать довольно медленно. Проблема в вызове index:count это всегда full-scan по индексу. Давайте закэшируем значения счетчиков.


queue._stats = {}for k,v in pairs(STATUS) do    queue._stats[v] = 0LLendfor _,t in box.space.queue:pairs() do    queue._stats[ t[F.status] ] = (queue._stats[ t[F.status] ] or 0LL)+1endfunction queue.stats()    return {        total   = box.space.queue:len(),        ready   = queue._stats[ STATUS.READY ],        waiting = queue._stats[ STATUS.WAITING ],        taken   = queue._stats[ STATUS.TAKEN ],    }end

Теперь эта функция начнет работать очень быстро, независимо от количества записей. Осталось обновлять счетчики при любых операциях. При каждой операции мы должны одно значение уменьшить, другое увеличить. Можно, конечно, вручную расставить апдейты по функциям, но это чревато ошибками и расхождениями. К счастью в Tarantool есть триггеры на спейсах. Триггер видит любое изменение в спейсе. Можно даже вручную выполнить space:update или space:delete, триггер это учтет и посчитает. Триггер будет учитывать все статусы именно по значению, по которому они хранятся в базе. При рестарте мы единоразово подсчитаем значения всех счётчиков.


box.space.queue:on_replace(function(old,new)    if old then        queue._stats[ old[ F.status ] ] = queue._stats[ old[ F.status ] ] - 1    end    if new then        queue._stats[ new[ F.status ] ] = queue._stats[ new[ F.status ] ] + 1    endend)

Осталась ещё одна операция, которую нельзя отловить непосредственно в спейсе, но которая влияет на его содержимое: space:truncate(). Отследить очистку спейса можно при помощи отдельного триггера в спейсе _truncate.


box.space._truncate:on_replace(function(old,new)    if new.id == box.space.queue.id then        for k,v in pairs(queue._stats) do            queue._stats[k] = 0LL        end    endend)

После этого всё начинает работать точно и консистентно. И теперь мы можем, например, отправлять статистику по сети. Вообще в Tarantool есть удобные неблокирующие сокеты. Работать с ними можно достаточно низкоуровнево, почти как в C.


Для демонстрации мы можем сделать отправку метрик в формате graphite по UDP:


local socket = require 'socket'local errno = require 'errno'local graphite_host = '127.0.0.1'local graphite_port = 2003local ai = socket.getaddrinfo(graphite_host, graphite_port, 1, { type = 'SOCK_DGRAM' })local addr,portfor _,info in pairs(ai) do   addr,port = info.host,info.port   breakendif not addr then error("Failed to resolve host") endqueue._monitor = fiber.create(function()    fiber.name('queue.monitor')    fiber.yield()    local remote = socket('AF_INET', 'SOCK_DGRAM', 'udp')    while true do        for k,v in pairs(queue.stats()) do            local msg = string.format("queue.stats.%s %s %s\n", k, tonumber(v), math.floor(fiber.time()))            local res = remote:sendto(addr, port, msg)            if not res then                log.error("Failed to send: %s", errno.strerror(errno()))            end        end        fiber.sleep(1)    endend)

или по TCP:


local socket = require 'socket'local errno = require 'errno'local graphite_host = '127.0.0.1'local graphite_port = 2003queue._monitor = fiber.create(function()    fiber.name('queue.monitor')    fiber.yield()    while true do        local remote =  require 'socket'.tcp_connect(graphite_host, graphite_port)        if not remote then            log.error("Failed to connect to graphite %s",errno.strerror())            fiber.sleep(1)        else            while true do                local data = {}                for k,v in pairs(queue.stats()) do                    table.insert(data,string.format("queue.stats.%s %s %s\n",k,tonumber(v),math.floor(fiber.time())))                end                data = table.concat(data,'')                if not remote:send(data) then                    log.error("%s",errno.strerror())                    break                end                fiber.sleep(1)            end        end    endend)

Горячая перезагрузка кода


И наконец, нельзя не упомянуть такую важную возможность платформы Tarantool, как горячая перезагрузка кода. В обычных приложениях такая функциональность не столь востребована, но когда у вас БД на гигабайты в памяти и любой перезапуск будет стоить вам времени запуска, это может сослужить отличную службу. Давайте рассмотрим, что необходимо сделать для горячей перезагрузки кода.


Когда Lua загружает какой-то код через require, содержимое этого файла интерпретируется и возвращённый результат кешируется в системной таблице package.loaded под именем модуля. Последующие вызовы require того же самого модуля не будут повторно читать файл, а будут возвращать закешированное значение. Чтобы заставить Lua повторно прочитать файл и загрузить его, достаточно просто стереть запись из package.loaded[...] и повторно вызвать require. Также мы должны запомнить то, что предзагружено самим рантаймом, потому что файлов для перезагрузки встроенных модулей не будет. Простейший сниппет кода для обработки релоада может выглядеть как-то так:


require'strict'.on()fiber = require 'fiber'box.cfg{    listen = '127.0.0.1:3301'}box.schema.user.grant('guest', 'super', nil, nil, { if_not_exists = true })local not_first_run = rawget(_G,'_NOT_FIRST_RUN')_NOT_FIRST_RUN = trueif not_first_run then   for k,v in pairs(package.loaded) do      if not preloaded[k] then         package.loaded[k] = nil      end   endelse   preloaded = {}   for k,v in pairs(package.loaded) do      preloaded[k] = true   endendqueue = require 'queue'require'console'.start()os.exit()

Поскольку релоад кода является довольно типичной и регулярной задачей, у нас уже есть готовый модуль package.reload, которым мы пользуемся в подавляющем большинстве приложений. Он сам запоминает, из какого файла всё было загружено, какие модули были предзагружены, и предоставляет удобный вызов для инициации перезагрузки: package.reload().


require'strict'.on()fiber = require 'fiber'box.cfg{    listen = '127.0.0.1:3301'}box.schema.user.grant('guest', 'super', nil, nil, { if_not_exists = true })require 'package.reload'queue = require 'queue'require'console'.start()os.exit()

Чтобы код можно было перезагружать, нужно писать его немного по-другому. Учитывайте, что код может исполняться повторно. Первый раз он исполняется при первой загрузке, остальные разы при перезагрузке. Соответственно, нам необходимо явно обрабатывать эту ситуацию.


local queue = {}local old = rawget(_G,'queue')if old then    queue.taken = old.taken    queue.bysid = old.bysid    queue._triggers = old._triggers    queue._stats = old._stats    queue._wait = old._wait    queue._runch = old._runch    queue._runat = old._runatelse    queue.taken = {}    queue.bysid = {}    queue._triggers = {}    queue._stats = {}    queue._wait = fiber.channel()    queue._runch = fiber.cond()    while true do        local t = box.space.queue.index.status:pairs({STATUS.TAKEN}):nth(1)        if not t then break end        box.space.queue:update({ t.id }, {{'=', F.status, STATUS.READY }})        log.info("Autoreleased %s at start", t.id)    end    for k,v in pairs(STATUS) do        queue._stats[v] = 0LL    end    for _,t in box.space.queue:pairs() do        queue._stats[ t[F.status] ] = (queue._stats[ t[F.status] ] or 0LL)+1    end    log.info("Perform initial stat counts %s", box.tuple.new{ queue._stats })end

Также учитывайте перезагрузку триггеров. Если оставить как было, то каждая перезагрузка будет порождать установку дополнительного триггера. Но триггеры поддерживают указание старой функции, при этом установка триггера возвращает её. Поэтому просто будем сохранять результат установки в переменную и передавать её в качестве аргумента. При первом запуске переменной не будет и установится новый триггер. При последующих загрузках триггер будет заменяться.


queue._triggers.on_replace = box.space.queue:on_replace(function(old,new)    if old then        queue._stats[ old[ F.status ] ] = queue._stats[ old[ F.status ] ] - 1    end    if new then        queue._stats[ new[ F.status ] ] = queue._stats[ new[ F.status ] ] + 1    endend, queue._triggers.on_replace)queue._triggers.on_truncate = box.space._truncate:on_replace(function(old,new)    if new.id == box.space.queue.id then        for k,v in pairs(queue._stats) do            queue._stats[k] = 0LL        end    endend, queue._triggers.on_truncate)queue._triggers.on_connect = box.session.on_connect(function()    box.session.storage.peer = box.session.peer()    log.info( "connected %s from %s", box.session.id(), box.session.storage.peer )end, queue._triggers.on_connect)queue._triggers.on_disconnect = box.session.on_disconnect(function()    log.info( "disconnected %s from %s", box.session.id(), box.session.storage.peer )    box.session.storage.destroyed = true    local sid = box.session.id()    local bysid = queue.bysid[ sid ]    if bysid then        while next(bysid) do            for key, id in pairs(bysid) do                log.info("Autorelease %s by disconnect", id);                queue.taken[key] = nil                bysid[key] = nil                local t = box.space.queue:get(id)                if t then                    if queue._wait:has_readers() then queue._wait:put(true,0) end                    box.space.queue:update({t.id},{{'=', F.status, STATUS.READY }})                end            end        end        queue.bysid[ sid ] = nil    endend, queue._triggers.on_disconnect)

Ещё один важный элемент перезагрузки это файберы. Файбер запускается в фоне, мы его никак не контролируем. В нём написано while ... true, он никогда не завершится и сам по себе не перезагрузится. Для того, чтобы взаимодействовать с ним, нам понадобится канал, а ещё лучше fiber.cond: condition variable, предназначенная для передачи сигналов файберам.


Есть несколько различных подходов к перезагрузке файберов. Например, можно уничтожать старые при помощи вызова fiber.kill, но такой подход не очень консистентен: мы можем вызвать kill в неподходящий момент. Поэтому в большинстве случаев мы пользуемся признаком поколения файбера: файбер продолжает свою работу только в том поколении, в котором он был создан. При перезагрузке кода поколение меняется и файбер чисто завершается. Также мы можем защититься от одновременной работы нескольких файберов: для этого мы можем смотреть на статус файбера предыдущего поколения.


queue._runat = fiber.create(function(queue, gen, old_fiber)    fiber.name('queue.runat.'..gen)    while package.reload.count == gen and old_fiber and old_fiber:status() ~= 'dead' do        log.info("Waiting for old to die")        queue._runch:wait(0.1)    end    log.info("Started...")    while package.reload.count == gen do        local remaining        local now = clock.realtime()        for _,t in box.space.queue.index.runat            :pairs( {0}, { iterator = 'GT' })        do            if t.runat > now then                remaining = t.runat - now                break            else                if t.status == STATUS.WAITING then                    log.info("Runat: W->R %s",t.id)                    if queue._wait:has_readers() then queue._wait:put(true,0) end                    box.space.queue:update({ t.id }, {                        { '=', F.status, STATUS.READY },                        { '=', F.runat, 0 },                    })                else                    log.error("Runat: bad status %s for %s", t.status, t.id)                    box.space.queue:update({ t.id },{{ '=', F.runat, 0 }})                end            end        end        if not remaining or remaining > 1 then remaining = 1 end        queue._runch:wait(remaining)    end    queue._runch:broadcast()    log.info("Finished")end, queue, package.reload.count, queue._runat)queue._runch:broadcast()

И напоследок: при перезагрузке кода у вас будет ошибка, что консоль уже запущена. Обработать эту ситуацию можно следующим способом:


if not fiber.self().storage.console then    require'console'.start()    os.exit()end

Подведем итог


Мы написали работающую сетевую очередь с возможностью отложенной обработки, с автовозвратом задач при помощи триггеров, с отправкой статистики в Graphite по TCP, и рассмотрели довольно много нюансов. На среднестатистическом современном железе такая очередь без проблем выдержит передачу от 20 тыс. сообщений в секунду. Она состоит примерно из 300 строк кода и пишется за день с изучением документации.


Результирующие файлы

queue.lua:


local clock = require 'clock'local errno = require 'errno'local fiber = require 'fiber'local log = require 'log'local msgpack = require 'msgpack'local socket = require 'socket'box.schema.create_space('queue',{ if_not_exists = true; })box.space.queue:format( {    { name = 'id';     type = 'number' },    { name = 'status'; type = 'string' },    { name = 'runat';  type = 'number' },    { name = 'data';   type = '*'      },} );local F = {}for no,def in pairs(box.space.queue:format()) do    F[no] = def.name    F[def.name] = noendbox.space.queue:create_index('primary', {   parts = { F.id, 'number' };   if_not_exists = true;})box.space.queue:create_index('status', {    parts = { F.status, 'string', F.id, 'number' };    if_not_exists = true;})box.space.queue:create_index('runat', {    parts = { F.runat, 'number', F.id, 'number' };    if_not_exists = true;})local STATUS = {}STATUS.READY = 'R'STATUS.TAKEN = 'T'STATUS.WAITING = 'W'local queue = {}local old = rawget(_G,'queue')if old then    queue.taken = old.taken    queue.bysid = old.bysid    queue._triggers = old._triggers    queue._stats = old._stats    queue._wait = old._wait    queue._runch = old._runch    queue._runat = old._runatelse    queue.taken = {}    queue.bysid = {}    queue._triggers = {}    queue._stats = {}    queue._wait = fiber.channel()    queue._runch = fiber.cond()    while true do        local t = box.space.queue.index.status:pairs({STATUS.TAKEN}):nth(1)        if not t then break end        box.space.queue:update({ t.id }, {{'=', F.status, STATUS.READY }})        log.info("Autoreleased %s at start", t.id)    end    for k,v in pairs(STATUS) do queue._stats[v] = 0LL end    for _,t in box.space.queue:pairs() do        queue._stats[ t[F.status] ] = (queue._stats[ t[F.status] ] or 0LL)+1    end    log.info("Perform initial stat counts %s", box.tuple.new{ queue._stats })endlocal function gen_id()    local new_id    repeat        new_id = clock.realtime64()    until not box.space.queue:get(new_id)    return new_idendlocal function keypack( key )    return msgpack.encode( key )endlocal function keyunpack( data )    return msgpack.decode( data )endqueue._triggers.on_replace = box.space.queue:on_replace(function(old,new)    if old then        queue._stats[ old[ F.status ] ] = queue._stats[ old[ F.status ] ] - 1    end    if new then        queue._stats[ new[ F.status ] ] = queue._stats[ new[ F.status ] ] + 1    endend, queue._triggers.on_replace)queue._triggers.on_truncate = box.space._truncate:on_replace(function(old,new)    if new.id == box.space.queue.id then        for k,v in pairs(queue._stats) do            queue._stats[k] = 0LL        end    endend, queue._triggers.on_truncate)queue._triggers.on_connect = box.session.on_connect(function()    box.session.storage.peer = box.session.peer()end, queue._triggers.on_connect)queue._triggers.on_disconnect = box.session.on_disconnect(function()    box.session.storage.destroyed = true    local sid = box.session.id()    local bysid = queue.bysid[ sid ]    if bysid then        log.info( "disconnected %s from %s", box.session.id(), box.session.storage.peer )        while next(bysid) do            for key, id in pairs(bysid) do                log.info("Autorelease %s by disconnect", id);                queue.taken[key] = nil                bysid[key] = nil                local t = box.space.queue:get(id)                if t then                    if queue._wait:has_readers() then queue._wait:put(true,0) end                    box.space.queue:update({t.id},{{'=', F.status, STATUS.READY }})                end            end        end        queue.bysid[ sid ] = nil    endend, queue._triggers.on_disconnect)queue._runat = fiber.create(function(queue, gen, old_fiber)    fiber.name('queue.runat.'..gen)    while package.reload.count == gen and old_fiber and old_fiber:status() ~= 'dead' do        log.info("Waiting for old to die")        queue._runch:wait(0.1)    end    log.info("Started...")    while package.reload.count == gen do        local remaining        local now = clock.realtime()        for _,t in box.space.queue.index.runat            :pairs( {0}, { iterator = 'GT' })        do            if t.runat > now then                remaining = t.runat - now                break            else                if t.status == STATUS.WAITING then                    log.info("Runat: W->R %s",t.id)                    if queue._wait:has_readers() then queue._wait:put(true,0) end                    box.space.queue:update({ t.id }, {                        { '=', F.status, STATUS.READY },                        { '=', F.runat, 0 },                    })                else                    log.error("Runat: bad status %s for %s", t.status, t.id)                    box.space.queue:update({ t.id },{{ '=', F.runat, 0 }})                end            end        end        if not remaining or remaining > 1 then remaining = 1 end        queue._runch:wait(remaining)    end    queue._runch:broadcast()    log.info("Finished")end, queue, package.reload.count, queue._runat)queue._runch:broadcast()local graphite_host = '127.0.0.1'local graphite_port = 2003queue._monitor = fiber.create(function(gen)    fiber.name('queue.mon.'..gen)    fiber.yield()    while package.reload.count == gen do        local remote =  require 'socket'.tcp_connect(graphite_host, graphite_port)        if not remote then            log.error("Failed to connect to graphite %s",errno.strerror())            fiber.sleep(1)        else            while package.reload.count == gen do                local data = {}                for k,v in pairs(queue.stats()) do                    table.insert(data,string.format("queue.stats.%s %s %s\n",k,tonumber(v),math.floor(fiber.time())))                end                data = table.concat(data,'')                if not remote:send(data) then                    log.error("%s",errno.strerror())                    break                end                fiber.sleep(1)            end        end    endend, package.reload.count)function queue.put(data, opts)    local id = gen_id()    local runat = 0    local status = STATUS.READY    if opts and opts.delay then        runat = clock.realtime() + tonumber(opts.delay)        status = STATUS.WAITING    else        if queue._wait:has_readers() then            queue._wait:put(true,0)        end    end    return box.space.queue        :insert{ id, status, runat, data }        :tomap{ names_only=true }endfunction queue.take(timeout)    if not timeout then timeout = 0 end    local now = fiber.time()    local found    while not found do        found = box.space.queue.index.status            :pairs({STATUS.READY},{ iterator = 'EQ' }):nth(1)        if not found then            local left = (now + timeout) - fiber.time()            if left <= 0 then return end            queue._wait:get(left)        end    end    if box.session.storage.destroyed then return end    local sid = box.session.id()    log.info("Register %s by %s", found.id, sid)    local key = keypack( found.id )    queue.taken[ key ] = sid    queue.bysid[ sid ] = queue.bysid[ sid ] or {}    queue.bysid[ sid ][ key ] = found.id    return box.space.queue        :update( {found.id}, {{'=', F.status, STATUS.TAKEN }})        :tomap{ names_only = true }endlocal function get_task( id )    if not id then error("Task id required", 2) end    id = tonumber64(id)    local key = keypack(id)    local t = box.space.queue:get{id}    if not t then        error(string.format( "Task {%s} was not found", id ), 2)    end    if not queue.taken[key] then        error(string.format( "Task %s not taken by anybody", id ), 2)    end    if queue.taken[key] ~= box.session.id() then        error(string.format( "Task %s taken by %d. Not you (%d)",            id, queue.taken[key], box.session.id() ), 2)    end    return t, keyendfunction queue.ack(id)    local t, key = get_task(id)    queue.taken[ key ] = nil    queue.bysid[ box.session.id() ][ key ] = nil    return box.space.queue:delete{t.id}:tomap{ names_only = true }endfunction queue.release(id, opts)    local t, key = get_task(id)    queue.taken[ key ] = nil    queue.bysid[ box.session.id() ][ key ] = nil    local runat = 0    local status = STATUS.READY    if opts and opts.delay then        runat = clock.realtime() + tonumber(opts.delay)        status = STATUS.WAITING    else        if queue._wait:has_readers() then queue._wait:put(true,0) end    end    return box.space.queue        :update({t.id},{{'=', F.status, status },{ '=', F.runat, runat }})        :tomap{ names_only = true }endfunction queue.stats()    return {        total   = box.space.queue:len(),        ready   = queue._stats[ STATUS.READY ],        waiting = queue._stats[ STATUS.WAITING ],        taken   = queue._stats[ STATUS.TAKEN ],    }endreturn queue

init.lua:


require'strict'.on()fiber = require 'fiber'require 'package.reload'box.cfg{    listen = '127.0.0.1:3301'}box.schema.user.grant('guest', 'super', nil, nil, { if_not_exists = true })queue = require 'queue'if not fiber.self().storage.console then    require'console'.start()    os.exit()end



14 июля можно будет посмотреть на это в деле на практикуме Rebrain & Tarantool: Разбираемся с отказоустойчивым application server Tarantool. Более подробная информация и регистрация по ссылке.

Источник: habr.com
К списку статей
Опубликовано: 10.07.2020 18:18:20
0

Сейчас читают

Комментариев (0)
Имя
Электронная почта

Блог компании mail.ru group

Lua

Алгоритмы

Высокая производительность

Tarantool

Queue

Tutorial

Категории

Последние комментарии

© 2006-2020, personeltest.ru