← Back to branch summary

~ubuntu-branches/debian/sid/golang-github-blevesearch-bleve/sid

« back to all changes in this revision

Viewing changes to registry/token_filter.go

  • Committer: Package Import Robot
  • Author(s): Michael Lustfield
  • Date: 2017-03-30 16:06:03 UTC
  • Revision ID: package-import@ubuntu.com-20170330160603-0oogmb960l7918jx
Tags: upstream-0.5.0+git20170324.202.4702785f
Import upstream version 0.5.0+git20170324.202.4702785f

Show diffs side-by-side

added added

removed removed

Lines of Context:
registry/token_filter.go
 
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
 
 
15
package registry
 
16
 
 
17
import (
 
18
        "fmt"
 
19
 
 
20
        "github.com/blevesearch/bleve/analysis"
 
21
)
 
22
 
 
23
func RegisterTokenFilter(name string, constructor TokenFilterConstructor) {
 
24
        _, exists := tokenFilters[name]
 
25
        if exists {
 
26
                panic(fmt.Errorf("attempted to register duplicate token filter named '%s'", name))
 
27
        }
 
28
        tokenFilters[name] = constructor
 
29
}
 
30
 
 
31
type TokenFilterConstructor func(config map[string]interface{}, cache *Cache) (analysis.TokenFilter, error)
 
32
type TokenFilterRegistry map[string]TokenFilterConstructor
 
33
 
 
34
type TokenFilterCache struct {
 
35
        *ConcurrentCache
 
36
}
 
37
 
 
38
func NewTokenFilterCache() *TokenFilterCache {
 
39
        return &TokenFilterCache{
 
40
                NewConcurrentCache(),
 
41
        }
 
42
}
 
43
 
 
44
func TokenFilterBuild(name string, config map[string]interface{}, cache *Cache) (interface{}, error) {
 
45
        cons, registered := tokenFilters[name]
 
46
        if !registered {
 
47
                return nil, fmt.Errorf("no token filter with name or type '%s' registered", name)
 
48
        }
 
49
        tokenFilter, err := cons(config, cache)
 
50
        if err != nil {
 
51
                return nil, fmt.Errorf("error building token filter: %v", err)
 
52
        }
 
53
        return tokenFilter, nil
 
54
}
 
55
 
 
56
func (c *TokenFilterCache) TokenFilterNamed(name string, cache *Cache) (analysis.TokenFilter, error) {
 
57
        item, err := c.ItemNamed(name, cache, TokenFilterBuild)
 
58
        if err != nil {
 
59
                return nil, err
 
60
        }
 
61
        return item.(analysis.TokenFilter), nil
 
62
}
 
63
 
 
64
func (c *TokenFilterCache) DefineTokenFilter(name string, typ string, config map[string]interface{}, cache *Cache) (analysis.TokenFilter, error) {
 
65
        item, err := c.DefineItem(name, typ, config, cache, TokenFilterBuild)
 
66
        if err != nil {
 
67
                if err == ErrAlreadyDefined {
 
68
                        return nil, fmt.Errorf("token filter named '%s' already defined", name)
 
69
                }
 
70
                return nil, err
 
71
        }
 
72
        return item.(analysis.TokenFilter), nil
 
73
}
 
74
 
 
75
func TokenFilterTypesAndInstances() ([]string, []string) {
 
76
        emptyConfig := map[string]interface{}{}
 
77
        emptyCache := NewCache()
 
78
        var types []string
 
79
        var instances []string
 
80
        for name, cons := range tokenFilters {
 
81
                _, err := cons(emptyConfig, emptyCache)
 
82
                if err == nil {
 
83
                        instances = append(instances, name)
 
84
                } else {
 
85
                        types = append(types, name)
 
86
                }
 
87
        }
 
88
        return types, instances
 
89
}