prod/net/jaekl/cfb/db/CfbSchema.java

   1 package net.jaekl.cfb.db;
   2
   3 // Copyright (C) 2015 Christian Jaekl
   4
   5 import static net.jaekl.cfb.db.Column.Null.*;
   6 import static net.jaekl.cfb.db.Column.Type.*;
   7
   8 import java.sql.Connection;
   9 import java.sql.SQLException;
  10 import java.util.Collection;
  11
  12 import net.jaekl.cfb.analyze.MessageMap;
  13 import net.jaekl.cfb.db.driver.DbDriver;
  14 import net.jaekl.cfb.xml.messages.BugCategory;
  15 import net.jaekl.cfb.xml.messages.BugPattern;
  16
  17
  18 public class CfbSchema extends Schema {
  19         MessageMap m_msgMap;
  20
  21         public static final Sequence BUG_SEQ      = new Sequence("BUG_SEQ");
  22         public static final Sequence CATEGORY_SEQ = new Sequence("CATEGORY_SEQ");
  23         public static final Sequence FOUND_SEQ    = new Sequence("FOUND_SEQ");
  24         public static final Sequence LOC_SEQ      = new Sequence("LOC_SEQ");
  25         public static final Sequence RUN_SEQ      = new Sequence("RUN_SEQ");
  26
  27         public static final Column BUGID       = new Column("BUGID",       INTEGER,      -1, NOT_NULL);
  28         public static final Column CATEGORY    = new Column("CATEGORY",    VARCHAR,      80, NOT_NULL);
  29         public static final Column CATEGORYID  = new Column("CATEGORYID",  INTEGER,      -1, NOT_NULL);
  30         public static final Column CLASSNAME   = new Column("CLASSNAME",   VARCHAR,     256, NOT_NULL);
  31         public static final Column ENDLINE     = new Column("ENDLINE",     INTEGER,      -1, NULL);
  32         public static final Column ENDTIME     = new Column("ENDTIME",     TIMESTAMPTZ,  -1, NOT_NULL);
  33         public static final Column FIRSTLOCID  = new Column("FIRSTLOCID",  INTEGER,      -1, NOT_NULL);
  34         public static final Column FOUNDID     = new Column("FOUNDID",     INTEGER,      -1, NOT_NULL);
  35         public static final Column LOCID       = new Column("LOCID",       INTEGER,      -1, NOT_NULL);
  36         public static final Column METHODNAME  = new Column("METHODNAME",  VARCHAR,     256, NULL);
  37         public static final Column RUNID       = new Column("RUNID",       INTEGER,      -1, NOT_NULL);
  38         public static final Column SECONDLOCID = new Column("SECONDLOCID", INTEGER,      -1, NULL);
  39         public static final Column STARTLINE   = new Column("STARTLINE",   INTEGER,      -1, NULL);
  40         public static final Column STARTTIME   = new Column("STARTTIME",   TIMESTAMPTZ,  -1, NOT_NULL);
  41         public static final Column THIRDLOCID  = new Column("THIRDLOCID",  INTEGER,      -1, NULL);
  42         public static final Column TYPE        = new Column("TYPE",        VARCHAR,      80, NOT_NULL );
  43         public static final Column VERSION     = new Column("VERSION",     VARCHAR,      32, NULL);
  44
  45         private static final Column[] BUGS_COLS       = { BUGID, TYPE };
  46         private static final Column[] CATEGORIES_COLS = { CATEGORYID, CATEGORY };
  47         private static final Column[] FOUND_COLS      = { FOUNDID, BUGID, CATEGORYID, FIRSTLOCID, SECONDLOCID, THIRDLOCID };
  48         private static final Column[] LOCATIONS_COLS  = { LOCID, CLASSNAME, METHODNAME, STARTLINE, ENDLINE };
  49         private static final Column[] RUNS_COLS       = { RUNID, VERSION, STARTTIME, ENDTIME };
  50
  51         public static final Table BUGS       = new Table("BUGS",       BUGS_COLS);
  52         public static final Table CATEGORIES = new Table("CATEGORIES", CATEGORIES_COLS);
  53         public static final Table FOUND      = new Table("FOUND",      FOUND_COLS);
  54         public static final Table LOCATIONS  = new Table("LOCATIONS",  LOCATIONS_COLS);
  55         public static final Table RUNS       = new Table("RUNS",       RUNS_COLS);
  56
  57         private static final Sequence[] SEQUENCES = {
  58                                         BUG_SEQ,
  59                                         CATEGORY_SEQ,
  60                                         FOUND_SEQ,
  61                                         LOC_SEQ,
  62                                         RUN_SEQ
  63                                 };
  64
  65         private static final Table[] TABLES = {
  66                                         BUGS,
  67                                         CATEGORIES,
  68                                         FOUND,
  69                                         LOCATIONS,
  70                                         RUNS
  71                                 };
  72
  73         public CfbSchema(DbDriver driver) {
  74                 super("CFB", driver);
  75
  76                 m_msgMap = null;
  77
  78                 addTables(TABLES);
  79                 addSequences(SEQUENCES);
  80         }
  81
  82         public void setMessageMap(MessageMap msgMap) {
  83                 m_msgMap = msgMap;
  84         }
  85
  86         @Override
  87         boolean postCreationInit(Connection con) throws SQLException {
  88                 assert(null != m_msgMap);
  89
  90                 if (! insertCategories(con)) {
  91                         return false;
  92                 }
  93                 if (! insertPatterns(con)) {
  94                         return false;
  95                 }
  96
  97                 return true;
  98         }
  99
 100         boolean insertCategories(Connection con) throws SQLException {
 101                 Collection<BugCategory> categories = m_msgMap.getColl().getCategories();
 102
 103                 Object[][] values = new Object[categories.size()][CATEGORIES.getNumColumns()];
 104
 105                 int row = 0;
 106                 for (BugCategory cat : categories) {
 107                         long categoryId = m_driver.nextVal(con, CATEGORY_SEQ);
 108                         cat.setId(categoryId);
 109
 110                         values[row][0] = Long.valueOf(categoryId);
 111                         values[row][1] = cat.getCategory();
 112                         row++;
 113                 }
 114
 115                 int count = m_driver.insert(con, CATEGORIES, values);
 116
 117                 return (categories.size() == count);
 118         }
 119
 120         boolean insertPatterns(Connection con) throws SQLException {
 121                 Collection<BugPattern> patterns = m_msgMap.getColl().getPatterns();
 122
 123                 Object[][] values = new Object[patterns.size()][BUGS.getNumColumns()];
 124
 125                 int row = 0;
 126                 for (BugPattern bug : patterns) {
 127                         long bugId = m_driver.nextVal(con, BUG_SEQ);
 128                         bug.setId(bugId);
 129
 130                         values[row][0] = Long.valueOf(bugId);
 131                         values[row][1] = bug.getType();
 132                         row++;
 133                 }
 134
 135                 int count = m_driver.insert(con, BUGS, values);
 136
 137                 return (patterns.size() == count);
 138         }
 139 }