binary_containers.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_BINARY_BINARY_CONTAINERS
#define _IGNITE_BINARY_BINARY_CONTAINERS

#include <stdint.h>

#include <ignite/common/utils.h>

#include "ignite/impl/binary/binary_writer_impl.h"
#include "ignite/impl/binary/binary_reader_impl.h"
#include "ignite/binary/binary_consts.h"

namespace ignite
{
    namespace binary
    {
        class IGNITE_IMPORT_EXPORT BinaryStringArrayWriter
        {
        public:
            BinaryStringArrayWriter(impl::binary::BinaryWriterImpl* impl, int32_t id);

            void Write(const char* val);

            void Write(const char* val, int32_t len);

            void Write(const std::string& val)
            {
                Write(val.c_str());
            }

            void Close();

        private:
            impl::binary::BinaryWriterImpl* impl; 

            const int32_t id;    
        };

        template<typename T>
        class IGNITE_IMPORT_EXPORT BinaryArrayWriter
        {
        public:
            BinaryArrayWriter(impl::binary::BinaryWriterImpl* impl, int32_t id) :
                impl(impl), id(id)
            {
                // No-op.
            }

            void Write(const T& val)
            {
                impl->WriteElement<T>(id, val);
            }

            void Close()
            {
                impl->CommitContainer(id);
            }

        private:
            impl::binary::BinaryWriterImpl* impl; 

            const int32_t id;      
        };

        template<typename T>
        class IGNITE_IMPORT_EXPORT BinaryCollectionWriter
        {
        public:
            BinaryCollectionWriter(impl::binary::BinaryWriterImpl* impl, int32_t id) :
                impl(impl), id(id)
            {
                // No-op.
            }

            void Write(const T& val)
            {
                impl->WriteElement<T>(id, val);
            }

            void Close()
            {
                impl->CommitContainer(id);
            }
        private:
            impl::binary::BinaryWriterImpl* impl; 

            const int32_t id;    
        };

        template<typename K, typename V>
        class IGNITE_IMPORT_EXPORT BinaryMapWriter
        {
        public:
            BinaryMapWriter(impl::binary::BinaryWriterImpl* impl, int32_t id) :
                impl(impl), id(id)
            {
                // No-op.
            }

            void Write(const K& key, const V& val)
            {
                impl->WriteElement<K, V>(id, key, val);
            }

            void Close()
            {
                impl->CommitContainer(id);
            }
        private:
            impl::binary::BinaryWriterImpl* impl; 

            const int32_t id;      
        };

        class IGNITE_IMPORT_EXPORT BinaryStringArrayReader
        {
        public:
            BinaryStringArrayReader(impl::binary::BinaryReaderImpl* impl, int32_t id, int32_t size);

            bool HasNext();

            int32_t GetNext(char* res, int32_t len);

            std::string GetNext()
            {
                int32_t len = GetNext(NULL, 0);

                if (len != -1)
                {
                    ignite::common::FixedSizeArray<char> arr(len + 1);

                    GetNext(arr.GetData(), static_cast<int32_t>(arr.GetSize()));

                    return std::string(arr.GetData());
                }
                else
                    return std::string();
            }

            int32_t GetSize() const;

            bool IsNull() const;

        private:
            impl::binary::BinaryReaderImpl* impl;  

            const int32_t id;

            const int32_t size;
        };

        template<typename T>
        class BinaryArrayReader
        {
        public:
            BinaryArrayReader(impl::binary::BinaryReaderImpl* impl, int32_t id, int32_t size) : 
                impl(impl), id(id), size(size)
            {
                // No-op.
            }

            bool HasNext()
            {
                return impl->HasNextElement(id);
            }

            T GetNext()
            {
                return impl->ReadElement<T>(id);
            }

            int32_t GetSize()
            {
                return size;
            }

            bool IsNull()
            {
                return size == -1;
            }
        private:
            impl::binary::BinaryReaderImpl* impl;

            const int32_t id;

            const int32_t size;
        };

        template<typename T>
        class BinaryCollectionReader
        {
        public:
            BinaryCollectionReader(impl::binary::BinaryReaderImpl* impl, int32_t id, 
                const CollectionType::Type type,  int32_t size) : impl(impl), id(id), type(type), size(size)
            {
                // No-op.
            }

            bool HasNext()
            {
                return impl->HasNextElement(id);
            }

            T GetNext()
            {
                return impl->ReadElement<T>(id);
            }
            
            CollectionType::Type GetType()
            {
                return type;
            }

            int32_t GetSize()
            {
                return size;
            }

            bool IsNull()
            {
                return size == -1;
            }
        private:
            impl::binary::BinaryReaderImpl* impl;  

            const int32_t id;     
            
            const CollectionType::Type type;  

            const int32_t size;                              
        };    

        template<typename K, typename V>
        class BinaryMapReader
        {
        public:
            BinaryMapReader(impl::binary::BinaryReaderImpl* impl, int32_t id, MapType::Type type,
                int32_t size) : impl(impl), id(id), type(type), size(size)
            {
                // No-op.
            }

            bool HasNext()
            {
                return impl->HasNextElement(id);
            }

            void GetNext(K& key, V& val)
            {
                return impl->ReadElement<K, V>(id, key, val);
            }

            MapType::Type GetType()
            {
                return type;
            }

            int32_t GetSize()
            {
                return size;
            }

            bool IsNull()
            {
                return size == -1;
            }
        private:
            impl::binary::BinaryReaderImpl* impl;  

            const int32_t id;     

            const MapType::Type type;

            const int32_t size;
        };
    }
}

#endif //_IGNITE_BINARY_BINARY_CONTAINERS