LbaasRepository.java

package io.fathom.cloud.lbaas.state;

import io.fathom.cloud.protobuf.LbaasModel.LbaasMappingData;
import io.fathom.cloud.protobuf.LbaasModel.LbaasServerData;
import io.fathom.cloud.server.model.Project;
import io.fathom.cloud.state.NumberedItemCollection;
import io.fathom.cloud.state.RepositoryBase;
import io.fathom.cloud.state.StateStore.StateNode;

import javax.inject.Singleton;

@Singleton
public class LbaasRepository extends RepositoryBase {
    public NumberedItemCollection<LbaasMappingData> getMappings(Project project) {
        StateNode node = stateStore.getRoot("lb_map");
        node = node.child(Long.toHexString(project.getId()));
        return NumberedItemCollection.builder(node, LbaasMappingData.class).idField(LbaasMappingData.ID_FIELD_NUMBER)
                .create();
    }

    public NumberedItemCollection<LbaasServerData> getServers(Project project) {
        StateNode node = stateStore.getRoot("lb_server");
        node = node.child(Long.toHexString(project.getId()));
        return NumberedItemCollection.builder(node, LbaasServerData.class).idField(LbaasServerData.ID_FIELD_NUMBER)
                .create();
    }

}