| 79 | } |
| 80 | |
| 81 | func (c Client) ZADD(key string, membersWithScores map[string]float64, flags Flags) (addedMembers []string, err error) { |
| 82 | for member, score := range membersWithScores { |
| 83 | builder := newExpresionBuilder() |
| 84 | builder.updateSetAV(c.skN, zScore{score}.ToAV()) |
| 85 | |
| 86 | if flags.has(IfNotExists) { |
| 87 | builder.addConditionNotExists(c.pk) |
| 88 | } |
| 89 | |
| 90 | if flags.has(IfAlreadyExists) { |
| 91 | builder.addConditionExists(c.pk) |
| 92 | } |
| 93 | |
| 94 | resp, err := c.ddbClient.UpdateItemRequest(&dynamodb.UpdateItemInput{ |
| 95 | ConditionExpression: builder.conditionExpression(), |
| 96 | ExpressionAttributeNames: builder.expressionAttributeNames(), |
| 97 | ExpressionAttributeValues: builder.expressionAttributeValues(), |
| 98 | Key: keyDef{pk: key, sk: member}.toAV(c), |
| 99 | ReturnValues: dynamodb.ReturnValueAllOld, |
| 100 | TableName: aws.String(c.table), |
| 101 | UpdateExpression: builder.updateExpression(), |
| 102 | }).Send(context.TODO()) |
| 103 | if conditionFailureError(err) { |
| 104 | continue |
| 105 | } |
| 106 | |
| 107 | if err != nil { |
| 108 | return addedMembers, err |
| 109 | } |
| 110 | |
| 111 | if len(resp.Attributes) == 0 { |
| 112 | addedMembers = append(addedMembers, member) |
| 113 | } |
| 114 | } |
| 115 | |
| 116 | return |
| 117 | } |
| 118 | |
| 119 | func (c Client) ZCARD(key string) (count int64, err error) { |
| 120 | return c.HLEN(key) |