|
|
|
@ -1,4 +1,4 @@ |
|
|
|
// Copyright (c) 2013 by Michael Dvorkin. All Rights Reserved.
|
|
|
|
// Copyright (c) 2013-2015 by Michael Dvorkin. All Rights Reserved.
|
|
|
|
// Use of this source code is governed by a MIT-style license that can
|
|
|
|
// Use of this source code is governed by a MIT-style license that can
|
|
|
|
// be found in the LICENSE file.
|
|
|
|
// be found in the LICENSE file.
|
|
|
|
|
|
|
|
|
|
|
|
@ -14,76 +14,133 @@ import ( |
|
|
|
// setup is rather lengthy; there should probably be more concise way
|
|
|
|
// setup is rather lengthy; there should probably be more concise way
|
|
|
|
// that uses reflection and avoids hardcoding the column names.
|
|
|
|
// that uses reflection and avoids hardcoding the column names.
|
|
|
|
type Sorter struct { |
|
|
|
type Sorter struct { |
|
|
|
profile *Profile // Pointer to where we store sort column and order.
|
|
|
|
profile *Profile // Pointer to where we store sort column and order.
|
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
type sortable []Stock |
|
|
|
type sortable []Stock |
|
|
|
func (list sortable) Len() int { return len(list) } |
|
|
|
|
|
|
|
|
|
|
|
func (list sortable) Len() int { return len(list) } |
|
|
|
func (list sortable) Swap(i, j int) { list[i], list[j] = list[j], list[i] } |
|
|
|
func (list sortable) Swap(i, j int) { list[i], list[j] = list[j], list[i] } |
|
|
|
|
|
|
|
|
|
|
|
type byTickerAsc struct { sortable } |
|
|
|
type byTickerAsc struct{ sortable } |
|
|
|
type byLastTradeAsc struct { sortable } |
|
|
|
type byLastTradeAsc struct{ sortable } |
|
|
|
type byChangeAsc struct { sortable } |
|
|
|
type byChangeAsc struct{ sortable } |
|
|
|
type byChangePctAsc struct { sortable } |
|
|
|
type byChangePctAsc struct{ sortable } |
|
|
|
type byOpenAsc struct { sortable } |
|
|
|
type byOpenAsc struct{ sortable } |
|
|
|
type byLowAsc struct { sortable } |
|
|
|
type byLowAsc struct{ sortable } |
|
|
|
type byHighAsc struct { sortable } |
|
|
|
type byHighAsc struct{ sortable } |
|
|
|
type byLow52Asc struct { sortable } |
|
|
|
type byLow52Asc struct{ sortable } |
|
|
|
type byHigh52Asc struct { sortable } |
|
|
|
type byHigh52Asc struct{ sortable } |
|
|
|
type byVolumeAsc struct { sortable } |
|
|
|
type byVolumeAsc struct{ sortable } |
|
|
|
type byAvgVolumeAsc struct { sortable } |
|
|
|
type byAvgVolumeAsc struct{ sortable } |
|
|
|
type byPeRatioAsc struct { sortable } |
|
|
|
type byPeRatioAsc struct{ sortable } |
|
|
|
type byDividendAsc struct { sortable } |
|
|
|
type byDividendAsc struct{ sortable } |
|
|
|
type byYieldAsc struct { sortable } |
|
|
|
type byYieldAsc struct{ sortable } |
|
|
|
type byMarketCapAsc struct { sortable } |
|
|
|
type byMarketCapAsc struct{ sortable } |
|
|
|
|
|
|
|
|
|
|
|
type byTickerDesc struct { sortable } |
|
|
|
type byTickerDesc struct{ sortable } |
|
|
|
type byLastTradeDesc struct { sortable } |
|
|
|
type byLastTradeDesc struct{ sortable } |
|
|
|
type byChangeDesc struct { sortable } |
|
|
|
type byChangeDesc struct{ sortable } |
|
|
|
type byChangePctDesc struct { sortable } |
|
|
|
type byChangePctDesc struct{ sortable } |
|
|
|
type byOpenDesc struct { sortable } |
|
|
|
type byOpenDesc struct{ sortable } |
|
|
|
type byLowDesc struct { sortable } |
|
|
|
type byLowDesc struct{ sortable } |
|
|
|
type byHighDesc struct { sortable } |
|
|
|
type byHighDesc struct{ sortable } |
|
|
|
type byLow52Desc struct { sortable } |
|
|
|
type byLow52Desc struct{ sortable } |
|
|
|
type byHigh52Desc struct { sortable } |
|
|
|
type byHigh52Desc struct{ sortable } |
|
|
|
type byVolumeDesc struct { sortable } |
|
|
|
type byVolumeDesc struct{ sortable } |
|
|
|
type byAvgVolumeDesc struct { sortable } |
|
|
|
type byAvgVolumeDesc struct{ sortable } |
|
|
|
type byPeRatioDesc struct { sortable } |
|
|
|
type byPeRatioDesc struct{ sortable } |
|
|
|
type byDividendDesc struct { sortable } |
|
|
|
type byDividendDesc struct{ sortable } |
|
|
|
type byYieldDesc struct { sortable } |
|
|
|
type byYieldDesc struct{ sortable } |
|
|
|
type byMarketCapDesc struct { sortable } |
|
|
|
type byMarketCapDesc struct{ sortable } |
|
|
|
|
|
|
|
|
|
|
|
func (list byTickerAsc) Less(i, j int) bool { return list.sortable[i].Ticker < list.sortable[j].Ticker } |
|
|
|
func (list byTickerAsc) Less(i, j int) bool { |
|
|
|
func (list byLastTradeAsc) Less(i, j int) bool { return list.sortable[i].LastTrade < list.sortable[j].LastTrade } |
|
|
|
return list.sortable[i].Ticker < list.sortable[j].Ticker |
|
|
|
func (list byChangeAsc) Less(i, j int) bool { return c(list.sortable[i].Change) < c(list.sortable[j].Change) } |
|
|
|
} |
|
|
|
func (list byChangePctAsc) Less(i, j int) bool { return c(list.sortable[i].ChangePct) < c(list.sortable[j].ChangePct) } |
|
|
|
func (list byLastTradeAsc) Less(i, j int) bool { |
|
|
|
func (list byOpenAsc) Less(i, j int) bool { return list.sortable[i].Open < list.sortable[j].Open } |
|
|
|
return list.sortable[i].LastTrade < list.sortable[j].LastTrade |
|
|
|
func (list byLowAsc) Less(i, j int) bool { return list.sortable[i].Low < list.sortable[j].Low } |
|
|
|
} |
|
|
|
func (list byHighAsc) Less(i, j int) bool { return list.sortable[i].High < list.sortable[j].High } |
|
|
|
func (list byChangeAsc) Less(i, j int) bool { |
|
|
|
func (list byLow52Asc) Less(i, j int) bool { return list.sortable[i].Low52 < list.sortable[j].Low52 } |
|
|
|
return c(list.sortable[i].Change) < c(list.sortable[j].Change) |
|
|
|
func (list byHigh52Asc) Less(i, j int) bool { return list.sortable[i].High52 < list.sortable[j].High52 } |
|
|
|
} |
|
|
|
func (list byVolumeAsc) Less(i, j int) bool { return list.sortable[i].Volume < list.sortable[j].Volume } |
|
|
|
func (list byChangePctAsc) Less(i, j int) bool { |
|
|
|
func (list byAvgVolumeAsc) Less(i, j int) bool { return list.sortable[i].AvgVolume < list.sortable[j].AvgVolume } |
|
|
|
return c(list.sortable[i].ChangePct) < c(list.sortable[j].ChangePct) |
|
|
|
func (list byPeRatioAsc) Less(i, j int) bool { return list.sortable[i].PeRatio < list.sortable[j].PeRatio } |
|
|
|
} |
|
|
|
func (list byDividendAsc) Less(i, j int) bool { return list.sortable[i].Dividend < list.sortable[j].Dividend } |
|
|
|
func (list byOpenAsc) Less(i, j int) bool { |
|
|
|
func (list byYieldAsc) Less(i, j int) bool { return list.sortable[i].Yield < list.sortable[j].Yield } |
|
|
|
return list.sortable[i].Open < list.sortable[j].Open |
|
|
|
func (list byMarketCapAsc) Less(i, j int) bool { return m(list.sortable[i].MarketCap) < m(list.sortable[j].MarketCap) } |
|
|
|
} |
|
|
|
|
|
|
|
func (list byLowAsc) Less(i, j int) bool { return list.sortable[i].Low < list.sortable[j].Low } |
|
|
|
func (list byTickerDesc) Less(i, j int) bool { return list.sortable[j].Ticker < list.sortable[i].Ticker } |
|
|
|
func (list byHighAsc) Less(i, j int) bool { |
|
|
|
func (list byLastTradeDesc) Less(i, j int) bool { return list.sortable[j].LastTrade < list.sortable[i].LastTrade } |
|
|
|
return list.sortable[i].High < list.sortable[j].High |
|
|
|
func (list byChangeDesc) Less(i, j int) bool { return c(list.sortable[j].ChangePct) < c(list.sortable[i].ChangePct) } |
|
|
|
} |
|
|
|
func (list byChangePctDesc) Less(i, j int) bool { return c(list.sortable[j].ChangePct) < c(list.sortable[i].ChangePct) } |
|
|
|
func (list byLow52Asc) Less(i, j int) bool { |
|
|
|
func (list byOpenDesc) Less(i, j int) bool { return list.sortable[j].Open < list.sortable[i].Open } |
|
|
|
return list.sortable[i].Low52 < list.sortable[j].Low52 |
|
|
|
func (list byLowDesc) Less(i, j int) bool { return list.sortable[j].Low < list.sortable[i].Low } |
|
|
|
} |
|
|
|
func (list byHighDesc) Less(i, j int) bool { return list.sortable[j].High < list.sortable[i].High } |
|
|
|
func (list byHigh52Asc) Less(i, j int) bool { |
|
|
|
func (list byLow52Desc) Less(i, j int) bool { return list.sortable[j].Low52 < list.sortable[i].Low52 } |
|
|
|
return list.sortable[i].High52 < list.sortable[j].High52 |
|
|
|
func (list byHigh52Desc) Less(i, j int) bool { return list.sortable[j].High52 < list.sortable[i].High52 } |
|
|
|
} |
|
|
|
func (list byVolumeDesc) Less(i, j int) bool { return list.sortable[j].Volume < list.sortable[i].Volume } |
|
|
|
func (list byVolumeAsc) Less(i, j int) bool { |
|
|
|
func (list byAvgVolumeDesc) Less(i, j int) bool { return list.sortable[j].AvgVolume < list.sortable[i].AvgVolume } |
|
|
|
return list.sortable[i].Volume < list.sortable[j].Volume |
|
|
|
func (list byPeRatioDesc) Less(i, j int) bool { return list.sortable[j].PeRatio < list.sortable[i].PeRatio } |
|
|
|
} |
|
|
|
func (list byDividendDesc) Less(i, j int) bool { return list.sortable[j].Dividend < list.sortable[i].Dividend } |
|
|
|
func (list byAvgVolumeAsc) Less(i, j int) bool { |
|
|
|
func (list byYieldDesc) Less(i, j int) bool { return list.sortable[j].Yield < list.sortable[i].Yield } |
|
|
|
return list.sortable[i].AvgVolume < list.sortable[j].AvgVolume |
|
|
|
func (list byMarketCapDesc) Less(i, j int) bool { return m(list.sortable[j].MarketCap) < m(list.sortable[i].MarketCap) } |
|
|
|
} |
|
|
|
|
|
|
|
func (list byPeRatioAsc) Less(i, j int) bool { |
|
|
|
|
|
|
|
return list.sortable[i].PeRatio < list.sortable[j].PeRatio |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
func (list byDividendAsc) Less(i, j int) bool { |
|
|
|
|
|
|
|
return list.sortable[i].Dividend < list.sortable[j].Dividend |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
func (list byYieldAsc) Less(i, j int) bool { |
|
|
|
|
|
|
|
return list.sortable[i].Yield < list.sortable[j].Yield |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
func (list byMarketCapAsc) Less(i, j int) bool { |
|
|
|
|
|
|
|
return m(list.sortable[i].MarketCap) < m(list.sortable[j].MarketCap) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
func (list byTickerDesc) Less(i, j int) bool { |
|
|
|
|
|
|
|
return list.sortable[j].Ticker < list.sortable[i].Ticker |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
func (list byLastTradeDesc) Less(i, j int) bool { |
|
|
|
|
|
|
|
return list.sortable[j].LastTrade < list.sortable[i].LastTrade |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
func (list byChangeDesc) Less(i, j int) bool { |
|
|
|
|
|
|
|
return c(list.sortable[j].ChangePct) < c(list.sortable[i].ChangePct) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
func (list byChangePctDesc) Less(i, j int) bool { |
|
|
|
|
|
|
|
return c(list.sortable[j].ChangePct) < c(list.sortable[i].ChangePct) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
func (list byOpenDesc) Less(i, j int) bool { |
|
|
|
|
|
|
|
return list.sortable[j].Open < list.sortable[i].Open |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
func (list byLowDesc) Less(i, j int) bool { return list.sortable[j].Low < list.sortable[i].Low } |
|
|
|
|
|
|
|
func (list byHighDesc) Less(i, j int) bool { |
|
|
|
|
|
|
|
return list.sortable[j].High < list.sortable[i].High |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
func (list byLow52Desc) Less(i, j int) bool { |
|
|
|
|
|
|
|
return list.sortable[j].Low52 < list.sortable[i].Low52 |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
func (list byHigh52Desc) Less(i, j int) bool { |
|
|
|
|
|
|
|
return list.sortable[j].High52 < list.sortable[i].High52 |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
func (list byVolumeDesc) Less(i, j int) bool { |
|
|
|
|
|
|
|
return list.sortable[j].Volume < list.sortable[i].Volume |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
func (list byAvgVolumeDesc) Less(i, j int) bool { |
|
|
|
|
|
|
|
return list.sortable[j].AvgVolume < list.sortable[i].AvgVolume |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
func (list byPeRatioDesc) Less(i, j int) bool { |
|
|
|
|
|
|
|
return list.sortable[j].PeRatio < list.sortable[i].PeRatio |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
func (list byDividendDesc) Less(i, j int) bool { |
|
|
|
|
|
|
|
return list.sortable[j].Dividend < list.sortable[i].Dividend |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
func (list byYieldDesc) Less(i, j int) bool { |
|
|
|
|
|
|
|
return list.sortable[j].Yield < list.sortable[i].Yield |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
func (list byMarketCapDesc) Less(i, j int) bool { |
|
|
|
|
|
|
|
return m(list.sortable[j].MarketCap) < m(list.sortable[i].MarketCap) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
// Returns new Sorter struct.
|
|
|
|
// Returns new Sorter struct.
|
|
|
|
func NewSorter(profile *Profile) *Sorter { |
|
|
|
func NewSorter(profile *Profile) *Sorter { |
|
|
|
@ -99,39 +156,39 @@ func (sorter *Sorter) SortByCurrentColumn(stocks []Stock) *Sorter { |
|
|
|
|
|
|
|
|
|
|
|
if sorter.profile.Ascending { |
|
|
|
if sorter.profile.Ascending { |
|
|
|
interfaces = []sort.Interface{ |
|
|
|
interfaces = []sort.Interface{ |
|
|
|
byTickerAsc { stocks }, |
|
|
|
byTickerAsc{stocks}, |
|
|
|
byLastTradeAsc { stocks }, |
|
|
|
byLastTradeAsc{stocks}, |
|
|
|
byChangeAsc { stocks }, |
|
|
|
byChangeAsc{stocks}, |
|
|
|
byChangePctAsc { stocks }, |
|
|
|
byChangePctAsc{stocks}, |
|
|
|
byOpenAsc { stocks }, |
|
|
|
byOpenAsc{stocks}, |
|
|
|
byLowAsc { stocks }, |
|
|
|
byLowAsc{stocks}, |
|
|
|
byHighAsc { stocks }, |
|
|
|
byHighAsc{stocks}, |
|
|
|
byLow52Asc { stocks }, |
|
|
|
byLow52Asc{stocks}, |
|
|
|
byHigh52Asc { stocks }, |
|
|
|
byHigh52Asc{stocks}, |
|
|
|
byVolumeAsc { stocks }, |
|
|
|
byVolumeAsc{stocks}, |
|
|
|
byAvgVolumeAsc { stocks }, |
|
|
|
byAvgVolumeAsc{stocks}, |
|
|
|
byPeRatioAsc { stocks }, |
|
|
|
byPeRatioAsc{stocks}, |
|
|
|
byDividendAsc { stocks }, |
|
|
|
byDividendAsc{stocks}, |
|
|
|
byYieldAsc { stocks }, |
|
|
|
byYieldAsc{stocks}, |
|
|
|
byMarketCapAsc { stocks }, |
|
|
|
byMarketCapAsc{stocks}, |
|
|
|
} |
|
|
|
} |
|
|
|
} else { |
|
|
|
} else { |
|
|
|
interfaces = []sort.Interface{ |
|
|
|
interfaces = []sort.Interface{ |
|
|
|
byTickerDesc { stocks }, |
|
|
|
byTickerDesc{stocks}, |
|
|
|
byLastTradeDesc { stocks }, |
|
|
|
byLastTradeDesc{stocks}, |
|
|
|
byChangeDesc { stocks }, |
|
|
|
byChangeDesc{stocks}, |
|
|
|
byChangePctDesc { stocks }, |
|
|
|
byChangePctDesc{stocks}, |
|
|
|
byOpenDesc { stocks }, |
|
|
|
byOpenDesc{stocks}, |
|
|
|
byLowDesc { stocks }, |
|
|
|
byLowDesc{stocks}, |
|
|
|
byHighDesc { stocks }, |
|
|
|
byHighDesc{stocks}, |
|
|
|
byLow52Desc { stocks }, |
|
|
|
byLow52Desc{stocks}, |
|
|
|
byHigh52Desc { stocks }, |
|
|
|
byHigh52Desc{stocks}, |
|
|
|
byVolumeDesc { stocks }, |
|
|
|
byVolumeDesc{stocks}, |
|
|
|
byAvgVolumeDesc { stocks }, |
|
|
|
byAvgVolumeDesc{stocks}, |
|
|
|
byPeRatioDesc { stocks }, |
|
|
|
byPeRatioDesc{stocks}, |
|
|
|
byDividendDesc { stocks }, |
|
|
|
byDividendDesc{stocks}, |
|
|
|
byYieldDesc { stocks }, |
|
|
|
byYieldDesc{stocks}, |
|
|
|
byMarketCapDesc { stocks }, |
|
|
|
byMarketCapDesc{stocks}, |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
@ -153,7 +210,7 @@ func c(str string) float32 { |
|
|
|
func m(str string) float32 { |
|
|
|
func m(str string) float32 { |
|
|
|
multiplier := 1.0 |
|
|
|
multiplier := 1.0 |
|
|
|
|
|
|
|
|
|
|
|
switch str[len(str)-1:len(str)] { // Check the last character.
|
|
|
|
switch str[len(str)-1 : len(str)] { // Check the last character.
|
|
|
|
case `B`: |
|
|
|
case `B`: |
|
|
|
multiplier = 1000000000.0 |
|
|
|
multiplier = 1000000000.0 |
|
|
|
case `M`: |
|
|
|
case `M`: |
|
|
|
@ -162,7 +219,7 @@ func m(str string) float32 { |
|
|
|
multiplier = 1000.0 |
|
|
|
multiplier = 1000.0 |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
trimmed := strings.Trim(str, ` $BMK`) // Get rid of non-numeric characters.
|
|
|
|
trimmed := strings.Trim(str, ` $BMK`) // Get rid of non-numeric characters.
|
|
|
|
value, _ := strconv.ParseFloat(trimmed, 32) |
|
|
|
value, _ := strconv.ParseFloat(trimmed, 32) |
|
|
|
|
|
|
|
|
|
|
|
return float32(value * multiplier) |
|
|
|
return float32(value * multiplier) |
|
|
|
|
Michael Dvorkin
10 years ago