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         public static final Sequence VARIABLE_SEQ = new Sequence("VARIABLE_SEQ");
  27
  28         public static final Column BUGID        = new Column("BUGID",       INTEGER,      -1, NOT_NULL);
  29         public static final Column CATEGORY     = new Column("CATEGORY",    VARCHAR,      80, NOT_NULL);
  30         public static final Column CATEGORYID   = new Column("CATEGORYID",  INTEGER,      -1, NOT_NULL);
  31         public static final Column CLASSNAME    = new Column("CLASSNAME",   VARCHAR,     256, NOT_NULL);
  32         public static final Column ENDLINE      = new Column("ENDLINE",     INTEGER,      -1, NULL);
  33         public static final Column ENDTIME      = new Column("ENDTIME",     TIMESTAMPTZ,  -1, NOT_NULL);
  34         public static final Column FIRSTLOCID   = new Column("FIRSTLOCID",  INTEGER,      -1, NOT_NULL);
  35         public static final Column FOUNDID      = new Column("FOUNDID",     INTEGER,      -1, NOT_NULL);
  36         public static final Column LOCID        = new Column("LOCID",       INTEGER,      -1, NOT_NULL);
  37         public static final Column METHODNAME   = new Column("METHODNAME",  VARCHAR,     256, NULL);
  38         public static final Column METHODROLE   = new Column("METHODROLE",  VARCHAR,      80, NULL);
  39         public static final Column NAME         = new Column("NAME",        VARCHAR,     256, NOT_NULL);
  40         public static final Column RUNID        = new Column("RUNID",       INTEGER,      -1, NOT_NULL);
  41         public static final Column SECONDLOCID  = new Column("SECONDLOCID", INTEGER,      -1, NULL);
  42         public static final Column STARTLINE    = new Column("STARTLINE",   INTEGER,      -1, NULL);
  43         public static final Column STARTTIME    = new Column("STARTTIME",   TIMESTAMPTZ,  -1, NOT_NULL);
  44         public static final Column THIRDLOCID   = new Column("THIRDLOCID",  INTEGER,      -1, NULL);
  45         public static final Column TYPE         = new Column("TYPE",        VARCHAR,      80, NOT_NULL);
  46         public static final Column VARID_FK     = new Column("VARID",       INTEGER,      -1, NULL);            // foreign key
  47         public static final Column VARID_PK     = new Column("VARID",       INTEGER,      -1, NOT_NULL);        // primary key
  48         public static final Column VARROLE      = new Column("VARROLE",     VARCHAR,      80, NOT_NULL);
  49         public static final Column VERSION      = new Column("VERSION",     VARCHAR,      32, NULL);
  50
  51         private static final Column[] BUGS_COLS       = { BUGID, TYPE };
  52         private static final Column[] CATEGORIES_COLS = { CATEGORYID, CATEGORY };
  53         private static final Column[] FOUND_COLS      = { FOUNDID, BUGID, CATEGORYID, FIRSTLOCID, SECONDLOCID, THIRDLOCID, VARID_FK };
  54         private static final Column[] LOCATIONS_COLS  = { LOCID, CLASSNAME, METHODNAME, METHODROLE, STARTLINE, ENDLINE };
  55         private static final Column[] RUNS_COLS       = { RUNID, VERSION, STARTTIME, ENDTIME };
  56         private static final Column[] VARIABLES_COLS  = { VARID_PK, NAME, VARROLE };
  57
  58         public static final Table BUGS       = new Table("BUGS",       BUGS_COLS);
  59         public static final Table CATEGORIES = new Table("CATEGORIES", CATEGORIES_COLS);
  60         public static final Table FOUND      = new Table("FOUND",      FOUND_COLS);
  61         public static final Table LOCATIONS  = new Table("LOCATIONS",  LOCATIONS_COLS);
  62         public static final Table RUNS       = new Table("RUNS",       RUNS_COLS);
  63         public static final Table VARIABLES  = new Table("VARIABLES",  VARIABLES_COLS);
  64
  65         private static final Sequence[] SEQUENCES = {
  66                                         BUG_SEQ,
  67                                         CATEGORY_SEQ,
  68                                         FOUND_SEQ,
  69                                         LOC_SEQ,
  70                                         RUN_SEQ,
  71                                         VARIABLE_SEQ
  72                                 };
  73
  74         private static final Table[] TABLES = {
  75                                         BUGS,
  76                                         CATEGORIES,
  77                                         FOUND,
  78                                         LOCATIONS,
  79                                         RUNS,
  80                                         VARIABLES
  81                                 };
  82
  83         public CfbSchema(DbDriver driver) {
  84                 super("CFB", driver);
  85
  86                 m_msgMap = null;
  87
  88                 addTables(TABLES);
  89                 addSequences(SEQUENCES);
  90         }
  91
  92         public void setMessageMap(MessageMap msgMap) {
  93                 m_msgMap = msgMap;
  94         }
  95
  96         @Override
  97         boolean postCreationInit(Connection con) throws SQLException {
  98                 assert(null != m_msgMap);
  99
 100                 if (! insertCategories(con)) {
 101                         return false;
 102                 }
 103                 if (! insertPatterns(con)) {
 104                         return false;
 105                 }
 106
 107                 return true;
 108         }
 109
 110         boolean insertCategories(Connection con) throws SQLException {
 111                 Collection<BugCategory> categories = m_msgMap.getColl().getCategories();
 112
 113                 Object[][] values = new Object[categories.size()][CATEGORIES.getNumColumns()];
 114
 115                 int row = 0;
 116                 for (BugCategory cat : categories) {
 117                         long categoryId = m_driver.nextVal(con, CATEGORY_SEQ);
 118                         cat.setId(categoryId);
 119
 120                         values[row][0] = Long.valueOf(categoryId);
 121                         values[row][1] = cat.getCategory();
 122                         row++;
 123                 }
 124
 125                 int count = m_driver.insert(con, CATEGORIES, values);
 126
 127                 return (categories.size() == count);
 128         }
 129
 130         boolean insertPatterns(Connection con) throws SQLException {
 131                 Collection<BugPattern> patterns = m_msgMap.getColl().getPatterns();
 132
 133                 Object[][] values = new Object[patterns.size()][BUGS.getNumColumns()];
 134
 135                 int row = 0;
 136                 for (BugPattern bug : patterns) {
 137                         long bugId = m_driver.nextVal(con, BUG_SEQ);
 138                         bug.setId(bugId);
 139
 140                         values[row][0] = Long.valueOf(bugId);
 141                         values[row][1] = bug.getType();
 142                         row++;
 143                 }
 144
 145                 int count = m_driver.insert(con, BUGS, values);
 146
 147                 return (patterns.size() == count);
 148         }
 149 }