Как сделать BasicAuthentication в SolrJ - CloudSolrClient? - PullRequest
Новая
       13

Как сделать BasicAuthentication в SolrJ - CloudSolrClient?

0 голосов
/ 28 октября 2018

Я использую Solr 7.5 Server, и я использовал External Zookeeper. Когда я просматриваю, используя Solr Admin UI, он запрашивает у меня аутентификацию.

Для клиента Java я использовал приведенный ниже код 

    BasicAuthSolrClientCache bs = new BasicAuthSolrClientCache("solr", "SolrRocks");

    CloudSolrClient solrCloudClient = bs.getCloudSolrClient(zkHost);
    solrCloudClient.setDefaultCollection("sample");

    SolrInputDocument doc = new SolrInputDocument();
    doc.addField("cat", "book");
    doc.addField("id", "book-1");
    doc.addField("name", "The Legend of the Hobbit part 1");
    solrCloudClient.add(doc);
    solrCloudClient.commit();
    solrCloudClient.close();

BasicAuthSolrClientCache.java 

 public class BasicAuthSolrClientCache extends SolrClientCache {

 private static final Logger log = 
 LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());

 private final Map<String, SolrClient> solrClients = new HashMap<>();
 private final String username;
 private final String password;

 public BasicAuthSolrClientCache(String username, String password) {
 this.username = username;
 this.password = password;
}

@Override
public synchronized CloudSolrClient getCloudSolrClient(String zkHost) {
CloudSolrClient client;
if (solrClients.containsKey(zkHost)) {
  client = (CloudSolrClient) solrClients.get(zkHost);
} else {
  client = new CloudSolrClient.Builder()
      .withZkHost(zkHost)
      .withHttpClient(getHttpClient())
      .build();
  client.connect();
  solrClients.put(zkHost, client);
}

return client;

} 

 @Override
 public synchronized HttpSolrClient getHttpSolrClient(String host) {
  HttpSolrClient client;
  if (solrClients.containsKey(host)) {
   client = (HttpSolrClient) solrClients.get(host);
  } else {
  client = new HttpSolrClient.Builder(host)
      .withHttpClient(getHttpClient())
      .build();
   solrClients.put(host, client);
  }
 return client;
}

@Override
public synchronized void close() {
 for(Map.Entry<String, SolrClient> entry : solrClients.entrySet()) {
  try {
    entry.getValue().close();
  } catch (IOException e) {
    log.error("Error closing SolrClient for " + entry.getKey(), e);
  }
}
solrClients.clear();
}

private HttpClient getHttpClient() {
 CredentialsProvider provider = new BasicCredentialsProvider();
 UsernamePasswordCredentials credentials = new 
 UsernamePasswordCredentials(this.username, this.password);
 provider.setCredentials(AuthScope.ANY, credentials);
 return 


HttpClientBuilder.create().setDefaultCredentialsProvider(provider).
build();
 }
}

Но это дает исключение, как показано ниже,

Исключение в потоке "main"org.apache.solr.

1 Ответ

0 голосов
/ 19 февраля 2019

Я нашел простой способ сделать это.

Вы добавляете такой перехватчик запросов, чтобы вам не пришлось беспокоиться о создании правильно сконфигурированного экземпляра HttpClient самостоятельно.Это просто добавит перехватчик к HttpClient по умолчанию, который создает Solrj. 

org.apache.solr.client.solrj.impl.HttpClientUtil.addRequestInterceptor(new SolrPreemptiveAuthInterceptor());

RequestInterceptor выглядит следующим образом: 

public class SolrPreemptiveAuthInterceptor implements HttpRequestInterceptor {

    final static Logger log = LoggerFactory.getLogger(SolrPreemptiveAuthInterceptor.class); 

    @Override
    public void process(HttpRequest request, HttpContext context) throws HttpException, IOException {
         AuthState authState = (AuthState) context.getAttribute(HttpClientContext.TARGET_AUTH_STATE);
            // If no auth scheme available yet, try to initialize it preemptively
            if (authState.getAuthScheme() == null) {
                log.info("No AuthState: set Basic Auth");

                HttpHost targetHost = (HttpHost) context.getAttribute(HttpCoreContext.HTTP_TARGET_HOST);
                AuthScope authScope = new AuthScope(targetHost.getHostName(), targetHost.getPort());

                CredentialsProvider credsProvider = (CredentialsProvider) context.getAttribute(HttpClientContext.CREDS_PROVIDER);

                Credentials creds = credsProvider.getCredentials(authScope);
                if(creds == null){
                    log.info("No Basic Auth credentials: add them");
                    creds = getCredentials(authScope);
                }
                authState.update(new BasicScheme(), creds);
            }

    }


    private Credentials getCredentials(AuthScope authScope) {

        String user = "";
        String password = "";
        UsernamePasswordCredentials creds = new UsernamePasswordCredentials(user, password);

        CredentialsProvider credsProvider = new BasicCredentialsProvider();
        credsProvider.setCredentials(authScope, creds);
        log.info("Creating Basic Auth credentials for user {}", user);

        return credsProvider.getCredentials(authScope);
    }
}
...