feat: update

This commit is contained in:
Rogee
2024-12-08 01:21:04 +08:00
parent c54d7896ef
commit 25df094481
15 changed files with 128 additions and 71 deletions

View File

@@ -66,6 +66,10 @@ func (svc *Service) GetByOpenID(ctx context.Context, openid string) (*model.User
func (svc *Service) GetOrNew(ctx context.Context, tenantID int64, openid string, authInfo pg.UserOAuth) (*model.Users, error) {
log := svc.log.WithField("method", "GetOrNew")
svc.log.Infof("get or new user for tenant: %d, openid: %s", tenantID, openid)
if openid == "" {
return nil, errors.New("openid is empty")
}
user, err := svc.GetByOpenID(ctx, openid)
if err == nil {
// check: if tenant has user
@@ -196,17 +200,17 @@ func (svc *Service) CreateTenantUser(ctx context.Context, userID, tenantID int64
}
// GetTenantIDBySlug
func (svc *Service) GetTenantIDBySlug(ctx context.Context, slug string) (int64, error) {
func (svc *Service) GetTenantBySlug(ctx context.Context, slug string) (*model.Tenants, error) {
log := svc.log.WithField("method", "GetTenantIDBySlug")
stmt := table.Tenants.SELECT(table.Tenants.ID).WHERE(table.Tenants.Slug.EQ(String(slug)))
stmt := table.Tenants.SELECT(table.Tenants.AllColumns).WHERE(table.Tenants.Slug.EQ(String(slug)))
log.Debug(stmt.DebugSql())
var id int64
if err := stmt.QueryContext(ctx, db.FromContext(ctx, svc.db), &id); err != nil {
return 0, errors.Wrap(err, "failed to query tenant id by slug")
var item model.Tenants
if err := stmt.QueryContext(ctx, db.FromContext(ctx, svc.db), &item); err != nil {
return nil, errors.Wrap(err, "failed to query tenant id by slug")
}
return id, nil
return &item, nil
}
// CreateTenant