summaryrefslogtreecommitdiff
path: root/cpp/src/IceStorm/DatabaseCache.cpp
blob: ee4374028c32c4940ee1540ba2f5f49815f923c2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
// **********************************************************************
//
// Copyright (c) 2003-2009 ZeroC, Inc. All rights reserved.
//
// This copy of Ice is licensed to you under the terms described in the
// ICE_LICENSE file included in this distribution.
//
// **********************************************************************

#include <IceStorm/DatabaseCache.h>
#include <IceStorm/Instance.h>
#ifdef QTSQL
#  include <Ice/Communicator.h>
#  include <Ice/Properties.h>
#endif

using namespace IceStorm;
using namespace std;

#ifdef QTSQL

DatabaseCache::DatabaseCache(const InstancePtr& instance) :
    IceSQL::DatabaseCache(instance->communicator(),
        instance->communicator()->getProperties()->getProperty(instance->serviceName() + ".SQL.DatabaseType"),
        instance->communicator()->getProperties()->getProperty(instance->serviceName() + ".SQL.DatabaseName"),
        instance->communicator()->getProperties()->getProperty(instance->serviceName() + ".SQL.HostName"),
        instance->communicator()->getProperties()->getProperty(instance->serviceName() + ".SQL.UserName"),
        instance->communicator()->getProperties()->getProperty(instance->serviceName() + ".SQL.Password"),
        false)
{
    Ice::PropertiesPtr properties = instance->properties();
    string serviceName = instance->serviceName();

    string tablePrefix = properties->getPropertyWithDefault(serviceName + ".InstanceName", "IceStorm");
    string id = properties->getProperty(serviceName + ".NodeId");
    if(!id.empty())
    {
        tablePrefix += "_";
        tablePrefix += id;
    }
    replace(tablePrefix.begin(), tablePrefix.end(), '.', '_');
    replace(tablePrefix.begin(), tablePrefix.end(), '-', '_');
    replace(tablePrefix.begin(), tablePrefix.end(), ' ', '_');
    replace(tablePrefix.begin(), tablePrefix.end(), ';', '_');

    IceSQL::DatabaseConnectionPtr connection = getConnection();
    IceSQL::TransactionHolder txn(connection);

    const_cast<SqlLLUPtr&>(llu) = new SqlLLU(connection, tablePrefix + "_LLU");
    const_cast<SqlSubscriberMapPtr&>(subscriberMap) =
        new SqlSubscriberMap(connection, tablePrefix + "_Subscribers", _communicator);

    txn.commit();
}

#else

DatabaseCache::DatabaseCache(const InstancePtr& instance) :
    _communicator(instance->communicator()),
    _envName(instance->serviceName()),
    _connection(Freeze::createConnection(_communicator, _envName))
{
}

DatabaseConnectionPtr
DatabaseCache::getConnection()
{
    return _connection;
}

DatabaseConnectionPtr
DatabaseCache::newConnection()
{
    return Freeze::createConnection(_communicator, _envName);
}


#endif