tgdrive/teldrive
1
1
Fork 0
mirror of https://github.com/tgdrive/teldrive.git synced 2025-09-06 14:37:49 +08:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: init user login in db transaction

Browse source
This commit is contained in:
divyam234 2024-07-18 15:50:15 +05:30
parent 826b67210e
commit 932eaa65bf
2 changed files with 17 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
internal/database/migrations/20240718155942_index.sql Normal file
View file

@ -0,0 +1,6 @@
-- +goose Up
-- +goose StatementBegin
ALTER TABLE teldrive.bots DROP CONSTRAINT IF EXISTS btoken_user_channel_un;
ALTER TABLE teldrive.bots DROP CONSTRAINT IF EXISTS bots_pkey;
ALTER TABLE teldrive.bots ADD CONSTRAINT bots_pkey PRIMARY KEY (user_id, token, channel_id);
-- +goose StatementEnd

23
pkg/services/auth.go
View file

@ -33,6 +33,7 @@ import (
"github.com/gotd/td/tg"
"github.com/gotd/td/tgerr"
"gorm.io/gorm"
"gorm.io/gorm/clause"
)
type AuthService struct {
@ -83,18 +84,11 @@ func (as *AuthService) LogIn(c *gin.Context, session *schemas.TgSession) (*schem
IsPremium: session.IsPremium,
}
var result []models.User
err = as.db.Transaction(func(tx *gorm.DB) error {
if err := as.db.Model(&models.User{}).Where("user_id = ?", session.UserID).
Find(&result).Error; err != nil {
return nil, &types.AppError{Error: err}
}
if len(result) == 0 {
if err := as.db.Create(&user).Error; err != nil {
return nil, &types.AppError{Error: err}
if err := as.db.Clauses(clause.OnConflict{DoNothing: true}).Create(&user).Error; err != nil {
return err
}
//Create root folder on first login
file := &models.File{
Name: "root",
Type: "folder",
@ -104,9 +98,14 @@ func (as *AuthService) LogIn(c *gin.Context, session *schemas.TgSession) (*schem
Status: "active",
ParentID: "root",
}
if err := as.db.Create(file).Error; err != nil {
return nil, &types.AppError{Error: err}
if err := as.db.Clauses(clause.OnConflict{DoNothing: true}).Create(file).Error; err != nil {
return err
}
return nil
})
if err != nil {
return nil, &types.AppError{Error: err}
}
client, _ := tgc.AuthClient(c, &as.cnf.TG, session.Sesssion)