]> granicus.if.org Git - icinga2/blobdiff - lib/base/value-operators.cpp
Implement comparison operators for the Array class
[icinga2] / lib / base / value-operators.cpp
index 2ef1541c84f5db6fa10fe44288fe8fa6994a046e..ad0039722a663cf35fa5bb6810702954c0b5e4a0 100644 (file)
@@ -1,6 +1,6 @@
 /******************************************************************************
  * Icinga 2                                                                   *
- * Copyright (C) 2012-2014 Icinga Development Team (http://www.icinga.org)    *
+ * Copyright (C) 2012-2016 Icinga Development Team (https://www.icinga.org/)  *
  *                                                                            *
  * This program is free software; you can redistribute it and/or              *
  * modify it under the terms of the GNU General Public License                *
@@ -20,6 +20,8 @@
 #include "base/value.hpp"
 #include "base/array.hpp"
 #include "base/dictionary.hpp"
+#include "base/datetime.hpp"
+#include "base/convert.hpp"
 #include "base/utility.hpp"
 #include "base/objectlock.hpp"
 #include <boost/foreach.hpp>
@@ -34,27 +36,32 @@ Value::operator double(void) const
        if (value)
                return *value;
 
+       const bool *fvalue = boost::get<bool>(&m_Value);
+
+       if (fvalue)
+               return *fvalue;
+
        if (IsEmpty())
                return 0;
 
-       return boost::lexical_cast<double>(m_Value);
+       try {
+               return boost::lexical_cast<double>(m_Value);
+       } catch (const std::exception&) {
+               std::ostringstream msgbuf;
+               msgbuf << "Can't convert '" << *this << "' to a floating point number.";
+               BOOST_THROW_EXCEPTION(std::invalid_argument(msgbuf.str()));
+       }
 }
 
 Value::operator String(void) const
 {
        Object *object;
-       double integral, fractional;
 
        switch (GetType()) {
                case ValueEmpty:
                        return String();
                case ValueNumber:
-                       fractional = std::modf(boost::get<double>(m_Value), &integral);
-
-                       if (fractional != 0)
-                               return boost::lexical_cast<std::string>(m_Value);
-                       else
-                               return boost::lexical_cast<std::string>((long)integral);
+                       return Convert::ToString(boost::get<double>(m_Value));
                case ValueBoolean:
                        if (boost::get<bool>(m_Value))
                                return "true";
@@ -72,7 +79,11 @@ Value::operator String(void) const
 
 std::ostream& icinga::operator<<(std::ostream& stream, const Value& value)
 {
-       stream << static_cast<String>(value);
+       if (value.IsBoolean())
+               stream << static_cast<int>(value);
+       else
+               stream << static_cast<String>(value);
+
        return stream;
 }
 
@@ -84,6 +95,12 @@ std::istream& icinga::operator>>(std::istream& stream, Value& value)
        return stream;
 }
 
+Value& Value::operator=(const Value& other)
+{
+       m_Value = other.m_Value;
+       return *this;
+}
+
 bool Value::operator==(bool rhs) const
 {
        return *this == Value(rhs);
@@ -138,7 +155,7 @@ bool Value::operator==(const Value& rhs) const
 {
        if (IsNumber() && rhs.IsNumber())
                return Get<double>() == rhs.Get<double>();
-       else if ((IsBoolean() || IsNumber() || IsEmpty()) && (rhs.IsBoolean() || rhs.IsNumber() || rhs.IsEmpty()) && !(IsEmpty() && rhs.IsEmpty()))
+       else if ((IsBoolean() || IsNumber()) && (rhs.IsBoolean() || rhs.IsNumber()) && !(IsEmpty() && rhs.IsEmpty()))
                return static_cast<double>(*this) == static_cast<double>(rhs);
 
        if (IsString() && rhs.IsString())
@@ -156,6 +173,13 @@ bool Value::operator==(const Value& rhs) const
                return false;
 
        if (IsObject()) {
+               if (IsObjectType<DateTime>() && rhs.IsObjectType<DateTime>()) {
+                       DateTime::Ptr dt1 = *this;
+                       DateTime::Ptr dt2 = rhs;
+
+                       return dt1->GetValue() == dt2->GetValue();
+               }
+
                if (IsObjectType<Array>() && rhs.IsObjectType<Array>()) {
                        Array::Ptr arr1 = *this;
                        Array::Ptr arr2 = rhs;
@@ -207,20 +231,22 @@ Value icinga::operator+(const String& lhs, const Value& rhs)
 
 Value icinga::operator+(const Value& lhs, const Value& rhs)
 {
-       if ((lhs.IsEmpty() || lhs.IsNumber()) && (rhs.IsEmpty() || rhs.IsNumber()) && !(lhs.IsEmpty() && rhs.IsEmpty()))
+       if ((lhs.IsEmpty() || lhs.IsNumber()) && !lhs.IsString() && (rhs.IsEmpty() || rhs.IsNumber()) && !rhs.IsString() && !(lhs.IsEmpty() && rhs.IsEmpty()))
                return static_cast<double>(lhs) + static_cast<double>(rhs);
-       if ((lhs.IsString() || lhs.IsEmpty() || lhs.IsNumber()) && (rhs.IsString() || rhs.IsEmpty() || rhs.IsNumber()) && !(lhs.IsEmpty() && rhs.IsEmpty()))
+       if ((lhs.IsString() || lhs.IsEmpty() || lhs.IsNumber()) && (rhs.IsString() || rhs.IsEmpty() || rhs.IsNumber()) && (!(lhs.IsEmpty() && rhs.IsEmpty()) || lhs.IsString() || rhs.IsString()))
                return static_cast<String>(lhs) + static_cast<String>(rhs);
-       else if ((lhs.IsNumber() || lhs.IsEmpty()) && (rhs.IsNumber() || rhs.IsEmpty()) && !(lhs.IsEmpty() && rhs.IsEmpty()))
+       else if ((lhs.IsNumber() || lhs.IsEmpty()) && (rhs.IsNumber() || rhs.IsEmpty()) && !(lhs.IsEmpty() && rhs.IsEmpty()) && !(lhs.IsEmpty() && rhs.IsEmpty()))
                return static_cast<double>(lhs) + static_cast<double>(rhs);
-       else if ((lhs.IsObjectType<Array>() || lhs.IsEmpty()) && (rhs.IsObjectType<Array>() || rhs.IsEmpty()) && !(lhs.IsEmpty() && rhs.IsEmpty())) {
+       else if (lhs.IsObjectType<DateTime>() && rhs.IsNumber())
+               return new DateTime(Convert::ToDateTimeValue(lhs) + rhs);
+       else if ((lhs.IsObjectType<Array>() || lhs.IsEmpty()) && (rhs.IsObjectType<Array>() || rhs.IsEmpty()) && !(lhs.IsEmpty() && rhs.IsEmpty()) && !(lhs.IsEmpty() && rhs.IsEmpty())) {
                Array::Ptr result = new Array();
                if (!lhs.IsEmpty())
                        static_cast<Array::Ptr>(lhs)->CopyTo(result);
                if (!rhs.IsEmpty())
                        static_cast<Array::Ptr>(rhs)->CopyTo(result);
                return result;
-       } else if ((lhs.IsObjectType<Dictionary>() || lhs.IsEmpty()) && (rhs.IsObjectType<Dictionary>() || rhs.IsEmpty()) && !(lhs.IsEmpty() && rhs.IsEmpty())) {
+       } else if ((lhs.IsObjectType<Dictionary>() || lhs.IsEmpty()) && (rhs.IsObjectType<Dictionary>() || rhs.IsEmpty()) && !(lhs.IsEmpty() && rhs.IsEmpty()) && !(lhs.IsEmpty() && rhs.IsEmpty())) {
                Dictionary::Ptr result = new Dictionary();
                if (!lhs.IsEmpty())
                        static_cast<Dictionary::Ptr>(lhs)->CopyTo(result);
@@ -254,8 +280,14 @@ Value icinga::operator+(int lhs, const Value& rhs)
 
 Value icinga::operator-(const Value& lhs, const Value& rhs)
 {
-       if ((lhs.IsNumber() || lhs.IsEmpty()) && (rhs.IsNumber() || rhs.IsEmpty()) && !(lhs.IsEmpty() && rhs.IsEmpty()))
+       if ((lhs.IsNumber() || lhs.IsEmpty()) && !lhs.IsString() && (rhs.IsNumber() || rhs.IsEmpty()) && !rhs.IsString() && !(lhs.IsEmpty() && rhs.IsEmpty()))
                return static_cast<double>(lhs) - static_cast<double>(rhs);
+       else if (lhs.IsObjectType<DateTime>() && rhs.IsNumber())
+               return new DateTime(Convert::ToDateTimeValue(lhs) - rhs);
+       else if (lhs.IsObjectType<DateTime>() && rhs.IsObjectType<DateTime>())
+               return Convert::ToDateTimeValue(lhs) - Convert::ToDateTimeValue(rhs);
+       else if ((lhs.IsObjectType<DateTime>() || lhs.IsEmpty()) && (rhs.IsObjectType<DateTime>() || rhs.IsEmpty()) && !(lhs.IsEmpty() && rhs.IsEmpty()) && !(lhs.IsEmpty() && rhs.IsEmpty()))
+               return new DateTime(Convert::ToDateTimeValue(lhs) - Convert::ToDateTimeValue(rhs));
        else if ((lhs.IsObjectType<Array>() || lhs.IsEmpty()) && (rhs.IsObjectType<Array>() || rhs.IsEmpty()) && !(lhs.IsEmpty() && rhs.IsEmpty())) {
                if (lhs.IsEmpty())
                        return new Array();
@@ -410,22 +442,22 @@ Value icinga::operator^(const Value& lhs, const Value& rhs)
 
 Value icinga::operator^(const Value& lhs, double rhs)
 {
-       return lhs & Value(rhs);
+       return lhs ^ Value(rhs);
 }
 
 Value icinga::operator^(double lhs, const Value& rhs)
 {
-       return Value(lhs) & rhs;
+       return Value(lhs) ^ rhs;
 }
 
 Value icinga::operator^(const Value& lhs, int rhs)
 {
-       return lhs & Value(rhs);
+       return lhs ^ Value(rhs);
 }
 
 Value icinga::operator^(int lhs, const Value& rhs)
 {
-       return Value(lhs) & rhs;
+       return Value(lhs) ^ rhs;
 }
 
 Value icinga::operator&(const Value& lhs, const Value& rhs)
@@ -545,10 +577,31 @@ bool icinga::operator<(const Value& lhs, const Value& rhs)
        if (lhs.IsString() && rhs.IsString())
                return static_cast<String>(lhs) < static_cast<String>(rhs);
        else if ((lhs.IsNumber() || lhs.IsEmpty()) && (rhs.IsNumber() || rhs.IsEmpty()) && !(lhs.IsEmpty() && rhs.IsEmpty()))
-               return static_cast<int>(lhs) < static_cast<int>(rhs);
-       else if (lhs.GetTypeName() != rhs.GetTypeName())
-               return lhs.GetTypeName() < rhs.GetTypeName();
-       else
+               return static_cast<double>(lhs) < static_cast<double>(rhs);
+       else if ((lhs.IsObjectType<DateTime>() || lhs.IsEmpty()) && (rhs.IsObjectType<DateTime>() || rhs.IsEmpty()) && !(lhs.IsEmpty() && rhs.IsEmpty()) && !(lhs.IsEmpty() && rhs.IsEmpty()))
+               return Convert::ToDateTimeValue(lhs) < Convert::ToDateTimeValue(rhs);
+       else if (lhs.IsObjectType<Array>() && rhs.IsObjectType<Array>()) {
+               Array::Ptr larr = lhs;
+               Array::Ptr rarr = rhs;
+
+               ObjectLock llock(larr);
+               ObjectLock rlock(rarr);
+
+               Array::SizeType llen = larr->GetLength();
+               Array::SizeType rlen = rarr->GetLength();
+
+               for (Array::SizeType i = 0; i < std::max(llen, rlen); i++) {
+                       Value lval = (i >= llen) ? Empty : larr->Get(i);
+                       Value rval = (i >= rlen) ? Empty : rarr->Get(i);
+
+                       if (lval < rval)
+                               return true;
+                       else if (lval > rval)
+                               return false;
+               }
+
+               return false;
+       } else
                BOOST_THROW_EXCEPTION(std::invalid_argument("Operator < cannot be applied to values of type '" + lhs.GetTypeName() + "' and '" + rhs.GetTypeName() + "'"));
 }
 
@@ -577,10 +630,31 @@ bool icinga::operator>(const Value& lhs, const Value& rhs)
        if (lhs.IsString() && rhs.IsString())
                return static_cast<String>(lhs) > static_cast<String>(rhs);
        else if ((lhs.IsNumber() || lhs.IsEmpty()) && (rhs.IsNumber() || rhs.IsEmpty()) && !(lhs.IsEmpty() && rhs.IsEmpty()))
-               return static_cast<int>(lhs) > static_cast<int>(rhs);
-       else if (lhs.GetTypeName() != rhs.GetTypeName())
-               return lhs.GetTypeName() > rhs.GetTypeName();
-       else
+               return static_cast<double>(lhs) > static_cast<double>(rhs);
+       else if ((lhs.IsObjectType<DateTime>() || lhs.IsEmpty()) && (rhs.IsObjectType<DateTime>() || rhs.IsEmpty()) && !(lhs.IsEmpty() && rhs.IsEmpty()) && !(lhs.IsEmpty() && rhs.IsEmpty()))
+               return Convert::ToDateTimeValue(lhs) > Convert::ToDateTimeValue(rhs);
+       else if (lhs.IsObjectType<Array>() && rhs.IsObjectType<Array>()) {
+               Array::Ptr larr = lhs;
+               Array::Ptr rarr = rhs;
+
+               ObjectLock llock(larr);
+               ObjectLock rlock(rarr);
+
+               Array::SizeType llen = larr->GetLength();
+               Array::SizeType rlen = rarr->GetLength();
+
+               for (Array::SizeType i = 0; i < std::max(llen, rlen); i++) {
+                       Value lval = (i >= llen) ? Empty : larr->Get(i);
+                       Value rval = (i >= rlen) ? Empty : rarr->Get(i);
+
+                       if (lval > rval)
+                               return true;
+                       else if (lval < rval)
+                               return false;
+               }
+
+               return false;
+       } else
                BOOST_THROW_EXCEPTION(std::invalid_argument("Operator > cannot be applied to values of type '" + lhs.GetTypeName() + "' and '" + rhs.GetTypeName() + "'"));
 }
 
@@ -609,9 +683,9 @@ bool icinga::operator<=(const Value& lhs, const Value& rhs)
        if (lhs.IsString() && rhs.IsString())
                return static_cast<String>(lhs) <= static_cast<String>(rhs);
        else if ((lhs.IsNumber() || lhs.IsEmpty()) && (rhs.IsNumber() || rhs.IsEmpty()) && !(lhs.IsEmpty() && rhs.IsEmpty()))
-               return static_cast<int>(lhs) <= static_cast<int>(rhs);
-       else if (lhs.GetTypeName() != rhs.GetTypeName())
-               return lhs.GetTypeName() <= rhs.GetTypeName();
+               return static_cast<double>(lhs) <= static_cast<double>(rhs);
+       else if ((lhs.IsObjectType<DateTime>() || lhs.IsEmpty()) && (rhs.IsObjectType<DateTime>() || rhs.IsEmpty()) && !(lhs.IsEmpty() && rhs.IsEmpty()) && !(lhs.IsEmpty() && rhs.IsEmpty()))
+               return Convert::ToDateTimeValue(lhs) <= Convert::ToDateTimeValue(rhs);
        else
                BOOST_THROW_EXCEPTION(std::invalid_argument("Operator <= cannot be applied to values of type '" + lhs.GetTypeName() + "' and '" + rhs.GetTypeName() + "'"));
 }
@@ -641,9 +715,9 @@ bool icinga::operator>=(const Value& lhs, const Value& rhs)
        if (lhs.IsString() && rhs.IsString())
                return static_cast<String>(lhs) >= static_cast<String>(rhs);
        else if ((lhs.IsNumber() || lhs.IsEmpty()) && (rhs.IsNumber() || rhs.IsEmpty()) && !(lhs.IsEmpty() && rhs.IsEmpty()))
-               return static_cast<int>(lhs) >= static_cast<int>(rhs);
-       else if (lhs.GetTypeName() != rhs.GetTypeName())
-               return lhs.GetTypeName() >= rhs.GetTypeName();
+               return static_cast<double>(lhs) >= static_cast<double>(rhs);
+       else if ((lhs.IsObjectType<DateTime>() || lhs.IsEmpty()) && (rhs.IsObjectType<DateTime>() || rhs.IsEmpty()) && !(lhs.IsEmpty() && rhs.IsEmpty()) && !(lhs.IsEmpty() && rhs.IsEmpty()))
+               return Convert::ToDateTimeValue(lhs) >= Convert::ToDateTimeValue(rhs);
        else
                BOOST_THROW_EXCEPTION(std::invalid_argument("Operator >= cannot be applied to values of type '" + lhs.GetTypeName() + "' and '" + rhs.GetTypeName() + "'"));
 }