51 lines
1.3 KiB
Go
51 lines
1.3 KiB
Go
|
package models
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/google/uuid"
|
||
|
|
"gorm.io/gorm"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type ModelBase struct {
|
||
|
|
ID uuid.UUID `json:"id" gorm:"type:uuid;primary_key;"`
|
||
|
|
CreatedAt time.Time `json:"created_at"`
|
||
|
|
UpdatedAt time.Time `json:"updated_at"`
|
||
|
|
DeletedAt *time.Time `json:"deleted_at,omitempty" gorm:"index"`
|
||
|
|
}
|
||
|
|
|
||
|
|
//https://medium.com/@the.hasham.ali/how-to-use-uuid-key-type-with-gorm-cc00d4ec7100
|
||
|
|
|
||
|
|
func (base *ModelBase) BeforeCreate(tx *gorm.DB) error {
|
||
|
|
base.ID = uuid.New()
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
type OriginBase struct {
|
||
|
|
ModelBase
|
||
|
|
User *User `json:"user,omitempty" gorm:"-"`
|
||
|
|
UserID uuid.UUID `json:"user_id"`
|
||
|
|
|
||
|
|
Source *SourceCredential `json:"source,omitempty" gorm:"-"`
|
||
|
|
SourceID uuid.UUID `json:"source_id" gorm:"not null;index:,unique,composite:source_resource_id"`
|
||
|
|
|
||
|
|
SourceResourceType string `json:"source_resource_type" gorm:"not null;index:,unique,composite:source_resource_id"`
|
||
|
|
SourceResourceID string `json:"source_resource_id" gorm:"not null;index:,unique,composite:source_resource_id"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func (o OriginBase) GetSourceID() uuid.UUID {
|
||
|
|
return o.SourceID
|
||
|
|
}
|
||
|
|
|
||
|
|
func (o OriginBase) GetSourceResourceType() string {
|
||
|
|
return o.SourceResourceType
|
||
|
|
}
|
||
|
|
func (o OriginBase) GetSourceResourceID() string {
|
||
|
|
return o.SourceResourceID
|
||
|
|
}
|
||
|
|
|
||
|
|
type OriginBaser interface {
|
||
|
|
GetSourceID() uuid.UUID
|
||
|
|
GetSourceResourceType() string
|
||
|
|
GetSourceResourceID() string
|
||
|
|
}
|