用sql解析通达信二进制day文件,得到历史股价数据

炒股软件通达信把日k线存放在vipdoc\sz\lday下面。sz表示深圳交易所的证券。sz000002.day表示万科的日k线数据,它是二进制的,不能直接阅读,需要一个解析的规则。网络上有一些用python的struct库解析的方法,这里尝试用spark sql来解析。

先看一下二进制文件长啥样:

val df = spark.read.format("binaryFile").load("C:/zd_xdzq/vipdoc/sz/lday/sz000002.day")
df.select("content").show()

 读出来是字节数组。每32个字节是一天的数据。这32个字节中,1-4是日期,5-8是开盘价,9-12是最高价,13-16是最低价,17-20是收盘价,21-24是成交额,25-28是成交量。

 这个[76 7A 33 01]用python struct解析出来是20150902,用sql

select hex(20150902)

得到的结果是

 1 33 7A 76跟[76 7A 33 01]刚好是反的(每个字节是反的)。所以sql解析的思路是,

第一步,把这长串二进制字符每32个字节切成一段

第二步,把每一段按字节反过来

第三步,转成十进制

每32个字节切成一段,没有现成的函数,split只能按标点符号或表达式来切,所以要写一个udf

def splitOnLength(len: Int) = udf((str: String) => {
    str.grouped(len).toSeq
  })

spark.udf.register("splitOnLen64", splitOnLength(64))
spark.udf.register("splitOnLen2", splitOnLength(2))

splitOnLen64就是每32个字节切一段。如果day文件有200个交易日,则生成一个包含200个元素的序列。然后用explode函数转成200行。

然后把每一段切成32小段,这可以用splitOnLen2函数。然后用slice函数,1-4是日期,5-8是开盘价等等。接着用reverse函数完成反转,用array_join函数完成拼接,最后用conv函数转成十进制。

完整sql如下

select conv(array_join(reverse(slice(v,1,4)),''),16,10) ymd,
       conv(array_join(reverse(slice(v,5,4)),''),16,10) open,
       conv(array_join(reverse(slice(v,9,4)),''),16,10) high,
       conv(array_join(reverse(slice(v,13,4)),''),16,10) low,
       conv(array_join(reverse(slice(v,17,4)),''),16,10) close,
       intBits2f(conv(array_join(reverse(slice(v,21,4)),''),16,10)) turnover,
       conv(array_join(reverse(slice(v,25,4)),''),16,10) volume
from
(select splitOnLen2(value) v
from
(select explode(splitOnLen64(hex(content))) value
from wanke
)
)

其中成交额是浮点数,需要一个额外的udf

val intBits2f = udf((bits:Int) => java.lang.Float.intBitsToFloat(bits))
spark.udf.register("intBits2f",intBits2f)

得到结果如下

  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值