f110255310
*feat: add arangodb init * fix: dao for sql + arangodb * fix: update error logs * fix: use db name dynamically
65 lines
1.5 KiB
Go
65 lines
1.5 KiB
Go
package db
|
|
|
|
import (
|
|
"log"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
"gorm.io/gorm/clause"
|
|
)
|
|
|
|
type Session struct {
|
|
Key string `json:"_key,omitempty"` // for arangodb
|
|
ObjectID string `json:"_id,omitempty"` // for arangodb & mongodb
|
|
ID string `gorm:"primaryKey;type:char(36)" json:"id"`
|
|
UserID string `gorm:"type:char(36)" json:"user_id"`
|
|
User User `json:"-"`
|
|
UserAgent string `json:"user_agent"`
|
|
IP string `json:"ip"`
|
|
CreatedAt int64 `gorm:"autoCreateTime" json:"created_at"`
|
|
UpdatedAt int64 `gorm:"autoUpdateTime" json:"updated_at"`
|
|
}
|
|
|
|
// AddSession function to save user sessiosn
|
|
func (mgr *manager) AddSession(session Session) error {
|
|
if session.ID == "" {
|
|
session.ID = uuid.New().String()
|
|
}
|
|
|
|
if session.CreatedAt == 0 {
|
|
session.CreatedAt = time.Now().Unix()
|
|
}
|
|
|
|
if session.UpdatedAt == 0 {
|
|
session.CreatedAt = time.Now().Unix()
|
|
}
|
|
|
|
if IsSQL {
|
|
// copy id as value for fields required for mongodb & arangodb
|
|
session.Key = session.ID
|
|
session.ObjectID = session.ID
|
|
res := mgr.sqlDB.Clauses(
|
|
clause.OnConflict{
|
|
DoNothing: true,
|
|
}).Create(&session)
|
|
if res.Error != nil {
|
|
log.Println(`error saving session`, res.Error)
|
|
return res.Error
|
|
}
|
|
}
|
|
|
|
if IsArangoDB {
|
|
|
|
session.CreatedAt = time.Now().Unix()
|
|
session.UpdatedAt = time.Now().Unix()
|
|
sessionCollection, _ := mgr.arangodb.Collection(nil, Collections.Session)
|
|
_, err := sessionCollection.CreateDocument(nil, session)
|
|
if err != nil {
|
|
log.Println(`error saving session`, err)
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|