//  Copyright 2021 Goldman Sachs
//
//  Licensed under the Apache License, Version 2.0 (the "License");
//  you may not use this file except in compliance with the License.
//  You may obtain a copy of the License at
//
//       http://www.apache.org/licenses/LICENSE-2.0
//
//  Unless required by applicable law or agreed to in writing, software
//  distributed under the License is distributed on an "AS IS" BASIS,
//  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//  See the License for the specific language governing permissions and
//  limitations under the License.
//

package org.finos.legend.depot.store.metrics;

import com.google.inject.PrivateModule;
import com.google.inject.Provides;
import com.google.inject.Singleton;
import org.finos.legend.depot.store.admin.services.schedules.SchedulesFactory;
import org.finos.legend.depot.store.metrics.api.ManageQueryMetrics;
import org.finos.legend.depot.store.metrics.resources.StoreMetricsResource;
import org.finos.legend.depot.store.metrics.store.mongo.MongoQueryMetrics;

import javax.inject.Named;
import java.time.LocalDateTime;

public class MetricsModule extends PrivateModule
{
    @Override
    protected void configure()
    {
        bind(StoreMetricsResource.class);
        bind(ManageQueryMetrics.class).to(MongoQueryMetrics.class);

        expose(ManageQueryMetrics.class);
        expose(StoreMetricsResource.class);
    }


    @Provides
    @Singleton
    @Named("persist-metrics")
    boolean scheduleMetricsPersistence(SchedulesFactory schedulesFactory, ManageQueryMetrics queryMetrics)
    {
        schedulesFactory.register("persist-metrics", LocalDateTime.now().plusSeconds(30), 5 * 60 * 1000, true, () ->
        {
            queryMetrics.persistMetrics();
            return true;
        });
        return true;
    }

}
