inserter.go に YAML 関連のをぶちこんだ

Author: mopeneko

pkg/inserter/inserter.go

@@ -17,6 +17,7 @@ import (
 // Inserter - Inserter
 type Inserter struct {
 	jsonInserter *JSONInserter
+	yamlInserter *YAMLInserter
 	jsInserter   *JSInserter
 }
 
@@ -26,6 +27,7 @@ func NewInserter(client *firestore.Client) *Inserter {
 	return &Inserter{
 		jsonInserter: NewJSONInserter(ci),
 		jsInserter:   NewJSInserter(ci),
+		yamlInserter: NewYAMLInserter(ci),
 	}
 }
 
@@ -61,6 +63,12 @@ func (i *Inserter) executeFile(ctx context.Context) func(path string, info os.Fi
 			if err != nil {
 				log.Printf("failed to insert json file: %s\n%+v", path, err)
 			}
+		case strings.HasSuffix(path, ".yaml"):
+		case strings.HasSuffix(path, ".yml"):
+			err = i.yamlInserter.Execute(ctx, cn, path)
+			if err != nil {
+				log.Printf("failed to insert yaml file: %s\n%+v", path, err)
+			}
 		case strings.HasSuffix(path, ".js"):
 			err = i.jsInserter.Execute(ctx, cn, path)
 			if err != nil {