package org.ovirt.engine.core.dao.gluster;

import java.util.List;

import javax.inject.Named;
import javax.inject.Singleton;

import org.ovirt.engine.core.common.businessentities.gluster.GlusterService;
import org.ovirt.engine.core.common.businessentities.gluster.ServiceType;
import org.ovirt.engine.core.common.utils.EnumUtils;
import org.ovirt.engine.core.compat.Guid;
import org.ovirt.engine.core.dao.DefaultReadDao;
import org.springframework.jdbc.core.RowMapper;
import org.springframework.jdbc.core.namedparam.MapSqlParameterSource;

@Named
@Singleton
public class GlusterServiceDaoImpl extends DefaultReadDao<GlusterService, Guid> implements GlusterServiceDao {

    private static final RowMapper<GlusterService> serviceRowMapper = (rs, rowNum) -> {
        GlusterService entity = new GlusterService();
        entity.setId(getGuidDefaultEmpty(rs, "id"));
        entity.setServiceType(ServiceType.valueOf(rs.getString("service_type")));
        entity.setServiceName(rs.getString("service_name"));
        return entity;
    };

    public GlusterServiceDaoImpl() {
        super("GlusterService");
    }

    @Override
    protected MapSqlParameterSource createIdParameterMapper(Guid id) {
        return getCustomMapSqlParameterSource().addValue("id", id);
    }

    @Override
    protected RowMapper<GlusterService> createEntityRowMapper() {
        return serviceRowMapper;
    }

    @Override
    public List<GlusterService> getByServiceType(ServiceType type) {
        return getCallsHandler().executeReadList("GetGlusterServicesByType",
                serviceRowMapper, getCustomMapSqlParameterSource().addValue("service_type", EnumUtils.nameOrNull(type)));
    }

    @Override
    public GlusterService getByServiceTypeAndName(ServiceType type, String name) {
        return getCallsHandler().executeRead("GetGlusterServiceByTypeAndName",
                serviceRowMapper,
                getCustomMapSqlParameterSource()
                        .addValue("service_type", EnumUtils.nameOrNull(type))
                        .addValue("service_name", name));
    }
}