//http://jalarms.svn.sourceforge.net/viewvc/jalarms/maven-project/src/main/java/com/solab/alarms/AlarmMemcachedClient.java?revision=59&content-type=text/plain&pathrev=59

/*
jAlarms A simple Java library to enable server apps to send alarms to sysadmins.
Copyright (C) 2009 Enrique Zamudio Lopez

This library is free software you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public
License as published by the Free Software Foundation either
version 2.1 of the License, or (at your option) any later version.

This library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
Lesser General Public License for more details.

You should have received a copy of the GNU Lesser General Public
License along with this library if not, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA
*/

package com.solab.alarms                                                                                       

import java.io.IOException                                                                              
import java.net.InetSocketAddress                                                                              
import java.util.List                                                                                          

import javax.annotation.PostConstruct                                                                          
import javax.annotation.PreDestroy                                                                             

import org.slf4j.Logger                                                                                        
import org.slf4j.LoggerFactory                                                                                 

import net.spy.memcached.MemcachedClient                                                                       
import net.spy.memcached.OperationTimeoutException                                                             


public class AlarmMemcachedClient implements AlarmCache                                                        

  private final Logger log = LoggerFactory.getLogger(getClass())                                               
  private MemcachedClient mc                                                                                   
  private InetSocketAddress[] servers                                                                          
  private int defint = 120                                                                                     

  public void setDefaultInterval(int value)                                                                    
    defint = value                                                                                             
                                                                                                               
  public int getDefaultInterval()                                                                              
    return defint                                                                                              
                                                                                                               

  public void setServers(List<String> value)                                                                   
    servers = new InetSocketAddress[value.size()]                                                              
    int pos = 0                                                                                                
    for (String s : value)                                                                                     
      int cp = value.indexOf(':')                                                                              
      if (cp > 0)                                                                                              
        servers[pos++] = new InetSocketAddress(s.substring(0, cp),
                                Integer.parseInt(s.substring(cp + 1)))                                         
      else                                                                                                     
        servers[pos++] = new InetSocketAddress(s, 11211)                                                       
                                                                                                               
                                                                                                               
                                                                                                               

  @PostConstruct
  public void init() throws IOException                                                                        
    mc = new MemcachedClient(servers)                                                                          
                                                                                                               

  @PreDestroy
  public void disconnect()                                                                                     
    mc.shutdown()                                                                                              
                                                                                                               

  @Override
  public void store(AlarmChannel channel, String source, String message)                                       
    if (mc == null)                                                                                            
      synchronized (this)                                                                                      
        if (mc == null)                                                                                        
          try                                                                                                  
            init()                                                                                              
          catch (IOException ex)                                                                               
            log.error("Initializing alarm memcached client", ex)                                               
            return                                                                                             
                                                                                                               
                                                                                                               
                                                                                                               
                                                                                                               
                                                                                                               

  @Override
  public boolean shouldResend(AlarmChannel channel, String source,
      String message)                                                                                          
    if (mc == null)                                                                                            
      return true                                                                                              
                                                                                                               
    try                                                                                                        
      return mc.get(k) == null                                                                                  
    catch (OperationTimeoutException ex)                                                                       
      log.error("Timeout waiting to retrieve  from memcached", k, ex)                                        
    catch (RuntimeException ex)                                                                                
      log.error("Retrieving key  from memcached", k, ex.getCause() == null ? ex : ex.getCause())             
    return true                                                                                                
                                                                                                               

  @Override
  public String toString()                                                                                     
    return String.format("Memcached(%s)", mc == null ? "disconnected" : mc.getAvailableServers())              
                                                                                                               

  @Override
  public void shutdown()                                                                                       
    if (mc != null)                                                                                            
      mc.shutdown()