9376db5508
Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com> Co-authored-by: marbar3778 <marbar3778@yahoo.com>
45 lines
1.1 KiB
Go
45 lines
1.1 KiB
Go
package postgres
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"cosmossdk.io/schema"
|
|
)
|
|
|
|
// ObjectIndexer is a helper struct that generates SQL for a given object type.
|
|
type ObjectIndexer struct {
|
|
moduleName string
|
|
typ schema.ObjectType
|
|
valueFields map[string]schema.Field
|
|
allFields map[string]schema.Field
|
|
options Options
|
|
}
|
|
|
|
// NewObjectIndexer creates a new ObjectIndexer for the given object type.
|
|
func NewObjectIndexer(moduleName string, typ schema.ObjectType, options Options) *ObjectIndexer {
|
|
allFields := make(map[string]schema.Field)
|
|
valueFields := make(map[string]schema.Field)
|
|
|
|
for _, field := range typ.KeyFields {
|
|
allFields[field.Name] = field
|
|
}
|
|
|
|
for _, field := range typ.ValueFields {
|
|
valueFields[field.Name] = field
|
|
allFields[field.Name] = field
|
|
}
|
|
|
|
return &ObjectIndexer{
|
|
moduleName: moduleName,
|
|
typ: typ,
|
|
allFields: allFields,
|
|
valueFields: valueFields,
|
|
options: options,
|
|
}
|
|
}
|
|
|
|
// TableName returns the name of the table for the object type scoped to its module.
|
|
func (tm *ObjectIndexer) TableName() string {
|
|
return fmt.Sprintf("%s_%s", tm.moduleName, tm.typ.Name)
|
|
}
|