compute_client.h

Go to the documentation of this file.

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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.
 */
 
#ifndef _IGNITE_THIN_COMPUTE_COMPUTE_CLIENT
#define _IGNITE_THIN_COMPUTE_COMPUTE_CLIENT
 
#include <ignite/common/concurrent.h>
 
namespace ignite
{
    namespace thin
    {
        namespace compute
        {
            struct ComputeClientFlags
            {
                enum Type
                {
                    NONE = 0,
                    NO_FAILOVER = 1,
                    NO_RESULT_CACHE = 2
                };
            };
 
            class IGNITE_IMPORT_EXPORT ComputeClient
            {
                typedef common::concurrent::SharedPointer<void> SP_Void;
            public:
                ComputeClient()
                {
                    // No-op.
                }
 
                ComputeClient(const SP_Void& impl) :
                    impl(impl),
                    flags(ComputeClientFlags::NONE),
                    timeout(0)
                {
                    // No-op.
                }
 
                ~ComputeClient()
                {
                    // No-op.
                }
 
                template<typename R, typename A>
                R ExecuteJavaTask(const std::string& taskName, const A& taskArg)
                {
                    R result;
 
                    impl::thin::WritableImpl<A> wrArg(taskArg);
                    impl::thin::ReadableImpl<R> rdResult(result);
 
                    InternalExecuteJavaTask(taskName, wrArg, rdResult);
 
                    return result;
                }
 
                template<typename R>
                R ExecuteJavaTask(const std::string& taskName)
                {
                    R result;
                    int* nullVal = 0;
 
                    impl::thin::WritableImpl<int*> wrArg(nullVal);
                    impl::thin::ReadableImpl<R> rdResult(result);
 
                    InternalExecuteJavaTask(taskName, wrArg, rdResult);
 
                    return result;
                }
 
                ComputeClient WithTimeout(int64_t timeoutMs)
                {
                    return ComputeClient(impl, flags, timeoutMs);
                }
 
                ComputeClient WithNoFailover()
                {
                    return ComputeClient(impl, flags | ComputeClientFlags::NO_FAILOVER, timeout);
                }
 
                ComputeClient WithNoResultCache()
                {
                    return ComputeClient(impl, flags | ComputeClientFlags::NO_RESULT_CACHE, timeout);
                }
 
            private:
                ComputeClient(const SP_Void& impl, int8_t flags, int64_t timeout) :
                    impl(impl),
                    flags(flags),
                    timeout(timeout)
                {
                    // No-op.
                }
 
                void InternalExecuteJavaTask(const std::string& taskName, impl::thin::Writable& wrArg,
                    impl::thin::Readable& res);
 
                SP_Void impl;
 
                int8_t flags;
 
                int64_t timeout;
            };
        }
    }
}
 
#endif // _IGNITE_THIN_COMPUTE_COMPUTE_CLIENT