migrate from github

internal/repository/invitation_repo/get_by_code.go

@@ -0,0 +1,37 @@
+package invitation_repo
+
+import (
+	"github.com/XingfenD/yoresee_doc/internal/model"
+	"gorm.io/gorm"
+)
+
+type InvitationGetByCodeOperation struct {
+	repo *InvitationRepository
+	code string
+	tx   *gorm.DB
+}
+
+func (r *InvitationRepository) GetByCode(code string) *InvitationGetByCodeOperation {
+	return &InvitationGetByCodeOperation{
+		repo: r,
+		code: code,
+	}
+}
+
+func (op *InvitationGetByCodeOperation) WithTx(tx *gorm.DB) *InvitationGetByCodeOperation {
+	op.tx = tx
+	return op
+}
+
+func (op *InvitationGetByCodeOperation) Exec() (*model.Invitation, error) {
+	var invitation model.Invitation
+	var err error
+
+	if op.tx != nil {
+		err = op.tx.Where("code = ?", op.code).First(&invitation).Error
+	} else {
+		err = op.repo.db.Where("code = ?", op.code).First(&invitation).Error
+	}
+
+	return &invitation, err
+}