summaryrefslogtreecommitdiffstats
path: root/sca-cpp/trunk/components/cache/partitioner.cpp
blob: 8469456fc86ae4c6e9793b81b3c656c66cd32e64 (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
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
/*
 * 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$ */

/**
 * A partitioner component implementation which forwards data access requests to a
 * dynamically selected data store component. The selection is externalized, performed
 * by a selector component, responsible for selecting the target data stores given the
 * data access request key and a list of references to available data store components.
 * This pattern can be used for sharding or load balancing for example.
 */

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

namespace tuscany {
namespace partitioner {

/**
 * Return a list of target partitions for a key.
 */
const failable<list<value> > partition(const value& key, const lvvlambda& selector, const list<value>& partitions) {

    // Call the selector component to convert the given key to a list of partitions
    const value p = selector(mklist<value>("get", key, partitions));
    if (isNull(p)) {
        ostringstream os;
        os << "Couldn't get partition: " << key;
        return mkfailure<list<value> >(str(os), -1, false);
    }
    return (list<value>)p;
}


/**
 * Get lists of items from a list of partitions.
 */
const failable<list<value> > getlist(const value& key, const list<value>& partitions) {
    if (isNull(partitions))
        return nilListValue;

    const lvvlambda l = car(partitions);
    const value val = l(mklist<value>("get", key));
    if (isNull(val))
        return getlist(key, cdr(partitions));
    if (!isList(val)) {
        ostringstream os;
        os << "Couldn't get list of entries from partition: " << key;
        return mkfailure<list<value> >(str(os), 500, false);
    }

    const failable<list<value> > cdrval = getlist(key, cdr(partitions));
    if (!hasContent(cdrval))
        return cdrval;

    return append<value>((list<value>)val, content(cdrval));
}

/**
 * Return the rank of a result item if available.
 */
const value itemrank(const value& val) {
    if (isList(val) && !isNull(val)) {
        const value e = car<value>(val);
        if (isList(e) && !isNull(e)) {
            if (car<value>(e) == "entry") {
                const list<value> ae = cdr<value>(e);
                const list<value> li = assoc<value>("id", ae);
                const list<value> lr = assoc<value>("rank", ae);
                if (!isNull(li) && !isNull(lr))
                    return mklist<value>(lr, li);
            }
        }
    }
    return val;
}

/**
 * Compare the ranks of two result items.
 */
const int rankitems(const value& a, const value& b) {
    const value ra = itemrank(a);
    const value rb = itemrank(b);
    if (ra == rb)
        return 0;
    if (ra < rb)
        return -1;
    return 1;
}

/**
 * Convert a key to a (param name, value) assoc.
 */
const list<value> keyparams(const list<value>& key) {
    if (isNull(key))
        return nilListValue;
    if (!isList(car(key)))
        return keyparams(cdr(key));
    return cons<value>(car(key), keyparams(cdr(key)));
}

/**
 * Get an item from a partition.
 */
const failable<value> get(const value& key, const lvvlambda& selector, const list<value>& partitions) {
    debug(key, "partitioner::get::key");

    // Select partition
    const failable<list<value> > fp = partition(key, selector, partitions);
    if (!hasContent(fp))
        return mkfailure<value>(fp);
    const list<value> p = content(fp);

    // Get a single item from the selected partition
    if (length(p) == 1) {
        const lvvlambda l = car(p);
        const value val = l(mklist<value>("get", key));
        if (isNull(val)) {
            ostringstream os;
            os << "Couldn't get entry from partition: " << key;
            return mkfailure<value>(str(os), 404, false);
        }
        debug(val, "partitioner::get::val");
        return val;
    }

    // Get list of items from the list of selected partitions
    const failable<list<value> > val = getlist(key, p);
    if (!hasContent(val))
        return mkfailure<value>(val);
    const list<value> cval = content(val);
    debug(cval, "partitioner::get::cval");

    // Apply any ranking sort and result limit
    const list<value> kparams = isList(key)? keyparams(key) : nilListValue;
    const list<value> sval = (!isNull(assoc<value>("textsearch", kparams)) || !isNull(assoc<value>("rank", kparams)))? reverse<value>(sort<value>(cval, rankitems)) : cval;
    const list<value> limit = assoc<value>("limit", kparams);
    const list<value> lval = !isNull(limit)? listHead(sval, (int)cadr(limit)) : sval;
    debug(lval, "partitioner::get::lval");

    return (value)lval;
}

/**
 * Post an item to a partition.
 */
const failable<value> post(const value& key, const value& val, const lvvlambda& selector, const list<value>& partitions) {
    const value id = append<value>(key, mklist(mkuuid()));

    // Select partition
    const failable<list<value> > p = partition(id, selector, partitions);
    if (!hasContent(p))
        return mkfailure<value>(p);

    // Put into select partition
    const lvvlambda l = car(content(p));
    l(mklist<value>("post", id, val));

    return id;
}

/**
 * Put an item into a partition.
 */
const failable<value> put(const value& key, const value& val, const lvvlambda& selector, const list<value>& partitions) {

    // Select partition
    const failable<list<value> > p = partition(key, selector, partitions);
    if (!hasContent(p))
        return mkfailure<value>(p);

    // Put into selected partition
    const lvvlambda l = car(content(p));
    l(mklist<value>("put", key, val));

    return trueValue;
}

/**
 * Patch an item in a partition.
 */
const failable<value> patch(const value& key, const value& val, const lvvlambda& selector, const list<value>& partitions) {

    // Select partition
    const failable<list<value> > p = partition(key, selector, partitions);
    if (!hasContent(p))
        return mkfailure<value>(p);

    // Path item in selected partition
    const lvvlambda l = car(content(p));
    l(mklist<value>("patch", key, val));

    return trueValue;
}

/**
 * Delete an item from a partition.
 */
const failable<value> del(const value& key, const lvvlambda& selector, const list<value>& partitions) {

    // Select partition
    const failable<list<value> > p = partition(key, selector, partitions);
    if (!hasContent(p))
        return mkfailure<value>(p);

    // Delete from selected partition
    const lvvlambda l = car(content(p));
    l(mklist<value>("delete", key));

    return trueValue;
}

}
}

extern "C" {

const tuscany::value apply(const tuscany::list<tuscany::value>& params) {
    const tuscany::value func(car(params));
    if (func == "get")
        return tuscany::partitioner::get(cadr(params), caddr(params), cdddr(params));
    if (func == "post")
        return tuscany::partitioner::post(cadr(params), caddr(params), cadddr(params), cddddr(params));
    if (func == "put")
        return tuscany::partitioner::put(cadr(params), caddr(params), cadddr(params), cddddr(params));
    if (func == "patch")
        return tuscany::partitioner::patch(cadr(params), caddr(params), cadddr(params), cddddr(params));
    if (func == "delete")
        return tuscany::partitioner::del(cadr(params), caddr(params), cdddr(params));
    return tuscany::mkfailure<tuscany::value>();
}

}