TweetGatewayRepositoryInMemory.java

/*
 * Copyright (c) 2014 Martin W. Kirst (nitram509 at bitkings dot de)
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */

package net.nitram509.gateways.repository;

import net.nitram509.gateways.api.Gateway;
import net.nitram509.gateways.api.GatewayId;
import net.nitram509.gateways.api.UserId;
import net.nitram509.gateways.api.UserProfile;

import java.util.*;

public class TweetGatewayRepositoryInMemory implements TweetGatewayRepository {

  private final List<UserProfile> profiles = new ArrayList<>();
  private final List<Gateway> gateways = new ArrayList<>();

  TweetGatewayRepositoryInMemory() {
    // singleton, make constructor private
  }

  @Override
  public void save(UserProfile userProfile) {
    profiles.add(userProfile);
  }

  @Override
  public void save(Gateway gateway) {
    gateways.add(gateway);
  }

  @Override
  public void update(GatewayId gatewayId, String suffix) {
    for (Gateway gateway : gateways) {
      if (gateway.getId().equals(gatewayId)) {
        gateway.setSuffix(suffix);
      }
    }
  }

  @Override
  public UserProfile getUser(UserId userId) {
    for (UserProfile profile : profiles) {
      if (profile.getId().equals(userId)) {
        return profile;
      }
    }
    throw new NoSuchElementException("UserProfile doesn't exists for " + userId);
  }

  @Override
  public List<Gateway> findGateways(UserId owner) {
    List<Gateway> result = new ArrayList<>(3);
    for (Gateway gateway : gateways) {
      if (gateway.getOwner().equals(owner)) {
        result.add(gateway);
      }
    }
    // order by GatewayId
    Collections.sort(result, new Comparator<Gateway>() {
      @Override
      public int compare(Gateway o1, Gateway o2) {
        return o1.compareTo(o2);
      }
    });
    return result;
  }

  @Override
  public Gateway getGateway(GatewayId gatewayId) {
    Gateway result = null;
    for (Gateway gateway : gateways) {
      if (gateway.getId().equals(gatewayId)) {
        result = gateway;
        break;
      }
    }
    return result;
  }

  @Override
  public void incrementActivity(GatewayId gatewayId) {
    Gateway gateway = getGateway(gatewayId);
    int activity = gateway.getActivity();
    gateway.setActivity(activity + 1);
    save(gateway);
  }

  @Override
  public void remove(GatewayId gatewayId) {
    Iterator<Gateway> iterator = gateways.iterator();
    while (iterator.hasNext()) {
      if (gatewayId.equals(iterator.next().getId())) {
        iterator.remove();
        break;
      }
    }
  }
}