com.tozny.mobiledemo.UserDAO.java Source code

Introduction

Here is the source code for com.tozny.mobiledemo.UserDAO.java

Source

/*
 * UserDAO.java
 *
 * Copyright (C) 2016, Tozny, LLC.
 * All Rights Reserved.
 *
 * Released under the Apache license. See the file "LICENSE"
 * for more information.
 */

package com.tozny.mobiledemo;

import com.google.common.collect.ImmutableMap;

import java.util.HashMap;
import java.util.Map;

import javax.annotation.Nullable;

/**
 * Interface to simple in-memory database.
 */
public class UserDAO {

    private ImmutableMap<String, User> userMap;

    public UserDAO() {
        userMap = ImmutableMap.of();
    }

    public void insertUser(User user) {
        userMap = new ImmutableMap.Builder<String, User>().putAll(userMap).put(user.getEmail(), user).build();
    }

    @Nullable
    public User findByEmail(String email) {
        return userMap.get(email);
    }

    @Nullable
    public User findByToznyId(String toznyId) {
        if (toznyId == null) {
            return null;
        }
        for (User user : userMap.values()) {
            String userId = user.getToznyId();
            if (userId != null && userId.equals(toznyId)) {
                return user;
            }
        }
        return null;
    }

}