mirror of https://gitee.com/bigwinds/arangodb
293 lines
8.7 KiB
JavaScript
293 lines
8.7 KiB
JavaScript
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief arango upgrade actions
|
|
///
|
|
/// @file
|
|
///
|
|
/// DISCLAIMER
|
|
///
|
|
/// Copyright 2004-2012 triAGENS GmbH, Cologne, Germany
|
|
///
|
|
/// Licensed under the Apache License, Version 2.0 (the "License");
|
|
/// you may not use this file except in compliance with the License.
|
|
/// You may obtain a copy of the License at
|
|
///
|
|
/// http://www.apache.org/licenses/LICENSE-2.0
|
|
///
|
|
/// Unless required by applicable law or agreed to in writing, software
|
|
/// distributed under the License is distributed on an "AS IS" BASIS,
|
|
/// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
/// See the License for the specific language governing permissions and
|
|
/// limitations under the License.
|
|
///
|
|
/// Copyright holder is triAGENS GmbH, Cologne, Germany
|
|
///
|
|
/// @author Jan Steemann
|
|
/// @author Copyright 2012, triAGENS GmbH, Cologne, Germany
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
// -----------------------------------------------------------------------------
|
|
// --SECTION-- arango-upgrade
|
|
// -----------------------------------------------------------------------------
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @addtogroup ArangoDB
|
|
/// @{
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief updates the database
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
function main (argv) {
|
|
var internal = require("internal");
|
|
var console = require("console");
|
|
var db = internal.db;
|
|
|
|
// path to the VERSION file
|
|
var versionFile = DATABASEPATH + "/VERSION";
|
|
|
|
var allTasks = [ ];
|
|
var activeTasks = [ ];
|
|
|
|
// assume we do not yet have a VERSION value
|
|
var currentVersion = 0;
|
|
if (FS_EXISTS(versionFile)) {
|
|
// VERSION file exists, read its contents
|
|
currentVersion = parseInt(SYS_READ(versionFile));
|
|
}
|
|
|
|
// helper function to define tasks
|
|
function addTask (description, maxVersion, code) {
|
|
// "description" is a textual description of the task that will be printed out on screen
|
|
// "maxVersion" is the maximum version number the task will be applied for
|
|
var task = { description: description, maxVersion: maxVersion, code: code };
|
|
|
|
allTasks.push(task);
|
|
|
|
// the maximum version number defined for the task is >= than the current VERSION number
|
|
// that means we will apply to task
|
|
if (currentVersion < parseInt(maxVersion)) {
|
|
activeTasks.push(task);
|
|
}
|
|
}
|
|
|
|
function getCollection (name) {
|
|
return db._collection(name);
|
|
}
|
|
|
|
function collectionExists (name) {
|
|
var collection = getCollection(name);
|
|
return (collection != undefined) && (collection != null) && (collection.name() == name);
|
|
}
|
|
|
|
function createSystemCollection (name, attributes) {
|
|
if (collectionExists(name)) {
|
|
return true;
|
|
}
|
|
|
|
var realAttributes = attributes || { };
|
|
realAttributes['isSystem'] = true;
|
|
|
|
if (db._create(name, realAttributes)) {
|
|
return true;
|
|
}
|
|
|
|
return collectionExists(name);
|
|
}
|
|
|
|
|
|
// --------------------------------------------------------------------------
|
|
// the actual upgrade tasks. all tasks defined here should be "re-entrant"
|
|
// --------------------------------------------------------------------------
|
|
|
|
// set up the collection _users
|
|
addTask("setup _users collection", 1, function () {
|
|
return createSystemCollection("_users", { waitForSync : true });
|
|
});
|
|
|
|
// create a unique index on username attribute in _users
|
|
addTask("create index on username attribute in _users collection", 1, function () {
|
|
var users = getCollection("_users");
|
|
if (! users) {
|
|
return false;
|
|
}
|
|
|
|
users.ensureUniqueConstraint("username");
|
|
|
|
return true;
|
|
});
|
|
|
|
// add a default root user with no passwd
|
|
addTask("add default root user", 1, function () {
|
|
var users = getCollection("_users");
|
|
if (! users) {
|
|
return false;
|
|
}
|
|
|
|
if (users.count() == 0) {
|
|
// only add account if user has not created his/her own accounts already
|
|
users.save({ user: "root", password: internal.encodePassword(""), active: true });
|
|
}
|
|
|
|
return true;
|
|
});
|
|
|
|
// set up the collection _graphs
|
|
addTask("setup _graphs collection", 1, function () {
|
|
return createSystemCollection("_graphs", { waitForSync : true });
|
|
});
|
|
|
|
// create a unique index on name attribute in _graphs
|
|
addTask("create index on name attribute in _graphs collection", 1, function () {
|
|
var graphs = getCollection("_graphs");
|
|
|
|
if (! graphs) {
|
|
return false;
|
|
}
|
|
|
|
graphs.ensureUniqueConstraint("name");
|
|
|
|
return true;
|
|
});
|
|
|
|
// make distinction between document and edge collections
|
|
addTask("set new collection type for edge collections and update collection version", 1, function () {
|
|
var collections = db._collections();
|
|
|
|
for (var i in collections) {
|
|
var collection = collections[i];
|
|
|
|
try {
|
|
if (collection.version() > 1) {
|
|
// already upgraded
|
|
continue;
|
|
}
|
|
|
|
if (collection.type() == 3) {
|
|
// already an edge collection
|
|
collection.setAttribute("version", 2);
|
|
continue;
|
|
}
|
|
|
|
if (collection.count() > 0) {
|
|
var isEdge = true;
|
|
// check the 1st 50 documents from a collection
|
|
var documents = collection.ALL(0, 50);
|
|
|
|
for (var j in documents) {
|
|
var doc = documents[j];
|
|
|
|
// check if documents contain both _from and _to attributes
|
|
if (! doc.hasOwnProperty("_from") || ! doc.hasOwnProperty("_to")) {
|
|
isEdge = false;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (isEdge) {
|
|
collection.setAttribute("type", 3);
|
|
console.log("made collection '" + collection.name() + " an edge collection");
|
|
}
|
|
}
|
|
collection.setAttribute("version", 2);
|
|
}
|
|
catch (e) {
|
|
console.error("could not upgrade collection '" + collection.name() + "'");
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
});
|
|
|
|
// create the _modules collection
|
|
addTask("setup _modules collection", 1, function () {
|
|
return createSystemCollection("_modules");
|
|
});
|
|
|
|
// create the _routing collection
|
|
addTask("setup _routing collection", 1, function () {
|
|
return createSystemCollection("_routing");
|
|
});
|
|
|
|
// create/update the VERSION file
|
|
addTask("create VERSION file", 2, function () {
|
|
// save "2" into VERSION file
|
|
SYS_SAVE(versionFile, "2");
|
|
return true;
|
|
});
|
|
|
|
// update markers in all collection datafiles to key markers
|
|
addTask("update markers in all collection datafiles", 2, function () {
|
|
var collections = db._collections();
|
|
|
|
for (var i in collections) {
|
|
var collection = collections[i];
|
|
|
|
try {
|
|
if (collection.version() >= 3) {
|
|
// already upgraded
|
|
continue;
|
|
}
|
|
|
|
if (collection.upgrade()) {
|
|
// success
|
|
collection.setAttribute("version", 3);
|
|
}
|
|
else {
|
|
// fail
|
|
console.error("could not upgrade collection datafiles for '" + collection.name() + "'");
|
|
return false;
|
|
}
|
|
}
|
|
catch (e) {
|
|
console.error("could not upgrade collection datafiles for '" + collection.name() + "'");
|
|
return false;
|
|
}
|
|
}
|
|
return true;
|
|
});
|
|
|
|
|
|
console.log("Upgrade script " + argv[0] + " started");
|
|
console.log("Server VERSION is: " + currentVersion);
|
|
|
|
// loop through all tasks and execute them
|
|
console.log("Found " + allTasks.length + " defined task(s), " + activeTasks.length + " task(s) to run");
|
|
|
|
var taskNumber = 0;
|
|
for (var i in activeTasks) {
|
|
var task = activeTasks[i];
|
|
|
|
console.log("Executing task #" + (++taskNumber) + ": " + task.description);
|
|
|
|
var result = task.code();
|
|
|
|
if (! result) {
|
|
console.error("Task failed. Aborting upgrade script.");
|
|
console.error("Please fix the problem and try running the upgrade script again.");
|
|
return 1;
|
|
}
|
|
else {
|
|
console.log("Task successful");
|
|
}
|
|
}
|
|
|
|
console.log("Upgrade script successfully finished");
|
|
|
|
return 0;
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @}
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
// -----------------------------------------------------------------------------
|
|
// --SECTION-- END-OF-FILE
|
|
// -----------------------------------------------------------------------------
|
|
|
|
// Local Variables:
|
|
// mode: outline-minor
|
|
// outline-regexp: "^\\(/// @brief\\|/// @addtogroup\\|// --SECTION--\\|/// @page\\|/// @}\\)"
|
|
// End:
|