Project

General

Profile

Feature #3400 » DboTestCompositeKey.C

Saif Rehman, 01/07/2016 07:50 PM

 
/*
* Copyright (C) 2009 Emweb bvba, Kessel-Lo, Belgium.
*
* See the LICENSE file for terms of use.
*/

#include <boost/test/unit_test.hpp>

#include <Wt/Dbo/Dbo>
#include <Wt/WDate>
#include <Wt/WDateTime>
#include <Wt/WTime>
#include <Wt/Dbo/WtSqlTraits>
#include <Wt/Dbo/ptr_tuple>

#include "DboFixture.h"

#include <string>
namespace dbo = Wt::Dbo;

//----------------------------------- DBOs -------------------------------------
//Fwd dec
class Page;
class Module;
struct PageKeys;
typedef Wt::Dbo::collection< Wt::Dbo::ptr<Page> > PageCollections;

//Module Dbo
class Module
{
public:
PageCollections PageCollection;

template<class Action>void persist(Action &a)
{
Wt::Dbo::hasMany(a, PageCollection, Wt::Dbo::ManyToOne, "Module");
}
};

//Page Dbo
struct PageKeys
{
long long id;
Wt::Dbo::ptr<Module> ModulePtr;

PageKeys();
PageKeys(long long id, Wt::Dbo::ptr<Module> ModulePtr);

bool operator< (const PageKeys &other) const;
bool operator== (const PageKeys &other) const;
};
std::ostream &operator<< (std::ostream &o, const PageKeys &c);


//Mapping for PageKeys composite key
namespace Wt
{
namespace Dbo
{
template<class Action>
void field(Action &action, PageKeys &Keys, const std::string &name, int size = -1)
{
field(action, Keys.id, name + "_page_id");
belongsTo(action, Keys.ModulePtr, name + "_Module", Wt::Dbo::OnDeleteCascade | Wt::Dbo::OnUpdateCascade | Wt::Dbo::NotNull);
}
template<>
struct dbo_traits<Page> : public dbo_default_traits
{
typedef PageKeys IdType;
static IdType invalidId();
static const char *surrogateIdField();
};
}
}

class Page
{
public:
PageCollections ChildrenPages;
Wt::Dbo::ptr<Page> ParentPage;

Page();
Page(long long id, Wt::Dbo::ptr<Module> ModulePtr = Wt::Dbo::ptr<Module>());

template<class Action>void persist(Action &a)
{
Wt::Dbo::id(a, _Id, "Page");

Wt::Dbo::hasMany(a, ChildrenPages, Wt::Dbo::ManyToOne, "Parent_Page");
Wt::Dbo::belongsTo(a, ParentPage, "Parent_Page", Wt::Dbo::OnDeleteCascade | Wt::Dbo::OnUpdateCascade);
}

private:
PageKeys _Id;
};

PageKeys::PageKeys()
: id(-1)
{ }
PageKeys::PageKeys(long long id, Wt::Dbo::ptr<Module> ModulePtr)
: id(id), ModulePtr(ModulePtr)
{ }
bool PageKeys::operator== (const PageKeys &other) const
{
return id == other.id && ModulePtr == other.ModulePtr;
}
bool PageKeys::operator< (const PageKeys &other) const
{
if(id < other.id)
{
return true;
}
else if(id == other.id)
{
return ModulePtr < other.ModulePtr;
}
else
{
return false;
}
}
std::ostream &operator<< (std::ostream &o, const PageKeys &c)
{
return o << "(" << c.id << ", " << c.ModulePtr << ")";
}

Wt::Dbo::dbo_traits<Page>::IdType Wt::Dbo::dbo_traits<Page>::invalidId()
{
return Wt::Dbo::dbo_traits<Page>::IdType();
}
const char* Wt::Dbo::dbo_traits<Page>::surrogateIdField()
{
return 0;
}

Page::Page()
{ }
Page::Page(long long id, Wt::Dbo::ptr<Module> ModulePtr)
: _Id(id, ModulePtr)
{ }
//------------------------------- End of DBOs ----------------------------------

struct Dbo4Fixture : DboFixtureBase
{
Dbo4Fixture() :
DboFixtureBase()
{
session_->mapClass<Module>("modules");
session_->mapClass<Page>("pages");

//Create
std::cout << "-------------------------- Create Table ----------------------" << std::endl;
session_->createTables();

//Drop
std::cout << "--------------------------- Drop Table -----------------------" << std::endl;
session_->dropTables();
}
};

BOOST_AUTO_TEST_CASE( dbo_test_composite_concat )
{
Dbo4Fixture f;
}
(2-2/2)