yx-dataset-server/vendor/github.com/open-dingtalk/dingtalk-stream-sdk-go/client/option.go

package client

import "github.com/open-dingtalk/dingtalk-stream-sdk-go/handler"

/**
 * @Author linya.jj
 * @Date 2023/3/22 14:48
 */

type ClientOption func(*StreamClient)

func WithAutoReconnect(autoReconnect bool) ClientOption {
	return func(c *StreamClient) {
		c.AutoReconnect = autoReconnect
	}
}

func WithAppCredential(cred *AppCredentialConfig) ClientOption {
	return func(c *StreamClient) {
		c.AppCredential = cred
	}
}

func WithSubscription(stype, stopic string, frameHandler handler.IFrameHandler) ClientOption {
	return func(c *StreamClient) {
		c.RegisterRouter(stype, stopic, frameHandler)
	}
}

func WithUserAgent(ua *UserAgentConfig) ClientOption {
	return func(c *StreamClient) {
		if ua.Valid() != nil {
			ua = NewDingtalkGoSDKUserAgent()
		}

		c.UserAgent = ua
	}
}

func WithExtras(extras map[string]string) ClientOption {
	return func(c *StreamClient) {
		c.extras = extras
	}
}

func WithOpenApiHost(host string) ClientOption {
	return func(c *StreamClient) {
		c.openApiHost = host
	}
}

func WithProxy(proxy string) ClientOption {
	return func(c *StreamClient) {
		c.proxy = proxy
	}
}