#!/usr/bin/env lua

local exec = require "tek.lib.exec"

local workers = { }
for i = 1, 50 do
	workers[i] = exec.run({ taskname = "worker" .. i, func = function()
		local exec = require "tek.lib.exec"
		while true do
			local order, sender = exec.waitmsg()
			if order == "getrandom" then
				assert(exec.sendmsg(sender, math.random(1, 10)))
			elseif order == "quit" then
				break
			end
		end
	end })
end
print(#workers .. " workers started")

local consumers = { }
for i = 1, 200 do
	consumers[i] = exec.run(function()
		local exec = require "tek.lib.exec"
		for i = 1, 10000 do
			local w = (i % arg[1]) + 1
			-- assert(exec.sendmsg("worker" .. w, "getrandom"))
			if exec.sendmsg("worker" .. w, "getrandom") then
				local number = exec.waitmsg()
				exec.sendmsg("*p", number)
			end
		end
	end, #workers)
end
print(#consumers .. " consumers started")

-- limit the number of replies to wait for:
local numres = #consumers * 1000 / 10
for i = 1, numres do
	exec.waitmsg()
end
print "received enough consumer goods"

for i = 1, #workers do
	workers[i]:sendmsg("quit")
end

print "workers signalled"
for i = 1, #workers do
	workers[i]:join()
end
print "workers joined"

print "consumers still running around die now..."