init repo

vendor/github.com/alibabacloud-go/tea/utils/assert.go

@@ -0,0 +1,64 @@
+package utils
+
+import (
+	"reflect"
+	"strings"
+	"testing"
+)
+
+func isNil(object interface{}) bool {
+	if object == nil {
+		return true
+	}
+
+	value := reflect.ValueOf(object)
+	kind := value.Kind()
+	isNilableKind := containsKind(
+		[]reflect.Kind{
+			reflect.Chan, reflect.Func,
+			reflect.Interface, reflect.Map,
+			reflect.Ptr, reflect.Slice},
+		kind)
+
+	if isNilableKind && value.IsNil() {
+		return true
+	}
+
+	return false
+}
+
+func containsKind(kinds []reflect.Kind, kind reflect.Kind) bool {
+	for i := 0; i < len(kinds); i++ {
+		if kind == kinds[i] {
+			return true
+		}
+	}
+
+	return false
+}
+
+func AssertEqual(t *testing.T, a, b interface{}) {
+	if !reflect.DeepEqual(a, b) {
+		t.Errorf("%v != %v", a, b)
+	}
+}
+
+func AssertNil(t *testing.T, object interface{}) {
+	if !isNil(object) {
+		t.Errorf("%v is not nil", object)
+	}
+}
+
+func AssertNotNil(t *testing.T, object interface{}) {
+	if isNil(object) {
+		t.Errorf("%v is nil", object)
+	}
+}
+
+func AssertContains(t *testing.T, contains string, msgAndArgs ...string) {
+	for _, value := range msgAndArgs {
+		if ok := strings.Contains(contains, value); !ok {
+			t.Errorf("%s does not contain %s", contains, value)
+		}
+	}
+}

vendor/github.com/alibabacloud-go/tea/utils/logger.go

@@ -0,0 +1,109 @@
+package utils
+
+import (
+	"io"
+	"log"
+	"strings"
+	"time"
+)
+
+type Logger struct {
+	*log.Logger
+	formatTemplate string
+	isOpen         bool
+	lastLogMsg     string
+}
+
+var defaultLoggerTemplate = `{time} {channel}: "{method} {uri} HTTP/{version}" {code} {cost} {hostname}`
+var loggerParam = []string{"{time}", "{start_time}", "{ts}", "{channel}", "{pid}", "{host}", "{method}", "{uri}", "{version}", "{target}", "{hostname}", "{code}", "{error}", "{req_headers}", "{res_body}", "{res_headers}", "{cost}"}
+var logChannel string
+
+func InitLogMsg(fieldMap map[string]string) {
+	for _, value := range loggerParam {
+		fieldMap[value] = ""
+	}
+}
+
+func (logger *Logger) SetFormatTemplate(template string) {
+	logger.formatTemplate = template
+
+}
+
+func (logger *Logger) GetFormatTemplate() string {
+	return logger.formatTemplate
+
+}
+
+func NewLogger(level string, channel string, out io.Writer, template string) *Logger {
+	if level == "" {
+		level = "info"
+	}
+
+	logChannel = "AlibabaCloud"
+	if channel != "" {
+		logChannel = channel
+	}
+	log := log.New(out, "["+strings.ToUpper(level)+"]", log.Lshortfile)
+	if template == "" {
+		template = defaultLoggerTemplate
+	}
+
+	return &Logger{
+		Logger:         log,
+		formatTemplate: template,
+		isOpen:         true,
+	}
+}
+
+func (logger *Logger) OpenLogger() {
+	logger.isOpen = true
+}
+
+func (logger *Logger) CloseLogger() {
+	logger.isOpen = false
+}
+
+func (logger *Logger) SetIsopen(isopen bool) {
+	logger.isOpen = isopen
+}
+
+func (logger *Logger) GetIsopen() bool {
+	return logger.isOpen
+}
+
+func (logger *Logger) SetLastLogMsg(lastLogMsg string) {
+	logger.lastLogMsg = lastLogMsg
+}
+
+func (logger *Logger) GetLastLogMsg() string {
+	return logger.lastLogMsg
+}
+
+func SetLogChannel(channel string) {
+	logChannel = channel
+}
+
+func (logger *Logger) PrintLog(fieldMap map[string]string, err error) {
+	if err != nil {
+		fieldMap["{error}"] = err.Error()
+	}
+	fieldMap["{time}"] = time.Now().Format("2006-01-02 15:04:05")
+	fieldMap["{ts}"] = getTimeInFormatISO8601()
+	fieldMap["{channel}"] = logChannel
+	if logger != nil {
+		logMsg := logger.formatTemplate
+		for key, value := range fieldMap {
+			logMsg = strings.Replace(logMsg, key, value, -1)
+		}
+		logger.lastLogMsg = logMsg
+		if logger.isOpen == true {
+			logger.Output(2, logMsg)
+		}
+	}
+}
+
+func getTimeInFormatISO8601() (timeStr string) {
+	gmt := time.FixedZone("GMT", 0)
+
+	return time.Now().In(gmt).Format("2006-01-02T15:04:05Z")
+}

vendor/github.com/alibabacloud-go/tea/utils/progress.go

@@ -0,0 +1,60 @@
+package utils
+
+// ProgressEventType defines transfer progress event type
+type ProgressEventType int
+
+const (
+	// TransferStartedEvent transfer started, set TotalBytes
+	TransferStartedEvent ProgressEventType = 1 + iota
+	// TransferDataEvent transfer data, set ConsumedBytes anmd TotalBytes
+	TransferDataEvent
+	// TransferCompletedEvent transfer completed
+	TransferCompletedEvent
+	// TransferFailedEvent transfer encounters an error
+	TransferFailedEvent
+)
+
+// ProgressEvent defines progress event
+type ProgressEvent struct {
+	ConsumedBytes int64
+	TotalBytes    int64
+	RwBytes       int64
+	EventType     ProgressEventType
+}
+
+// ProgressListener listens progress change
+type ProgressListener interface {
+	ProgressChanged(event *ProgressEvent)
+}
+
+// -------------------- Private --------------------
+
+func NewProgressEvent(eventType ProgressEventType, consumed, total int64, rwBytes int64) *ProgressEvent {
+	return &ProgressEvent{
+		ConsumedBytes: consumed,
+		TotalBytes:    total,
+		RwBytes:       rwBytes,
+		EventType:     eventType}
+}
+
+// publishProgress
+func PublishProgress(listener ProgressListener, event *ProgressEvent) {
+	if listener != nil && event != nil {
+		listener.ProgressChanged(event)
+	}
+}
+
+func GetProgressListener(obj interface{}) ProgressListener {
+	if obj == nil {
+		return nil
+	}
+	listener, ok := obj.(ProgressListener)
+	if !ok {
+		return nil
+	}
+	return listener
+}
+
+type ReaderTracker struct {
+	CompletedBytes int64
+}