| // Copyright 2016 The Vanadium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style |
| // license that can be found in the LICENSE file. |
| |
| package io.v.todos.persistence.syncbase; |
| |
| import android.app.Activity; |
| import android.util.Log; |
| |
| import com.google.common.util.concurrent.Futures; |
| import com.google.common.util.concurrent.ListenableFuture; |
| |
| import java.util.ArrayList; |
| import java.util.HashMap; |
| import java.util.List; |
| import java.util.Map; |
| import java.util.concurrent.Callable; |
| |
| import javax.annotation.Nullable; |
| |
| import io.v.impl.google.services.syncbase.SyncbaseServer; |
| import io.v.todos.model.ListMetadata; |
| import io.v.todos.model.ListSpec; |
| import io.v.todos.model.TaskSpec; |
| import io.v.todos.persistence.ListEventListener; |
| import io.v.todos.persistence.MainPersistence; |
| import io.v.v23.InputChannel; |
| import io.v.v23.InputChannelCallback; |
| import io.v.v23.InputChannels; |
| import io.v.v23.VFutures; |
| import io.v.v23.services.syncbase.BatchOptions; |
| import io.v.v23.services.syncbase.KeyValue; |
| import io.v.v23.syncbase.Batch; |
| import io.v.v23.syncbase.BatchDatabase; |
| import io.v.v23.syncbase.ChangeType; |
| import io.v.v23.syncbase.Collection; |
| import io.v.v23.syncbase.RowRange; |
| import io.v.v23.syncbase.WatchChange; |
| import io.v.v23.vdl.VdlAny; |
| import io.v.v23.verror.VException; |
| import io.v.v23.vom.VomUtil; |
| |
| public class SyncbaseMain extends SyncbasePersistence implements MainPersistence { |
| private static final String |
| TAG = SyncbaseMain.class.getSimpleName(), |
| LISTS_PREFIX = "lists_"; |
| |
| private final IdGenerator mIdGenerator = new IdGenerator(IdAlphabets.COLLECTION_ID, true); |
| private final Map<String, MainListTracker> mTaskTrackers = new HashMap<>(); |
| |
| /** |
| * This constructor blocks until the instance is ready for use. |
| */ |
| public SyncbaseMain(Activity activity, final ListEventListener<ListMetadata> listener) |
| throws VException, SyncbaseServer.StartException { |
| super(activity); |
| |
| InputChannel<WatchChange> watch = getDatabase().watch( |
| mVContext, getUserCollection().id(), LISTS_PREFIX); |
| trap(InputChannels.withCallback(watch, new InputChannelCallback<WatchChange>() { |
| @Override |
| public ListenableFuture<Void> onNext(WatchChange change) { |
| String listId = change.getRowName(); |
| |
| if (change.getChangeType() == ChangeType.DELETE_CHANGE) { |
| trap(mTaskTrackers.remove(listId).deleteList(mVContext)); |
| } else { |
| mIdGenerator.registerId(change.getRowName().substring(LISTS_PREFIX.length())); |
| |
| MainListTracker listTracker = new MainListTracker( |
| mVContext, getDatabase(), listId, listener); |
| if (mTaskTrackers.put(listId, listTracker) != null) { |
| // List entries in the main collection are just ( list ID => nil ), so we |
| // never expect updates other than an initial add... |
| Log.w(TAG, "Unexpected update to " + USER_COLLECTION_NAME + " collection " + |
| "for list " + listId); |
| } |
| |
| trap(listTracker.watchFuture); |
| } |
| return null; |
| } |
| })); |
| } |
| |
| @Override |
| public void addTodoList(final ListSpec listSpec) { |
| final String listName = LISTS_PREFIX + mIdGenerator.generateTailId(); |
| final Collection listCollection = getDatabase().getCollection(mVContext, listName); |
| Futures.addCallback(listCollection.create(mVContext, null), |
| new TrappingCallback<Void>(mActivity) { |
| @Override |
| public void onSuccess(@Nullable Void result) { |
| // These can happen in either order |
| trap(getUserCollection().put(mVContext, listName, null, VdlAny.class)); |
| trap(listCollection.put(mVContext, SyncbaseTodoList.LIST_METADATA_ROW_NAME, |
| listSpec, ListSpec.class)); |
| } |
| }); |
| } |
| |
| @Override |
| public void deleteTodoList(String key) { |
| trap(getUserCollection().delete(mVContext, key)); |
| } |
| |
| @Override |
| public void setCompletion(ListMetadata listMetadata, final boolean done) { |
| final String listId = listMetadata.key; |
| trap(Batch.runInBatch(mVContext, getDatabase(), new BatchOptions(), |
| new Batch.BatchOperation() { |
| @Override |
| public ListenableFuture<Void> run(final BatchDatabase db) { |
| return sExecutor.submit(new Callable<Void>() { |
| @Override |
| public Void call() throws Exception { |
| final Collection list = db.getCollection(mVContext, listId); |
| |
| InputChannel<KeyValue> scan = list.scan(mVContext, |
| RowRange.prefix(SyncbaseTodoList.TASKS_PREFIX)); |
| |
| List<ListenableFuture<Void>> puts = new ArrayList<>(); |
| for (KeyValue kv : InputChannels.asIterable(scan)) { |
| TaskSpec taskSpec = (TaskSpec) VomUtil.decode(kv.getValue()); |
| if (taskSpec.getDone() != done) { |
| taskSpec.setDone(done); |
| puts.add(list.put(mVContext, kv.getKey(), taskSpec, |
| TaskSpec.class)); |
| } |
| } |
| |
| if (!puts.isEmpty()) { |
| puts.add(SyncbaseTodoList.updateListTimestamp(mVContext, list)); |
| } |
| VFutures.sync(Futures.allAsList(puts)); |
| return null; |
| } |
| }); |
| } |
| })); |
| } |
| } |