pingginp / cqlc Goto Github PK

func Map{{$StructType}}(iter *gocql.Iter, callback func(t {{$StructType}}) (bool, error)) error {
        columns := iter.Columns()
        row := make([]interface{}, len(columns))

        for {
            t := {{$StructType}}{}

            for i := 0; i < len(columns); i++ {
                switch columns[i].Name {
                {{range $_, $col := $cf.Columns}}
                    case "{{$col.Name}}": row[i] = &t.{{snakeToCamel $col.Name}}
                {{end}}
                default:
                    log.Fatal("unhandled column: ", columns[i].Name)
                }
            }
    }

const (
	NoCollectionType CollectionType = iota
	ListType
	SetType
	MapType
)

func appendList(c *Context, col ListColumn, values interface{}) {
	b := ColumnBinding{Column: col, Value: values, Incremental: true, CollectionType: ListType, CollectionOperationType: Append}
	c.Bindings = append(c.Bindings, b)
}

func prependList(c *Context, col ListColumn, values interface{}) {
	b := ColumnBinding{Column: col, Value: values, Incremental: true, CollectionType: ListType, CollectionOperationType: Prepend}
	c.Bindings = append(c.Bindings, b)
}

func removeList(c *Context, col ListColumn, values interface{}) {
	b := ColumnBinding{Column: col, Value: values, Incremental: true, CollectionType: ListType, CollectionOperationType: RemoveByValue}
	c.Bindings = append(c.Bindings, b)
}

func renderUpdate(ctx *Context, buf *bytes.Buffer, counterTable bool) {
           // ....
              switch binding.CollectionType {
			case ListType:
				switch binding.CollectionOperationType {
				case Append:
					setFragments[i] = fmt.Sprintf("%s = %s + ?", col, col)
				case Prepend:
					setFragments[i] = fmt.Sprintf("%s = ? + %s", col, col)
				case RemoveByValue:
					setFragments[i] = fmt.Sprintf("%s = %s - ?", col, col)
				}
			default:
				setFragments[i] = fmt.Sprintf("%s = ?", col)
			}
}