controllers/util/solr_api/v2.go

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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 solr_api

import (
	"bytes"
	"context"
	"encoding/json"
	"fmt"
	solr "github.com/apache/solr-operator/api/v1beta1"
	"io"
	"k8s.io/apimachinery/pkg/api/errors"
	"net/http"
	"net/url"
	"strings"
)

type SolrRebalanceRequest struct {
	Nodes []string `json:"nodes,omitempty"`

	WaitForFinalState bool `json:"waitForFinalState,omitempty"`

	Async string `json:"async,omitempty"`
}

func CallCollectionsApiV2(ctx context.Context, cloud *solr.SolrCloud, urlMethod string, urlPath string, urlParams url.Values, body interface{}, response interface{}) (err error) {
	client := noVerifyTLSHttpClient
	if mTLSHttpClient != nil {
		client = mTLSHttpClient
	}

	cloudUrl := solr.InternalURLForCloud(cloud)
	if !strings.HasPrefix(urlPath, "/") {
		urlPath = "/" + urlPath
	}

	cloudUrl += urlPath
	if len(urlParams) > 0 {
		cloudUrl += "?" + urlParams.Encode()
	}

	resp := &http.Response{}

	var b *bytes.Buffer
	if body != nil {
		b = new(bytes.Buffer)
		if err = json.NewEncoder(b).Encode(body); err != nil {
			return
		}
	}
	var req *http.Request
	if req, err = http.NewRequestWithContext(ctx, urlMethod, cloudUrl, b); err != nil {
		return err
	}

	// Any custom HTTP headers passed through the Context
	if httpHeaders, hasHeaders := ctx.Value(HTTP_HEADERS_CONTEXT_KEY).(map[string]string); hasHeaders {
		for key, header := range httpHeaders {
			req.Header.Add(key, header)
		}
	}
	req.Header.Set("Content-Type", "application/json")
	req.Header.Set("Accept", "application/json")

	if resp, err = client.Do(req); err != nil {
		return err
	}

	defer resp.Body.Close()

	if err == nil && resp.StatusCode >= 400 {
		b, _ := io.ReadAll(resp.Body)
		err = errors.NewServiceUnavailable(fmt.Sprintf("Recieved bad response code of %d from solr with response: %s", resp.StatusCode, string(b)))
		// try to read the response, just in case Solr returned an error that we can read
		json.NewDecoder(bytes.NewReader(b)).Decode(&response)
	}

	if err == nil {
		err = json.NewDecoder(resp.Body).Decode(&response)
	}

	return err
}

func IsNotSupportedApiError(errorBody *SolrErrorResponse) bool {
	return errorBody.Code == 404 &&
		(strings.Contains(errorBody.Message, "Cannot find API for the path") || strings.Contains(errorBody.Message, "no core retrieved for null"))
}