2020-03-13 17:21:30 +08:00
|
|
|
// Copyright 2017 gf Author(https://github.com/gogf/gf). All Rights Reserved.
|
|
|
|
//
|
|
|
|
// This Source Code Form is subject to the terms of the MIT License.
|
|
|
|
// If a copy of the MIT was not distributed with this file,
|
|
|
|
// You can obtain one at https://github.com/gogf/gf.
|
|
|
|
|
|
|
|
package gdb
|
|
|
|
|
|
|
|
import (
|
2020-04-27 21:18:42 +08:00
|
|
|
"fmt"
|
2020-03-13 17:21:30 +08:00
|
|
|
"github.com/gogf/gf/container/gset"
|
|
|
|
"github.com/gogf/gf/text/gstr"
|
|
|
|
)
|
|
|
|
|
|
|
|
// Filter marks filtering the fields which does not exist in the fields of the operated table.
|
2020-06-04 17:29:16 +08:00
|
|
|
// Note that this function supports only single table operations.
|
2020-03-13 17:21:30 +08:00
|
|
|
func (m *Model) Filter() *Model {
|
|
|
|
if gstr.Contains(m.tables, " ") {
|
|
|
|
panic("function Filter supports only single table operations")
|
|
|
|
}
|
|
|
|
model := m.getModel()
|
|
|
|
model.filter = true
|
|
|
|
return model
|
|
|
|
}
|
|
|
|
|
|
|
|
// Fields sets the operation fields of the model, multiple fields joined using char ','.
|
|
|
|
func (m *Model) Fields(fields string) *Model {
|
|
|
|
model := m.getModel()
|
|
|
|
model.fields = fields
|
|
|
|
return model
|
|
|
|
}
|
|
|
|
|
|
|
|
// FieldsEx sets the excluded operation fields of the model, multiple fields joined using char ','.
|
2020-06-04 17:29:16 +08:00
|
|
|
// Note that this function supports only single table operations.
|
2020-03-13 17:21:30 +08:00
|
|
|
func (m *Model) FieldsEx(fields string) *Model {
|
|
|
|
if gstr.Contains(m.tables, " ") {
|
|
|
|
panic("function FieldsEx supports only single table operations")
|
|
|
|
}
|
2020-04-27 21:18:42 +08:00
|
|
|
tableFields, err := m.db.TableFields(m.tables)
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
if len(tableFields) == 0 {
|
|
|
|
panic(fmt.Sprintf(`empty table fields for table "%s"`, m.tables))
|
|
|
|
}
|
2020-03-13 17:21:30 +08:00
|
|
|
model := m.getModel()
|
|
|
|
model.fieldsEx = fields
|
|
|
|
fieldsExSet := gset.NewStrSetFrom(gstr.SplitAndTrim(fields, ","))
|
2020-04-27 21:18:42 +08:00
|
|
|
fieldsArray := make([]string, len(tableFields))
|
|
|
|
for k, v := range tableFields {
|
|
|
|
fieldsArray[v.Index] = k
|
|
|
|
}
|
|
|
|
model.fields = ""
|
|
|
|
for _, k := range fieldsArray {
|
|
|
|
if fieldsExSet.Contains(k) {
|
|
|
|
continue
|
2020-03-13 17:21:30 +08:00
|
|
|
}
|
2020-04-27 21:18:42 +08:00
|
|
|
if len(model.fields) > 0 {
|
|
|
|
model.fields += ","
|
|
|
|
}
|
|
|
|
model.fields += k
|
2020-03-13 17:21:30 +08:00
|
|
|
}
|
2020-04-27 21:18:42 +08:00
|
|
|
model.fields = model.db.QuoteString(model.fields)
|
2020-03-13 17:21:30 +08:00
|
|
|
return model
|
|
|
|
}
|
|
|
|
|
|
|
|
// FieldsStr retrieves and returns all fields from the table, joined with char ','.
|
|
|
|
// The optional parameter <prefix> specifies the prefix for each field, eg: FieldsStr("u.").
|
|
|
|
func (m *Model) FieldsStr(prefix ...string) string {
|
|
|
|
prefixStr := ""
|
|
|
|
if len(prefix) > 0 {
|
|
|
|
prefixStr = prefix[0]
|
|
|
|
}
|
2020-04-27 21:18:42 +08:00
|
|
|
tableFields, err := m.db.TableFields(m.tables)
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
if len(tableFields) == 0 {
|
|
|
|
panic(fmt.Sprintf(`empty table fields for table "%s"`, m.tables))
|
|
|
|
}
|
|
|
|
fieldsArray := make([]string, len(tableFields))
|
|
|
|
for k, v := range tableFields {
|
|
|
|
fieldsArray[v.Index] = k
|
|
|
|
}
|
|
|
|
newFields := ""
|
|
|
|
for _, k := range fieldsArray {
|
|
|
|
if len(newFields) > 0 {
|
|
|
|
newFields += ","
|
2020-03-13 17:21:30 +08:00
|
|
|
}
|
2020-04-27 21:18:42 +08:00
|
|
|
newFields += prefixStr + k
|
2020-03-13 17:21:30 +08:00
|
|
|
}
|
2020-04-27 21:18:42 +08:00
|
|
|
newFields = m.db.QuoteString(newFields)
|
|
|
|
return newFields
|
2020-03-13 17:21:30 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// FieldsExStr retrieves and returns fields which are not in parameter <fields> from the table,
|
|
|
|
// joined with char ','.
|
|
|
|
// The parameter <fields> specifies the fields that are excluded.
|
|
|
|
// The optional parameter <prefix> specifies the prefix for each field, eg: FieldsExStr("id", "u.").
|
|
|
|
func (m *Model) FieldsExStr(fields string, prefix ...string) string {
|
|
|
|
prefixStr := ""
|
|
|
|
if len(prefix) > 0 {
|
|
|
|
prefixStr = prefix[0]
|
|
|
|
}
|
2020-04-27 21:18:42 +08:00
|
|
|
tableFields, err := m.db.TableFields(m.tables)
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
if len(tableFields) == 0 {
|
|
|
|
panic(fmt.Sprintf(`empty table fields for table "%s"`, m.tables))
|
|
|
|
}
|
|
|
|
fieldsExSet := gset.NewStrSetFrom(gstr.SplitAndTrim(fields, ","))
|
|
|
|
fieldsArray := make([]string, len(tableFields))
|
|
|
|
for k, v := range tableFields {
|
|
|
|
fieldsArray[v.Index] = k
|
|
|
|
}
|
|
|
|
newFields := ""
|
|
|
|
for _, k := range fieldsArray {
|
|
|
|
if fieldsExSet.Contains(k) {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
if len(newFields) > 0 {
|
|
|
|
newFields += ","
|
2020-03-13 17:21:30 +08:00
|
|
|
}
|
2020-04-27 21:18:42 +08:00
|
|
|
newFields += prefixStr + k
|
2020-03-13 17:21:30 +08:00
|
|
|
}
|
2020-04-27 21:18:42 +08:00
|
|
|
newFields = m.db.QuoteString(newFields)
|
|
|
|
return newFields
|
2020-03-13 17:21:30 +08:00
|
|
|
}
|