dds_env_prop.h

Go to the documentation of this file.

// Copyright 2015 GSI, Inc. All rights reserved.
//
//
//
#ifndef DDSENVPROP_H_
#define DDSENVPROP_H_

// STD
#include <cstdlib>
#include <string>

namespace dds
{
    enum EEnvProp
    {
        task_id,    
        task_index, 
        task_name,  
        task_path,  
        collection_index, 
        collection_name,  
        group_name,       
        dds_location      
    };

    template <EEnvProp T>
    inline typename std::enable_if<T == task_id, uint64_t>::type env_prop()
    {
        size_t ret(0);
        std::string envName;
        switch (T)
        {
            case task_id:
                envName = "DDS_TASK_ID";
                break;
            default:
                return 0;
        }
        const char* env = std::getenv(envName.c_str());
        if (nullptr == env)
            return ret;

        try
        {
            ret = std::stoul(env);
        }
        catch (...)
        {
            return 0;
        }

        return ret;
    }

    template <EEnvProp T>
    inline typename std::enable_if<T == task_index || T == collection_index, size_t>::type env_prop()
    {
        size_t ret(0);
        std::string envName;
        switch (T)
        {
            case task_index:
                envName = "DDS_TASK_INDEX";
                break;
            case collection_index:
                envName = "DDS_COLLECTION_INDEX";
                break;
            default:
                return 0;
        }
        const char* env = std::getenv(envName.c_str());
        if (nullptr == env)
            return ret;

        try
        {
            ret = std::stoi(env);
        }
        catch (...)
        {
            return 0;
        }

        return ret;
    }

    template <EEnvProp T>
    inline typename std::enable_if<T == task_name || T == collection_name || T == group_name || T == dds_location ||
                                       T == task_path,
                                   std::string>::type
        env_prop()
    {
        std::string envName;
        switch (T)
        {
            case task_name:
                envName = "DDS_TASK_NAME";
                break;
            case collection_name:
                envName = "DDS_COLLECTION_NAME";
                break;
            case group_name:
                envName = "DDS_GROUP_NAME";
                break;
            case dds_location:
                envName = "DDS_LOCATION";
                break;
            case task_path:
                envName = "DDS_TASK_PATH";
                break;
            default:
                return 0;
        }
        const char* env = std::getenv(envName.c_str());
        if (nullptr == env)
            return "";

        return std::string(env);
    }
};

#endif /*DDSENVPROP_H_*/