diff options
Diffstat (limited to 'vendor/xorm.io/builder/builder_update.go')
-rw-r--r-- | vendor/xorm.io/builder/builder_update.go | 46 |
1 files changed, 46 insertions, 0 deletions
diff --git a/vendor/xorm.io/builder/builder_update.go b/vendor/xorm.io/builder/builder_update.go new file mode 100644 index 0000000000..37b4551526 --- /dev/null +++ b/vendor/xorm.io/builder/builder_update.go @@ -0,0 +1,46 @@ +// Copyright 2016 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 ( + "fmt" +) + +// Update creates an update Builder +func Update(updates ...Eq) *Builder { + builder := &Builder{cond: NewCond()} + return builder.Update(updates...) +} + +func (b *Builder) updateWriteTo(w Writer) error { + if len(b.from) <= 0 { + return ErrNoTableName + } + if len(b.updates) <= 0 { + return ErrNoColumnToUpdate + } + + if _, err := fmt.Fprintf(w, "UPDATE %s SET ", b.from); err != nil { + return err + } + + for i, s := range b.updates { + if err := s.opWriteTo(",", w); err != nil { + return err + } + + if i != len(b.updates)-1 { + if _, err := fmt.Fprint(w, ","); err != nil { + return err + } + } + } + + if _, err := fmt.Fprint(w, " WHERE "); err != nil { + return err + } + + return b.cond.WriteTo(w) +} |