nl.knaw.huygens.timbuctoo.storage.file.UserFileCollection.java Source code

Introduction

Here is the source code for nl.knaw.huygens.timbuctoo.storage.file.UserFileCollection.java

Source

package nl.knaw.huygens.timbuctoo.storage.file;

/*
 * #%L
 * Timbuctoo core
 * =======
 * Copyright (C) 2012 - 2015 Huygens ING
 * =======
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as
 * published by the Free Software Foundation, either version 3 of the 
 * License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public 
 * License along with this program.  If not, see
 * <http://www.gnu.org/licenses/gpl-3.0.html>.
 * #L%
 */

import java.util.LinkedList;
import java.util.List;
import java.util.Map;

import nl.knaw.huygens.timbuctoo.model.User;
import nl.knaw.huygens.timbuctoo.storage.StorageIterator;
import nl.knaw.huygens.timbuctoo.storage.StorageIteratorStub;

import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
import com.fasterxml.jackson.databind.annotation.JsonSerialize;
import com.google.common.collect.Lists;
import com.google.common.collect.Maps;

@JsonSerialize(using = FileCollectionSerializer.class)
@JsonDeserialize(using = UserFileCollectionDeserializer.class)
public class UserFileCollection extends FileCollection<User> {

    private final Map<String, User> idUserMap;
    private final Map<String, String> persistentIdIdMap;

    public UserFileCollection() {
        this(Lists.<User>newArrayList());
    }

    public UserFileCollection(List<User> users) {
        // I'm not sure if this is needed, better safe than sorry.
        idUserMap = Maps.newConcurrentMap();
        persistentIdIdMap = Maps.newConcurrentMap();
        initialize(users);
    }

    private void initialize(List<User> users) {
        for (User user : users) {
            String id = user.getId();
            idUserMap.put(id, user);
            indexPersistenId(id, user.getPersistentId());
        }
    }

    @Override
    public String add(User user) {
        String id = createId(User.ID_PREFIX);
        user.setId(id);

        String persistentId = user.getPersistentId();
        if (persistentId != null && persistentIdIdMap.containsKey(persistentId)) {
            return persistentIdIdMap.get(persistentId);
        }

        idUserMap.put(id, user);

        indexPersistenId(id, persistentId);

        return id;
    }

    private void indexPersistenId(String id, String persistentId) {
        if (persistentId != null) {
            persistentIdIdMap.put(persistentId, id);
        }
    }

    @Override
    protected LinkedList<String> getIds() {
        return Lists.newLinkedList(idUserMap.keySet());
    }

    /**
     * Find the user by persistentId.
     * 
     * @param user
     *        the user that contains the persistentId
     * @return the user when found, null if the user has no persistent id.
     */
    @Override
    public User findItem(User user) {
        if (user == null || user.getPersistentId() == null) {
            return null;
        }
        String persistentId = user.getPersistentId();
        String id = persistentIdIdMap.get(persistentId);
        return this.get(id);
    }

    @Override
    public User get(String id) {
        return id != null ? idUserMap.get(id) : null;
    }

    @Override
    public StorageIterator<User> getAll() {
        return StorageIteratorStub.newInstance(Lists.newArrayList(idUserMap.values()));
    }

    @Override
    public void updateItem(User item) {
        if (item.getId() != null) {
            idUserMap.put(item.getId(), item);
        }
    }

    @Override
    public void deleteItem(User item) {
        if (item.getId() != null) {
            idUserMap.remove(item.getId());
        }
    }

    @Override
    public User[] asArray() {
        return idUserMap.values().toArray(new User[] {});
    }

}