digital-forensics-labs/SecGen
1
0
Fork 0
mirror of https://github.com/cliffe/SecGen.git synced 2026-02-20 13:50:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Print id when adding batch job

Browse Source
This commit is contained in:
Z. Cliffe Schreuders
2018-07-30 17:07:20 +01:00
parent b53408bce8
commit 0a0d050020
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
lib/batch/batch_secgen.rb
View File

@@ -306,10 +306,11 @@ def insert_row(db_conn, prepared_statements, statement_id, secgen_args)
secgen_args = '--shutdown ' + secgen_args.strip
Print.info "Adding to queue: '#{statement}' '#{secgen_args}' 'todo'"
unless prepared_statements.include? statement
db_conn.prepare(statement, 'insert into queue (secgen_args, status) values ($1, $2)')
db_conn.prepare(statement, 'insert into queue (secgen_args, status) values ($1, $2) returning id')
prepared_statements << statement
end
db_conn.exec_prepared(statement, [secgen_args, 'todo'])
result = db_conn.exec_prepared(statement, [secgen_args, 'todo'])
Print.info "id: #{result.first['id']}"
end
def select_all(db_conn)