youngee_b_api/app/dao/enterprise_dao.go

package dao

import "youngee_b_api/app/entity"

type EnterpriseDao struct{}

func (EnterpriseDao) GetenterpriseIdbyuserid(userid string) (string, error) {
	var enterprise entity.Enterprise
	err := Db.Model(&entity.Enterprise{}).Select("enterprise_id").Where("user_id = ?", userid).First(&enterprise).Error
	return enterprise.EnterpriseID, err
}

func (EnterpriseDao) GetenterpriseIdbynickname(business_name string) (string, error) {
	var enterprise entity.Enterprise
	err := Db.Model(&entity.Enterprise{}).Select("enterprise_id").Where("business_name= ?", business_name).First(&enterprise).Error
	return enterprise.EnterpriseID, err
}