Commit to metadata table for state analysis (#106)

* Commit to metadata table

* Fix schema

* Fix schema

* Fix schema

Author: yzang2019

tools/cmd/seidb/operations/state_size.go

@@ -25,7 +25,7 @@ func StateSizeCmd() *cobra.Command {
 
 	// DynamoDB export flags
 	cmd.PersistentFlags().Bool("export-dynamodb", false, "Export results to DynamoDB instead of printing")
-	cmd.PersistentFlags().String("dynamodb-table", "state-size-analysis", "DynamoDB table name")
+	cmd.PersistentFlags().String("dynamodb-table", "state_size_analysis", "DynamoDB table name")
 	cmd.PersistentFlags().String("aws-region", "us-east-2", "AWS region for DynamoDB")
 
 	return cmd
@@ -218,7 +218,9 @@ func exportResultsToDynamoDB(moduleResults map[string]*ModuleResult, height int6
 		return fmt.Errorf("failed to export analyses to DynamoDB: %w", err)
 	}
 
-	return nil
+	metadataTableName := tableName + "_metadata"
+	_, err = dynamoClient.UpdateLatestHeightIfGreater(metadataTableName, height)
+	return err
 }
 
 // printResultsToConsole prints the collected results to console

tools/utils/dynamo.go

@@ -1,9 +1,11 @@
 package utils
 
 import (
+	"errors"
 	"fmt"
 
 	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/aws/awserr"
 	"github.com/aws/aws-sdk-go/aws/session"
 	"github.com/aws/aws-sdk-go/service/dynamodb"
 	"github.com/aws/aws-sdk-go/service/dynamodb/dynamodbattribute"
@@ -50,6 +52,7 @@ type ContractSizeEntry struct {
 	KeyCount  uint64 `json:"key_count"`
 }
 
+// PrefixSize is a helper structure kept for completeness (unused here)
 type PrefixSize struct {
 	KeySize   uint64 `json:"key_size"`
 	ValueSize uint64 `json:"value_size"`
@@ -88,3 +91,32 @@ func (d *DynamoDBClient) ExportMultipleAnalyses(analyses []*StateSizeAnalysis) e
 	}
 	return nil
 }
+
+// UpdateLatestHeightIfGreater keeps exactly one item in the metadata table using the schema:
+// Partition key: keyname (S). Numeric attribute: height (N) stores the latest height.
+// Upserts the row keyname = "latest_height" and sets height = :h only if missing or lower.
+func (d *DynamoDBClient) UpdateLatestHeightIfGreater(metadataTable string, height int64) (bool, error) {
+	input := &dynamodb.UpdateItemInput{
+		TableName: aws.String(metadataTable),
+		Key: map[string]*dynamodb.AttributeValue{
+			"keyname": {S: aws.String("latest_height")},
+		},
+		UpdateExpression:    aws.String("SET height = :h"),
+		ConditionExpression: aws.String("attribute_not_exists(height) OR height < :h"),
+		ExpressionAttributeValues: map[string]*dynamodb.AttributeValue{
+			":h": {N: aws.String(fmt.Sprintf("%d", height))},
+		},
+		ReturnValues: aws.String("NONE"),
+	}
+
+	_, err := d.client.UpdateItem(input)
+	if err != nil {
+		var aerr awserr.Error
+		if errors.As(err, &aerr) && aerr.Code() == dynamodb.ErrCodeConditionalCheckFailedException {
+			return false, nil
+		}
+		return false, fmt.Errorf("failed to update latest height: %w", err)
+	}
+	fmt.Printf("Updated Dynamodb with latest height %d\n", height)
+	return true, nil
+}