feat: add features

backend/modules/users/service.go

@@ -3,9 +3,12 @@ package users
 import (
 	"context"
 	"database/sql"
+	"time"
 
+	"backend/common/consts"
 	"backend/database/models/qvyun/public/model"
 	"backend/database/models/qvyun/public/table"
+	"backend/pkg/db"
 	"backend/pkg/pg"
 
 	. "github.com/go-jet/jet/v2/postgres"
@@ -42,25 +45,118 @@ func (svc *Service) GetByOpenID(ctx context.Context, openid string) (*model.User
 }
 
 // GetOrNew
-func (svc *Service) GetOrNew(ctx context.Context, openid string, authInfo pg.UserOAuth) (*model.Users, error) {
+func (svc *Service) GetOrNew(ctx context.Context, tenantID int64, openid string, authInfo pg.UserOAuth) (*model.Users, error) {
+	log := svc.log.WithField("method", "GetOrNew")
+
 	user, err := svc.GetByOpenID(ctx, openid)
 	if err == nil {
+		// check: if tenant has user
 		return user, nil
 	}
 
-	if err != nil {
-		if errors.Is(err, sql.ErrNoRows) {
-			// user = &model.Users{
-			// 	OpenID:  openid,
-			// 	OAuth:,authInfo
-			// }
-			// if err := user.Insert(ctx, svc.db, table.Users); err != nil {
-			// 	return nil, errors.Wrap(err, "failed to insert user")
-			// }
+	if errors.Is(err, sql.ErrNoRows) {
+		user = &model.Users{
+			OpenID:   openid,
+			OAuth:    authInfo,
+			ExpireIn: time.Now().Add(time.Minute * time.Duration(authInfo.ExpiresIn)),
 		}
-		return nil, errors.Wrap(err, "failed to get user by openid")
 
+		tx, err := svc.db.BeginTx(ctx, nil)
+		if err != nil {
+			return nil, errors.Wrap(err, "failed to begin transaction")
+		}
+		defer tx.Rollback()
+		ctx = context.WithValue(ctx, consts.CtxKeyTx, tx)
+
+		user, err := svc.CreateFromModel(ctx, user)
+		if err != nil {
+			return nil, err
+		}
+
+		// create user-tenant relation
+		if err := svc.CreateTenantUser(ctx, user.ID, tenantID); err != nil {
+			return nil, errors.Wrap(err, "failed to create user-tenant relation")
+		}
+
+		if err := tx.Commit(); err != nil {
+			return nil, errors.Wrap(err, "failed to commit transaction")
+		}
+
+		log.Infof("create new user for tenant: %d success, openID: %s", tenantID, openid)
+		return user, nil
 	}
 
-	return nil, errors.New("unknown error")
+	return nil, errors.Wrap(err, "failed to get user by openid")
+}
+
+// CreateFromModel create user from model
+func (svc *Service) CreateFromModel(ctx context.Context, user *model.Users) (*model.Users, error) {
+	log := svc.log.WithField("method", "CreateFromModel")
+
+	stmt := table.Users.INSERT().MODEL(user).RETURNING(table.Users.AllColumns)
+	log.Debug(stmt.DebugSql())
+
+	// get tx from context
+
+	var item model.Users
+	if err := stmt.QueryContext(ctx, db.FromContext(ctx, svc.db), &item); err != nil {
+		return nil, errors.Wrap(err, "failed to create user")
+	}
+
+	return &item, nil
+}
+
+// GetTenantByID
+func (svc *Service) GetTenantByID(ctx context.Context, id int64) (*model.Tenants, error) {
+	log := svc.log.WithField("method", "GetTenantByID")
+
+	stmt := table.Tenants.SELECT(table.Tenants.AllColumns).WHERE(table.Tenants.ID.EQ(Int64(id)))
+	log.Debug(stmt.DebugSql())
+
+	var item model.Tenants
+	if err := stmt.QueryContext(ctx, db.FromContext(ctx, svc.db), &item); err != nil {
+		return nil, errors.Wrapf(err, "failed to query tenant by id %d", id)
+	}
+	return &item, nil
+}
+
+// TenantHasUser
+func (svc *Service) TenantHasUser(ctx context.Context, userID, tenantID int64) (bool, error) {
+	log := svc.log.WithField("method", "TenantHasUser")
+
+	tbl := table.UsersTenants
+	stmt := tbl.
+		SELECT(COUNT(tbl.ID)).
+		WHERE(
+			tbl.UserID.EQ(Int64(userID)).AND(
+				tbl.TenantID.EQ(Int64(tenantID)),
+			),
+		)
+	log.Debug(stmt.DebugSql())
+
+	var cnt int
+	if err := stmt.QueryContext(ctx, db.FromContext(ctx, svc.db), &cnt); err != nil {
+		return false, errors.Wrap(err, "failed to query user-tenant relation")
+	}
+
+	return cnt > 0, nil
+}
+
+// CreateTenantUser
+func (svc *Service) CreateTenantUser(ctx context.Context, userID, tenantID int64) error {
+	log := svc.log.WithField("method", "CreateTenantUser")
+
+	stmt := table.UsersTenants.INSERT(
+		table.UsersTenants.UserID,
+		table.UsersTenants.TenantID,
+	).VALUES(
+		Int64(userID),
+		Int64(tenantID),
+	)
+	log.Debug(stmt.DebugSql())
+
+	if _, err := stmt.ExecContext(ctx, db.FromContext(ctx, svc.db)); err != nil {
+		return errors.Wrap(err, "failed to create user-tenant relation")
+	}
+	return nil
 }