IM-server-new/broker/src/main/java/io/moquette/persistence/GroupLoader.java

/*
 * This file is part of the Wildfire Chat package.
 * (c) Heavyrain2012 <[email protected]>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

package io.moquette.persistence;

import cn.wildfirechat.proto.WFCMessage;
import com.hazelcast.core.MapLoader;
import com.hazelcast.core.MapStore;
import io.moquette.server.Server;

import java.util.Collection;
import java.util.Map;

public class GroupLoader implements MapStore<String, WFCMessage.GroupInfo> {
    private DatabaseStore getDatabaseStore() {
        return Server.getServer().getStore().messagesStore().getDatabaseStore();
    }

    /**
     * Loads the value of a given key. If distributed map doesn't contain the value
     * for the given key then Hazelcast will call implementation's load (key) method
     * to obtain the value. Implementation can use any means of loading the given key;
     * such as an O/R mapping tool, simple SQL or reading a file etc.
     *
     * @param key@return value of the key, value cannot be null
     */
    @Override
    public WFCMessage.GroupInfo load(String key) {
        return getDatabaseStore().getPersistGroupInfo(key);
    }

    /**
     * Loads given keys. This is batch load operation so that implementation can
     * optimize the multiple loads.
     * <p>
     * For any key in the input keys, there should be a single mapping in the resulting map. Also the resulting
     * map should not have any keys that are not part of the input keys.
     * <p>
     * The given collection should not contain any <code>null</code> keys.
     * The returned Map should not contain any <code>null</code> keys or values.
     *
     * @param keys keys of the values entries to load
     * @return map of loaded key-value pairs.
     */
    @Override
    public Map<String, WFCMessage.GroupInfo> loadAll(Collection<String> keys) {
        return null;
    }


    @Override
    public Iterable<String> loadAllKeys() {
        return null;
    }

    @Override
    public void store(String key, WFCMessage.GroupInfo value) {
        getDatabaseStore().persistGroupInfo(value);
    }

    @Override
    public void storeAll(Map<String, WFCMessage.GroupInfo> map) {
        for (WFCMessage.GroupInfo value : map.values()) {
            getDatabaseStore().persistGroupInfo(value);
        }
    }

    @Override
    public void delete(String key) {
        getDatabaseStore().removeGroupInfoFromDB(key);
    }

    @Override
    public void deleteAll(Collection<String> keys) {

    }
}