JNIEXPORT jbyteArray JNICALL Java_org_nessdb_DB_stats(JNIEnv *jenv, jobject clazz, jlong ptr)
{
(void)clazz;
char buf[1024 * 10] = {0};
struct slice stats;
struct nessdb *_db = (struct nessdb *)ptr;
if (!_db) {
__ERROR("info-->db is null, pls open first\n");
return NULL;
}
stats.len = 1024 * 10;
stats.data = buf;
db_stats(_db, &stats);
jbyteArray jval = (*jenv)->NewByteArray(jenv, stats.len);
(*jenv)->SetByteArrayRegion(jenv, jval, 0, stats.len, (jbyte*)stats.data);
return jval;
}
/*
* This is the function called from the main polling loop.
*
* It calls command functions to get strings of data, and sends them to the server.
*
*/
int collect_and_send_metrics(int cycle) {
int retval;
char* command;
StringInfoData commands;
pgstat_report_activity(STATE_RUNNING, "Collecting metrics");
initStringInfo(&commands);
/*
* Populate first cycle command data. These are executed on the first cycle
* of a restart. The bgworker restars every N cycles, as listed at the bottom
* of the main loop in pgsampler.c.
*/
if (cycle == 0) {
command = restart_gucs();
appendStringInfoString(&commands, command);
if (strcmp(output_mode, "csv") == 0)
write_to_csv("restart_gucs", command);
pfree(command);
command = relation_info();
appendStringInfoString(&commands, command);
if (strcmp(output_mode, "csv") == 0)
write_to_csv("stat_class", command);
pfree(command);
command = database_list();
appendStringInfoString(&commands, command);
if (strcmp(output_mode, "csv") == 0)
write_to_csv("databases", command);
pfree(command);
command = column_info();
appendStringInfoString(&commands, command);
if (strcmp(output_mode, "csv") == 0)
write_to_csv("columns", command);
pfree(command);
command = index_info();
appendStringInfoString(&commands, command);
if (strcmp(output_mode, "csv") == 0)
write_to_csv("indexes", command);
pfree(command);
command = column_stats();
appendStringInfoString(&commands, command);
if (strcmp(output_mode, "csv") == 0)
write_to_csv("stat_column", command);
pfree(command);
command = db_stats();
appendStringInfoString(&commands, command);
if (strcmp(output_mode, "csv") == 0)
write_to_csv("stat_database", command);
pfree(command);
}
/* HEARTBEAT */
if (cycle % heartbeat_seconds == 0) {
command = heartbeat();
appendStringInfoString(&commands, command);
if (strcmp(output_mode, "csv") == 0)
write_to_csv("heartbeats", command);
pfree(command);
}
/* SYSTEM INFO */
if (cycle % system_seconds == 0) {
command = system_info();
appendStringInfoString(&commands, command);
if (strcmp(output_mode, "csv") == 0)
write_to_csv("stat_system", command);
pfree(command);
command = fs_info();
appendStringInfoString(&commands, command);
if (strcmp(output_mode, "csv") == 0)
write_to_csv("stat_filesystem", command);
pfree(command);
}
/* */
if (cycle % activity_seconds == 0) {
command = activity_stats();
appendStringInfoString(&commands, command);
if (strcmp(output_mode, "csv") == 0)
write_to_csv("stat_activity", command);
pfree(command);
}
if (cycle % replication_seconds == 0) {
command = replication_stats();
appendStringInfoString(&commands, command);
if (strcmp(output_mode, "csv") == 0)
write_to_csv("stat_replication", command);
pfree(command);
}
/* */
if (cycle % bgwriter_seconds == 0) {
command = bgwriter_stats();
appendStringInfoString(&commands, command);
if (strcmp(output_mode, "csv") == 0)
write_to_csv("stat_bgwriter", command);
pfree(command);
}
if (cycle % guc_seconds == 0) {
command = transient_gucs();
appendStringInfoString(&commands, command);
if (strcmp(output_mode, "csv") == 0)
write_to_csv("transient_gucs", command);
pfree(command);
}
if (cycle % statements_seconds == 0) {
command = stat_statements();
appendStringInfoString(&commands, command);
if (strcmp(output_mode, "csv") == 0)
write_to_csv("stat_statements", command);
pfree(command);
}
/* */
if (cycle % relation_seconds == 0) {
command = table_stats();
appendStringInfoString(&commands, command);
if (strcmp(output_mode, "csv") == 0)
write_to_csv("stat_table", command);
pfree(command);
command = index_stats();
appendStringInfoString(&commands, command);
if (strcmp(output_mode, "csv") == 0)
write_to_csv("stat_index", command);
pfree(command);
command = table_io_stats();
appendStringInfoString(&commands, command);
if (strcmp(output_mode, "csv") == 0)
write_to_csv("statio_user_tables", command);
pfree(command);
command = index_io_stats();
appendStringInfoString(&commands, command);
if (strcmp(output_mode, "csv") == 0)
write_to_csv("statio_user_indexes", command);
pfree(command);
command = function_stats();
appendStringInfoString(&commands, command);
if (strcmp(output_mode, "csv") == 0)
write_to_csv("stat_function", command);
pfree(command);
}
/* Send / Write metrics based on output_mode */
if (strcmp(output_mode, "network") == 0) {
pgstat_report_activity(STATE_RUNNING, "Sending metrics to antenna");
retval = send_data(commands.data);
if (retval == NO_DATA_SENT) { //close socket and retry establishing connection and sending data.
// elog(LOG, "reseting..."); //just a note to say reseting socket
if (sockfd != 0)
shutdown(sockfd, SHUT_RDWR);
sockfd = 0;
retval = send_data(commands.data); // we ignore success or failure here. drops data if fails.
}
}
return 0;
}
