io.antmedia.console.datastore.AbstractConsoleDataStore Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of ant-media-server Show documentation

Ant Media Server supports RTMP, RTSP, MP4, HLS, WebRTC, Adaptive Streaming, etc.

There is a newer version: 2.11.3

package io.antmedia.console.datastore;

import io.antmedia.datastore.db.types.User;
import io.antmedia.rest.model.UserType;

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


public abstract class AbstractConsoleDataStore {
	
	public static final String SERVER_STORAGE_FILE = "server.db";
	public static final String SERVER_STORAGE_MAP_NAME = "serverdb";
	
	private Map invalidLoginCountMap = new HashMap<>();

	private Map blockTimeMap = new HashMap<>();

	private Map isBlockedMap = new HashMap<>();

	public abstract boolean addUser(User user);

	public abstract boolean editUser(User user);

	public abstract boolean deleteUser(String username);
	
	public abstract boolean doesUsernameExist(String username);

	public abstract boolean doesUserExist(String username, String password);

	public abstract List getUserList();

	public abstract User getUser(String username);

	public abstract void clear();

	public abstract void close();
	
	public abstract int getNumberOfUserRecords();
	
	/**
	 * Return if data store is available. DataStore is available if it's initialized and not closed. 
	 * It's not available if it's closed. 
	 * @return availability of the datastore
	 */
	public abstract boolean isAvailable();
	
	
	public long getBlockTime(String usermail) {
		return blockTimeMap.containsKey(usermail) ? blockTimeMap.get(usermail) : 0;
	}
	
	public int getInvalidLoginCount(String usermail){
		return  getInvalidLoginCountMap().containsKey(usermail) ? getInvalidLoginCountMap().get(usermail) : 0;
	}
	
	public boolean isUserBlocked(String usermail){
		return getIsBlockedMap().containsKey(usermail);
	}

	public void setBlockTime(String usermail, long blockTime){
		blockTimeMap.put(usermail,blockTime);
	}
	
	public void incrementInvalidLoginCount(String usermail){
		getInvalidLoginCountMap().put(usermail, getInvalidLoginCount(usermail)+1);
	}

	public void resetInvalidLoginCount(String usermail){
		getInvalidLoginCountMap().remove(usermail);
	}
	
	public void setBlocked(String usermail){
		getIsBlockedMap().put(usermail, true);
	}
	
	public void setUnBlocked(String usermail){
		getIsBlockedMap().remove(usermail);
	}

	public Map getIsBlockedMap() {
		return isBlockedMap;
	}

	public Map getInvalidLoginCountMap() {
		return invalidLoginCountMap;
	}
}