export the cmd util funcs

This commit is contained in:
lionsoul2014 2026-04-08 10:56:37 +08:00
parent 5244ebad08
commit e3094c8ef2
6 changed files with 13 additions and 13 deletions

View File

@ -13,7 +13,7 @@ func Bench() {
var err error var err error
var dbFile, srcFile, ipVersion, logLevel = "", "", "", "" var dbFile, srcFile, ipVersion, logLevel = "", "", "", ""
var ignoreError = false var ignoreError = false
var fErr = iterateFlags(func(key string, val string) error { var fErr = IterateFlags(func(key string, val string) error {
switch key { switch key {
case "db": case "db":
dbFile = val dbFile = val
@ -65,7 +65,7 @@ func Bench() {
} }
// check and apply the log level // check and apply the log level
err = applyLogLevel(logLevel) err = ApplyLogLevel(logLevel)
if err != nil { if err != nil {
slog.Error("failed to apply log level", "error", err) slog.Error("failed to apply log level", "error", err)
return return

View File

@ -21,7 +21,7 @@ import (
func Edit() { func Edit() {
var err error var err error
var srcFile, ipVersion = "", "" var srcFile, ipVersion = "", ""
var fErr = iterateFlags(func(key string, val string) error { var fErr = IterateFlags(func(key string, val string) error {
switch key { switch key {
case "src": case "src":
srcFile = val srcFile = val

View File

@ -20,7 +20,7 @@ func Generate() {
var srcFile, dstFile = "", "" var srcFile, dstFile = "", ""
var ipVersion, fieldList, logLevel = "", "", "info" var ipVersion, fieldList, logLevel = "", "", "info"
var indexPolicy = xdb.VectorIndexPolicy var indexPolicy = xdb.VectorIndexPolicy
var fErr = iterateFlags(func(key string, val string) error { var fErr = IterateFlags(func(key string, val string) error {
switch key { switch key {
case "src": case "src":
srcFile = val srcFile = val
@ -72,13 +72,13 @@ func Generate() {
} }
// check and apply the log level // check and apply the log level
err = applyLogLevel(logLevel) err = ApplyLogLevel(logLevel)
if err != nil { if err != nil {
slog.Error("failed to apply log level", "error", err) slog.Error("failed to apply log level", "error", err)
return return
} }
fields, err := getFilterFields(fieldList) fields, err := GetFilterFields(fieldList)
if err != nil { if err != nil {
slog.Error("failed to get filter fields", "error", err) slog.Error("failed to get filter fields", "error", err)
return return

View File

@ -22,7 +22,7 @@ func Process() {
var fieldList, logLevel = "", "" var fieldList, logLevel = "", ""
var clearBasedIndex = -1 var clearBasedIndex = -1
var clearValueEqual, clearValueExcept = "", "" var clearValueEqual, clearValueExcept = "", ""
var fErr = iterateFlags(func(key string, val string) error { var fErr = IterateFlags(func(key string, val string) error {
switch key { switch key {
case "src": case "src":
srcFile = val srcFile = val
@ -79,13 +79,13 @@ func Process() {
} }
// check and apply the log level // check and apply the log level
err = applyLogLevel(logLevel) err = ApplyLogLevel(logLevel)
if err != nil { if err != nil {
slog.Error("failed to apply log level", "error", err) slog.Error("failed to apply log level", "error", err)
return return
} }
fields, err := getFilterFields(fieldList) fields, err := GetFilterFields(fieldList)
if err != nil { if err != nil {
slog.Error("failed to get filter fields", "error", err) slog.Error("failed to get filter fields", "error", err)
return return

View File

@ -21,7 +21,7 @@ import (
func Search() { func Search() {
var err error var err error
var dbFile = "" var dbFile = ""
var fErr = iterateFlags(func(key string, val string) error { var fErr = IterateFlags(func(key string, val string) error {
if key == "db" { if key == "db" {
dbFile = val dbFile = val
} else { } else {

View File

@ -22,7 +22,7 @@ func PrintHelp() {
} }
// Iterate the cli flags // Iterate the cli flags
func iterateFlags(cb func(key string, val string) error) error { func IterateFlags(cb func(key string, val string) error) error {
for i := 2; i < len(os.Args); i++ { for i := 2; i < len(os.Args); i++ {
r := os.Args[i] r := os.Args[i]
if len(r) < 5 { if len(r) < 5 {
@ -46,7 +46,7 @@ func iterateFlags(cb func(key string, val string) error) error {
return nil return nil
} }
func applyLogLevel(logLevel string) error { func ApplyLogLevel(logLevel string) error {
// check and apply the log level // check and apply the log level
var levelLog = slog.LevelInfo var levelLog = slog.LevelInfo
switch strings.ToLower(logLevel) { switch strings.ToLower(logLevel) {
@ -71,7 +71,7 @@ func applyLogLevel(logLevel string) error {
var pattern = regexp.MustCompile(`^(\d+(-\d+)?)$`) var pattern = regexp.MustCompile(`^(\d+(-\d+)?)$`)
func getFilterFields(fieldList string) ([]int, error) { func GetFilterFields(fieldList string) ([]int, error) {
if len(fieldList) == 0 { if len(fieldList) == 0 {
return []int{}, nil return []int{}, nil
} }