src/main/java/at/ac/tuwien/sbc/valesriegler/xvsm/GroupAgentXVSM.java

   1 package at.ac.tuwien.sbc.valesriegler.xvsm;
   2
   3 import at.ac.tuwien.sbc.valesriegler.common.Util;
   4 import at.ac.tuwien.sbc.valesriegler.group.DeliveryGroup;
   5 import at.ac.tuwien.sbc.valesriegler.group.GroupAgent;
   6 import at.ac.tuwien.sbc.valesriegler.group.gui.DeliveryOverviewModel;
   7 import at.ac.tuwien.sbc.valesriegler.types.DeliveryGroupData;
   8 import at.ac.tuwien.sbc.valesriegler.types.GroupData;
   9 import at.ac.tuwien.sbc.valesriegler.types.PizzaType;
  10 import at.ac.tuwien.sbc.valesriegler.xvsm.spacehelpers.SpaceAction;
  11 import org.mozartspaces.core.MzsConstants.RequestTimeout;
  12 import org.slf4j.Logger;
  13 import org.slf4j.LoggerFactory;
  14
  15 import javax.swing.*;
  16 import java.io.Serializable;
  17 import java.util.ArrayList;
  18 import java.util.Arrays;
  19 import java.util.List;
  20 import java.util.Set;
  21
  22 import static java.lang.String.valueOf;
  23
  24 public class GroupAgentXVSM extends AbstractXVSMConnector {
  25         private static final Logger log = LoggerFactory.getLogger(GroupAgentXVSM.class);
  26
  27         public GroupAgentXVSM() {
  28                 super(Util.GROUP_AGENT_PORT);
  29         groupAgentInfoContainer = useContainer(Util.GROUP_AGENT_INFO);
  30         }
  31
  32     public void listenForNewPizzerias() {
  33         getDefaultBuilder().setCref(groupAgentInfoContainer).setLookaround(true).setSpaceAction(new SpaceAction() {
  34
  35             @Override
  36             public void onEntriesWritten(List<? extends Serializable> entries) throws Exception {
  37                 final Set<String> pizzeriaIdentifiers = GroupAgent.getInstance().getPizzeriaIdentifiers();
  38
  39                 synchronized (pizzeriaIdentifiers) {
  40                     final List<PizzeriaRegistration> pizzeriaRegistrations = castEntries(entries);
  41                     final List<String> pizzeriaIds = new ArrayList<>();
  42                     for (PizzeriaRegistration registration : pizzeriaRegistrations) {
  43                         final String pizzeriaId = valueOf(registration.pizzeriaSpacePort);
  44                         if (!pizzeriaIdentifiers.contains(pizzeriaId)) {
  45                             pizzeriaIds.add(pizzeriaId);
  46                             listenToPizzeria(pizzeriaId);
  47                         }
  48                     }
  49
  50                     SwingUtilities.invokeLater(new Runnable() {
  51                         @Override
  52                         public void run() {
  53                             pizzeriaIdentifiers.addAll(pizzeriaIds);
  54                         }
  55                     });
  56                 }
  57             }
  58         }).createSpaceListenerImpl();
  59
  60     }
  61
  62     public void listenToPizzeria(String pizzeriaId) {
  63         final int port = Integer.parseInt(pizzeriaId);
  64         listenToGroupUpdates(port);
  65         listenToDeliveryUpdates(port);
  66     }
  67
  68     private void listenToDeliveryUpdates(int port) {
  69         getDefaultBuilder().setCref(useContainerOfSpaceWithPort(Util.PIZZERIA_DELIVERY, port)).setSpaceAction(new SpaceAction() {
  70             @Override
  71             public void onEntriesWritten(List<? extends Serializable> entries) throws Exception {
  72                 final List<DeliveryGroupData> groups = castEntries(entries);
  73
  74                 SwingUtilities.invokeLater(new Runnable() {
  75
  76                     @Override
  77                     public void run() {
  78                         GroupAgent.getInstance().getDeliveryModel().addDeliveries(groups);
  79                     }
  80                 });
  81             }
  82         }).createSpaceListenerImpl();
  83     }
  84
  85     private void listenToGroupUpdates(int port) {
  86         getDefaultBuilder().setCref(useContainerOfSpaceWithPort(Util.PIZZERIA_GROUP, port)).setSpaceAction(new SpaceAction() {
  87             @Override
  88             public void onEntriesWritten(List<? extends Serializable> entries) throws Exception {
  89                 final List<GroupData> groups = castEntries(entries);
  90
  91                 SwingUtilities.invokeLater(new Runnable() {
  92
  93                     @Override
  94                     public void run() {
  95                         GroupAgent.getInstance().getGroupModel().addGroupData(groups);
  96                     }
  97                 });
  98             }
  99         }).createSpaceListenerImpl();
 100     }
 101
 102     public void sendNewGroupsToSpace(List<GroupData> groupData, int pizzeriaSpacePort) {
 103         sendItemsToContainer(groupData, useContainerOfSpaceWithPort(Util.PIZZERIA_GROUP, pizzeriaSpacePort), RequestTimeout.DEFAULT, null);
 104         sendItemsToContainer(groupData, useContainerOfSpaceWithPort(Util.ASSIGN_TABLE, pizzeriaSpacePort), RequestTimeout.DEFAULT, null);
 105     }
 106
 107     public void sendNewDeliveriesToSpace(List<DeliveryGroupData> groupData, int pizzeriaSpacePort) {
 108         sendItemsToContainer(groupData, useContainerOfSpaceWithPort(Util.PIZZERIA_DELIVERY, pizzeriaSpacePort), RequestTimeout.DEFAULT, null);
 109         sendItemsToContainer(groupData, useContainerOfSpaceWithPort(Util.PHONE_CALLS, pizzeriaSpacePort), RequestTimeout.DEFAULT, null);
 110     }
 111
 112     public void runSimulation() {
 113         new Thread(new Runnable() {
 114             @Override
 115             public void run() {
 116                 List<DeliveryGroup> newDeliveryGroups = new ArrayList<>();
 117                 List<PizzaType> pizzaTypes1 = Arrays.asList(PizzaType.CARDINALE, PizzaType.SALAMI, PizzaType.MARGHERITA, PizzaType.MARGHERITA);
 118                 List<PizzaType> pizzaTypes2 = Arrays.asList(PizzaType.CARDINALE, PizzaType.SALAMI, PizzaType.MARGHERITA);
 119                 List<PizzaType> pizzaTypes3 = Arrays.asList(PizzaType.SALAMI, PizzaType.MARGHERITA);
 120
 121                 final String pizzeria1 = "9875";
 122                 final String pizzeria2 = "9874";
 123                 List<DeliveryGroup> groups1 = GroupAgent.getInstance().createGroups(pizzaTypes1, pizzeria1, 4);
 124                 List<DeliveryGroup> groups2 = GroupAgent.getInstance().createGroups(pizzaTypes2, pizzeria2, 3);
 125                 List<DeliveryGroup> groups3 = GroupAgent.getInstance().createGroups(pizzaTypes3, pizzeria2, 2);
 126
 127                 final DeliveryOverviewModel deliveryModel = GroupAgent.getInstance().getDeliveryModel();
 128                 deliveryModel.addItems(groups1);
 129                 deliveryModel.addItems(groups2);
 130                 deliveryModel.addItems(groups3);
 131
 132                 GroupAgent.getInstance().onDeliveryGroupsCreated(groups2);
 133                 GroupAgent.getInstance().onDeliveryGroupsCreated(groups3);
 134                 GroupAgent.getInstance().onDeliveryGroupsCreated(groups1);
 135
 136                 log.info("ATTENTION: Now let's wait 60 seconds!");
 137
 138                 try {
 139                     Thread.sleep(60000);
 140                 } catch (InterruptedException e) {
 141                     e.printStackTrace();  //To change body of catch statement use File | Settings | File Templates.
 142                 }
 143
 144                 log.info("ATTENTION: It's all over now!");
 145                 notificationMgr.shutdown();
 146
 147 //                        System.exit(1);
 148             }
 149         }).start();
 150     }
 151 }