-
Notifications
You must be signed in to change notification settings - Fork 340
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #85 from Azure-Samples/sql_api
Adding support for Azure CosmosDB SQL API
- Loading branch information
Showing
9 changed files
with
708 additions
and
286 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,196 @@ | ||
| package main | ||
|
|
||
| import ( | ||
| "context" | ||
| "encoding/json" | ||
| "log" | ||
| "strings" | ||
|
|
||
| "github.com/Azure/azure-sdk-for-go/sdk/data/azcosmos" | ||
| "github.com/gofrs/uuid" | ||
| ) | ||
|
|
||
| type PartitionKey struct { | ||
| Key string | ||
| Value string | ||
| } | ||
|
|
||
| type CosmosDBOrderRepo struct { | ||
| db *azcosmos.ContainerClient | ||
| partitionKey PartitionKey | ||
| } | ||
|
|
||
| func NewCosmosDBOrderRepo(cosmosDbEndpoint string, dbName string, containerName string, cosmosDbKey string, partitionKey PartitionKey) (*CosmosDBOrderRepo, error) { | ||
| cred, err := azcosmos.NewKeyCredential(cosmosDbKey) | ||
| if err != nil { | ||
| log.Printf("failed to create cosmosdb key credential: %v\n", err) | ||
| return nil, err | ||
| } | ||
|
|
||
| // create a cosmos client | ||
| client, err := azcosmos.NewClientWithKey(cosmosDbEndpoint, cred, nil) | ||
| if err != nil { | ||
| log.Printf("failed to create cosmosdb client: %v\n", err) | ||
| return nil, err | ||
| } | ||
|
|
||
| // create a cosmos container | ||
| container, err := client.NewContainer(dbName, containerName) | ||
| if err != nil { | ||
| log.Printf("failed to create cosmosdb container: %v\n", err) | ||
| return nil, err | ||
| } | ||
|
|
||
| return &CosmosDBOrderRepo{container, partitionKey}, nil | ||
| } | ||
|
|
||
| func (r *CosmosDBOrderRepo) GetPendingOrders() ([]Order, error) { | ||
| var orders []Order | ||
|
|
||
| pk := azcosmos.NewPartitionKeyString(r.partitionKey.Value) | ||
| opt := &azcosmos.QueryOptions{ | ||
| QueryParameters: []azcosmos.QueryParameter{ | ||
| {Name: "@status", Value: Pending}, | ||
| }, | ||
| } | ||
| queryPager := r.db.NewQueryItemsPager("SELECT * FROM o WHERE o.status = @status", pk, opt) | ||
|
|
||
| for queryPager.More() { | ||
| queryResponse, err := queryPager.NextPage(context.Background()) | ||
| if err != nil { | ||
| log.Printf("failed to get next page: %v\n", err) | ||
| return nil, err | ||
| } | ||
|
|
||
| for _, item := range queryResponse.Items { | ||
| var order Order | ||
| err := json.Unmarshal(item, &order) | ||
| if err != nil { | ||
| log.Printf("failed to deserialize order: %v\n", err) | ||
| return nil, err | ||
| } | ||
| orders = append(orders, order) | ||
| } | ||
| } | ||
| return orders, nil | ||
| } | ||
|
|
||
| func (r *CosmosDBOrderRepo) GetOrder(id string) (Order, error) { | ||
| pk := azcosmos.NewPartitionKeyString(r.partitionKey.Value) | ||
| opt := &azcosmos.QueryOptions{ | ||
| QueryParameters: []azcosmos.QueryParameter{ | ||
| {Name: "@orderId", Value: id}, | ||
| }, | ||
| } | ||
| queryPager := r.db.NewQueryItemsPager("SELECT * FROM o WHERE o.orderId = @orderId", pk, opt) | ||
|
|
||
| for queryPager.More() { | ||
| queryResponse, err := queryPager.NextPage(context.Background()) | ||
| if err != nil { | ||
| log.Printf("failed to get next page: %v\n", err) | ||
| return Order{}, err | ||
| } | ||
|
|
||
| for _, item := range queryResponse.Items { | ||
| var order Order | ||
| err := json.Unmarshal(item, &order) | ||
| if err != nil { | ||
| log.Printf("failed to deserialize order: %v\n", err) | ||
| return Order{}, err | ||
| } | ||
| return order, nil | ||
| } | ||
| } | ||
| return Order{}, nil | ||
| } | ||
|
|
||
| func (r *CosmosDBOrderRepo) InsertOrders(orders []Order) error { | ||
| var counter = 0 | ||
|
|
||
| for _, o := range orders { | ||
| pk := azcosmos.NewPartitionKeyString(r.partitionKey.Value) | ||
|
|
||
| marshalledOrder, err := json.Marshal(o) | ||
| if err != nil { | ||
| log.Printf("failed to marshal order: %v\n", err) | ||
| return err | ||
| } | ||
|
|
||
| var order map[string]interface{} | ||
| err = json.Unmarshal(marshalledOrder, &order) | ||
| if err != nil { | ||
| log.Printf("failed to unmarshal order: %v\n", err) | ||
| return err | ||
| } | ||
|
|
||
| // add id with value of uuid.NewV4() to marhsalled order | ||
| uuidWithHyphen, err := uuid.NewV4() | ||
| if err != nil { | ||
| log.Printf("failed to generate uuid: %v\n", err) | ||
| return err | ||
| } | ||
| uuid := strings.Replace(uuidWithHyphen.String(), "-", "", -1) | ||
| order["id"] = uuid | ||
|
|
||
| order[r.partitionKey.Key] = r.partitionKey.Value | ||
|
|
||
| marshalledOrder, err = json.Marshal(order) | ||
| if err != nil { | ||
| log.Printf("failed to marshal order: %v\n", err) | ||
| return err | ||
| } | ||
|
|
||
| _, err = r.db.CreateItem(context.Background(), pk, marshalledOrder, nil) | ||
| if err != nil { | ||
| log.Printf("failed to create item: %v\n", err) | ||
| return err | ||
| } | ||
|
|
||
| // increment counter for each order inserted | ||
| counter++ | ||
| } | ||
|
|
||
| log.Printf("Inserted %v documents into database\n", counter) | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| func (r *CosmosDBOrderRepo) UpdateOrder(order Order) error { | ||
| var existingOrderId string | ||
| pk := azcosmos.NewPartitionKeyString(r.partitionKey.Value) | ||
| opt := &azcosmos.QueryOptions{ | ||
| QueryParameters: []azcosmos.QueryParameter{ | ||
| {Name: "@orderId", Value: order.OrderID}, | ||
| }, | ||
| } | ||
| queryPager := r.db.NewQueryItemsPager("SELECT * FROM o WHERE o.orderId = @orderId", pk, opt) | ||
|
|
||
| for queryPager.More() { | ||
| queryResponse, err := queryPager.NextPage(context.Background()) | ||
| if err != nil { | ||
| break | ||
| } | ||
|
|
||
| for _, item := range queryResponse.Items { | ||
| var order map[string]interface{} | ||
| err = json.Unmarshal(item, &order) | ||
| if err != nil { | ||
| log.Printf("failed to deserialize order: %v\n", err) | ||
| return err | ||
| } | ||
| existingOrderId = order["id"].(string) | ||
| break | ||
| } | ||
| } | ||
|
|
||
| patch := azcosmos.PatchOperations{} | ||
| patch.AppendReplace("/status", order.Status) | ||
|
|
||
| _, err := r.db.PatchItem(context.Background(), pk, existingOrderId, patch, nil) | ||
| if err != nil { | ||
| log.Printf("failed to replace item: %v\n", err) | ||
| return err | ||
| } | ||
|
|
||
| return nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.