1

     2

#include <stdlib.h>

     3

     4

#include "memcache.h"

     5

#include "server.h"

     6

#include "../common.h"

     7

     8

struct memcache *memcache_alloc (memcache_cb cb_fn) {

     9

    struct memcache *mc = NULL;

    10

    11

    if ((mc = calloc(1, sizeof(*mc))) == NULL)

    12

        ERROR("calloc");

    13

    14

    // store callback

    15

    mc->cb_fn = cb_fn;

    16

    17

    // init server list

    18

    LIST_INIT(&mc->server_list);

    19

    20

    // success

    21

    return mc;

    22

    23

error:

    24

    if (mc)

    25

        free(mc);

    26

    27

    return NULL;

    28

}

    29

    30

int memcache_add_server (struct memcache *mc, struct config_endpoint *endpoint, int max_connections) {

    31

    struct memcache_server *server = NULL;

    32

    33

    // alloc the server

    34

    if ((server = memcache_server_alloc(endpoint, max_connections)) == NULL)

    35

        goto error;

    36

    37

    // enlist it

    38

    LIST_INSERT_HEAD(&mc->server_list, server, serverlist_node);

    39

    40

    // done

    41

    return 0;

    42

    43

error:

    44

    return -1;

    45

}

    46

    47

struct memcache_server *memcache_choose_server (struct memcache *mc, const struct memcache_key *key) {

    48

    // XXX: support multiple servers

    49

    return mc->server_list.lh_first;

    50

}

    51

    52

struct memcache_req *memcache_fetch (struct memcache *mc, const struct memcache_key *key, void *cb_arg) {

    53

    struct memcache_req *req = NULL;

    54

    struct memcache_server *server = NULL;

    55

    56

    // alloc the request

    57

    if ((req = memcache_req_alloc(key, cb_arg)) == NULL)

    58

        ERROR("failed to allocate request");

    59

    60

    // pick a server

    61

    if ((server = memcache_choose_server(mc, key)) == NULL)

    62

        ERROR("failed to find a server to use");

    63

    64

    // enqueue it!

    65

    if (memcache_server_add_req(server, req))

    66

        ERROR("failed to hand over the request for processing");

    67

    68

    // success!

    69

    return 0;

    70

    71

error:

    72

    if (req)

    73

        memcache_req_free(req);

    74

    75

    return -1;

    76

}  

    77