Merge branch 'master' into servlets
[proteocache.git] / datadb / compbio / cassandra / JpredParserLocalFile.java
1 package compbio.cassandra;
2
3 import java.io.BufferedReader;
4 import java.io.File;
5 import java.io.FileNotFoundException;
6 import java.io.IOException;
7 import java.io.InputStreamReader;
8 import java.net.HttpURLConnection;
9 import java.net.MalformedURLException;
10 import java.io.FileInputStream;
11 import java.text.ParseException;
12 import java.text.SimpleDateFormat;
13 import java.util.ArrayList;
14 import java.util.Calendar;
15 import java.util.Date;
16 import java.util.List;
17
18 public class JpredParserLocalFile implements JpredParser {
19         private CassandraNativeConnector cc = new CassandraNativeConnector();
20         private String dirprefix;
21
22         public void setSource(String newsourceprefix) {
23                 this.dirprefix = newsourceprefix;
24         }
25
26         JpredParserLocalFile() {
27                 this.dirprefix = "/home/asherstnev/Projects/Java.projects/proteocache/data_stress_test/data.dat";
28         }
29
30         JpredParserLocalFile(String sourceurl) {
31                 this.dirprefix = sourceurl;
32         }
33
34         public void Parsing(String source, int nDays) throws IOException {
35                 Calendar cal = Calendar.getInstance();
36                 cal.add(Calendar.DATE, -nDays);
37                 List<String> alljobs = new ArrayList<String>();
38                 File file = new File(source);
39                 BufferedReader alljobsfile = new BufferedReader(new InputStreamReader(new FileInputStream(file)));
40                 String line;
41
42                 while ((line = alljobsfile.readLine()) != null) {
43                         alljobs.add(line);
44                 }
45                 alljobsfile.close();
46
47                 System.out.println("Inserting jobs for " + nDays + " days, " + alljobs.size() + " jobs in total");
48                 final long startTime = System.currentTimeMillis();
49                 for (int i = 0; i < nDays; ++i) {
50                         cal.add(Calendar.DATE, 1);
51                         int month = cal.get(Calendar.MONTH) + 1;
52                         int year = cal.get(Calendar.YEAR);
53                         int day = cal.get(Calendar.DATE);
54                         String date = year + "/" + month + "/" + day;
55                         ParsingForDate(alljobs, date);
56                 }
57                 final long execTime = System.currentTimeMillis() - startTime;
58                 System.out.println("Execution Time = " + execTime + " ms");
59         }
60
61         private void ParsingForDate(List<String> input, String date) {
62                 int totalcount = 0;
63                 int countNoData = 0;
64                 int countUnclearFASTAid = 0;
65                 int countinsertions = 0;
66                 int countinserted = 0;
67                 int counAlignments = 0;
68                 int countStrange = 0;
69
70                 System.out.println("Inserting jobs for " + date);
71                 for (String in : input) {
72                         if (in.matches(date + ":(.*)jp_[^\\s]+")) {
73                                 String[] table = in.split("\\s+");
74                                 String starttime = table[0];
75                                 String finishtime = table[1];
76                                 String ip = table[2];
77                                 String id = table[table.length - 1];
78                                 totalcount++;
79                                 String confilename = dirprefix + "/" + id + "/" + id + ".concise";
80                                 File confile = new File(confilename);
81                                 if (confile.exists()) {
82                                         try {
83                                                 final FastaReader fr = new FastaReader(confilename);
84                                                 final List<FastaSequence> seqs = new ArrayList<FastaSequence>();
85                                                 String newprotein = "";
86                                                 while (fr.hasNext()) {
87                                                         final FastaSequence fs = fr.next();
88                                                         if (fs.getId().equals("QUERY") || fs.getId().equals(id))
89                                                                 newprotein = fs.getSequence().replaceAll("\n", "");
90                                                         else if (fs.getId().equals("jnetpred") || fs.getId().equals("JNETPRED")) {
91                                                                 seqs.add(fs);
92                                                         }
93                                                 }
94                                                 if (newprotein.equals("")) {
95                                                         countUnclearFASTAid++;
96                                                 } else {
97                                                         SimpleDateFormat formatter = new SimpleDateFormat("yyyy/MM/dd");
98                                                         String dateInString1 = starttime.substring(0, starttime.indexOf(":"));
99                                                         long insertdate = 0;
100                                                         try {
101                                                                 Date dat = formatter.parse(dateInString1);
102                                                                 insertdate = dat.getTime();
103                                                         } catch (ParseException e) {
104                                                                 e.printStackTrace();
105                                                         }
106                                                         countinsertions += cc.FormQueryTables(insertdate, starttime, finishtime, ip, id, "OK", "OK", newprotein, seqs);
107                                                 }
108                                                 fr.close();
109                                         } catch (IOException e) {
110                                                 e.printStackTrace();
111                                         }
112                                 } else {
113                                         countNoData++;
114                                 }
115                         } else {
116                                 if (in.matches(date + "(.*)Sequence0/(.*)")) {
117                                         ++counAlignments;
118                                 } else {
119                                         ++countStrange;
120                                 }
121                         }
122                 }
123                 if (true) {
124                         System.out.println("Total number of jobs = " + totalcount);
125                         System.out.println("   " + countinserted + " jobs inserted already");
126                         System.out.println("   " + counAlignments + " jalview jobs");
127                         System.out.println("   " + countStrange + " not analysed jobs");
128                         System.out.println("   " + countNoData + " jobs without *.concise.fasta file");
129                         System.out.println("   " + countUnclearFASTAid + " jobs with unclear FASTA protein id in *.concise.fasta");
130                         System.out.println("   " + countinsertions + " new job insertions\n");
131                 }
132         }
133
134 }