summaryrefslogtreecommitdiff
path: root/src/com/p4square/grow/backend/CassandraGrowData.java
blob: 22a771676af9ae4c6da9c573f99b76536bd98a4b (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
/*
 * Copyright 2014 Jesse Morgan
 */

package com.p4square.grow.backend;

import java.io.IOException;

import com.p4square.grow.config.Config;

import com.p4square.grow.backend.db.CassandraDatabase;
import com.p4square.grow.backend.db.CassandraKey;
import com.p4square.grow.backend.db.CassandraProviderImpl;
import com.p4square.grow.backend.db.CassandraCollectionProvider;
import com.p4square.grow.backend.db.CassandraTrainingRecordProvider;

import com.p4square.grow.model.Message;
import com.p4square.grow.model.MessageThread;
import com.p4square.grow.model.Playlist;
import com.p4square.grow.model.Question;
import com.p4square.grow.model.TrainingRecord;
import com.p4square.grow.model.UserRecord;

import com.p4square.grow.provider.CollectionProvider;
import com.p4square.grow.provider.DelegateCollectionProvider;
import com.p4square.grow.provider.DelegateProvider;
import com.p4square.grow.provider.Provider;

/**
 *
 * @author Jesse Morgan <jesse@jesterpm.net>
 */
class CassandraGrowData implements GrowData {
    private static final String DEFAULT_COLUMN = "value";

    private final Config mConfig;
    private final CassandraDatabase mDatabase;

    private final Provider<String, UserRecord> mUserRecordProvider;

    private final Provider<String, Question> mQuestionProvider;
    private final CassandraTrainingRecordProvider mTrainingRecordProvider;
    private final CollectionProvider<String, String, String> mVideoProvider;

    private final CollectionProvider<String, String, MessageThread> mFeedThreadProvider;
    private final CollectionProvider<String, String, Message> mFeedMessageProvider;

    private final Provider<String, String> mStringProvider;

    private final CollectionProvider<String, String, String> mAnswerProvider;

    public CassandraGrowData(final Config config) {
        mConfig = config;
        mDatabase = new CassandraDatabase();

        mUserRecordProvider = new DelegateProvider<String, CassandraKey, UserRecord>(
                new CassandraProviderImpl<UserRecord>(mDatabase, UserRecord.class)) {
            @Override
            public CassandraKey makeKey(String userid) {
                return new CassandraKey("accounts", userid, DEFAULT_COLUMN);
            }
        };

        mQuestionProvider = new DelegateProvider<String, CassandraKey, Question>(
                new CassandraProviderImpl<Question>(mDatabase, Question.class)) {
            @Override
            public CassandraKey makeKey(String questionId) {
                return new CassandraKey("strings", "/questions/" + questionId, DEFAULT_COLUMN);
            }
        };

        mFeedThreadProvider = new CassandraCollectionProvider<MessageThread>(mDatabase,
                "feedthreads", MessageThread.class);
        mFeedMessageProvider = new CassandraCollectionProvider<Message>(mDatabase,
                "feedmessages", Message.class);

        mTrainingRecordProvider = new CassandraTrainingRecordProvider(mDatabase);

        mVideoProvider = new DelegateCollectionProvider<String, String, String, String, String>(
                new CassandraCollectionProvider<String>(mDatabase, "strings", String.class)) {
            @Override
            public String makeCollectionKey(String key) {
                return "/training/" + key;
            }

            @Override
            public String makeKey(String key) {
                return key;
            }

            @Override
            public String unmakeKey(String key) {
                return key;
            }
        };

        mStringProvider = new DelegateProvider<String, CassandraKey, String>(
                new CassandraProviderImpl<String>(mDatabase, String.class)) {
            @Override
            public CassandraKey makeKey(String id) {
                return new CassandraKey("strings", id, DEFAULT_COLUMN);
            }
        };

        mAnswerProvider = new CassandraCollectionProvider<String>(
                mDatabase, "assessments", String.class);
    }

    @Override
    public void start() throws Exception {
        mDatabase.setClusterName(mConfig.getString("clusterName", "Dev Cluster"));
        mDatabase.setKeyspaceName(mConfig.getString("keyspace", "GROW"));
        mDatabase.init();
    }

    @Override
    public void stop() throws Exception {
        mDatabase.close();
    }

    /**
     * @return the current database.
     */
    public CassandraDatabase getDatabase() {
        return mDatabase;
    }

    @Override
    public Provider<String, UserRecord> getUserRecordProvider() {
        return mUserRecordProvider;
    }

    @Override
    public Provider<String, Question> getQuestionProvider() {
        return mQuestionProvider;
    }

    @Override
    public Provider<String, TrainingRecord> getTrainingRecordProvider() {
        return mTrainingRecordProvider;
    }

    @Override
    public CollectionProvider<String, String, String> getVideoProvider() {
        return mVideoProvider;
    }

    @Override
    public Playlist getDefaultPlaylist() throws IOException {
        return mTrainingRecordProvider.getDefaultPlaylist();
    }

    @Override
    public CollectionProvider<String, String, MessageThread> getThreadProvider() {
        return mFeedThreadProvider;
    }

    @Override
    public CollectionProvider<String, String, Message> getMessageProvider() {
        return mFeedMessageProvider;
    }

    @Override
    public Provider<String, String> getStringProvider() {
        return mStringProvider;
    }

    @Override
    public CollectionProvider<String, String, String> getAnswerProvider() {
        return mAnswerProvider;
    }
}