PHON-5: Use Mongo Orchestration to manage MongoDB servers

This commit is contained in:
Hannes Magnusson
2014-12-15 17:05:25 -08:00
parent bbbda03d10
commit 2d88ca250f
3 changed files with 170 additions and 3 deletions
+3
View File
@@ -44,6 +44,9 @@ composer:
exit 1; \
fi
test-bootstrap:
php scripts/start-servers.php
testunit: composer
@command -v phpunit >/dev/null 2>&1; \
if test $$? -eq 0; then \
+12 -3
View File
@@ -1,14 +1,23 @@
<?php
require __DIR__ . "/" . "orchestration.php";
require __DIR__ . "/" . "tools.php";
$host = "http://localhost:8889";
if ($_ENV && isset($_ENV["ORCHESTRATION"])) {
$host = $_ENV["ORCHESTRATION"];
}
$orch = new Mongo\Orchestration($host);
if (!$orch->ping()) {
exit("skip Mongo Orchestration not running on {$host} - run 'make test-bootstrap'");
}
$consts = array(
"MONGODB_URI" => "mongodb://localhost:27017",
"MONGODB_CLEANUP_URI" => "mongodb://localhost:27017",
"DATABASE_NAME" => "phongo_test",
"MONGODB_URI" => $orch->hasStandalone() ?: $orch->startStandalone(),
"MONGODB_CLEANUP_URI" => $orch->hasStandalone() ?: $orch->startStandalone(),
"DATABASE_NAME" => "phongo",
"COLLECTION_NAME" => makeCollectionNameFromFilename($_SERVER["SCRIPT_FILENAME"]),
"DEBUG_DIR" => sys_get_temp_dir() . "/PHONGO-TESTS/",
);
+155
View File
@@ -0,0 +1,155 @@
<?php
namespace Mongo;
class Orchestration {
protected $baseuri;
protected $conf;
function __construct($baseuri) {
$this->baseuri = $baseuri;
$this->conf = array(
"timeout" => 35,
);
}
function ping() {
try {
$data = $this->get("");
} catch(\Exception $e) {
return false;
}
}
function stopAll() {
$servers = $this->get("servers");
foreach($servers["servers"] as $server) {
$this->stopServer($server["id"]);
}
}
function hasStandalone() {
try {
$data = $this->get("servers/STANDALONE");
} catch(\Exception $e) {
return false;
}
return $data["mongodb_uri"];
}
function startStandalone() {
$retval = $this->post("servers", ["name" => "mongod", "id" => "STANDALONE"]);
if ($retval["procInfo"]["alive"]) {
return $retval["mongodb_uri"];
}
return false;
}
function stopStandalone() {
return $this->stopServer("STANDALONE");
}
function getServerInfo($id) {
$data = $this->get("servers/" . $id);
return $data;
}
function stopServer($id) {
try {
$retval = $this->delete("servers/$id");
return true;
} catch(\Exception $e) {
if ($e->getCode() == 204) {
return true;
}
return false;
}
}
function getTimeout() {
return $this->conf["timeout"];
}
function delete($target) {
$opts = [
"http" => [
"timeout" => $this->getTimeout(),
"method" => "DELETE",
"header" => "Accept: application/json\r\n" .
"Content-type: application/x-www-form-urlencoded",
"ignore_errors" => true,
],
];
$data = $this->_sendAndReceive($target, $opts);
return $data;
}
function get($target) {
$opts = [
"http" => [
"timeout" => $this->getTimeout(),
"method" => "GET",
"header" => "Accept: application/json\r\n" .
"Content-type: application/x-www-form-urlencoded",
"ignore_errors" => true,
],
];
$data = $this->_sendAndReceive($target, $opts);
return $data;
}
function post($target, $array) {
$postdata = json_encode($array);
$opts = [
"http" => [
"timeout" => $this->getTimeout(),
"method" => "POST",
"header" => "Accept: application/json\r\n" .
"Content-type: application/x-www-form-urlencoded",
"content" => $postdata,
"ignore_errors" => true,
],
];
return $this->_sendAndReceive($target, $opts);
}
protected function _sendAndReceive($target, $opts) {
$context = stream_context_create($opts);
$url = $this->baseuri . "/" . $target;
$http = $opts["http"];
$hdr = "";
foreach(explode("\r\n", $http["header"]) as $header) {
$hdr .= "--header='{$header}' ";
}
$debug = sprintf("wget --body-data='%s' --method='%s' %s %s\n",
isset($http["content"]) ? $http["content"] : "",
$http["method"],
$hdr,
$url
);
if (defined("DUMP_WGET") && DUMP_WGET) {
echo $debug;
}
$data = @file_get_contents($url, false, $context);
if ($data) {
return json_decode($data, true);
}
if (!empty($http_response_header)) {
sscanf($http_response_header[0], "HTTP/%f %i %s", $proto, $code, $desc);
throw new \RuntimeException(json_encode($http_response_header), $code);
}
throw new \RuntimeException(error_get_last()["message"]);
}
}