migrations/20241007210540_2_rating.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

package migrations

import (
	"context"
	"fmt"

	"github.com/uptrace/bun"
	"sh.org.ru/pkg/model"
)

func init() {
	Migrations.MustRegister(func(ctx context.Context, db *bun.DB) error {
		fmt.Print(" [up migration] ")
		if _, err := db.NewAddColumn().
			Model((*model.Quote)(nil)).
			ColumnExpr("rating INT NOT NULL DEFAULT '0'").
			Exec(ctx); err != nil {
			return err
		}

		return nil
	}, func(ctx context.Context, db *bun.DB) error {
		fmt.Print(" [down migration] ")
		if _, err := db.NewDropColumn().
			Model((*model.Quote)(nil)).
			Column("rating").
			Exec(ctx); err != nil {
			return err
		}

		return nil
	})
}