コード例 #1
0
ファイル: fake.go プロジェクト: waterlink/rebecca
// Where is for fetching specific records
func (d *Driver) Where(tablename string, fields []field.Field, ctx context.Context, where string, args ...interface{}) ([][]field.Field, error) {
	if tx := ctx.GetTx(); tx != nil {
		return tx.(*Driver).Where(tablename, fields, ctx.SetTx(nil), where, args...)
	}

	result := [][]field.Field{}

	fn, ok := d.whereRegistry[where]
	if !ok {
		return nil, fmt.Errorf(
			"Fake driver has no '%s' where query registerd, please register with RegisterWhere",
			where,
		)
	}

	for _, record := range d.getTable(tablename) {
		ok, err := fn(record, args...)
		if err != nil {
			return nil, fmt.Errorf("Registered query '%s' returned error - %s", where, err)
		}

		if ok {
			result = append(result, record)
		}
	}

	return result, nil
}
コード例 #2
0
ファイル: fake.go プロジェクト: waterlink/rebecca
// All is for fetching all records
func (d *Driver) All(tablename string, fields []field.Field, ctx context.Context) ([][]field.Field, error) {
	if tx := ctx.GetTx(); tx != nil {
		return tx.(*Driver).All(tablename, fields, ctx.SetTx(nil))
	}

	return d.getTable(tablename), nil
}
コード例 #3
0
ファイル: pg.go プロジェクト: waterlink/rebecca
// All is for fetching all records in current context
func (d *Driver) All(tablename string, fields []field.Field, ctx context.Context) ([][]field.Field, error) {
	names := fieldNames(fields)

	query := "SELECT %s FROM %s %s"
	query = fmt.Sprintf(query, namesRepr(names), tablename, contextFor(ctx))

	return d.readRows(ctx.GetTx(), fields, query)
}
コード例 #4
0
ファイル: pg.go プロジェクト: waterlink/rebecca
// First is for fetching only first specific record from current context matching given where query and arguments
func (d *Driver) First(tablename string, fields []field.Field, ctx context.Context, where string, args ...interface{}) ([]field.Field, error) {
	firstCtx := ctx.SetLimit(1)
	names := fieldNames(fields)

	query := "SELECT %s FROM %s WHERE %s %s"
	query = fmt.Sprintf(query, namesRepr(names), tablename, where, contextFor(firstCtx))
	return d.readRow(ctx.GetTx(), fields, query, args...)
}
コード例 #5
0
ファイル: fake.go プロジェクト: waterlink/rebecca
// First is for fetching first specific record
func (d *Driver) First(tablename string, fields []field.Field, ctx context.Context, where string, args ...interface{}) ([]field.Field, error) {
	if tx := ctx.GetTx(); tx != nil {
		return tx.(*Driver).First(tablename, fields, ctx.SetTx(nil), where, args...)
	}

	records, err := d.Where(tablename, fields, ctx, where, args...)
	if err != nil {
		return nil, fmt.Errorf("Unable to get first record - %s", err)
	}

	if len(records) == 0 {
		return nil, fmt.Errorf("Record not found with where query '%s'", where)
	}

	return records[0], nil
}
コード例 #6
0
ファイル: pg.go プロジェクト: waterlink/rebecca
func contextFor(ctx context.Context) string {
	queryCtx := ""

	if group := ctx.GetGroup(); group != "" {
		queryCtx = queryCtx + fmt.Sprintf(" GROUP BY %s", group)
	}

	if order := ctx.GetOrder(); order != "" {
		queryCtx = queryCtx + fmt.Sprintf(" ORDER BY %s", order)
	}

	if limit := ctx.GetLimit(); limit > 0 {
		queryCtx = queryCtx + fmt.Sprintf(" LIMIT %d", limit)
	}

	if skip := ctx.GetSkip(); skip > 0 {
		queryCtx = queryCtx + fmt.Sprintf(" OFFSET %d", skip)
	}

	return queryCtx
}