1 files changed, 187 insertions, 0 deletions
diff --git a/slicer/db/sqlSelectDeserializer.cpp b/slicer/db/sqlSelectDeserializer.cpp
new file mode 100644
index 0000000..2a499b6
--- /dev/null
+++ b/slicer/db/sqlSelectDeserializer.cpp
@@ -0,0 +1,187 @@
+#include "sqlSelectDeserializer.h"
+#include <boost/algorithm/string/predicate.hpp>
+#include <boost/numeric/conversion/cast.hpp>
+
+namespace Slicer {
+	NoRowsReturned::NoRowsReturned() : std::runtime_error("No rows returned") { }
+
+	TooManyRowsReturned::TooManyRowsReturned() : std::runtime_error("Too many rows returned") { }
+
+	class SqlSource : public Slicer::ValueSource,
+			public Slicer::TValueSource<boost::posix_time::time_duration>,
+			public Slicer::TValueSource<boost::posix_time::ptime>
+	{
+		public:
+			SqlSource(const DB::Column & c) :
+				column(c)
+			{
+			}
+
+			bool isNull() const
+			{
+				return column.isNull();
+			}
+
+			void set(boost::posix_time::ptime & b) const override
+			{
+				column >> b;
+			}
+
+			void set(boost::posix_time::time_duration & b) const override
+			{
+				column >> b;
+			}
+
+			void set(bool & b) const override
+			{
+				column >> b;
+			}
+
+			void set(Ice::Byte & b) const override
+			{
+				int64_t cb;
+				column >> cb;
+				b = boost::numeric_cast<Ice::Byte>(cb);
+			}
+
+			void set(Ice::Short & b) const override
+			{
+				int64_t cb;
+				column >> cb;
+				b = boost::numeric_cast<Ice::Byte>(cb);
+			}
+
+			void set(Ice::Int & b) const override
+			{
+				int64_t cb;
+				column >> cb;
+				b = boost::numeric_cast<Ice::Int>(cb);
+			}
+
+			void set(Ice::Long & b) const override
+			{
+				column >> b;
+			}
+
+			void set(Ice::Float & b) const override
+			{
+				double cb;
+				column >> cb;
+				b = boost::numeric_cast<Ice::Float>(cb);
+			}
+
+			void set(Ice::Double & b) const override
+			{
+				column >> b;
+			}
+
+			void set(std::string & b) const override
+			{
+				column >> b;
+			}
+
+		private:
+			const DB::Column & column;
+	};
+	typedef IceUtil::Handle<SqlSource> SqlSourcePtr;
+
+	SqlSelectDeserializer::SqlSelectDeserializer(DB::SelectCommand & c, IceUtil::Optional<std::string> tc) :
+		cmd(c),
+		typeIdColName(tc)
+	{
+	}
+
+	void
+	SqlSelectDeserializer::Deserialize(Slicer::ModelPartPtr mp)
+	{
+		cmd.execute();
+		columnCount = cmd.columnCount();
+		if (typeIdColName) {
+			typeIdColIdx = cmd.getOrdinal(*typeIdColName);
+		}
+		switch (mp->GetType()) {
+			case Slicer::mpt_Sequence:
+				DeserializeSequence(mp);
+				return;
+			case Slicer::mpt_Complex:
+				DeserializeObject(mp);
+				return;
+			case Slicer::mpt_Simple:
+				DeserializeSimple(mp);
+				return;
+			default:
+				throw std::invalid_argument("Unspported model type");
+		}
+	}
+
+	void
+	SqlSelectDeserializer::DeserializeSimple(Slicer::ModelPartPtr mp)
+	{
+		auto fmp = mp->GetAnonChild();
+		if (!cmd.fetch()) {
+			throw NoRowsReturned();
+		}
+		SqlSourcePtr h = new SqlSource(cmd[0]);
+		if (!h->isNull()) {
+			fmp->Create();
+			fmp->SetValue(h);
+			fmp->Complete();
+		}
+		if (cmd.fetch()) {
+			throw TooManyRowsReturned();
+		}
+	}
+
+	void
+	SqlSelectDeserializer::DeserializeSequence(Slicer::ModelPartPtr mp)
+	{
+		mp = mp->GetAnonChild();
+		while (cmd.fetch()) {
+			DeserializeRow(mp);
+		}
+	}
+
+	void
+	SqlSelectDeserializer::DeserializeObject(Slicer::ModelPartPtr mp)
+	{
+		if (!cmd.fetch()) {
+			throw NoRowsReturned();
+		}
+		DeserializeRow(mp);
+		if (cmd.fetch()) {
+			while (cmd.fetch()) ;
+			throw TooManyRowsReturned();
+		}
+	}
+
+	void
+	SqlSelectDeserializer::DeserializeRow(Slicer::ModelPartPtr mp)
+	{
+		auto rmp = mp->GetAnonChild();
+		if (rmp) {
+			if (typeIdColIdx) {
+				std::string subclass;
+				cmd[*typeIdColIdx] >> subclass;
+				rmp = rmp->GetSubclassModelPart(subclass);
+			}
+			rmp->Create();
+			for (auto col = 0u; col < columnCount; col += 1) {
+				const DB::Column & c = cmd[col];
+				SqlSourcePtr h = new SqlSource(c);
+				auto fmpr = rmp->GetAnonChildRef([&c](Slicer::HookCommonPtr h) {
+						return boost::iequals(c.name.raw(), h->PartName());
+					});
+				if (fmpr) {
+					auto fmp = fmpr->Child();
+					if (!h->isNull()) {
+						fmp->Create();
+						fmp->SetValue(h);
+						fmp->Complete();
+					}
+				}
+			}
+			rmp->Complete();
+		}
+	}
+}
+