Add default timeout for go-ora library

nxshock · May 21, 2023 · ab02e12 · ab02e12
1 parent 67d0cf8
commit ab02e12
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/consts.go b/consts.go
@@ -1,8 +1,12 @@
 package main
 
+import "time"
+
 const (
 	defaultConfigFileName = "gron.conf"
 
+	defaultDbTimeout = 24 * time.Hour
+
 	defaultOnSuccessMessageFmt = "Job {{.JobName}} finished."
 	defaultOnErrorMessageFmt   = "Job {{.JobName}} failed:\n\n{{.Error}}"
 )
diff --git a/job.go b/job.go
@@ -1,6 +1,7 @@
 package main
 
 import (
+	"context"
 	"database/sql"
 	"errors"
 	"fmt"
@@ -245,7 +246,10 @@ func (j *Job) runSql(jobLogFile *os.File) error {
 	}
 	defer db.Close()
 
-	_, err = db.Exec(j.JobConfig.SqlText)
+	ctx, cancel := context.WithTimeout(context.Background(), defaultDbTimeout)
+	defer cancel()
+
+	_, err = db.ExecContext(ctx, j.JobConfig.SqlText)
 	if err != nil {
 		return err
 	}