Reputation: 53
What is the best way of writing a query based on filter inputs(vendor_id, customer_id, order_status_id ... or all nonempt field at the same time) in one function(Golang, Gocql, Cassandra)? I don't want to write new function in every case like the below function which filter only by customerID:
func (repo *orderRepo) FindByCustomerId(req *pb.FindRequest) (*pb.OrdersResponse, error) {
offset := req.GetOffset()
limit := req.GetLimit()
customerID := req.GetCustomerId()
//vendorID := req.GetVendorId()
//orderStatusID := req.GetOrderStatusId()
var (
orders []*pb.Order
count int32
)
if err := repo.session.Query(`SELECT count(1) FROM orders WHERE customer_id = ?`, customerID).
Scan(&count); err != nil {
return nil, err
}
if offset < count {
query := repo.session.Query(`SELECT
id,
customer_id,
customer_name,
vendor_id,
toJSON(items),
paid_at,
created_at,
order_status_id
FROM orders WHERE customer_id = ?`, customerID)
defer query.Release()
var ps []byte
iter := query.PageState(ps).PageSize(int(offset % limit)).Iter()
ps = iter.PageState()
page := offset / limit
var i int32
for i = 0; i < page; i++ {
iter = query.PageState(ps).PageSize(int(limit)).Iter()
ps = iter.PageState()
}
iter = query.PageState(ps).PageSize(int(limit)).Iter()
rows := iter.Scanner()
for rows.Next() {
var (
order pb.Order
createdAt time.Time
paidAt time.Time
itemsJSON []byte
)
if err := rows.Scan(
&order.Id,
&order.CustomerId,
&order.CustomerName,
&order.VendorId,
&itemsJSON,
&paidAt,
&createdAt,
&order.OrderStatusId,
); err != nil {
return nil, err
}
if err := json.Unmarshal(itemsJSON, &order.Items); err != nil {
return nil, err
}
order.PaidAt = paidAt.String()
order.CreatedAt = createdAt.String()
orders = append(orders, &order)
}
if err := iter.Close(); err != nil {
return nil, err
}
}
return &pb.OrdersResponse{
Orders: orders,
Count: count,
}, nil
}
I need to add parameters dynamically to the above query
Upvotes: 2
Views: 809
Reputation: 53
Here is my solution to my own question that I arose in 3 days:) I think that would help others)
func (repo *orderRepo) Find(req *pb.FindRequest) (*pb.OrdersResponse, error) {
offset := req.GetOffset()
limit := req.GetLimit()
vendorID := req.GetVendorId()
customerID := req.GetCustomerId()
orderStatusID := req.GetOrderStatusId()
var (
orders []*pb.Order
count int32
)
var filter string
var values []interface{}
if len(vendorID) > 0 {
if len(filter) > 0 {
filter = filter + " AND vendor_id = ?"
} else {
filter = " WHERE vendor_id = ?"
}
values = append(values, vendorID)
}
if len(customerID) > 0 {
if len(filter) > 0 {
filter = filter + " AND customer_id = ?"
} else {
filter = " WHERE customer_id = ?"
}
values = append(values, customerID)
}
if len(orderStatusID) > 0 {
if len(filter) > 0 {
filter = filter + " AND order_status_id = ?"
} else {
filter = " WHERE order_status_id = ?"
}
values = append(values, orderStatusID)
}
fmt.Println(filter)
if err := repo.session.Query(`SELECT count(1) FROM orders`+filter, values...).
Scan(&count); err != nil {
return nil, err
}
if offset < count {
query := repo.session.Query(`SELECT
id,
customer_id,
customer_name,
vendor_id,
toJSON(items),
paid_at,
created_at,
order_status_id
FROM orders`+filter, values...)
defer query.Release()
var ps []byte
iter := query.PageState(ps).PageSize(int(offset % limit)).Iter()
ps = iter.PageState()
page := offset / limit
var i int32
for i = 0; i < page; i++ {
iter = query.PageState(ps).PageSize(int(limit)).Iter()
ps = iter.PageState()
}
iter = query.PageState(ps).PageSize(int(limit)).Iter()
rows := iter.Scanner()
for rows.Next() {
var (
order pb.Order
createdAt time.Time
paidAt time.Time
itemsJSON []byte
)
if err := rows.Scan(
&order.Id,
&order.CustomerId,
&order.CustomerName,
&order.VendorId,
&itemsJSON,
&paidAt,
&createdAt,
&order.OrderStatusId,
); err != nil {
return nil, err
}
if err := json.Unmarshal(itemsJSON, &order.Items); err != nil {
return nil, err
}
order.PaidAt = paidAt.String()
order.CreatedAt = createdAt.String()
orders = append(orders, &order)
}
if err := iter.Close(); err != nil {
return nil, err
}
}
return &pb.OrdersResponse{
Orders: orders,
Count: count,
}, nil
}
Upvotes: 0