/**
* @brief
* Save a queue to the database
*
* @param[in] pque - Pointer to the queue to save
* @param[in] mode:
* QUE_SAVE_FULL - Save full queue information (update)
* QUE_SAVE_NEW - Save new queue information (insert)
*
* @return Error code
* @retval 0 - Success
* @retval 1 - Failure
*
*/
int
que_save_db(pbs_queue *pque, int mode)
{
pbs_db_que_info_t dbque;
pbs_db_attr_info_t attr_info;
pbs_db_obj_info_t obj;
pbs_db_conn_t *conn = (pbs_db_conn_t *) svr_db_conn;
int flag=0;
svr_to_db_que(pque, &dbque);
obj.pbs_db_obj_type = PBS_DB_QUEUE;
obj.pbs_db_un.pbs_db_que = &dbque;
/* load que_qs */
if (pbs_db_begin_trx(conn, 0, 0) !=0)
goto db_err;
if (mode == QUE_SAVE_NEW) {
if (pbs_db_insert_obj(conn, &obj) != 0)
goto db_err;
flag = 1;
}
else if (mode == QUE_SAVE_FULL) {
/*
* Delete the queue and write it afresh
* Deleting the queue removes all attributes automatically
* Thus when reinserting, we add only the set attributes
*/
pbs_db_delete_obj(conn, &obj);
if (pbs_db_insert_obj(conn, &obj) != 0)
goto db_err;
flag = 1;
} else {
/* que_qs*/
if (pbs_db_update_obj(conn, &obj) != 0)
goto db_err;
}
/* que_attrs */
attr_info.parent_obj_type = PARENT_TYPE_QUE_ALL; /* que attr */
attr_info.parent_id = dbque.qu_name;
if (save_attr_db(conn, &attr_info, que_attr_def, pque->qu_attr, (int)QA_ATR_LAST, flag) !=0)
goto db_err;
if (pbs_db_end_trx(conn, PBS_DB_COMMIT) != 0)
goto db_err;
return (0);
db_err:
strcpy(log_buffer, "que_save failed ");
if (conn->conn_db_err != NULL)
strncat(log_buffer, conn->conn_db_err, LOG_BUF_SIZE - strlen(log_buffer) - 1);
log_err(-1, __func__, log_buffer);
(void) pbs_db_end_trx(conn, PBS_DB_ROLLBACK);
panic_stop_db(log_buffer);
return (-1);
}
int
sched_save_db(struct sched *ps, int mode)
{
pbs_db_conn_t *conn = (pbs_db_conn_t *) svr_db_conn;
pbs_db_sched_info_t dbsched;
pbs_db_attr_info_t attr_info;
pbs_db_obj_info_t obj;
int flag=0;
svr_to_db_sched(ps, &dbsched);
obj.pbs_db_obj_type = PBS_DB_SCHED;
obj.pbs_db_un.pbs_db_sched = &dbsched;
if (mode == SVR_SAVE_QUICK) {
/* save server_qs */
if (pbs_db_update_obj(conn, &obj) != 0)
goto db_err;
} else { /* SVR_SAVE_FULL Save */
if (pbs_db_begin_trx(conn, 0, 0) !=0)
goto db_err;
if (mode == SVR_SAVE_NEW) {
if (pbs_db_insert_obj(conn, &obj) != 0)
goto db_err;
flag = 1;
} else {
/* server_qs */
if (pbs_db_update_obj(conn, &obj) != 0) {
if (pbs_db_insert_obj(conn, &obj) != 0)
goto db_err;
}
}
/* svr_attrs */
attr_info.parent_obj_type = PARENT_TYPE_SCHED; /* svr attr */
attr_info.parent_id = pbs_server_id;
if (save_attr_db(conn, &attr_info, sched_attr_def, ps->sch_attr, (int)SCHED_ATR_LAST, flag) !=0)
goto db_err;
if (pbs_db_end_trx(conn, PBS_DB_COMMIT) != 0)
goto db_err;
}
return (0);
db_err:
strcpy(log_buffer, schedemsg);
if (conn->conn_db_err != NULL)
strncat(log_buffer, conn->conn_db_err, LOG_BUF_SIZE - strlen(log_buffer) - 1);
log_err(-1, __func__, log_buffer);
(void) pbs_db_end_trx(conn, PBS_DB_ROLLBACK);
panic_stop_db(log_buffer);
return (-1);
}
int
svr_save_db(struct server *ps, int mode)
{
pbs_db_conn_t *conn = (pbs_db_conn_t *) svr_db_conn;
pbs_db_svr_info_t dbsvr;
pbs_db_obj_info_t obj;
int savetype = PBS_UPDATE_DB_FULL;
int rc;
ps->sv_qs.sv_savetm = time_now;
/* as part of the server save, update svrlive file now,
* used in failover
*/
if (update_svrlive() !=0)
return -1;
if (mode == SVR_SAVE_FULL)
savetype = PBS_UPDATE_DB_FULL;
else if (mode == SVR_SAVE_QUICK)
savetype = PBS_UPDATE_DB_QUICK;
else
savetype = PBS_INSERT_DB;
if (svr_to_db_svr(ps, &dbsvr, savetype) != 0)
goto db_err;
obj.pbs_db_obj_type = PBS_DB_SVR;
obj.pbs_db_un.pbs_db_svr = &dbsvr;
rc = pbs_db_save_obj(conn, &obj, savetype);
if (rc != 0) {
savetype = PBS_INSERT_DB;
rc = pbs_db_save_obj(conn, &obj, savetype);
}
pbs_db_reset_obj(&obj);
if (rc != 0)
goto db_err;
return (0);
db_err:
strcpy(log_buffer, msg_svdbnosv);
if (conn->conn_db_err != NULL)
strncat(log_buffer, conn->conn_db_err, LOG_BUF_SIZE - strlen(log_buffer) - 1);
log_err(-1, __func__, log_buffer);
panic_stop_db(log_buffer);
return (-1);
}
int
sched_save_db(pbs_sched *ps, int mode)
{
pbs_db_conn_t *conn = (pbs_db_conn_t *) svr_db_conn;
pbs_db_sched_info_t dbsched;
pbs_db_obj_info_t obj;
int savetype = PBS_UPDATE_DB_FULL;
int rc;
if (mode == SVR_SAVE_FULL)
savetype = PBS_UPDATE_DB_FULL;
else if (mode == SVR_SAVE_QUICK)
savetype = PBS_UPDATE_DB_QUICK;
else
savetype = PBS_INSERT_DB;
if (svr_to_db_sched(ps, &dbsched, savetype) != 0)
goto db_err;
obj.pbs_db_obj_type = PBS_DB_SCHED;
obj.pbs_db_un.pbs_db_sched = &dbsched;
rc = pbs_db_save_obj(conn, &obj, savetype);
if (rc != 0) {
savetype = PBS_INSERT_DB;
rc = pbs_db_save_obj(conn, &obj, savetype);
}
/* free the attribute list allocated by encode_attrs */
pbs_db_reset_obj(&obj);
if (rc != 0)
goto db_err;
return (0);
db_err:
strcpy(log_buffer, schedemsg);
if (conn->conn_db_err != NULL)
strncat(log_buffer, conn->conn_db_err, LOG_BUF_SIZE - strlen(log_buffer) - 1);
log_err(-1, __func__, log_buffer);
panic_stop_db(log_buffer);
return (-1);
}
/**
* @brief
* Save a queue to the database
*
* @param[in] pque - Pointer to the queue to save
* @param[in] mode:
* QUE_SAVE_FULL - Save full queue information (update)
* QUE_SAVE_NEW - Save new queue information (insert)
*
* @return Error code
* @retval 0 - Success
* @retval 1 - Failure
*
*/
int
que_save_db(pbs_queue *pque, int mode)
{
pbs_db_que_info_t dbque;
pbs_db_obj_info_t obj;
pbs_db_conn_t *conn = (pbs_db_conn_t *) svr_db_conn;
int savetype = PBS_UPDATE_DB_FULL;
if (svr_to_db_que(pque, &dbque, savetype) != 0)
goto db_err;
obj.pbs_db_obj_type = PBS_DB_QUEUE;
obj.pbs_db_un.pbs_db_que = &dbque;
if (mode == QUE_SAVE_NEW)
savetype = PBS_INSERT_DB;
if (pbs_db_save_obj(conn, &obj, savetype) != 0)
goto db_err;
pbs_db_reset_obj(&obj);
return (0);
db_err:
/* free the attribute list allocated by encode_attrs */
free(dbque.attr_list.attributes);
strcpy(log_buffer, "que_save failed ");
if (conn->conn_db_err != NULL)
strncat(log_buffer, conn->conn_db_err, LOG_BUF_SIZE - strlen(log_buffer) - 1);
log_err(-1, __func__, log_buffer);
panic_stop_db(log_buffer);
return (-1);
}
int
svr_save_db(struct server *ps, int mode)
{
pbs_db_conn_t *conn = (pbs_db_conn_t *) svr_db_conn;
pbs_db_svr_info_t dbsvr;
pbs_db_attr_info_t attr_info;
pbs_db_obj_info_t obj;
int flag=0;
ps->sv_qs.sv_savetm = time_now;
/* as part of the server save, update svrlive file now,
* used in failover
*/
if (update_svrlive() !=0)
return -1;
svr_to_db_svr(ps, &dbsvr);
obj.pbs_db_obj_type = PBS_DB_SVR;
obj.pbs_db_un.pbs_db_svr = &dbsvr;
if (mode == SVR_SAVE_QUICK) {
/* save server_qs */
if (pbs_db_update_obj(conn, &obj) != 0)
goto db_err;
} else { /* SVR_SAVE_FULL Save */
if (pbs_db_begin_trx(conn, 0, 0) !=0)
goto db_err;
if (mode == SVR_SAVE_NEW) {
if (pbs_db_insert_obj(conn, &obj) != 0)
goto db_err;
flag = 1;
} else { /* FULL SAVE */
/*
* remove all old attributes
* and insert them back again
*/
pg_db_delete_svrattr(conn, &obj);
/* server_qs */
if (pbs_db_update_obj(conn, &obj) != 0)
goto db_err;
flag = 1; /* so all set attributes are re-inserted back */
}
/* svr_attrs */
attr_info.parent_obj_type = PARENT_TYPE_SERVER; /* svr attr */
attr_info.parent_id = pbs_server_id;
if (save_attr_db(conn, &attr_info, svr_attr_def, ps->sv_attr, (int)SRV_ATR_LAST, flag) !=0)
goto db_err;
if (pbs_db_end_trx(conn, PBS_DB_COMMIT) != 0)
goto db_err;
}
return (0);
db_err:
strcpy(log_buffer, msg_svdbnosv);
if (conn->conn_db_err != NULL)
strncat(log_buffer, conn->conn_db_err, LOG_BUF_SIZE - strlen(log_buffer) - 1);
log_err(-1, __func__, log_buffer);
(void) pbs_db_end_trx(conn, PBS_DB_ROLLBACK);
panic_stop_db(log_buffer);
return (-1);
}
