X-Git-Url: https://repo.niif.hu/gitweb/gitweb.cgi?p=java-idp.git;a=blobdiff_plain;f=src%2Fedu%2Finternet2%2Fmiddleware%2Fshibboleth%2Fidp%2Fsession%2Fimpl%2FSessionImpl.java;h=ffebd4b5f2b1ff543319a7e65a9127f31ca5fd04;hp=7074bfa8348abeba1e4c56339062aab2c02a6e3a;hb=92fabdf36fb525fc110f7f8f565ebab007ae9ac0;hpb=beeb6427ef2d3586fb326ea3a3d260814a69f56d

diff --git a/src/edu/internet2/middleware/shibboleth/idp/session/impl/SessionImpl.java b/src/edu/internet2/middleware/shibboleth/idp/session/impl/SessionImpl.java
index 7074bfa..ffebd4b 100644
--- a/src/edu/internet2/middleware/shibboleth/idp/session/impl/SessionImpl.java
+++ b/src/edu/internet2/middleware/shibboleth/idp/session/impl/SessionImpl.java
@@ -17,9 +17,10 @@
 package edu.internet2.middleware.shibboleth.idp.session.impl;
 
 import java.net.InetAddress;
+import java.util.ArrayList;
+import java.util.HashMap;
 import java.util.List;
-
-import javolution.util.FastList;
+import java.util.Map;
 
 import edu.internet2.middleware.shibboleth.common.session.impl.AbstractSession;
 import edu.internet2.middleware.shibboleth.idp.session.AuthenticationMethodInformation;
@@ -38,7 +39,7 @@ public class SessionImpl extends AbstractSession implements Session {
     private List<AuthenticationMethodInformation> authnMethods;
 
     /** The list of services to which the user has logged in. */
-    private List<ServiceInformation> servicesInformation;
+    private Map<String, ServiceInformation> servicesInformation;
 
     /**
      * Default constructor.
@@ -49,18 +50,27 @@ public class SessionImpl extends AbstractSession implements Session {
     public SessionImpl(InetAddress presenter, String principal) {
         super(presenter, principal);
 
-        authnMethods = new FastList<AuthenticationMethodInformation>();
-        servicesInformation = new FastList<ServiceInformation>();
+        authnMethods = new ArrayList<AuthenticationMethodInformation>();
+        servicesInformation = new HashMap<String, ServiceInformation>();
     }
 
     /** {@inheritDoc} */
     public List<AuthenticationMethodInformation> getAuthenticationMethods() {
         return authnMethods;
     }
+    
+    /** {@inheritDoc} */
+    public ServiceInformation getServiceInformation(String entityId) {
+        return servicesInformation.get(entityId);
+    }
 
     /** {@inheritDoc} */
     public List<ServiceInformation> getServicesInformation() {
-        return servicesInformation;
+        ArrayList<ServiceInformation> info = new ArrayList<ServiceInformation>();
+        for(Map.Entry<String, ServiceInformation> entry : servicesInformation.entrySet()){
+            info.add(entry.getValue());
+        }
+        
+        return info;
     }
-
-}
+}
\ No newline at end of file