ston1th/docstore
1
1
Fork 0
Code Issues 3 Releases Activity

simplify tags

Browse source
This commit is contained in:
ston1th 2019-08-25 12:34:10 +02:00
commit 3c5be25cb3
6 changed files with 28 additions and 31 deletions
Download patch file Download diff file Expand all files Collapse all files

15
pkg/db/index.go
View file

@ -6,7 +6,6 @@ import (
"errors"
"git.giftfish.de/ston1th/docstore/pkg/core"
"git.giftfish.de/ston1th/docstore/pkg/log"
"git.giftfish.de/ston1th/docstore/pkg/tags"
)
const (
@ -20,9 +19,9 @@ const (
pathPrefix = "path/"
)
func (db *DB) Reindex(path string, match []core.RTag) (err error) {
if match == nil {
match, err = db.GetAllRTags()
func (db *DB) Reindex(path string, tags core.RTags) (err error) {
if tags == nil {
tags, err = db.GetAllRTags()
if err != nil {
return
}
@ -36,7 +35,7 @@ func (db *DB) Reindex(path string, match []core.RTag) (err error) {
if err != nil {
return
}
doc.Tags = tags.Find(doc.Text, match)
doc.Tags = tags.Match(doc.Text)
err = db.Index.Reindex(id, doc)
if err != nil {
return
@ -53,12 +52,12 @@ func (db *DB) ReindexAll() (err error) {
if err != nil {
return
}
match, err := db.GetAllRTags()
tags, err := db.GetAllRTags()
if err != nil {
return
}
for _, p := range paths {
e := db.Reindex(p, match)
e := db.Reindex(p, tags)
if e != nil {
if err == nil {
err = errors.New("reindexAll had errors")
@ -69,7 +68,7 @@ func (db *DB) ReindexAll() (err error) {
return
}
func (db *DB) GetPaths(res []core.Result) {
func (db *DB) GetPaths(res core.Results) {
for i, v := range res {
err := db.store.Get(pathPrefix+v.ID, &res[i].Path)
if err != nil {

6
pkg/db/tags.go
View file

@ -12,7 +12,7 @@ const (
tagPrefix = "tag/"
)
var defaultTags = []core.Tag{
var defaultTags = core.Tags{
{"invoice", `(?i)[^e]rechnung|quittung|beleg|invoice|bill|check`},
{"tax", `(?i)steuer|lohn|tax|salary`},
{"contract", `(?i)vertrag|kündigung|vereinbarung|contract|termination|agreement`},
@ -29,7 +29,7 @@ func (db *DB) addDefaultTags() (err error) {
return
}
func (db *DB) GetAllTags() (tags []core.Tag, err error) {
func (db *DB) GetAllTags() (tags core.Tags, err error) {
err = db.store.ForEachPrefix(tagsPrefix, func(k string, _ []byte) error {
var t core.Tag
err := db.store.Get(tagsPrefix+k, &t)
@ -41,7 +41,7 @@ func (db *DB) GetAllTags() (tags []core.Tag, err error) {
return
}
func (db *DB) GetAllRTags() (rtags []core.RTag, err error) {
func (db *DB) GetAllRTags() (rtags core.RTags, err error) {
tags, err := db.GetAllTags()
if err != nil {
return