cas ticket 存储方式-memcache

修改 ticketRegistry.xml :

<bean id="ticketRegistry" class="com.olymtech.cas.ticket.registry.MemCacheTicketRegistry">
  <constructor-arg index="0">
   <list>
    <value>${memcache.url}</value>
   </list>
  </constructor-arg>
  <constructor-arg index="1" type="int" value="21600" />
  <constructor-arg index="2" type="int" value="300" />
 </bean>

 

MemCacheTicketRegistry.java

 

package com.olymtech.cas.ticket.registry;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.concurrent.Future;

import net.spy.memcached.MemcachedClient;

import org.jasig.cas.ticket.ServiceTicket;
import org.jasig.cas.ticket.Ticket;
import org.jasig.cas.ticket.TicketGrantingTicket;
import org.springframework.beans.factory.DisposableBean;

import javax.validation.constraints.Min;

import com.olymtech.common.cache.MemCachedFactory;

/**
 * Memcache (or Repcache) backed ticket registry.
 *
 */
public final class MemCacheTicketRegistry extends AbstractDistributedTicketRegistry implements DisposableBean {

 private final MemcachedClient client;

 @Min(0)
 private final int tgtTimeout;

 @Min(0)
 private final int stTimeout;

 private boolean synchronizeUpdatesToRegistry = false;

 /**
  * @param hostnames
  * @param ticketGrantingTicketTimeOut
  * @param serviceTicketTimeOut
  */
 public MemCacheTicketRegistry(final String[] hostnames, final int ticketGrantingTicketTimeOut, final int serviceTicketTimeOut) {
 
  System.out.println("use memcached...");
  this.tgtTimeout = ticketGrantingTicketTimeOut;
  this.stTimeout = serviceTicketTimeOut;
  final List<InetSocketAddress> addresses = new ArrayList<InetSocketAddress>();

  for (final String hostname : hostnames) {
   String[] hostPort = hostname.split(":");
   addresses.add(new InetSocketAddress(hostPort[0], Integer.parseInt(hostPort[1])));
  }

  try {
   this.client = new MemcachedClient(addresses);
  } catch (final IOException e) {
   throw new IllegalStateException(e);
  }
  
 }

 /**
  * This alternative constructor takes time in milliseconds. It has the
  * timeout parameters first because, otherwise it would be almost impossible
  * to differentiate the two.
  *
  * @param ticketGrantingTicketTimeOut
  * @param serviceTicketTimeOut
  * @param hostnames
  */
 public MemCacheTicketRegistry(final long ticketGrantingTicketTimeOut, final long serviceTicketTimeOut, final String[] hostnames) {
  this(hostnames, (int) (ticketGrantingTicketTimeOut / 1000), (int) (serviceTicketTimeOut / 1000));
 }

 private void handleSynchronousRequest(final Future f) {
  try {
   if (this.synchronizeUpdatesToRegistry) {
    f.get();
   }
  } catch (final Exception e) {
   // ignore these.
  }
 }

 protected void updateTicket(final Ticket ticket) {
  if (ticket instanceof TicketGrantingTicket) {
   handleSynchronousRequest(this.client.replace(ticket.getId(), this.tgtTimeout, ticket));
     }

  if (ticket instanceof ServiceTicket) {
   handleSynchronousRequest(this.client.replace(ticket.getId(), this.stTimeout, ticket));
     }
 }

 public void addTicket(final Ticket ticket) {
  if (ticket instanceof TicketGrantingTicket) {
   handleSynchronousRequest(this.client.add(ticket.getId(), this.tgtTimeout, ticket));
     }

  if (ticket instanceof ServiceTicket) {
   handleSynchronousRequest(this.client.add(ticket.getId(), this.stTimeout, ticket));
   }
 }

 public boolean deleteTicket(final String ticketId) {

  Future<Boolean> f = this.client.delete(ticketId);
  try {
   return f.get().booleanValue();
  } catch (final Exception e) {
   log.error(e.getMessage(), e);
   return false;
  }
    
 }

 public Ticket getTicket(final String ticketId) {
  final Ticket t = (Ticket) this.client.get(ticketId);
  if (t == null) {
   return null;
  }

  return getProxiedTicketInstance(t);
 }


 public Collection<Ticket> getTickets() {
  throw new UnsupportedOperationException("GetTickets not supported.");
 }

 public void destroy() throws Exception {
  this.client.shutdown();
 }

 public void setSynchronizeUpdatesToRegistry(final boolean b) {
  this.synchronizeUpdatesToRegistry = b;
 }
}

posted on 2010-08-05 17:53  horus  阅读(3693)  评论(3编辑  收藏  举报