youngee_b_api/app/dao/enterprise_dao.go

package dao

import "youngee_b_api/app/entity"

type EnterpriseDao struct{}

func (d EnterpriseDao) GetEnterpriseInfo(enterpriseId string) (*entity.Enterprise, error) {
	var enterprise entity.Enterprise
	err := Db.Debug().Model(&entity.Enterprise{}).Where("enterprise_id = ?", enterpriseId).First(&enterprise).Error
	if err != nil {
		return nil, err
	}
	return &enterprise, nil
}

func (d EnterpriseDao) GetEnterprise(enterpriseId string) (*entity.Enterprise, error) {
	var enterprise entity.Enterprise
	err := Db.Model(&entity.Enterprise{}).Where("enterprise_id = ?", enterpriseId).Select("business_name, user_id").First(&enterprise).Error
	if err != nil {
		return nil, err
	}
	return &enterprise, nil
}

func (d EnterpriseDao) GetEnterprisePhone(enterpriseId string) (string, error) {
	var phone string
	err := Db.Debug().Model(&entity.Enterprise{}).Where("enterprise_id = ?", enterpriseId).Select("phone").First(&phone).Error
	if err != nil {
		return "", err
	}
	return phone, nil
}

// 更新账户余额
func (d EnterpriseDao) UpdateEnterpriseBalance(enterpriseId string, balance float64) (*string, error) {
	var enterprise entity.Enterprise
	var err error
	err = Db.Debug().Model(&entity.Enterprise{}).Where("enterprise_id = ?", enterpriseId).First(&enterprise).Error
	if err != nil {
		return nil, err
	}
	enterprise.Balance += balance
	enterprise.AvailableBalance += balance
	err = Db.Debug().Model(&entity.Enterprise{}).Where("enterprise_id = ?", enterpriseId).Updates(enterprise).Error
	if err != nil {
		return nil, err
	}
	return &enterpriseId, nil
}