summaryrefslogtreecommitdiffstats
path: root/sca-cpp/trunk/components/sqldb/sqldb.cpp
blob: 75be2c0624b364ffa552e8e7cd551854b9f7b038 (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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

/* $Rev$ $Date$ */

/**
 * PostgreSQL-based database component implementation.
 */

#define WANT_HTTPD_LOG 1
#include "string.hpp"
#include "function.hpp"
#include "list.hpp"
#include "value.hpp"
#include "monad.hpp"
#include "parallel.hpp"
#include "pgsql.hpp"

namespace tuscany {
namespace sqldb {

/**
 * Get an item from the database.
 */
const failable<value> get(const list<value>& params, const pgsql::PGSql& pg) {
    return pgsql::get(car(params), pg);
}

/**
 * Post an item to the database.
 */
const failable<value> post(const list<value>& params, const pgsql::PGSql& pg) {
    const value id = append<value>(car(params), mklist(mkuuid()));
    const failable<bool> val = pgsql::post(id, cadr(params), pg);
    if (!hasContent(val))
        return mkfailure<value>(val);
    return id;
}

/**
 * Put an item into the database.
 */
const failable<value> put(const list<value>& params, const pgsql::PGSql& pg) {
    const failable<bool> val = pgsql::put(car(params), cadr(params), pg);
    if (!hasContent(val))
        return mkfailure<value>(val);
    return value(content(val));
}

/**
 * Patch an item in the database.
 */
const failable<value> patch(const list<value>& params, const pgsql::PGSql& pg) {
    // Read patch
    value p = assoc<value>("patch", assoc<value>("content", car<value>(cadr(params))));
    if (isNil(p))
        return mkfailure<value>("Couldn't read patch script");
    const string script = cadr<value>(p);
    debug(script, "pgsql::patch::script");

    const lambda<const failable<value>(const value&, const pgsql::PGSql&, const string&, const int)> tryPatch = [&tryPatch](const value& key, const pgsql::PGSql& pg, const string& script, const int count) -> const failable<value> {

        // Begin database transaction
        const failable<bool> brc = pgsql::begin(pg);
        if (!hasContent(brc))
            return mkfailure<value>(brc);

        // Get existing value from database
        const failable<value> ival = pgsql::get(key, pg);
        if (!hasContent(ival) && rcode(ival) != 404) {
            pgsql::rollback(pg);
            return mkfailure<value>(ival);
        }

        // Apply patch
        istringstream is(script);
        scheme::Env env = scheme::setupEnvironment();
        const value pval = scheme::evalScript(cons<value>("patch", scheme::quotedParameters(mklist<value>(key, hasContent(ival)? content(ival) : (value)list<value>()))), is, env);
        if (isNil(pval)) {
            ostringstream os;
            os << "Couldn't patch postgresql entry: " << key;
            return mkfailure<value>(str(os), 404, false);
        }

        // Push patched value to database
        const failable<bool> val = pgsql::patch(key, pval, pg);
        if (!hasContent(val)) {
            pgsql::rollback(pg);

            // Retry on a transaction serialization error
            if (rcode(val) == 409 && count > 0)
                return tryPatch(key, pg, script, count - 1);
            return mkfailure<value>(val);
        }

        // Commit database transaction
        const failable<bool> crc = pgsql::commit(pg);
        if (!hasContent(crc))
            return mkfailure<value>(crc);

        return value(content(val));
    };

    // Try patching the entry and automatically retry a few times on transaction
    // serialization errors
    return tryPatch(car(params), pg, script, 5);
}

/**
 * Delete an item from the database.
 */
const failable<value> del(const list<value>& params, const pgsql::PGSql& pg) {
    const failable<bool> val = pgsql::del(car(params), pg);
    if (!hasContent(val))
        return mkfailure<value>(val);
    return value(content(val));
}

/**
 * Start the component.
 */
const failable<value> start(const list<value>& params) {
    // Connect to the configured database and table
    debug("sqldb::start");
    const gc_pool cp(gc_current_pool());
    const value conninfo = ((lvvlambda)car(params))(nilListValue);
    const value table = ((lvvlambda)cadr(params))(nilListValue);
    const lambda<const gc_ptr<pgsql::PGSql>()> newPGSql = [conninfo, table, cp]() -> const gc_ptr<pgsql::PGSql> {
        debug("sqldb::newPGSql");
        const gc_scoped_pool sp(pool(cp));
        return new (gc_new<pgsql::PGSql>()) pgsql::PGSql(conninfo, table);
    };
    const perthread_ptr<pgsql::PGSql> pg = *(new (gc_new<perthread_ptr<pgsql::PGSql> >()) perthread_ptr<pgsql::PGSql>(newPGSql));

    // Return the component implementation lambda function
    const lvvlambda applySqldb = [pg](const list<value>& params) -> const value {
        const value func(car(params));
        if (func == "get")
            return get(cdr(params), *pg);
        if (func == "post")
            return post(cdr(params), *pg);
        if (func == "put")
            return put(cdr(params), *pg);
        if (func == "patch")
            return patch(cdr(params), *pg);
        if (func == "delete")
            return del(cdr(params), *pg);
        return mkfailure<value>();
    };
    return value(applySqldb);
}

}
}

extern "C" {

const tuscany::value apply(const tuscany::list<tuscany::value>& params) {
    const tuscany::value func(car(params));
    if (func == "start")
        return tuscany::sqldb::start(cdr(params));
    return tuscany::mkfailure<tuscany::value>();
}

}