Merge pull request #12 from Tinkoff/dev

add waiting close flag

Author: jstalex

README.md

@@ -119,7 +119,7 @@ func main() {
 	// результат подписки на инструменты это канал с определенным типом информации, при повторном вызове функции
 	// подписки(например на свечи), возвращаемый канал можно игнорировать, так как при первом вызове он уже был получен
 	firstInstrumetsGroup := []string{"BBG004730N88", "BBG00475KKY8", "BBG004RVFCY3"}
-	candleChan, err := firstMDStream.SubscribeCandle(firstInstrumetsGroup, pb.SubscriptionInterval_SUBSCRIPTION_INTERVAL_ONE_MINUTE)
+	candleChan, err := firstMDStream.SubscribeCandle(firstInstrumetsGroup, pb.SubscriptionInterval_SUBSCRIPTION_INTERVAL_ONE_MINUTE, true)
 	if err != nil {
 		logger.Errorf(err.Error())
 	}

examples/md_stream.go

@@ -62,7 +62,7 @@ func main() {
 	// результат подписки на инструменты это канал с определенным типом информации, при повторном вызове функции
 	// подписки(например на свечи), возвращаемый канал можно игнорировать, так как при первом вызове он уже был получен
 	firstInstrumetsGroup := []string{"BBG004730N88", "BBG00475KKY8", "BBG004RVFCY3"}
-	candleChan, err := firstMDStream.SubscribeCandle(firstInstrumetsGroup, pb.SubscriptionInterval_SUBSCRIPTION_INTERVAL_ONE_MINUTE)
+	candleChan, err := firstMDStream.SubscribeCandle(firstInstrumetsGroup, pb.SubscriptionInterval_SUBSCRIPTION_INTERVAL_ONE_MINUTE, true)
 	if err != nil {
 		logger.Errorf(err.Error())
 	}

investgo/client.go

@@ -72,12 +72,23 @@ func NewClient(ctx context.Context, conf Config, l Logger) (*Client, error) {
 		return nil, err
 	}
 
-	return &Client{
+	client := &Client{
 		conn:   conn,
 		Config: conf,
 		Logger: l,
 		ctx:    ctx,
-	}, nil
+	}
+
+	if conf.AccountId == "" {
+		s := client.NewSandboxServiceClient()
+		resp, err := s.OpenSandboxAccount()
+		if err != nil {
+			return nil, err
+		}
+		client.Config.AccountId = resp.GetAccountId()
+	}
+
+	return client, nil
 }
 
 func setDefaultConfig(conf *Config) {

investgo/config.go

@@ -17,7 +17,7 @@ type Config struct {
 	// AppName - Название вашего приложения, по умолчанию = tinkoff-api-go-sdk
 	AppName string `yaml:"AppName"`
 	// AccountId - Если уже есть аккаунт для апи можно указать напрямую,
-	// для песочницы создастся и запишется автоматически
+	// по умолчанию откроется новый счет в песочнице
 	AccountId string `yaml:"AccountId"`
 	// DisableResourceExhaustedRetry - Если true, то сдк не пытается ретраить, после получения ошибки об исчерпывании
 	// лимита запросов, если false, то сдк ждет нужное время и пытается выполнить запрос снова. По умолчанию = false

investgo/instruments.go

@@ -371,6 +371,24 @@ func (is *InstrumentsServiceClient) InstrumentByPositionUid(id string) (*Instrum
 	return is.instrumentBy(id, pb.InstrumentIdType_INSTRUMENT_ID_TYPE_POSITION_UID, "")
 }
 
+// LotByUid - Метод получения лотности инструмента по его Uid
+func (is *InstrumentsServiceClient) LotByUid(uid string) (int64, error) {
+	resp, err := is.InstrumentByUid(uid)
+	if err != nil {
+		return 0, err
+	}
+	return int64(resp.GetInstrument().GetLot()), nil
+}
+
+// LotByFigi - Метод получения лотности инструмента по его FIGI
+func (is *InstrumentsServiceClient) LotByFigi(figi string) (int64, error) {
+	resp, err := is.InstrumentByFigi(figi)
+	if err != nil {
+		return 0, err
+	}
+	return int64(resp.GetInstrument().GetLot()), nil
+}
+
 func (is *InstrumentsServiceClient) instrumentBy(id string, idType pb.InstrumentIdType, classCode string) (*InstrumentResponse, error) {
 	var header, trailer metadata.MD
 	resp, err := is.pbClient.GetInstrumentBy(is.ctx, &pb.InstrumentRequest{

investgo/md_stream.go

@@ -29,29 +29,34 @@ type MarketDataStream struct {
 	subs subscriptions
 }
 
+type candleSub struct {
+	interval     pb.SubscriptionInterval
+	waitingClose bool
+}
+
 type subscriptions struct {
-	candles         map[string]pb.SubscriptionInterval
+	candles         map[string]candleSub
 	orderBooks      map[string]int32
 	trades          map[string]struct{}
 	tradingStatuses map[string]struct{}
 	lastPrices      map[string]struct{}
 }
 
 // SubscribeCandle - Метод подписки на свечи с заданным интервалом
-func (mds *MarketDataStream) SubscribeCandle(ids []string, interval pb.SubscriptionInterval) (<-chan *pb.Candle, error) {
-	err := mds.sendCandlesReq(ids, interval, pb.SubscriptionAction_SUBSCRIPTION_ACTION_SUBSCRIBE)
+func (mds *MarketDataStream) SubscribeCandle(ids []string, interval pb.SubscriptionInterval, waitingClose bool) (<-chan *pb.Candle, error) {
+	err := mds.sendCandlesReq(ids, interval, pb.SubscriptionAction_SUBSCRIPTION_ACTION_SUBSCRIBE, waitingClose)
 	if err != nil {
 		return nil, err
 	}
 	for _, id := range ids {
-		mds.subs.candles[id] = interval
+		mds.subs.candles[id] = candleSub{interval: interval, waitingClose: waitingClose}
 	}
 	return mds.candle, nil
 }
 
 // UnSubscribeCandle - Метод отписки от свечей
-func (mds *MarketDataStream) UnSubscribeCandle(ids []string, interval pb.SubscriptionInterval) error {
-	err := mds.sendCandlesReq(ids, interval, pb.SubscriptionAction_SUBSCRIPTION_ACTION_UNSUBSCRIBE)
+func (mds *MarketDataStream) UnSubscribeCandle(ids []string, interval pb.SubscriptionInterval, waitingClose bool) error {
+	err := mds.sendCandlesReq(ids, interval, pb.SubscriptionAction_SUBSCRIPTION_ACTION_UNSUBSCRIBE, waitingClose)
 	if err != nil {
 		return err
 	}
@@ -61,7 +66,7 @@ func (mds *MarketDataStream) UnSubscribeCandle(ids []string, interval pb.Subscri
 	return nil
 }
 
-func (mds *MarketDataStream) sendCandlesReq(ids []string, interval pb.SubscriptionInterval, act pb.SubscriptionAction) error {
+func (mds *MarketDataStream) sendCandlesReq(ids []string, interval pb.SubscriptionInterval, act pb.SubscriptionAction, waitingClose bool) error {
 	instruments := make([]*pb.CandleInstrument, 0, len(ids))
 	for _, id := range ids {
 		instruments = append(instruments, &pb.CandleInstrument{
@@ -70,14 +75,12 @@ func (mds *MarketDataStream) sendCandlesReq(ids []string, interval pb.Subscripti
 		})
 	}
 
-	WCFlag := interval == pb.SubscriptionInterval_SUBSCRIPTION_INTERVAL_ONE_MINUTE
-
 	return mds.stream.Send(&pb.MarketDataRequest{
 		Payload: &pb.MarketDataRequest_SubscribeCandlesRequest{
 			SubscribeCandlesRequest: &pb.SubscribeCandlesRequest{
 				SubscriptionAction: act,
 				Instruments:        instruments,
-				WaitingClose:       WCFlag,
+				WaitingClose:       waitingClose,
 			}}})
 }
 
@@ -307,14 +310,14 @@ func (mds *MarketDataStream) Stop() {
 func (mds *MarketDataStream) UnSubscribeAll() error {
 	ids := make([]string, 0)
 	if len(mds.subs.candles) > 0 {
-		intervals := make(map[pb.SubscriptionInterval][]string, 0)
+		candleSubs := make(map[candleSub][]string, 0)
 
-		for id, interval := range mds.subs.candles {
-			intervals[interval] = append(intervals[interval], id)
+		for id, c := range mds.subs.candles {
+			candleSubs[c] = append(candleSubs[c], id)
 			delete(mds.subs.candles, id)
 		}
-		for interval, ids := range intervals {
-			err := mds.UnSubscribeCandle(ids, interval)
+		for c, ids := range candleSubs {
+			err := mds.UnSubscribeCandle(ids, c.interval, c.waitingClose)
 			if err != nil {
 				return err
 			}

investgo/md_stream_client.go

@@ -29,7 +29,7 @@ func (c *MarketDataStreamClient) MarketDataStream() (*MarketDataStream, error) {
 		lastPrice:     make(chan *pb.LastPrice, 1),
 		tradingStatus: make(chan *pb.TradingStatus, 1),
 		subs: subscriptions{
-			candles:         make(map[string]pb.SubscriptionInterval, 0),
+			candles:         make(map[string]candleSub, 0),
 			orderBooks:      make(map[string]int32, 0),
 			trades:          make(map[string]struct{}, 0),
 			tradingStatuses: make(map[string]struct{}, 0),

retry/retry.go

@@ -114,7 +114,7 @@ func UnaryClientInterceptorRE(optFuncs ...CallOption) grpc.UnaryClientIntercepto
 			switch {
 			case status.Code(lastErr) == codes.ResourceExhausted:
 				duration, err := durationFromTrailer(trailer)
-				log.Printf("dur = %v\n", duration.String())
+				log.Printf("Resource Exhausted, sleep for %v...\n", duration.String())
 				if err != nil {
 					return err
 				}