2014-07-11 20:49:59 +02:00
|
|
|
// Copyright (c) 2014 Couchbase, Inc.
|
|
|
|
// 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.
|
2014-09-02 16:54:50 +02:00
|
|
|
|
2014-07-30 18:30:38 +02:00
|
|
|
package http
|
2014-07-11 20:49:59 +02:00
|
|
|
|
|
|
|
import (
|
2014-07-30 18:30:38 +02:00
|
|
|
"sync"
|
|
|
|
|
2014-08-28 21:38:57 +02:00
|
|
|
"github.com/blevesearch/bleve"
|
2014-07-11 20:49:59 +02:00
|
|
|
)
|
|
|
|
|
2014-07-30 18:30:38 +02:00
|
|
|
var indexNameMapping map[string]bleve.Index
|
|
|
|
var indexNameMappingLock sync.RWMutex
|
2014-10-02 20:12:22 +02:00
|
|
|
var indexStats = bleve.IndexStats{}
|
2014-07-11 20:49:59 +02:00
|
|
|
|
2014-10-02 20:12:22 +02:00
|
|
|
func RegisterIndexName(name string, idx bleve.Index) {
|
2014-07-30 18:30:38 +02:00
|
|
|
indexNameMappingLock.Lock()
|
|
|
|
defer indexNameMappingLock.Unlock()
|
2014-07-11 20:49:59 +02:00
|
|
|
|
2014-07-30 18:30:38 +02:00
|
|
|
if indexNameMapping == nil {
|
|
|
|
indexNameMapping = make(map[string]bleve.Index)
|
|
|
|
}
|
2014-10-02 20:12:22 +02:00
|
|
|
indexNameMapping[name] = idx
|
|
|
|
indexStats[name] = idx.Stats()
|
2014-07-11 20:49:59 +02:00
|
|
|
}
|
|
|
|
|
2014-08-25 15:08:27 +02:00
|
|
|
func UnregisterIndexByName(name string) bleve.Index {
|
|
|
|
indexNameMappingLock.Lock()
|
|
|
|
defer indexNameMappingLock.Unlock()
|
|
|
|
|
|
|
|
if indexNameMapping == nil {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
rv := indexNameMapping[name]
|
|
|
|
if rv != nil {
|
|
|
|
delete(indexNameMapping, name)
|
|
|
|
}
|
2014-10-02 20:12:22 +02:00
|
|
|
delete(indexStats, name)
|
2014-08-25 15:08:27 +02:00
|
|
|
return rv
|
|
|
|
}
|
|
|
|
|
2014-07-30 18:30:38 +02:00
|
|
|
func IndexByName(name string) bleve.Index {
|
|
|
|
indexNameMappingLock.RLock()
|
|
|
|
defer indexNameMappingLock.RUnlock()
|
|
|
|
|
|
|
|
return indexNameMapping[name]
|
2014-07-11 20:49:59 +02:00
|
|
|
}
|
2014-08-25 15:08:27 +02:00
|
|
|
|
|
|
|
func IndexNames() []string {
|
|
|
|
indexNameMappingLock.RLock()
|
|
|
|
defer indexNameMappingLock.RUnlock()
|
|
|
|
|
|
|
|
rv := make([]string, len(indexNameMapping))
|
|
|
|
count := 0
|
2014-09-04 00:47:02 +02:00
|
|
|
for k := range indexNameMapping {
|
2014-08-25 15:08:27 +02:00
|
|
|
rv[count] = k
|
|
|
|
count++
|
|
|
|
}
|
|
|
|
return rv
|
|
|
|
}
|
2014-10-02 20:12:22 +02:00
|
|
|
|
|
|
|
func IndexStats() bleve.IndexStats {
|
|
|
|
return indexStats
|
|
|
|
}
|