initial commit

models/credit_cards.go

@@ -0,0 +1,50 @@
+package models
+
+import (
+	"context"
+	"time"
+
+	"github.com/uptrace/bun"
+)
+
+type CreditCard struct {
+	bun.BaseModel `bun:"table:credit_cards"`
+	CreditCardID  int64     `bun:",pk,autoincrement" json:"credit_card_id" yaml:"credit_card_id"`
+	UserID        int64     `json:"user_id" yaml:"user_id"`
+	Title         string    `bun:"title,type:text" json:"title,omitempty" yaml:"title,omitempty"`
+	Bank          string    `bun:"bank,type:text" json:"bank,omitempty" yaml:"bank,omitempty"`
+	Type          string    `bun:"type,type:text" json:"type,omitempty" yaml:"type,omitempty"`
+	Digits        string    `bun:"digits,type:text,unique" json:"digits,omitempty" yaml:"digits,omitempty"`
+	Created       time.Time `bun:",nullzero,notnull,default:current_timestamp" json:"created" yaml:"created"`
+	Updated       time.Time `json:"updated" yaml:"updated"`
+}
+
+type CreditCards []*CreditCard
+
+var _ bun.AfterCreateTableHook = (*CreditCard)(nil)
+
+func (*CreditCard) AfterCreateTable(ctx context.Context, query *bun.CreateTableQuery) error {
+	_, err := query.DB().NewCreateIndex().Model((*CreditCard)(nil)).Index("idx_credit_card_title").Column("title").Exec(ctx)
+	if err != nil {
+		return err
+	}
+	_, err = query.DB().NewCreateIndex().Model((*CreditCard)(nil)).Index("idx_credit_card_type").Column("type").Exec(ctx)
+	if err != nil {
+		return err
+	}
+	_, err = query.DB().NewCreateIndex().Model((*CreditCard)(nil)).Index("idx_credit_card_digits").Column("digits").Exec(ctx)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+// Add .
+func (cc *CreditCard) Add(db *bun.DB, ctx context.Context) error {
+	_, err := db.NewInsert().Model(cc).On("CONFLICT (digits) DO UPDATE").Set("updated = EXCLUDED.updated").Exec(ctx)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}