package christie.example.PrefixTree;

import christie.annotation.Peek;
import christie.annotation.Take;
import christie.codegear.CodeGear;
import christie.codegear.CodeGearManager;
import christie.topology.Message;
import christie.topology.TopologyDataGear;

public class CountMessage extends CodeGear {
    @Take
    Message ack;

    @Peek
    int totalNodeNum;

    @Peek
    TopologyDataGear topoDG;

    @Take
    int count;

    @Override
    protected void run(CodeGearManager cgm) {
        count ++;
        if(count == totalNodeNum){
            System.out.print("success");
             for(String nodeName : topoDG.getConnectionList()) {
                put(nodeName, "start", new Message());
            }

            cgm.setup(new CountCalculate());
        }else{
            cgm.setup(new CountMessage());
            put("count",count);
        }

    }
}