package org.infinispan.ec2demo.web;
import org.infinispan.Cache;
import org.infinispan.ec2demo.CacheBuilder;
import org.infinispan.ec2demo.Influenza_N_P_CR_Element;
import org.infinispan.ec2demo.Nucleotide_Protein_Element;
import org.infinispan.remoting.transport.Address;
import org.infinispan.util.logging.Log;
import org.infinispan.util.logging.LogFactory;
import javax.servlet.RequestDispatcher;
import javax.servlet.ServletConfig;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/**
* Servlet implementation class CacheSearchServlet
*/
public class CacheSearchServlet extends HttpServlet {
private Log myLogger = LogFactory.getLog(CacheSearchServlet.class);
private static final long serialVersionUID = 1L;
private Cache<String, Influenza_N_P_CR_Element> influenzaCache;
private Cache<String, Nucleotide_Protein_Element> proteinCache;
private Cache<String, Nucleotide_Protein_Element> nucleiodCache;
@Override
public void init(ServletConfig config) throws ServletException {
super.init(config);
CacheBuilder cacheBuilder = (CacheBuilder) getServletContext().getAttribute("cacheBuilder");
influenzaCache = cacheBuilder.getCacheManager().getCache("InfluenzaCache");
proteinCache = cacheBuilder.getCacheManager().getCache("ProteinCache");
nucleiodCache = cacheBuilder.getCacheManager().getCache("NucleotideCache");
}
/**
* @see HttpServlet#doGet(HttpServletRequest request, HttpServletResponse
* response)
*/
@Override
protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
RequestDispatcher dispatcher = getServletContext().getRequestDispatcher("/jsp/displayVirusDetails.jsp");
dispatcher.forward(request, response);
}
@Override
public void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
CacheBuilder cacheBuilder = (CacheBuilder) getServletContext().getAttribute("cacheBuilder");
String searchGBAN = request.getParameter("vGBAN");
request.setAttribute("total", searchGBAN);
//check the influenza cache first
Influenza_N_P_CR_Element myRec = influenzaCache.get(searchGBAN);
if (myRec != null) {
myLogger.trace("Searching nucleiodCache for " + myRec.getGanNucleoid());
Nucleotide_Protein_Element nucldet = nucleiodCache.get(myRec.getGanNucleoid());
request.setAttribute("Nucleotide", nucldet);
// Display the protein details
Map<String, String> myProt = myRec.getProtein_Data();
Map<String, String> myMap = new HashMap<String, String>();
for (String x : myProt.keySet()) {
myLogger.trace("Searching proteinCache for " + x);
Nucleotide_Protein_Element myProtdet = proteinCache.get(x);
String protein_CR = myProt.get(x);
myMap.put(myProtdet.getGenbankAccessionNumber(), protein_CR);
}
request.setAttribute("PMap", myMap);
}
//Retrieve the cache cluster memebers
List<Address> myList = cacheBuilder.getCacheManager().getMembers();
List<String> k = new ArrayList<String>();
for (Address ad : myList)
k.add(ad.toString());
request.setAttribute("CMap", k);
RequestDispatcher dispatcher = getServletContext().getRequestDispatcher("/jsp/VirusDetails.jsp");
dispatcher.forward(request, response);
}
}