Class: Pacct::Log

Inherits:

Object

• Object
• Pacct::Log

Defined in:

ext/pacct/pacct_c.c

Class Method Summary

• .new(filename) ⇒ Object

  Creates a new Pacct::Log using the given accounting file.

Instance Method Summary

• #close ⇒ Object

  Closes the log file.

• #each_entry([start]) {|entry, index| ... } ⇒ Object

  Yields each entry in the file to the given block.

• #initialize(filename, mode) ⇒ Object constructor
• #last_entry ⇒ Object

  Returns the last entry in the file.

• #num_entries ⇒ Object

  Returns the number of entries in the file.

• #write_entry(entry) ⇒ Object

  Appends the given entry to the file.

Constructor Details

#initialize(filename, mode) ⇒ Object

# File 'ext/pacct/pacct_c.c', line 140

static VALUE pacct_log_init(VALUE self, VALUE filename, VALUE mode) {
  PacctLog* log;
  FILE* acct;
  long length;
  char* c_filename = StringValueCStr(filename);
  size_t c_filename_len;
  const char* c_mode = "rb";
  
  if(mode != Qnil) {
    int isValidMode = 0;
    size_t i;
    c_mode = StringValueCStr(mode);
    for(i = 0; i < sizeof(validFileModes) / sizeof(char*); ++i) {
      if(strcmp(c_mode, validFileModes[i]) == 0) {
        isValidMode = 1;
        break;
      }
    }
    if(!isValidMode) {
      rb_raise(rb_eArgError, "Invalid mode for Pacct::File: '%s'", c_mode);
    }
  }
  
  acct = fopen(c_filename, c_mode);
  if(!acct) {
    rb_raise(rb_eIOError, "Unable to open file '%s'", c_filename);
  }
  
  Data_Get_Struct(self, PacctLog, log);
  
  log->file = acct;
  c_filename_len = strlen(c_filename);
  log->filename = malloc(c_filename_len + 1);
  ENSURE_ALLOCATED(log->filename);
  strncpy(log->filename, c_filename, c_filename_len);
  log->filename[c_filename_len] = '\0';
  
  CHECK_CALL(fseek(acct, 0, SEEK_END), 0);
  length = ftell(acct);
  rewind(acct);
  
  if(length % sizeof(struct acct_v3) != 0) {
    rb_raise(rb_eIOError, "Accounting file '%s' appears to be the wrong size.", c_filename);
  }
  
  log->num_entries = length / sizeof(struct acct_v3);
  
  return self;
}

Class Method Details

.new(filename) ⇒ Object

Creates a new Pacct::Log using the given accounting file

# File 'ext/pacct/pacct_c.c', line 123

static VALUE pacct_log_new(int argc, VALUE* argv, VALUE class) {
  VALUE log;
  VALUE init_args[2];
  PacctLog* ptr;
  
  init_args[1] = Qnil;
  rb_scan_args(argc, argv, "11", init_args, init_args + 1);
  
  log = Data_Make_Struct(class, PacctLog, 0, pacct_log_free, ptr);
  
  ptr->file = NULL;
  ptr->num_entries = 0;
  
  rb_obj_call_init(log, 2, init_args);
  return log;
}

Instance Method Details

#close ⇒ Object

Closes the log file

# File 'ext/pacct/pacct_c.c', line 199

static VALUE pacct_log_close(VALUE self) {
  PacctLog* log;
  
  Data_Get_Struct(self, PacctLog, log);
  
  if(log->file) {
    fclose(log->file);
    log->file = NULL;
  }
  
  return Qnil;
}

#each_entry([start]) {|entry, index| ... } ⇒ Object

Yields each entry in the file to the given block

If start is given, iteration starts at the entry with that index.

Yields:

• (entry, index)

# File 'ext/pacct/pacct_c.c', line 242

static VALUE each_entry(int argc, VALUE* argv, VALUE self) {
  PacctLog* log;
  VALUE start_value;
  long start = 0;
  int i = 0;
  
  rb_scan_args(argc, argv, "01", &start_value);
  if(argc && start_value != Qnil) {
    start = NUM2UINT(start_value);
  }
  
  Data_Get_Struct(self, PacctLog, log);
  
  pacct_log_check_closed(log);
  
  if(start > log->num_entries) {
    rb_raise(rb_eRangeError, "Index %li is out of range", start);
  }
  
  CHECK_CALL(fseek(log->file, start * sizeof(struct acct_v3), SEEK_SET), 0);
  
  for(i = start; i < log->num_entries; ++i) {
    VALUE entry = pacct_entry_new(log);
    rb_yield(entry);
  }

  return Qnil;
}

#last_entry ⇒ Object

Returns the last entry in the file

# File 'ext/pacct/pacct_c.c', line 274

static VALUE last_entry(VALUE self) {
  PacctLog* log;
  long pos;
  VALUE entry;
  
  Data_Get_Struct(self, PacctLog, log);
  
  pacct_log_check_closed(log);
  
  if(log->num_entries == 0) {
    return Qnil;
  }
  
  pos = ftell(log->file);
  CHECK_CALL(fseek(log->file, -sizeof(struct acct_v3), SEEK_END), 0);
  
  entry = pacct_entry_new(log);
  
  CHECK_CALL(fseek(log->file, pos, SEEK_SET), 0);
  
  return entry;
}

#num_entries ⇒ Object

Returns the number of entries in the file

# File 'ext/pacct/pacct_c.c', line 300

static VALUE get_num_entries(VALUE self) {
  PacctLog* log;
  
  Data_Get_Struct(self, PacctLog, log);
  
  return INT2NUM(log->num_entries);
}

#write_entry(entry) ⇒ Object

Appends the given entry to the file

# File 'ext/pacct/pacct_c.c', line 314

static VALUE write_entry(VALUE self, VALUE entry) {
  //To do consider: verification?
  PacctLog* log;
  long pos;
  struct acct_v3* acct;
  
  Data_Get_Struct(self, PacctLog, log);
  pacct_log_check_closed(log);
  Data_Get_Struct(entry, struct acct_v3, acct);
  
  pos = ftell(log->file);
  CHECK_CALL(fseek(log->file, 0, SEEK_END), 0);
  
  if(fwrite(acct, sizeof(struct acct_v3), 1, log->file) != 1) {
    rb_raise(rb_eIOError, "Unable to write to accounting file '%s'", log->filename);
  }
  
  ++(log->num_entries);
  
  CHECK_CALL(fseek(log->file, pos, SEEK_SET), 0);
  
  return Qnil;
}