Change something?
This commit is contained in:
8
main.go
8
main.go
@@ -82,7 +82,7 @@ func main() {
|
|||||||
}
|
}
|
||||||
mainLogger.Debug("Database connection established")
|
mainLogger.Debug("Database connection established")
|
||||||
|
|
||||||
workdone, err := HandleSpecialArgs(args, err, db)
|
workdone, err := HandleSpecialArgs(args, db)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
mainLogger.Error("Failed to handle special args: %v", err)
|
mainLogger.Error("Failed to handle special args: %v", err)
|
||||||
return
|
return
|
||||||
@@ -366,13 +366,13 @@ func main() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func HandleSpecialArgs(args []string, err error, db utils.DB) (bool, error) {
|
func HandleSpecialArgs(args []string, db utils.DB) (bool, error) {
|
||||||
handleSpecialArgsLogger := logger.Default.WithPrefix("HandleSpecialArgs")
|
handleSpecialArgsLogger := logger.Default.WithPrefix("HandleSpecialArgs")
|
||||||
handleSpecialArgsLogger.Debug("Handling special arguments: %v", args)
|
handleSpecialArgsLogger.Debug("Handling special arguments: %v", args)
|
||||||
switch args[0] {
|
switch args[0] {
|
||||||
case "reset":
|
case "reset":
|
||||||
handleSpecialArgsLogger.Info("Resetting all files")
|
handleSpecialArgsLogger.Info("Resetting all files")
|
||||||
err = utils.ResetAllFiles(db)
|
err := utils.ResetAllFiles(db)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
handleSpecialArgsLogger.Error("Failed to reset all files: %v", err)
|
handleSpecialArgsLogger.Error("Failed to reset all files: %v", err)
|
||||||
return true, err
|
return true, err
|
||||||
@@ -381,7 +381,7 @@ func HandleSpecialArgs(args []string, err error, db utils.DB) (bool, error) {
|
|||||||
return true, nil
|
return true, nil
|
||||||
case "dump":
|
case "dump":
|
||||||
handleSpecialArgsLogger.Info("Dumping all files from database")
|
handleSpecialArgsLogger.Info("Dumping all files from database")
|
||||||
err = db.RemoveAllFiles()
|
err := db.RemoveAllFiles()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
handleSpecialArgsLogger.Error("Failed to remove all files from database: %v", err)
|
handleSpecialArgsLogger.Error("Failed to remove all files from database: %v", err)
|
||||||
return true, err
|
return true, err
|
||||||
|
|||||||
Reference in New Issue
Block a user