using Distributed
global numWorkers = 5
addprocs(numWorkers)

@everywhere function workerInit(_size, firstLeader)
        global workload = round(rand(), digits=1)
        global workloads = Vector{Float64}(undef, _size)
        global participant = false
        global currLeader = firstLeader
        global size = _size
        work()
    end

@everywhere function dataFromWorker()
        myid_ = myid()
            println("Workloads: $(workloads)")
    end

@everywhere function work()
        while true
            if !participant
                sendWorkload()
            end
            sleep(2)
            dataFromWorker()
            checkPossibleElection()
        end
    end

@everywhere function sendWorkload()
        global workload =  round(rand(), digits=1)
        id = myid()
        for w in workers()
            @spawnat(w, updateWorkload(id, workload))
        end
    end

@everywhere function updateWorkload(id, newWorkload)
        global workloads[id-1] = newWorkload
    end

@everywhere function checkPossibleElection()
        id = myid()
        if !participant && workloads[currLeader-1] >= 0.8 && workloads[id-1] < 0.8
            global participant = true
            local next = nextWorker()
            id = myid()
            @spawnat next propagateElection(workloads[id-1], id, id)
        end
    end

@everywhere function propagateElection(best, bestId, starterId)
        id = myid()
        next = nextWorker()
        if id != starterId
            global participant = true
            if (best > workloads[id-1]) || (best == workloads[id-1] && id > bestId)
                @spawnat next propagateElection(workloads[id-1], id, starterId)
            else
                @spawnat next propagateElection(best, bestId, starterId)
            end
        else
            for w in workers()
                @spawnat w updateLeader(bestId)
            end
        end
    end

@everywhere function updateLeader(newLeader)
        if participant
            id = myid()
            global participant = false
            if id == newLeader
                println("New leader elected! Worker $(id)")
            end
            global currLeader = newLeader
        end
    end

@everywhere function nextWorker()
        local id = myid()
        id = ((id-1)%size)+2
        return id
    end

@sync for w in workers()
    @spawn workerInit(numWorkers, 2)
end