forked from Kitware/VTK
-
Notifications
You must be signed in to change notification settings - Fork 0
/
DatabaseSchemaWith2Tables.cxx
106 lines (88 loc) · 4.2 KB
/
DatabaseSchemaWith2Tables.cxx
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
/*=========================================================================
Program: Visualization Toolkit
Module: DatabaseSchemaWith2Tables.cxx
Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
All rights reserved.
See Copyright.txt or http://www.kitware.com/Copyright.htm for details.
This software is distributed WITHOUT ANY WARRANTY; without even
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE. See the above copyright notice for more information.
=========================================================================*/
/*----------------------------------------------------------------------------
Copyright (c) Sandia Corporation
See Copyright.txt or http://www.paraview.org/HTML/Copyright.html for details.
----------------------------------------------------------------------------*/
// .SECTION Thanks
// Thanks to Philippe Pebay from Sandia National Laboratories for implementing
// this example of a database schema.
#include "DatabaseSchemaWith2Tables.h"
#include <stdexcept>
#include "vtkSQLDatabaseSchema.h"
DatabaseSchemaWith2Tables::DatabaseSchemaWith2Tables()
{
this->Create();
}
DatabaseSchemaWith2Tables::~DatabaseSchemaWith2Tables()
{
if(this->Schema)
{
this->Schema->Delete();
}
}
void DatabaseSchemaWith2Tables::Create()
{
cerr << "@@ Creating a schema...";
this->Schema = vtkSQLDatabaseSchema::New();
this->Schema->SetName( "TestSchema" );
// Create PostgreSQL-specific preambles to load the PL/PGSQL language and create a function
// with this language. These will be ignored by other backends.
this->Schema->AddPreamble( "dropplpgsql", "DROP EXTENSION IF EXISTS PLPGSQL", VTK_SQL_POSTGRESQL );
this->Schema->AddPreamble( "loadplpgsql", "CREATE LANGUAGE PLPGSQL", VTK_SQL_POSTGRESQL );
this->Schema->AddPreamble( "createsomefunction",
"CREATE OR REPLACE FUNCTION somefunction() RETURNS TRIGGER AS $btable$ "
"BEGIN "
"INSERT INTO btable (somevalue) VALUES (NEW.somenmbr); "
"RETURN NEW; "
"END; $btable$ LANGUAGE PLPGSQL",
VTK_SQL_POSTGRESQL );
// Insert in alphabetical order so that SHOW TABLES does not mix handles
this->TableAHandle = this->Schema->AddTableMultipleArguments( "atable",
vtkSQLDatabaseSchema::COLUMN_TOKEN, vtkSQLDatabaseSchema::SERIAL, "tablekey", 0, "",
vtkSQLDatabaseSchema::COLUMN_TOKEN, vtkSQLDatabaseSchema::VARCHAR, "somename", 64, "NOT NULL",
vtkSQLDatabaseSchema::COLUMN_TOKEN, vtkSQLDatabaseSchema::BIGINT, "somenmbr", 17, "DEFAULT 0",
vtkSQLDatabaseSchema::INDEX_TOKEN, vtkSQLDatabaseSchema::PRIMARY_KEY, "bigkey",
vtkSQLDatabaseSchema::INDEX_COLUMN_TOKEN, "tablekey",
vtkSQLDatabaseSchema::END_INDEX_TOKEN,
vtkSQLDatabaseSchema::INDEX_TOKEN, vtkSQLDatabaseSchema::UNIQUE, "reverselookup",
vtkSQLDatabaseSchema::INDEX_COLUMN_TOKEN, "somename",
vtkSQLDatabaseSchema::INDEX_COLUMN_TOKEN, "somenmbr",
vtkSQLDatabaseSchema::END_INDEX_TOKEN,
vtkSQLDatabaseSchema::TRIGGER_TOKEN, vtkSQLDatabaseSchema::AFTER_INSERT,
"inserttrigger", "DO NOTHING",
VTK_SQL_SQLITE,
vtkSQLDatabaseSchema::TRIGGER_TOKEN, vtkSQLDatabaseSchema::AFTER_INSERT,
"inserttrigger", "FOR EACH ROW EXECUTE PROCEDURE somefunction ()",
VTK_SQL_POSTGRESQL,
vtkSQLDatabaseSchema::TRIGGER_TOKEN, vtkSQLDatabaseSchema::AFTER_INSERT,
"inserttrigger", "FOR EACH ROW INSERT INTO btable SET somevalue = NEW.somenmbr",
VTK_SQL_MYSQL,
vtkSQLDatabaseSchema::END_TABLE_TOKEN
);
if(this->TableAHandle < 0 )
{
throw std::runtime_error("Could not create test schema: Failed to create atable");
}
this->TableBHandle = this->Schema->AddTableMultipleArguments( "btable",
vtkSQLDatabaseSchema::COLUMN_TOKEN, vtkSQLDatabaseSchema::SERIAL, "tablekey", 0, "",
vtkSQLDatabaseSchema::COLUMN_TOKEN, vtkSQLDatabaseSchema::BIGINT, "somevalue", 12, "DEFAULT 0",
vtkSQLDatabaseSchema::INDEX_TOKEN, vtkSQLDatabaseSchema::PRIMARY_KEY, "",
vtkSQLDatabaseSchema::INDEX_COLUMN_TOKEN, "tablekey",
vtkSQLDatabaseSchema::END_INDEX_TOKEN,
vtkSQLDatabaseSchema::END_TABLE_TOKEN
);
if ( this->TableBHandle < 0 )
{
throw std::runtime_error("Could not create test schema: Failed to create btable");
}
cerr << " done." << endl;
}