com.talanlabs.mybatis.rsql.engine.orderby.registry.DefaultSortDirectionManagerRegistry Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of component-mybatis-rsql Show documentation

The newest version!

package com.talanlabs.mybatis.rsql.engine.orderby.registry;

import com.talanlabs.mybatis.rsql.configuration.IRsqlConfiguration;
import com.talanlabs.mybatis.rsql.engine.orderby.ISortDirectionManager;
import com.talanlabs.mybatis.rsql.engine.orderby.sorts.StandardSortDirectionManager;
import com.talanlabs.mybatis.rsql.sort.SortDirection;
import com.talanlabs.mybatis.rsql.sort.SortDirections;

import java.util.HashMap;
import java.util.Map;

public class DefaultSortDirectionManagerRegistry implements ISortDirectionManagerRegistry {

    private final Map comparisonFactoryMap;

    public DefaultSortDirectionManagerRegistry(IRsqlConfiguration configuration) {
        super();

        StandardSortDirectionManager standardSortDirectionManager = new StandardSortDirectionManager(configuration);

        this.comparisonFactoryMap = new HashMap<>();

        registry(SortDirections.ASC, standardSortDirectionManager);
        registry(SortDirections.ASC_NULLS_FIRST, standardSortDirectionManager);
        registry(SortDirections.ASC_NULLS_LAST, standardSortDirectionManager);
        registry(SortDirections.DESC, standardSortDirectionManager);
        registry(SortDirections.DESC_NULLS_FIRST, standardSortDirectionManager);
        registry(SortDirections.DESC_NULLS_LAST, standardSortDirectionManager);
    }

    /**
     * Registry a sort direction
     *
     * @param sortDirection        direction
     * @param sortDirectionFactory factory
     */
    public void registry(SortDirection sortDirection, ISortDirectionManager sortDirectionFactory) {
        this.comparisonFactoryMap.put(sortDirection, sortDirectionFactory);
    }

    @Override
    public ISortDirectionManager getSortDirectionManager(SortDirection sortDirection) {
        return comparisonFactoryMap.get(sortDirection);
    }
}