-
Notifications
You must be signed in to change notification settings - Fork 1
/
schemabuilder.go
103 lines (95 loc) · 2.76 KB
/
schemabuilder.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
package tantivy_go
//#include "bindings.h"
import "C"
import (
"errors"
)
type (
SchemaBuilder struct{ ptr *C.SchemaBuilder }
Schema struct{ ptr *C.Schema }
)
const (
// IndexRecordOptionBasic specifies that only basic indexing information should be used.
IndexRecordOptionBasic = iota
// IndexRecordOptionWithFreqs specifies that indexing should include term frequencies.
IndexRecordOptionWithFreqs
// IndexRecordOptionWithFreqsAndPositions specifies that indexing should include term frequencies and term positions.
IndexRecordOptionWithFreqsAndPositions
)
const DefaultTokenizer = "default"
const (
Arabic = "ar"
Danish = "da"
Dutch = "nl"
English = "en"
Finnish = "fi"
French = "fr"
German = "de"
Greek = "el"
Hungarian = "hu"
Italian = "it"
Norwegian = "no"
Portuguese = "pt"
Romanian = "ro"
Russian = "ru"
Spanish = "es"
Swedish = "sv"
Tamil = "ta"
Turkish = "tr"
)
// NewSchemaBuilder creates a new SchemaBuilder instance.
// Returns a pointer to the SchemaBuilder and an error if creation fails.
func NewSchemaBuilder() (*SchemaBuilder, error) {
ptr := C.schema_builder_new()
if ptr == nil {
return nil, errors.New("failed to create schema builder")
}
return &SchemaBuilder{ptr: ptr}, nil
}
// AddTextField adds a text field to the schema being built.
//
// Parameters:
// - name: The name of the field.
// - stored: Whether the field should be stored in the index.
// - isText: Whether the field should be treated as tantivy text or string for full-text search.
// - isFast: Whether the field should be indexed as tantivy quick field.
// - indexRecordOption: The indexing option to be used (e.g., basic, with frequencies, with frequencies and positions).
// - tokenizer: The name of the tokenizer to be used for the field.
//
// Returns an error if the field could not be added.
func (b *SchemaBuilder) AddTextField(
name string,
stored bool,
isText bool,
isFast bool,
indexRecordOption int,
tokenizer string,
) error {
cName := C.CString(name)
cTokenizer := C.CString(tokenizer)
defer C.string_free(cName)
defer C.string_free(cTokenizer)
var errBuffer *C.char
C.schema_builder_add_text_field(
b.ptr,
cName,
C._Bool(stored),
C._Bool(isText),
C._Bool(isFast),
pointerCType(indexRecordOption),
cTokenizer,
&errBuffer,
)
return tryExtractError(errBuffer)
}
// BuildSchema finalizes the schema building process and returns the resulting Schema.
// Returns a pointer to the Schema and an error if the schema could not be built.
func (b *SchemaBuilder) BuildSchema() (*Schema, error) {
var errBuffer *C.char
ptr := C.schema_builder_build(b.ptr, &errBuffer)
if ptr == nil {
defer C.string_free(errBuffer)
return nil, errors.New(C.GoString(errBuffer))
}
return &Schema{ptr: ptr}, nil
}