Calcium-Ion
6dc4030fdf
* feat: support ClickHouse log database * feat(log): optimize log deletion process for ClickHouse
23 lines
644 B
Go
23 lines
644 B
Go
package model
|
|
|
|
import "github.com/QuantumNous/new-api/common"
|
|
|
|
// GetDBTimestamp returns a UNIX timestamp from database time.
|
|
// Falls back to application time on error.
|
|
func GetDBTimestamp() int64 {
|
|
var ts int64
|
|
var err error
|
|
switch {
|
|
case common.UsingMainDatabase(common.DatabaseTypePostgreSQL):
|
|
err = DB.Raw("SELECT EXTRACT(EPOCH FROM NOW())::bigint").Scan(&ts).Error
|
|
case common.UsingMainDatabase(common.DatabaseTypeSQLite):
|
|
err = DB.Raw("SELECT strftime('%s','now')").Scan(&ts).Error
|
|
default:
|
|
err = DB.Raw("SELECT UNIX_TIMESTAMP()").Scan(&ts).Error
|
|
}
|
|
if err != nil || ts <= 0 {
|
|
return common.GetTimestamp()
|
|
}
|
|
return ts
|
|
}
|