Arakne / Araknemu

 * This file is part of Araknemu.

 *

 * Araknemu is free software: you can redistribute it and/or modify

 * it under the terms of the GNU Lesser General Public License as published by

 * the Free Software Foundation, either version 3 of the License, or

 * (at your option) any later version.

 *

 * Araknemu is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU Lesser General Public License for more details.

 *

 * You should have received a copy of the GNU Lesser General Public License

 * along with Araknemu.  If not, see <https://www.gnu.org/licenses/>.

 *

 * Copyright (c) 2017-2020 Vincent Quatrevieux

 */

package fr.quatrevieux.araknemu.data.world.repository.implementation.sql;

import fr.arakne.utils.value.Interval;

import fr.quatrevieux.araknemu.core.dbal.executor.QueryExecutor;

import fr.quatrevieux.araknemu.core.dbal.repository.EntityNotFoundException;

import fr.quatrevieux.araknemu.core.dbal.repository.Record;

import fr.quatrevieux.araknemu.core.dbal.repository.RepositoryException;

import fr.quatrevieux.araknemu.core.dbal.repository.RepositoryUtils;

import fr.quatrevieux.araknemu.data.world.entity.monster.MonsterRewardData;

import fr.quatrevieux.araknemu.data.world.repository.monster.MonsterRewardRepository;

import fr.quatrevieux.araknemu.util.Asserter;

import java.sql.ResultSet;

import java.sql.SQLException;

import java.util.Arrays;

import java.util.List;

import java.util.Optional;

/**

 * SQL implementation for monster reward repository

 */

final class SqlMonsterRewardRepository implements MonsterRewardRepository {

    private final QueryExecutor executor;

    private final RepositoryUtils<MonsterRewardData> utils;

    public SqlMonsterRewardRepository(QueryExecutor executor) {

        this.executor = executor;

        utils = new RepositoryUtils<>(this.executor, new SqlMonsterRewardRepository.Loader());

    }

    @Override

    public void initialize() throws RepositoryException {

        try {

            executor.query(

                "CREATE TABLE `MONSTER_REWARD` (" +

                    "`MONSTER_ID` INTEGER PRIMARY KEY," +

                    "`MIN_KAMAS` INTEGER," +

                    "`MAX_KAMAS` INTEGER," +

                    "`EXPERIENCES` VARCHAR(200)" +

                ")"

            );

        } catch (SQLException e) {

            throw new RepositoryException(e);

        }

    }

    @Override

    public void destroy() throws RepositoryException {

        try {

            executor.query("DROP TABLE MONSTER_REWARD");

        } catch (SQLException e) {

            throw new RepositoryException(e);

        }

    }

    @Override

    public MonsterRewardData get(MonsterRewardData entity) throws RepositoryException {

        return get(entity.id()).orElseThrow(EntityNotFoundException::new);

    }

    }

    @Override

    public boolean has(MonsterRewardData entity) throws RepositoryException {

        return utils.aggregate(

            "SELECT COUNT(*) FROM MONSTER_REWARD WHERE MONSTER_ID = ?",

            stmt -> stmt.setInt(1, entity.id())

        ) > 0;

    }

    @Override

    public List<MonsterRewardData> all() {

        return utils.findAll("SELECT * FROM MONSTER_REWARD");

    }

    private class Loader implements RepositoryUtils.Loader<MonsterRewardData> {

        @Override

        @SuppressWarnings("argument") // LongStream do not handle @NonNegative parameter

        public MonsterRewardData create(Record record) throws SQLException {

            return new MonsterRewardData(

                record.getInt("MONSTER_ID"),

                new Interval(

                    record.getNonNegativeInt("MIN_KAMAS"),

                    record.getNonNegativeInt("MAX_KAMAS")

                ),

                Arrays.stream(record.getCsvArray("EXPERIENCES", '|'))

                    .mapToLong(Long::parseLong)

                    .map(Asserter::assertNonNegative)

                    .toArray()

            );

        }

        @Override

        public MonsterRewardData fillKeys(MonsterRewardData entity, ResultSet keys) {

            throw new RepositoryException("Read-only entity");

        }

    }

}