Add loglevel access. Fix timer labels.

This commit is contained in:
Jan Eitzinger 2023-06-09 15:02:22 +02:00
parent 71db1f8cdd
commit 918f1993f2
3 changed files with 22 additions and 10 deletions

View File

@ -6,10 +6,10 @@ package repository
import ( import (
"database/sql" "database/sql"
"log"
"sync" "sync"
"time" "time"
"github.com/ClusterCockpit/cc-backend/pkg/log"
"github.com/jmoiron/sqlx" "github.com/jmoiron/sqlx"
"github.com/mattn/go-sqlite3" "github.com/mattn/go-sqlite3"
"github.com/qustavo/sqlhooks/v2" "github.com/qustavo/sqlhooks/v2"
@ -48,15 +48,17 @@ func Connect(driver string, db string) {
switch driver { switch driver {
case "sqlite3": case "sqlite3":
sql.Register("sqlite3WithHooks", sqlhooks.Wrap(&sqlite3.SQLiteDriver{}, &Hooks{}))
// - Set WAL mode (not strictly necessary each time because it's persisted in the database, but good for first run) // - Set WAL mode (not strictly necessary each time because it's persisted in the database, but good for first run)
// - Set busy timeout, so concurrent writers wait on each other instead of erroring immediately // - Set busy timeout, so concurrent writers wait on each other instead of erroring immediately
// - Enable foreign key checks // - Enable foreign key checks
opts.URL += "?_journal=WAL&_timeout=5000&_fk=true" opts.URL += "?_journal=WAL&_timeout=5000&_fk=true"
if log.Loglevel() == "debug" {
sql.Register("sqlite3WithHooks", sqlhooks.Wrap(&sqlite3.SQLiteDriver{}, &Hooks{}))
dbHandle, err = sqlx.Open("sqlite3WithHooks", opts.URL) dbHandle, err = sqlx.Open("sqlite3WithHooks", opts.URL)
// dbHandle, err = sqlx.Open("sqlite3", opts.URL) } else {
dbHandle, err = sqlx.Open("sqlite3", opts.URL)
}
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
} }

View File

@ -171,7 +171,7 @@ func (r *JobRepository) JobsStatsGrouped(
} }
} }
log.Infof("Timer JobStatistics %s", time.Since(start)) log.Infof("Timer JobsStatsGrouped %s", time.Since(start))
return stats, nil return stats, nil
} }
@ -212,7 +212,7 @@ func (r *JobRepository) JobsStats(
TotalAccHours: totalAccHours}) TotalAccHours: totalAccHours})
} }
log.Infof("Timer JobStatistics %s", time.Since(start)) log.Infof("Timer JobStats %s", time.Since(start))
return stats, nil return stats, nil
} }
@ -251,7 +251,7 @@ func (r *JobRepository) JobCountGrouped(
} }
} }
log.Infof("Timer JobStatistics %s", time.Since(start)) log.Infof("Timer JobCountGrouped %s", time.Since(start))
return stats, nil return stats, nil
} }
@ -300,7 +300,7 @@ func (r *JobRepository) AddJobCountGrouped(
} }
} }
log.Infof("Timer JobStatistics %s", time.Since(start)) log.Infof("Timer AddJobCountGrouped %s", time.Since(start))
return stats, nil return stats, nil
} }
@ -343,7 +343,7 @@ func (r *JobRepository) AddJobCount(
} }
} }
log.Infof("Timer JobStatistics %s", time.Since(start)) log.Infof("Timer JobJobCount %s", time.Since(start))
return stats, nil return stats, nil
} }
@ -351,6 +351,7 @@ func (r *JobRepository) AddHistograms(
ctx context.Context, ctx context.Context,
filter []*model.JobFilter, filter []*model.JobFilter,
stat *model.JobsStatistics) (*model.JobsStatistics, error) { stat *model.JobsStatistics) (*model.JobsStatistics, error) {
start := time.Now()
castType := r.getCastType() castType := r.getCastType()
var err error var err error
@ -367,6 +368,7 @@ func (r *JobRepository) AddHistograms(
return nil, err return nil, err
} }
log.Infof("Timer AddHistograms %s", time.Since(start))
return stat, nil return stat, nil
} }

View File

@ -41,6 +41,8 @@ var (
CritLog *log.Logger = log.New(CritWriter, CritPrefix, log.LstdFlags|log.Llongfile) CritLog *log.Logger = log.New(CritWriter, CritPrefix, log.LstdFlags|log.Llongfile)
) )
var loglevel string = "info"
/* CONFIG */ /* CONFIG */
func Init(lvl string, logdate bool) { func Init(lvl string, logdate bool) {
@ -78,6 +80,8 @@ func Init(lvl string, logdate bool) {
ErrLog = log.New(ErrWriter, ErrPrefix, log.LstdFlags|log.Llongfile) ErrLog = log.New(ErrWriter, ErrPrefix, log.LstdFlags|log.Llongfile)
CritLog = log.New(CritWriter, CritPrefix, log.LstdFlags|log.Llongfile) CritLog = log.New(CritWriter, CritPrefix, log.LstdFlags|log.Llongfile)
} }
loglevel = lvl
} }
/* PRINT */ /* PRINT */
@ -170,6 +174,10 @@ func Fatalf(format string, v ...interface{}) {
os.Exit(1) os.Exit(1)
} }
func Loglevel() string {
return loglevel
}
/* SPECIAL */ /* SPECIAL */
// func Finfof(w io.Writer, format string, v ...interface{}) { // func Finfof(w io.Writer, format string, v ...interface{}) {