add function drops

Author: siherrmann

database/dbJob.go

@@ -134,6 +134,14 @@ func (r JobDBHandler) DropTables() error {
 		return helper.NewError("job_archive table", err)
 	}
 
+	for _, functionName := range loadSql.JobFunctions {
+		dropFunctionQuery := fmt.Sprintf(`DROP FUNCTION IF EXISTS %s;`, functionName)
+		_, err = r.db.Instance.ExecContext(ctx, dropFunctionQuery)
+		if err != nil {
+			return helper.NewError("drop function "+functionName, err)
+		}
+	}
+
 	r.db.Logger.Info("Dropped table job")
 
 	return nil

database/dbMaster.go

@@ -96,6 +96,14 @@ func (r MasterDBHandler) DropTable() error {
 		return helper.NewError("drop", err)
 	}
 
+	for _, functionName := range loadSql.MasterFunctions {
+		dropFunctionQuery := fmt.Sprintf(`DROP FUNCTION IF EXISTS %s;`, functionName)
+		_, err = r.db.Instance.ExecContext(ctx, dropFunctionQuery)
+		if err != nil {
+			return helper.NewError("drop function "+functionName, err)
+		}
+	}
+
 	r.db.Logger.Info("Dropped table master")
 
 	return nil

database/dbWorker.go

@@ -110,6 +110,14 @@ func (r WorkerDBHandler) DropTable() error {
 		return helper.NewError("worker table", err)
 	}
 
+	for _, functionName := range loadSql.WorkerFunctions {
+		dropFunctionQuery := fmt.Sprintf(`DROP FUNCTION IF EXISTS %s;`, functionName)
+		_, err = r.db.Instance.ExecContext(ctx, dropFunctionQuery)
+		if err != nil {
+			return helper.NewError("drop function "+functionName, err)
+		}
+	}
+
 	r.db.Logger.Info("Dropped table worker")
 
 	return nil