sws_xcx/service/sysadminservice.go

package service

import (
	"sws_xcx/models"

	"github.com/jinzhu/gorm"
)

type SysAdminService struct {
	rdb *gorm.DB
	wdb *gorm.DB
}

func NewSysAdminService() *SysAdminService {
	admin := &models.SysAdmin{}
	return &SysAdminService{
		rdb: readDb.Model(admin),
		wdb: writeDb.Model(admin),
	}
}

func (s *SysAdminService) GetByUserName(userName string) (*models.SysAdmin, error) {
	var admin models.SysAdmin
	err := s.rdb.Where("user_name = ?", userName).First(&admin).Error
	if err == gorm.ErrRecordNotFound {
		return &admin, nil
	}
	return &admin, err
}