search-algolia/initsearch.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 algolia

import (
	"encoding/json"
	"github.com/algolia/algoliasearch-client-go/v4/algolia/search"
)

// initSettings update algolia search settings
func (s *SearchAlgolia) initSettings() (err error) {
	var (
		settings = search.IndexSettings{}
	)
	err = json.Unmarshal([]byte(AlgoliaSearchServerConfig), &settings)
	if err != nil {
		return
	}

	// point virtual index to sort
	settings.Replicas = []string{
		"virtual(" + s.getIndexName(NewestIndex) + ")",
		"virtual(" + s.getIndexName(ActiveIndex) + ")",
		"virtual(" + s.getIndexName(ScoreIndex) + ")",
	}

	_, err = s.client.SetSettings(s.client.NewApiSetSettingsRequest(s.getIndexName(""), &settings).WithForwardToReplicas(true))
	if err != nil {
		return
	}
	err = s.initVirtualReplicaSetting()
	return
}

// initVirtualReplicaSetting init virtual index replica setting
func (s *SearchAlgolia) initVirtualReplicaSetting() (err error) {
	_, err = s.client.SetSettings(
		s.client.NewApiSetSettingsRequest(
			s.getIndexName(NewestIndex),
			search.NewEmptyIndexSettings().
				SetCustomRanking([]string{
					"desc(created)",
					"desc(content)",
					"desc(title)",
				}),
		),
	)
	if err != nil {
		return
	}

	_, err = s.client.SetSettings(
		s.client.NewApiSetSettingsRequest(
			s.getIndexName(ActiveIndex),
			search.NewEmptyIndexSettings().
				SetCustomRanking([]string{
					"desc(active)",
					"desc(content)",
					"desc(title)",
				}),
		),
	)
	if err != nil {
		return
	}

	_, err = s.client.SetSettings(
		s.client.NewApiSetSettingsRequest(
			s.getIndexName(ScoreIndex),
			search.NewEmptyIndexSettings().
				SetCustomRanking([]string{
					"desc(score)",
					"desc(content)",
					"desc(title)",
				}),
		),
	)
	if err != nil {
		return
	}
	return
}