zeni/go-chart
1
0
Fork 0
Code Issues Pull requests Projects Releases 2 Packages Wiki Activity Actions

refinements.

Browse source
This commit is contained in:
Will Charczuk 2016-07-23 13:01:38 -07:00
parent d41c9313aa
commit 78645130e4
1 changed files with 10 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
date/util.go
View file

@ -292,18 +292,18 @@ func NextMarketClose(after, closeTime time.Time, isHoliday HolidayProvider) time
// CalculateMarketSecondsBetween calculates the number of seconds the market was open between two dates. // CalculateMarketSecondsBetween calculates the number of seconds the market was open between two dates.
func CalculateMarketSecondsBetween(start, end, marketOpen, marketClose time.Time, isHoliday HolidayProvider) (seconds int64) { func CalculateMarketSecondsBetween(start, end, marketOpen, marketClose time.Time, isHoliday HolidayProvider) (seconds int64) {
se := start.In(Eastern()) startEastern := start.In(Eastern())
ee := end.In(Eastern()) endEastern := end.In(Eastern())
startMarketOpen := NextMarketOpen(se, marketOpen, isHoliday) startMarketOpen := NextMarketOpen(startEastern, marketOpen, isHoliday)
startMarketClose := NextMarketClose(se, marketClose, isHoliday) startMarketClose := NextMarketClose(startEastern, marketClose, isHoliday)
if (se.Equal(startMarketOpen) || se.After(startMarketOpen)) && se.Before(startMarketClose) { if (startEastern.Equal(startMarketOpen) || startEastern.After(startMarketOpen)) && startEastern.Before(startMarketClose) {
seconds += int64(startMarketClose.Sub(se) / time.Second) seconds += int64(startMarketClose.Sub(startEastern) / time.Second)
} }
cursor := NextMarketOpen(startMarketClose, marketOpen, isHoliday) cursor := NextMarketOpen(startMarketClose, marketOpen, isHoliday)
for BeforeDate(cursor, ee) { for BeforeDate(cursor, endEastern) {
if IsWeekDay(cursor.Weekday()) && !isHoliday(cursor) { if IsWeekDay(cursor.Weekday()) && !isHoliday(cursor) {
close := NextMarketClose(cursor, marketClose, isHoliday) close := NextMarketClose(cursor, marketClose, isHoliday)
seconds += int64(close.Sub(cursor) / time.Second) seconds += int64(close.Sub(cursor) / time.Second)
@ -313,9 +313,9 @@ func CalculateMarketSecondsBetween(start, end, marketOpen, marketClose time.Time
finalMarketOpen := NextMarketOpen(cursor, marketOpen, isHoliday) finalMarketOpen := NextMarketOpen(cursor, marketOpen, isHoliday)
finalMarketClose := NextMarketClose(cursor, marketClose, isHoliday) finalMarketClose := NextMarketClose(cursor, marketClose, isHoliday)
if end.After(finalMarketOpen) { if endEastern.After(finalMarketOpen) {
if end.Before(finalMarketClose) { if endEastern.Before(finalMarketClose) {
seconds += int64(end.Sub(finalMarketOpen) / time.Second) seconds += int64(endEastern.Sub(finalMarketOpen) / time.Second)
} else { } else {
seconds += int64(finalMarketClose.Sub(finalMarketOpen) / time.Second) seconds += int64(finalMarketClose.Sub(finalMarketOpen) / time.Second)
} }