initial commit

models/merchants.go

@@ -0,0 +1,47 @@
+package models
+
+import (
+	"context"
+	"time"
+
+	"github.com/uptrace/bun"
+)
+
+type Merchant struct {
+	bun.BaseModel `bun:"table:merchants"`
+	MerchantID    int64     `bun:",pk,autoincrement" json:"merchant_id" yaml:"merchant_id"`
+	Title         string    `bun:"title" json:"title" yaml:"title"`
+	Address       string    `bun:"address,type:text,unique" json:"address,omitempty" yaml:"address,omitempty"`
+	Phone         string    `bun:"phone,type:text,unique" json:"phone,omitempty" yaml:"phone,omitempty"`
+	Created       time.Time `bun:",nullzero,notnull,default:current_timestamp" json:"created" yaml:"created"`
+	Updated       time.Time `json:"updated" yaml:"updated"`
+}
+
+var _ bun.AfterCreateTableHook = (*Merchant)(nil)
+
+func (*Merchant) AfterCreateTable(ctx context.Context, query *bun.CreateTableQuery) error {
+	_, err := query.DB().NewCreateIndex().Model((*Merchant)(nil)).Index("idx_merchant_title").Column("title").Exec(ctx)
+	if err != nil {
+		return err
+	}
+	_, err = query.DB().NewCreateIndex().Model((*Merchant)(nil)).Index("idx_merchant_address").Column("address").Exec(ctx)
+	if err != nil {
+		return err
+	}
+	_, err = query.DB().NewCreateIndex().Model((*Merchant)(nil)).Index("idx_merchant_phone").Column("phone").Exec(ctx)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+// Add .
+func (m *Merchant) Add(db *bun.DB, ctx context.Context) error {
+	_, err := db.NewInsert().Model(m).On("CONFLICT (phone) DO UPDATE").Set("updated = EXCLUDED.updated").Exec(ctx)
+	// _, err := db.NewInsert().Model(m).Exec(ctx)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}