new readers for the queries
[proteocache.git] / server / compbio / controllers / SequenceController.java
1 package compbio.controllers;
2
3 import java.util.List;
4 import java.util.Map;
5 import java.util.regex.Pattern;
6
7 import org.springframework.stereotype.Controller;
8 import org.springframework.web.bind.annotation.RequestMapping;
9 import org.springframework.web.bind.annotation.RequestMethod;
10 import org.springframework.web.bind.annotation.RequestParam;
11
12 import compbio.beans.ProteinBean;
13 import compbio.beans.TotalByCounterBean;
14 import compbio.cassandra.readers.ReaderByCounter;
15 import compbio.cassandra.readers.SequenceReader;
16 import compbio.statistic.CassandraRequester;
17
18 /**
19  * Spring controller for sequence queries. This version works in the servlet
20  * style.
21  * 
22  * @author Alexander Sherstnev
23  * @author Natasha Sherstneva
24  * 
25  * @since 0.5
26  * @version 1.0 December 2013
27  */
28 @Controller
29 @RequestMapping("/sequence")
30 public class SequenceController extends BasicController {
31
32         /**
33          * pattern for NON-protein alphabet symbols
34          */
35         private final Pattern NONPROTEIN = Pattern.compile("[^ARNDCQEGHILKMFPSTWYV]+", Pattern.CASE_INSENSITIVE);
36
37         /**
38          * form a query page for search protein sequence. The servlet should be
39          * available for users and admins only. 
40          * 
41          * @param model
42          *            MVC model
43          * @return link to the JSP query page
44          */
45         @RequestMapping(value = "query", method = RequestMethod.GET)
46         public String formSequenceQuery(Map<String, Object> model) {
47                 model.put("username", getPrincipalName());
48                 model.put("value", "AAAAA");
49                 return "query/Sequence";
50         }
51
52         /**
53          * form a query page for statistics: Protein by job count. The servlet should be
54          * available for users and admins only. 
55          * 
56          * @param model
57          *            MVC model
58          * @return link to the JSP query page
59          */
60         @RequestMapping(value = "counts/query", method = RequestMethod.GET)
61         public String formCounterQuery(Map<String, Object> model) {
62                 model.put("username", getPrincipalName());
63                 model.put("value", 5);
64                 return "query/SequenceCounts";
65         }
66
67         /**
68          * form a report page for search protein sequence.
69          * 
70          * @param model
71          *            MVC model object
72          * @param sequence
73          *            protein sequence or part of sequence
74          * @param searchtype
75          *            defined whether the whole sequence or part of sequence would be searched
76          * @return link to the report JSP page
77          */
78         @RequestMapping(value = "sequence/results", method = RequestMethod.GET)
79         public String findSequence(@RequestParam("sequence") String sequence, @RequestParam("searchtype") String searchtype,
80                         Map<String, Object> model) {
81                 model.put("username", getPrincipalName());
82                 final long startTime = System.currentTimeMillis();
83
84                 // input checks
85                 String trimmedsequence = sequence.replaceAll("\\s", "");
86                 if (trimmedsequence.equalsIgnoreCase("")) {
87                         model.put("error", "The sequence cann't be empty");
88                         model.put("value", sequence);
89                         return "query/Sequence";
90                 }
91                 if (NONPROTEIN.matcher(trimmedsequence).find()) {
92                         model.put("error", "The sequence contains symbols not from the standard protein alphabet");
93                         model.put("value", sequence);
94                         return "query/Sequence";
95                 }
96
97                 model.put("njobs", 0);
98                 model.put("prot", trimmedsequence);
99                 model.put("searchtype", searchtype);
100
101                 if (0 < trimmedsequence.length()) {
102                         SequenceReader reader = new SequenceReader();
103                         List<ProteinBean> result = reader.readProteins(trimmedsequence, searchtype);
104                         model.put("results", result);
105                         if (null != result) {
106                                 if (searchtype.equals("whole"))
107                                         model.put("njobs", result.get(0).getJobid().size());
108                                 else
109                                         model.put("njobs", result.size());
110                         }
111                 }
112                 final long endTime = System.currentTimeMillis();
113                 model.put("timeExecution", (endTime - startTime));
114                 return "reportProteinSequences";
115         }
116
117         /**
118          * form a report page for statistics: Protein by job count.
119          * 
120          * @param model
121          *            MVC model object
122          * @param counter
123          *           
124          * @return link to the report JSP page
125          */
126         @RequestMapping(value = "counts/results", method = RequestMethod.GET)
127         public String countSequences(@RequestParam("counterJob") String counter, Map<String, Object> model) {
128                 model.put("username", getPrincipalName());
129                 final long startTime = System.currentTimeMillis();
130
131                 if (counter.equals("")) {
132                         model.put("error", "The value must not be empty");
133                         model.put("value", counter);
134                         return "query/SequenceCounts";
135                 }
136
137                 int realcounter;
138                 try {
139                         realcounter = Integer.parseInt(counter.trim());
140                 } catch (NumberFormatException e) {
141                         model.put("error", "The value must be an integer number");
142                         model.put("value", counter);
143                         return "query/SequenceCounts";
144                 }
145
146                 if (realcounter < 1) {
147                         model.put("error", "The value must be greater than 0");
148                         model.put("value", counter);
149                         return "query/SequenceCounts";
150                 }
151
152                 ReaderByCounter reader = new ReaderByCounter();
153
154                 List<TotalByCounterBean> r = reader.readProteinByCounter(realcounter);
155                 model.put("results", r);
156                 model.put("njobs", 0);
157                 if (null != r) {
158                         model.put("njobs", r.size());
159                 }
160                 final long endTime = System.currentTimeMillis();
161                 model.put("timeExecution", (endTime - startTime));
162                 model.put("counter", realcounter);
163                 return "reportProteinSequencesCounter";
164         }
165
166 }