0904

internal/cmd/uid.go

@@ -0,0 +1,61 @@
+package cmd
+
+import (
+	"fmt"
+
+	"github.com/pkg/errors"
+	"github.com/rs/xid"
+	"github.com/urfave/cli/v2"
+
+	"github.com/tamakyi/TamaBox/internal/conf"
+	"github.com/tamakyi/TamaBox/internal/db"
+)
+
+var Uid = &cli.Command{
+	Name:   "uid",
+	Usage:  "Create uid for user",
+	Action: runUid,
+}
+
+func runUid(ctx *cli.Context) error {
+	if err := conf.Init(); err != nil {
+		return errors.Wrap(err, "load configuration")
+	}
+
+	dbType := conf.Database.Type
+	if dbType == "postgres" {
+		conf.Database.DSN = conf.PostgresDsn()
+	} else {
+		conf.Database.DSN = conf.MySQLDsn()
+	}
+
+	database, err := db.Init(dbType, conf.Database.DSN)
+	if err != nil {
+		return errors.Wrap(err, "connect to database")
+	}
+
+	var users []*db.User
+	if err := database.WithContext(ctx.Context).Unscoped().Find(&users).Error; err != nil {
+		return errors.Wrap(err, "query users")
+	}
+
+	for idx, user := range users {
+		user := user
+		if user.UID != "" {
+			continue
+		}
+
+		if idx%1000 == 0 {
+			fmt.Printf("Processing user %d/%d\n", idx, len(users))
+		}
+
+		uid := xid.NewWithTime(user.CreatedAt).String()
+		if err := database.WithContext(ctx.Context).Unscoped().
+			Model(&db.User{}).
+			Omit("updated_at").
+			Where("id = ?", user.ID).UpdateColumn("uid", uid).Error; err != nil {
+			return errors.Wrap(err, "update uid")
+		}
+	}
+	return nil
+}