使用BinData从二进制数据创建用户定义的基元类型?

时间:2012-11-05 00:37:30

标签: ruby bindata

我有一组文件,我必须从传统的Cobol系统下载 每晚。我将这些文件从二进制数据文件转换为 MySql表。

我写了一个Ruby程序来使用BinData为个人做这件事 文件结构。每个文件中都包含几个字段 压缩十进制数据(Cobol COMP-3)。以下代码适用于阅读一个代码 二进制文件和我编写的代码将字段amt1转换为 一个浮点十进制字段。

此代码的问题是,对于每个打包字段,我必须重复 用于字段转换的代码甚至更难的硬编码小数位数 将每个字段放入代码中(参见程序中的注释代码)。

代码示例:

require 'bindata'
require 'bigdecimal'

class WangRec < BinData::Record
  string  :cust_no,         :read_length => 6
  string  :doc_date,        :read_length => 6
  string  :doc_no,          :read_length => 6
  string  :doc_type,        :read_length => 1
  string  :apply_to_no,     :read_length => 6
  string  :cust_no_alt,     :read_length => 6
  string  :apply_to_no_alt, :read_length => 6
  string  :doc_due_date,    :read_length => 6
  string  :amt1,            :read_length => 6 
  string  :amt2,            :read_length => 5
  string  :ref,             :read_length => 30
  string  :slsmn1,          :read_length => 3
  string  :slsmn2,          :read_length => 3
  string  :slsmn3,          :read_length => 3
  string  :amt3,            :read_length => 5
end

def packed(packed_field, dec_pos) 
  unpkd_field = packed_field.unpack('H12')
  num, sign = unpkd_field[0][0..-2], unpkd_field[-1]
  unless sign == 'f'
    amt = num.insert(0, '-')
  end

  if dec_pos > 0
    dec_amt = amt.insert((dec_pos + 1) * -1, '.')
  end

  return dec_amt.to_f 

end

wang_aropnfile = File.open('../data/wangdata/AROPNFIL.bin', 'rb')

count = 0

while !wang_aropnfile.eof?
  rec = WangRec.read(wang_aropnfile)

# The following line of code would have to be repeated for each
# packed field along with the decimal places
  amt1 = packed(rec.amt1, 2)

  puts "#{rec.cust_no} #{rec.doc_type} #{rec.doc_date} #{amt1}"

  count += 1
end

puts count

如何创建我自己的名为pkddec的数据类型原语,该原语采用read_lengthdec_pos参数并创建class PackedDecimal << BinData ::Primitive

1 个答案:

答案 0 :(得分:1)

实际上,我不能因为回答我的问题而感到荣幸,但是感谢“BinData”的创建者Dion Mendel在Ruby Forge的支持中回答这个问题。我在芝加哥时间接近午夜的时间深夜提交了这个问题,今天早上醒来,找到了Dion Mendel的回答,大约3个小时后回答了这个回答。我想与社区分享他的答案,并展示工作代码。

require 'bindata'
require 'bigdecimal'

class PkdDec < BinData::Primitive
  mandatory_parameter :length
  default_parameter   :dec_pos => 0

  string  :str, :read_length => :length

  def get
    str_length = eval_parameter(:length)
    dec_pos    = eval_parameter(:dec_pos)
    unpkd_field = str.unpack("H#{str_length * 2}").first
    num, sign = unpkd_field[0..-2], unpkd_field[-1]
    unless sign == 'f'
      num = num.insert(0, '-')
    end

    if dec_pos > 0
      dec_amt = num.insert((dec_pos + 1) * -1, '.')
    else
      dec_amt = num
    end

    return dec_amt.to_f 

  end

  def set(dec_val)

    #  Not concerned about going the other way
    #  Reverse the get process above

  end

end

class WangRec < BinData::Record
  string  :cust_no, :read_length => 6
  string  :doc_date, :read_length => 6
  string  :doc_no,  :read_length => 6
  string  :doc_type,  :read_length => 1
  string  :apply_to_no, :read_length => 6
  string  :cust_no_alt, :read_length => 6
  string  :apply_to_no_alt, :read_length => 6
  string  :doc_due_date,    :read_length => 6
  PkdDec  :amt1,            :length => 6,  :dec_pos => 2 
  PkdDec  :amt2,            :length => 5,  :dec_pos => 2
  string  :ref,             :read_length => 30
  string  :slsmn1,          :read_length => 3
  string  :slsmn2,          :read_length => 3
  string  :slsmn3,          :read_length => 3
  PkdDec  :amt3,            :length => 5,  :dec_pos => 2

end

wang_aropnfile = File.open('../data/wangdata/AROPNFIL.bin', 'rb')

count = 0

while !wang_aropnfile.eof?
  rec = WangRec.read(wang_aropnfile)
  puts "#{rec.cust_no} #{rec.doc_type} #{rec.amt1} #{rec.amt2} #{rec.amt3}"

  count += 1
end

puts count

再一次,感谢Dion Mendel