1、websocket
用户量量大,数据量大,而且要求实时更新数据的时候,需要使用websocket。
tradingview正好就是这样的应用场景。
2、tradingview和websocket结合
- getBars方法。tradingview图表插件响应用户操作,根据用户界面渲染需要的数据时间段,调用getBars方法,传递参数function(symbolInfo, resolution, rangeStartDate, rangeEndDate, onLoadedCallback),我们需要构造出预期的参数返回给onLoadedCallback方法,即onLoadedCallback(data)。
- onLoadedCallback方法。该方法接收的data有两种,一种是list数组。另一种是newBars对象。list是历史数据,rangeStartDate - rangeEndDate时间段的数据,根据时间颗粒resolution来划分。newBars是增量数据,同样是rangeStartDate - rangeEndDate时间段的数据,但是这个数据必定是在当前的resolution内。当我们把数据返回给onLoadedCallback,图表会渲染这些数据。
3、数据获取和填充
- 页面初始化,实例化图表,连接websocket。
- 从websocket得到的数据存入缓存cacheData。
- 因为getBars是图表自己调用的,所以当我们存入数据到cacheData的时候,在getBars方法中调用onLoadedCallback就可以。
4、实例
图表调用的getBars方法:
TVjsApi.prototype.getBars = function(symbolInfo, resolution, rangeStartDate, rangeEndDate, onLoadedCallback) {
//console.log(' >> :', this.formatt(rangeStartDate*1000), this.formatt(rangeEndDate*1000))
var ticker = this.symbol + "-" + resolution;
var tickerload = ticker + "load";
var tickerstate = ticker + "state";
if(!this.cacheData[ticker] && !this.cacheData[tickerstate]){
//如果缓存没有数据,而且未发出请求,记录当前节点开始时间
this.cacheData[tickerload] = rangeStartDate;
//发起请求,从websocket获取当前时间段的数据
this.initMessage(symbolInfo, resolution, rangeStartDate, rangeEndDate, onLoadedCallback);
//设置状态为true
this.cacheData[tickerstate] = !0;
return false;
}
if(!this.cacheData[tickerload] || this.cacheData[tickerload] > rangeStartDate){
//如果缓存有数据,但是没有当前时间段的数据,更新当前节点时间
this.cacheData[tickerload] = rangeStartDate;
//发起请求,从websocket获取当前时间段的数据
this.initMessage(symbolInfo, resolution, rangeStartDate, rangeEndDate, onLoadedCallback);
//设置状态为true
this.cacheData[tickerstate] = !0;
return false;
}
if(this.cacheData[tickerstate]){
//正在从websocket获取数据,禁止一切操作
return false;
}
//ticker = this.symbol + "-" + this.interval;
//如果缓存有数据,且长度不为0,构造数据
if (this.cacheData[ticker] && this.cacheData[ticker].length) {
this.isLoading = false
var newBars = []
this.cacheData[ticker].forEach(item => {
if (item.time >= rangeStartDate * 1000 && item.time <= rangeEndDate * 1000) {
newBars.push(item)
}
})
onLoadedCallback(newBars)
} else {
//如果没有数据,等待10ms,再执行一次
var self = this
this.getBarTimer = setTimeout(function() {
self.getBars(symbolInfo, resolution, rangeStartDate, rangeEndDate, onLoadedCallback)
}, 10)
}
}
获取历史数据list的init方法:
TVjsApi.prototype.initMessage = function(symbolInfo, resolution, rangeStartDate, rangeEndDate, onLoadedCallback){
//console.log('发起请求,从websocket获取当前时间段的数据');
var that = this;
//保留当前回调
that.cacheData['onLoadedCallback'] = onLoadedCallback;
//获取需要请求的数据数目
var limit = that.initLimit(resolution, rangeStartDate, rangeEndDate);
//商品名
var symbol = that.symbol;
//如果当前时间颗粒已经改变,停止上一个时间颗粒的订阅,修改时间节点值
if(that.interval !== resolution){
that.unSubscribe(that.interval)
that.interval = resolution;
}
//获取当前时间段的数据,在onMessage中执行回调onLoadedCallback
if (that.interval < 60) {
that.socket.send({
cmd: 'req',
args: ["candle.M"+resolution+"."+symbol, limit, rangeEndDate],
id: 'trade.M'+ resolution +'.'+ symbol.toLowerCase()
})
} else if (that.interval >= 60) {
that.socket.send({
cmd: 'req',
args: ["candle.H"+(resolution/60)+"."+symbol, limit, rangeEndDate],
id: 'trade.H'+ (resolution / 60) +'.'+ symbol.toLowerCase()
})
} else {
that.socket.send({
cmd: 'req',
args: ["candle.D1."+symbol, limit, rangeEndDate],
id: 'trade.D1.'+ symbol.toLowerCase()
})
}
}
响应websocket的Message方法:
TVjsApi.prototype.onMessage = function(data) {
var thats = this;
// console.log("这是后台返回的数据"+count+":"+JSON.stringify(data) )
if (data.data && data.data.length) {
//websocket返回的值,数组代表时间段历史数据,不是增量
var list = []
var ticker = thats.symbol + "-" + thats.interval;
var tickerstate = ticker + "state";
//var that = thats;
//遍历数组,构造缓存数据
data.data.forEach(function(element) {
list.push({
time: element.id*1000,
open: element.open,
high: element.high,
low: element.low,
close: element.close,
volume: element.quote_vol
})
}, thats)
//如果没有缓存数据,则直接填充,发起订阅
if(!thats.cacheData[ticker]){
/*thats.cacheData[ticker] = thats.cacheData[ticker].concat(list);
thats.cacheData['onLoadedCallback'](list);
}else{*/
thats.cacheData[ticker] = list;
thats.subscribe()
}
//新数据即当前时间段需要的数据,直接喂给图表插件
if(thats.cacheData['onLoadedCallback']){
thats.cacheData['onLoadedCallback'](list);
}
//请求完成,设置状态为false
thats.cacheData[tickerstate] = !1;
//记录当前缓存时间,即数组最后一位的时间
thats.lastTime = thats.cacheData[ticker][thats.cacheData[ticker].length - 1].time
}
if (data.type && data.type.indexOf(thats.symbol.toLowerCase()) !== -1) {
// console.log(' >> sub:', data.type)
// data带有type,即返回的是订阅数据,
//缓存的key
var ticker = thats.symbol + "-" + thats.interval;
//构造增量更新数据
var barsData = {
time: data.id * 1000,
open: data.open,
high: data.high,
low: data.low,
close: data.close,
volume: data.quote_vol
}
/*if (barsData.time >= thats.lastTime && thats.cacheData[ticker] && thats.cacheData[ticker].length) {
thats.cacheData[ticker][thats.cacheData[ticker].length - 1] = barsData
}*/
//如果增量更新数据的时间大于缓存时间,而且缓存有数据,数据长度大于0
if (barsData.time > thats.lastTime && thats.cacheData[ticker] && thats.cacheData[ticker].length) {
//增量更新的数据直接加入缓存数组
thats.cacheData[ticker].push(barsData)
//修改缓存时间
thats.lastTime = barsData.time
}else if(barsData.time == thats.lastTime && thats.cacheData[ticker].length){
//如果增量更新的时间等于缓存时间,即在当前时间颗粒内产生了新数据,更新当前数据
thats.cacheData[ticker][thats.cacheData[ticker].length - 1] = barsData
}
// 通知图表插件,可以开始增量更新的渲染了
thats.datafeeds.barsUpdater.updateData()
}
}
代码逻辑如下:
- getBars方法由图表插件调用,带有当前需要渲染的时间参数、时间颗粒和callback。
- 如果缓存中有当前时间段的数据,构造newBars,调用onLoadedCallback(newBars)。
- 如果缓存中没有数据,判断当前需要的数据类型,如果是list历史记录,那么我们需要通过websocket来获取,执行initMessage。
- 判断当前缓存中的时间颗粒resolution是否已经改变,如果已经改变,需要清除缓存数据,更新缓存cacheData = list,返回图表onLoadedCallback(list),另外需要订阅当前时间颗粒的增量数据,取消上一次订阅。
- 如果时间颗粒resolution没有改变,websocket得到的历史数据不需要存入缓存,直接调用onLoadedCallback(list)就可以,因为图表插件的渲染是根本得到的数据时间段来执行的,即:图表插件会缓存已经渲染过的数据。
- 如果当前请求的数据为增量数据,但是缓存中没有,等待10ms,因为增量数据来源于websocket,我们已经订阅,需要等待websocket返回,相当于pending。
以上逻辑基本含括了用户的两个重要操作:切换时间颗粒、查看历史记录。
可运行代码在GitHub上已经更新,可预览。