antlinker / go-dirtyfilter Goto Github PK

package main

import (
	"fmt"

	"github.com/antlinker/go-dirtyfilter"
	"github.com/antlinker/go-dirtyfilter/store"
)

var (
	filterText = `我是需要过滤的内容，内容为：**文@@件，需要过滤。。。hello`
)

func main() {
	memStore, err := store.NewMemoryStore(store.MemoryConfig{
		DataSource: []string{"哈哈"},
	})
	if err != nil {
		panic(err)
	}
	filterManage := filter.NewDirtyManager(memStore)
	result, err := filterManage.Filter().Replace(filterText, '*')
	if err != nil {
		panic(err)
	}
	fmt.Println(result)
}

func (nf *nodeFilter) Replace(text string, delim rune) (string, error) {
	uchars := []rune(text)
	idexs := nf.doIndexes(uchars)
	if len(idexs) == 0 {
		// return "", nil
		// 合法内容，原样输出
		return text, nil
	}
	for i := 0; i < len(idexs); i++ {
		uchars[idexs[i]] = rune(delim)
	}
	return string(uchars), nil
}