feat: allow database fulltext parser

AlistGo · Aug 24, 2024 · 0232f9c · 0232f9c
1 parent 868b0ec
commit 0232f9c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/internal/conf/config.go b/internal/conf/config.go
@@ -18,6 +18,7 @@ type Database struct {
 	TablePrefix string `json:"table_prefix" env:"TABLE_PREFIX"`
 	SSLMode     string `json:"ssl_mode" env:"SSL_MODE"`
 	DSN         string `json:"dsn" env:"DSN"`
+	Parser      string `json:"parser" env:"PARSER"`
 }
 
 type Meilisearch struct {

diff --git a/internal/search/db/init.go b/internal/search/db/init.go
@@ -19,19 +19,26 @@ var config = searcher.Config{
 func init() {
 	searcher.RegisterSearcher(config, func() (searcher.Searcher, error) {
 		db := db.GetDb()
+		var parser string
 		switch conf.Conf.Database.Type {
 		case "mysql":
+			if conf.Conf.Database.Parser != "" {
+				parser = fmt.Sprintf(" WITH PARSER %s", conf.Conf.Database.Parser)
+			}
 			tableName := fmt.Sprintf("%ssearch_nodes", conf.Conf.Database.TablePrefix)
-			tx := db.Exec(fmt.Sprintf("CREATE FULLTEXT INDEX idx_%s_name_fulltext ON %s(name);", tableName, tableName))
+			tx := db.Exec(fmt.Sprintf("CREATE FULLTEXT INDEX idx_%s_name_fulltext%s ON %s(name);", tableName, parser, tableName))
 			if err := tx.Error; err != nil && !strings.Contains(err.Error(), "Error 1061 (42000)") { // duplicate error
 				log.Errorf("failed to create full text index: %v", err)
 				return nil, err
 			}
 		case "postgres":
+			if conf.Conf.Database.Parser != "" {
+				parser = fmt.Sprintf(" gin_%s", conf.Conf.Database.Parser)
+			}
 			db.Exec("CREATE EXTENSION pg_trgm;")
 			db.Exec("CREATE EXTENSION btree_gin;")
 			tableName := fmt.Sprintf("%ssearch_nodes", conf.Conf.Database.TablePrefix)
-			tx := db.Exec(fmt.Sprintf("CREATE INDEX idx_%s_name ON %s USING GIN (name);", tableName, tableName))
+			tx := db.Exec(fmt.Sprintf("CREATE INDEX idx_%s_name ON %s USING GIN (name%s);", tableName, tableName, parser))
 			if err := tx.Error; err != nil && !strings.Contains(err.Error(), "SQLSTATE 42P07") {
 				log.Errorf("failed to create index using GIN: %v", err)
 				return nil, err