Update README.md

fullpipe · Oct 30, 2024 · ae28173 · ae28173
1 parent e3b06fc
commit ae28173
Showing 1 changed file with 7 additions and 11 deletions.
diff --git a/README.md b/README.md
@@ -65,6 +65,11 @@ bundle, err := mf.NewBundle(
     mf.WithLangFallback(language.BritishEnglish, language.English),
     mf.WithLangFallback(language.Portuguese, language.Spanish),
 
+    // Load all yaml files in directory as messages
+    mf.WithYamlProvider(messagesDir),
+
+    // or you could use your own custom message provider
+    // mf.WithProvider(sqlMessageProvider),
 
     // We assume that the translated messages are mostly correct.
     // However, if any errors occur during translation,
@@ -77,12 +82,6 @@ bundle, err := mf.NewBundle(
     }),
 )
 
-if err != nil {
-    log.Fatal(err)
-}
-
-// Load all yaml files in directory as messages
-err = bundle.LoadDir(messagesDir)
 if err != nil {
     log.Fatal(err)
 }
@@ -131,6 +130,8 @@ func main() {
 		mf.WithLangFallback(language.BritishEnglish, language.English),
 		mf.WithLangFallback(language.Portuguese, language.Spanish),
 
+                mf.WithYamlProvider(messagesDir),
+
 		mf.WithErrorHandler(func(err error, id string, ctx map[string]any) {
 			slog.Error(err.Error(), slog.String("id", id), slog.Any("ctx", ctx))
 		}),
@@ -140,11 +141,6 @@ func main() {
 		log.Fatal(err)
 	}
 
-	err = bundle.LoadDir(messagesDir)
-	if err != nil {
-		log.Fatal(err)
-	}
-
 	tr := bundle.Translator("es")
 
 	slog.Info(tr.Trans("say_hello", mf.Arg("name", "Bob")))