/*
    * Copyright (C) 2024 B3Partners B.V.
    *
    * SPDX-License-Identifier: MIT
    */
   package org.tailormap.api.solr;
   
   import io.micrometer.core.instrument.Metrics;
   import java.io.IOException;
  import java.util.concurrent.TimeUnit;
  import org.apache.solr.client.solrj.SolrClient;
  import org.apache.solr.client.solrj.SolrServerException;
  import org.apache.solr.client.solrj.impl.Http2SolrClient;
  import org.apache.solr.client.solrj.response.SolrPingResponse;
  import org.springframework.beans.factory.annotation.Value;
  import org.springframework.stereotype.Service;
  
  @Service
  public class SolrService {
    @Value("${tailormap-api.solr-url}")
    private String solrUrl;
  
    @Value("${tailormap-api.solr-core-name:tailormap}")
    private String solrCoreName;
  
    @Value("${tailormap-api.solr-connection-timeout-seconds:60}")
    private int solrConnectionTimeout;
  
    @Value("${tailormap-api.solr-request-timeout-seconds:240}")
    private int solrRequestTimeout;
  
    @Value("${tailormap-api.solr-idle-timeout-seconds:10}")
    private int solrIdleTimeout;
  
    /**
     * Get a concurrent update Solr client for bulk operations.
     *
     * @return the Solr client
     */
    public SolrClient getSolrClientForIndexing() {
      return new Http2SolrClient.Builder(this.solrUrl + this.solrCoreName)
          .withFollowRedirects(true)
          .withConnectionTimeout(solrConnectionTimeout, TimeUnit.SECONDS)
          .withRequestTimeout(solrRequestTimeout, TimeUnit.SECONDS)
          // Set maxConnectionsPerHost for http1 connections,
          // maximum number http2 connections is limited to 4
          // .withMaxConnectionsPerHost(10)
          .withIdleTimeout(solrIdleTimeout, TimeUnit.SECONDS)
          .build();
    }
  
    /**
     * Get a Solr client for searching.
     *
     * @return the Solr client
     */
    public SolrClient getSolrClientForSearching() {
      return new Http2SolrClient.Builder(this.solrUrl + this.solrCoreName)
          .withConnectionTimeout(solrConnectionTimeout, TimeUnit.SECONDS)
          .withFollowRedirects(true)
          .build();
    }
  
    public void setSolrUrl(String solrUrl) {
      this.solrUrl = solrUrl;
    }
  
    public boolean isSolrServiceAvailable() {
      try (SolrClient client = getSolrClientForIndexing()) {
        SolrPingResponse response = client.ping();
        Metrics.timer("tailormap_solr_ping").record(response.getElapsedTime(), TimeUnit.MILLISECONDS);
        return true;
      } catch (SolrServerException | IOException e) {
        return false;
      }
    }
  }