aboutsummaryrefslogtreecommitdiffstats
path: root/vendor/xorm.io/builder/writer.go
blob: fb4fae5c5144cb776f4d6851e9734f2ca7f2a0d4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
// Copyright 2019 The Xorm Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package builder

import (
	"io"
	"strings"
)

// Writer defines the interface
type Writer interface {
	io.Writer
	Append(...interface{})
}

var _ Writer = NewWriter()

// BytesWriter implments Writer and save SQL in bytes.Buffer
type BytesWriter struct {
	*strings.Builder
	args []interface{}
}

// NewWriter creates a new string writer
func NewWriter() *BytesWriter {
	w := &BytesWriter{
		Builder: &strings.Builder{},
	}
	return w
}

// Append appends args to Writer
func (w *BytesWriter) Append(args ...interface{}) {
	w.args = append(w.args, args...)
}

// Args returns args
func (w *BytesWriter) Args() []interface{} {
	return w.args
}