package main import ( "context" "fmt" "os" "path/filepath" "time" cfgpkg "dev.mediocregopher.com/mediocre-blog.git/src/cfg" "dev.mediocregopher.com/mediocre-blog.git/src/post" "dev.mediocregopher.com/mediocre-blog.git/src/post/asset" "dev.mediocregopher.com/mediocre-go-lib.git/mctx" "dev.mediocregopher.com/mediocre-go-lib.git/mlog" "gopkg.in/yaml.v3" ) type testData struct { PublishedPosts []post.Post `yaml:"published_posts"` Assets map[string]string } func loadTestData(path string) (testData, error) { f, err := os.Open(path) if err != nil { return testData{}, fmt.Errorf("opening file: %w", err) } defer f.Close() var res testData if err := yaml.NewDecoder(f).Decode(&res); err != nil { return testData{}, fmt.Errorf("decoding file contents: %w", err) } return res, nil } func main() { ctx := context.Background() cfg := cfgpkg.NewBlogCfg(cfgpkg.Params{}) var dataDir cfgpkg.DataDir dataDir.SetupCfg(cfg) defer dataDir.Close() ctx = mctx.WithAnnotator(ctx, &dataDir) testDataPath := cfg.String("test-data-path", "./test-data.yml", "File containing the data to be loaded in") // initialization err := cfg.Init(ctx) logger := mlog.NewLogger(nil) defer logger.Close() logger.Info(ctx, "process started") defer logger.Info(ctx, "process exiting") if err != nil { logger.Fatal(ctx, "initializing", err) } testDataDir := filepath.Dir(*testDataPath) testData, err := loadTestData(*testDataPath) if err != nil { logger.Fatal(ctx, "loading test data", err) } postDB, err := post.NewSQLDB(dataDir) if err != nil { logger.Fatal(ctx, "initializing post sql db", err) } defer postDB.Close() { postStore := post.NewStore(postDB) now := time.Now().Truncate(time.Hour) for _, post := range testData.PublishedPosts { ctx := mctx.Annotate(ctx, "postID", post.ID, "now", now) if _, err := postStore.Set(post, now); err != nil { logger.Fatal(ctx, "setting post", err) } logger.Info(ctx, "set post") now = now.Add(-1 * time.Hour) } } { assetStore := asset.NewStore(postDB) setAsset := func(assetID, assetPath string) error { assetFullPath := filepath.Join(testDataDir, assetPath) f, err := os.Open(assetFullPath) if err != nil { return fmt.Errorf("opening %q for reading: %w", assetFullPath, err) } defer f.Close() return assetStore.Set(assetID, f) } for assetID, assetPath := range testData.Assets { ctx := mctx.Annotate(ctx, "assetID", assetID, "assetPath", assetPath) if err := setAsset(assetID, assetPath); err != nil { logger.Fatal(ctx, "setting asset", err) } logger.Info(ctx, "set asset") } } }