本站源代码
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

52 lines
1.7KB

  1. // Copyright (c) 2014 Couchbase, Inc.
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. package registry
  15. import (
  16. "fmt"
  17. "github.com/blevesearch/bleve/index/store"
  18. )
  19. func RegisterKVStore(name string, constructor KVStoreConstructor) {
  20. _, exists := stores[name]
  21. if exists {
  22. panic(fmt.Errorf("attempted to register duplicate store named '%s'", name))
  23. }
  24. stores[name] = constructor
  25. }
  26. // KVStoreConstructor is used to build a KVStore of a specific type when
  27. // specificied by the index configuration. In addition to meeting the
  28. // store.KVStore interface, KVStores must also support this constructor.
  29. // Note that currently the values of config must
  30. // be able to be marshaled and unmarshaled using the encoding/json library (used
  31. // when reading/writing the index metadata file).
  32. type KVStoreConstructor func(mo store.MergeOperator, config map[string]interface{}) (store.KVStore, error)
  33. type KVStoreRegistry map[string]KVStoreConstructor
  34. func KVStoreConstructorByName(name string) KVStoreConstructor {
  35. return stores[name]
  36. }
  37. func KVStoreTypesAndInstances() ([]string, []string) {
  38. var types []string
  39. var instances []string
  40. for name := range stores {
  41. types = append(types, name)
  42. }
  43. return types, instances
  44. }
上海开阖软件有限公司 沪ICP备12045867号-1