package ru.ensemplix.storage;

import com.avaje.ebean.Ebean;
import com.avaje.ebean.Query;
import lombok.AllArgsConstructor;

import java.util.Map;

@AllArgsConstructor
public class DatabaseStorage<V> implements Storage<V> {

    private Class cls;

    private String server;

    public DatabaseStorage(Class cls) {
        this.cls = cls;
    }

    @Override
    public void add(V value) {
        Ebean.save(value);
    }

    @Override
    public void remove(V value) {
        Ebean.delete(value);
    }

    @Override
    @SuppressWarnings("unchecked")
    public Map<String, V> get(String key) {
        Query query = Ebean.find(cls);
        query.setMapKey(key);

        if(server != null) {
            query.where().eq("server", server);
        }

        return query.findMap();
    }
}