1 package compbio.cassandra;
3 import java.text.ParseException;
4 import java.text.SimpleDateFormat;
5 import java.util.Calendar;
9 import org.apache.log4j.Logger;
11 import com.datastax.driver.core.ResultSet;
12 import com.datastax.driver.core.Row;
13 import com.datastax.driver.core.Session;
15 public class CassandraRemover {
16 private Session session;
17 static SimpleDateFormat dateformatter = new SimpleDateFormat("yyyy/MM/dd");
18 private static Logger log = Logger.getLogger(CassandraNativeConnector.class);
20 public CassandraRemover() {
21 Session inis = CassandraNativeConnector.getSession();
25 public void setSession(Session s) {
31 * delete a record from CF for current jobId
33 private void RemoveJob(String jobid, long date) {
34 String com0 = "DELETE FROM ProteinLog WHERE JobID = '" + jobid + "';";
35 System.out.println("Command: " + com0);
36 session.execute(com0);
37 String com1 = "DELETE FROM ProteinRow WHERE JobID = '" + jobid + "';";
38 System.out.println("Command: " + com1);
39 session.execute(com1);
40 String com2 = "DELETE FROM ProteinData WHERE JobID = '" + jobid + "' AND jobtime = " + date + ";";
41 System.out.println("Command: " + com2);
42 session.execute(com2);
43 String com3 = "UPDATE jpredarchive SET finalstatus = 'DELETED' WHERE JobID = '" + jobid + "' ;";
44 System.out.println("Command: " + com3);
45 session.execute(com3);
46 String com = "SELECT total FROM JobDateInfo WHERE jobday = " + date + ";";
47 System.out.println("Command: " + com);
48 ResultSet results = session.execute(com);
49 long njobs = results.one().getLong("total");
50 System.out.println("njobs: " + njobs);
51 String com4 = "INSERT INTO JobDateInfo " + "(jobday, Total)" + " VALUES (" + date + "," + (njobs -1) + ");";
52 System.out.println("Command: " + com4);
53 session.execute(com4);
54 System.out.println("Remove jobs: " + jobid);
57 public void RemoveJobById (String jobid) {
60 Long date = FindDate(jobid);
63 RemoveJob(jobid, date);
66 public void RemoveJobByDate (String date1, String date2) {
67 System.out.println("Start " + date1 + ", " + date2);
68 if (date1 == null || date2 == null)
70 Long dateBegin = convertDate(date1);
71 Long dateEnd = convertDate(date2);
72 System.out.println("Date to long done!: ");
73 if (dateBegin == null || dateEnd == null)
75 Calendar start = Calendar.getInstance();
76 start.setTime(new Date(dateBegin));
77 Calendar end = Calendar.getInstance();
78 end.setTime(new Date(dateEnd));
79 System.out.println("Date to cal done!: ");
80 for (Date date = start.getTime(); !start.after(end); start.add(Calendar.DATE, 1), date = start.getTime()) {
81 String com = "SELECT JobID FROM ProteinData WHERE jobtime = " + date.getTime() + ";";
82 System.out.println("Command: " + com);
83 ResultSet results = session.execute(com);
84 if (!results.isExhausted()) {
85 List<Row> rows = results.all();
87 String jobid = r.getString("JobID");
89 RemoveJob(jobid, date.getTime());
96 public void RemoveJobByIp (String ip) {
99 String com = "SELECT databegin, JobID FROM ProteinLog WHERE ip = '" + ip + "';";
100 System.out.println("Command: " + com);
101 ResultSet results = session.execute(com);
102 if (!results.isExhausted()) {
103 List<Row> rows = results.all();
105 Long date = convertDate(r.getString("databegin"));
106 String jobid = r.getString("JobID");
107 if (date == null || jobid == null)
109 RemoveJob(jobid, date);
114 public void RemoveJobBySequence (String seq) {
117 String com = "SELECT JobID FROM ProteinRow WHERE Protein = '" + seq + "';";
118 System.out.println("Command: " + com);
119 ResultSet results = session.execute(com);
120 if (!results.isExhausted()) {
121 List<Row> rows = results.all();
123 String jobid = r.getString("JobID");
126 Long date = FindDate(jobid);
129 RemoveJob(jobid, date);
134 private Long FindDate(String jobid) {
135 String com = "SELECT databegin FROM ProteinLog WHERE JobID = '" + jobid + "';";
136 System.out.println("Command: " + com);
137 ResultSet results = session.execute(com);
138 if (results.isExhausted())
140 Long date = convertDate(results.one().getString("databegin"));
145 protected long convertDate (String d) {
148 Date startdate = dateformatter.parse(d);
149 return startdate.getTime();
151 } catch (ParseException e) {