Add found bugs to database.
[cfb.git] / prod / net / jaekl / cfb / store / DbStore.java
index 7a1e6102febaf6916ad34c705033d01757451893..2238568c001bec9d9207419eb36806055a8d0abd 100644 (file)
 package net.jaekl.cfb.store;
 
 import java.sql.Connection;
+import java.sql.SQLException;
+import java.util.ArrayList;
+import java.util.List;
 
 import net.jaekl.cfb.analyze.Analysis;
+import net.jaekl.cfb.db.CfbSchema;
+import net.jaekl.cfb.db.Column;
+import net.jaekl.cfb.db.Condition;
+import net.jaekl.cfb.db.Operation;
+import net.jaekl.cfb.db.Row;
+import net.jaekl.cfb.db.Table;
+import net.jaekl.cfb.db.TypeMismatchException;
+import net.jaekl.cfb.db.driver.DbDriver;
+import net.jaekl.cfb.xml.BugClass;
+import net.jaekl.cfb.xml.BugInstance;
+import net.jaekl.cfb.xml.BugMethod;
+import net.jaekl.cfb.xml.SourceLine;
+import net.jaekl.cfb.xml.messages.MessageCollection;
 
 public class DbStore {
-       Connection m_conn;
+       Connection m_con;
+       DbDriver m_driver;
+       MessageCollection m_msgColl;
        
-       public DbStore(Connection conn) {
-               m_conn = conn;
+       public DbStore(Connection con, DbDriver driver, MessageCollection msgColl) {
+               m_con = con;
+               m_driver = driver;
+               m_msgColl = msgColl;
        }
        
-       public boolean put(Analysis analysis) {
+       public boolean put(Analysis analysis) throws SQLException, TypeMismatchException {
                if (null == analysis) {
                        return false;
                }
                
+               // ----------------------------------
+               // Add a run record for this analysis
                
+               long runId = m_driver.nextVal(m_con, CfbSchema.RUN_SEQ);
+               Object[][] values = { 
+                                                               {
+                                                                       Long.valueOf(runId),
+                                                                       analysis.getBuildNumber(),
+                                                                       analysis.getStart(),
+                                                                       analysis.getEnd() 
+                                                               } 
+                                                       };
+               int count = m_driver.insert(m_con, CfbSchema.RUNS, values);
+               if (1 != count) {
+                       return false;
+               }
+               
+               // -------------------------------------
+               // Add a found record for each bug found
+               
+               List<BugInstance> bugs = analysis.getBugCollection().getBugs();
+               values = new Object[bugs.size()][CfbSchema.FOUND.getNumColumns()];
+               
+               int row = 0;
+               for (BugInstance bug : bugs)
+               {
+                       Long foundId = Long.valueOf(m_driver.nextVal(m_con, CfbSchema.FOUND_SEQ));
+                       Long bugId = Long.valueOf(m_msgColl.getPattern(bug.getType()).getId());
+                       Long categoryId = Long.valueOf(m_msgColl.getCategory(bug.getCategory()).getId());
+                       Location[] locs = computeLocations(bug);
+                       Location firstLoc  = (locs.length > 0) ? locs[0] : null;
+                       Location secondLoc = (locs.length > 1) ? locs[1] : null;
+                       Location thirdLoc  = (locs.length > 2) ? locs[2] : null;
+                       
+                       values[row][0] = foundId;
+                       values[row][1] = bugId;
+                       values[row][2] = categoryId;
+                       values[row][3] = getLocId(firstLoc);
+                       values[row][4] = getLocId(secondLoc);
+                       values[row][5] = getLocId(thirdLoc);
+                       row++;
+               }
+               
+               count = m_driver.insert(m_con, CfbSchema.FOUND, values);
+               return (bugs.size() == count);
+       }
+       
+       Location[] computeLocations(BugInstance bug)
+       {
+               ArrayList<Location> locs = new ArrayList<Location>();
+               
+               for (SourceLine line : bug.getLines()) {
+                       locs.add(new Location(line));
+               }
+               for (BugMethod method : bug.getMethods()) {
+                       locs.add(new Location(method));
+               }
+               for (BugClass clazz : bug.getClasses()) {
+                       locs.add(new Location(clazz));
+               }
+               
+               return locs.toArray(new Location[locs.size()]);
+       }
+       
+       Long getLocId(Location loc) throws SQLException, TypeMismatchException 
+       {
+               if (null == loc) {
+                       return null;
+               }
+               Long locId = findLocId(loc);
+               if (null != locId) {
+                       return locId;
+               }
+
+               return storeLoc(loc);
+       }
+       
+       Long findLocId(Location loc) throws SQLException, TypeMismatchException
+       {
+               Column[] columns = { CfbSchema.LOCATIONS.getColumn(CfbSchema.LOCID) };
+               Table[] tables = { CfbSchema.LOCATIONS };
+               Condition[] conditions = { 
+                                               new Condition( CfbSchema.LOCATIONS.getColumn(CfbSchema.CLASSNAME),
+                                                              loc.getClassName(),
+                                                              Operation.EQUAL 
+                                                            ),
+                                               new Condition( CfbSchema.LOCATIONS.getColumn(CfbSchema.METHODNAME),
+                                                                      loc.getMethodName(),
+                                                                      Operation.EQUAL
+                                                                    ),
+                                               new Condition( CfbSchema.LOCATIONS.getColumn(CfbSchema.STARTLINE),
+                                                                      loc.getStart(),
+                                                                      Operation.EQUAL
+                                                                    ),
+                                               new Condition( CfbSchema.LOCATIONS.getColumn(CfbSchema.ENDLINE),
+                                                                      loc.getEnd(),
+                                                                      Operation.EQUAL
+                                                                    )
+                                       };
+               List<Row> rows = m_driver.select(m_con, columns, tables, conditions);
+               if (rows.size() > 0) {
+                       assert(1 == rows.size());       // should only have one match
+                       
+                       return rows.get(0).getLong(0);
+               }
+               
+               return null;    // not found
+       }
+       
+       Long storeLoc(Location loc) throws SQLException
+       {
+               long locId = m_driver.nextVal(m_con, CfbSchema.LOC_SEQ);
+               
+               Object[][] values = { { 
+                                                       Long.valueOf(locId),
+                                                       loc.getClassName(),
+                                                       loc.getMethodName(),
+                                                       Long.valueOf(loc.getStart()),
+                                                       Long.valueOf(loc.getEnd())
+                                               } };
+               m_driver.insert(m_con, CfbSchema.LOCATIONS, values);
                
-               return true;
+               return Long.valueOf(locId);
        }
 }