Merge branch 'tgrabiec/distributed' of github.com:cloudius-systems/seastar-dev

distributed::start() fix, from Tomasz.
This commit is contained in:
Avi Kivity
2015-03-25 11:19:13 +02:00
5 changed files with 99 additions and 75 deletions

View File

@@ -105,7 +105,7 @@ tests = [
'tests/shared_ptr_test',
'tests/slab_test',
'tests/fstream_test',
'tests/map_reduce_test',
'tests/distributed_test',
'tests/rpc',
]
@@ -237,7 +237,7 @@ deps = {
'tests/shared_ptr_test': ['tests/shared_ptr_test.cc'] + core,
'tests/slab_test': ['tests/slab_test.cc'] + core,
'tests/fstream_test': ['tests/fstream_test.cc'] + core,
'tests/map_reduce_test': ['tests/map_reduce_test.cc'] + core,
'tests/distributed_test': ['tests/distributed_test.cc'] + core,
'tests/rpc': ['tests/rpc.cc'] + core + libnet,
}

View File

@@ -174,7 +174,7 @@ distributed<Service>::start(Args&&... args) {
unsigned c = 0;
return parallel_for_each(_instances.begin(), _instances.end(),
[this, &c, args = std::make_tuple(std::forward<Args>(args)...)] (Service*& inst) mutable {
return smp::submit_to(c++, [&inst, args = std::move(args)] () mutable {
return smp::submit_to(c++, [&inst, args] () mutable {
inst = apply([] (Args&&... args) {
return new Service(std::forward<Args>(args)...);
}, std::move(args));

View File

@@ -70,7 +70,7 @@ if __name__ == "__main__":
for test in boost_tests:
test_to_run.append((os.path.join(prefix, test),'boost'))
test_to_run.append(('tests/memcached/test.py --mode ' + mode + (' --fast' if args.fast else ''),'other'))
test_to_run.append((os.path.join(prefix, 'map_reduce_test') + ' -c 2','other'))
test_to_run.append((os.path.join(prefix, 'distributed_test') + ' -c 2','other'))
allocator_test_path = os.path.join(prefix, 'allocator_test')

95
tests/distributed_test.cc Normal file
View File

@@ -0,0 +1,95 @@
/*
* This file is open source software, licensed to you under the terms
* of the Apache License, Version 2.0 (the "License"). See the NOTICE file
* distributed with this work for additional information regarding copyright
* ownership. 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 (C) 2015 Cloudius Systems, Ltd.
*/
#include "core/app-template.hh"
#include "core/distributed.hh"
#include "core/future-util.hh"
struct X {
sstring echo(sstring arg) {
return arg;
}
future<> stop() { return make_ready_future<>(); }
};
template <typename T, typename Func>
future<> do_with_distributed(Func&& func) {
auto x = make_shared<distributed<T>>();
return func(*x).finally([x] {
return x->stop();
}).finally([x]{});
}
future<> test_that_each_core_gets_the_arguments() {
return do_with_distributed<X>([] (auto& x) {
return x.start().then([&x] {
return x.map_reduce([] (sstring msg){
if (msg != "hello") {
throw std::runtime_error("wrong message");
}
}, &X::echo, sstring("hello"));
});
});
}
future<> test_functor_version() {
return do_with_distributed<X>([] (auto& x) {
return x.start().then([&x] {
return x.map_reduce([] (sstring msg){
if (msg != "hello") {
throw std::runtime_error("wrong message");
}
}, [] (X& x) { return x.echo("hello"); });
});
});
}
struct Y {
sstring s;
Y(sstring s) : s(std::move(s)) {}
future<> stop() { return make_ready_future<>(); }
};
future<> test_constructor_argument_is_passed_to_each_core() {
return do_with_distributed<Y>([] (auto& y) {
return y.start(sstring("hello")).then([&y] {
return y.invoke_on_all([] (Y& y) {
if (y.s != "hello") {
throw std::runtime_error(sprint("expected message mismatch, is \"%s\"", y.s));
}
});
});
});
}
int main(int argc, char** argv) {
app_template app;
return app.run(argc, argv, [] {
test_that_each_core_gets_the_arguments().then([] {
return test_functor_version();
}).then([] {
return test_constructor_argument_is_passed_to_each_core();
}).then([] {
return engine().exit(0);
}).or_terminate();
});
}

View File

@@ -1,71 +0,0 @@
/*
* This file is open source software, licensed to you under the terms
* of the Apache License, Version 2.0 (the "License"). See the NOTICE file
* distributed with this work for additional information regarding copyright
* ownership. 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 (C) 2015 Cloudius Systems, Ltd.
*/
#include "core/app-template.hh"
#include "core/distributed.hh"
#include "core/future-util.hh"
struct X {
sstring echo(sstring arg) {
return arg;
}
future<> stop() { return make_ready_future<>(); }
};
template <typename T, typename Func>
future<> do_with_distributed(Func&& func) {
auto x = make_shared<distributed<T>>();
return func(*x).finally([x] {
return x->stop();
}).finally([x]{});
}
future<> test_that_each_core_gets_the_arguments() {
return do_with_distributed<X>([] (auto& x) {
return x.map_reduce([] (sstring msg){
if (msg != "hello") {
throw std::runtime_error("wrong message");
}
}, &X::echo, sstring("hello"));
});
}
future<> test_functor_version() {
return do_with_distributed<X>([] (auto& x) {
return x.map_reduce([] (sstring msg){
if (msg != "hello") {
throw std::runtime_error("wrong message");
}
}, [] (X& x) { return x.echo("hello"); });
});
}
int main(int argc, char** argv) {
app_template app;
return app.run(argc, argv, [] {
test_that_each_core_gets_the_arguments().then([] {
return test_functor_version();
}).then([] {
return engine().exit(0);
}).or_terminate();
});
}