Forzing units to mass (kg or t)

chaincode/controller/operations.go

@@ -71,12 +71,13 @@ func ContextOperations(m shared.AbstractChaincodeOperationManager) error {
 				"asset-register",
 				shared.WRITE_OP,
 				[]shared.MiddlewareInterface{
+					m2.ValidateUnits,
+					m2.ValidateStatus,
+					m2.ValidateType,
 					middleware.InjectAssetIdIfNone,
 					middleware.RejectIfAssetExists,
 					middleware.MarkAssetAsCreated,
 					middleware.InjectAssetOwnerData,
-					m2.ValidateStatus,
-					m2.ValidateType,
 				},
 				nil,
 				assetController.SaveAbstractAsset,
@@ -99,11 +100,12 @@ func ContextOperations(m shared.AbstractChaincodeOperationManager) error {
 				"asset-update",
 				shared.WRITE_OP,
 				[]shared.MiddlewareInterface{
+					m2.ValidateUnits,
+					m2.ValidateOptionalStatus,
+					m2.ValidateOptionalType,
 					middleware.RejectIfNoId,
 					middleware.RejectIfAssetNotExist,
 					middleware.MarkAssetAsModified,
-					m2.ValidateOptionalStatus,
-					m2.ValidateOptionalType,
 				},
 				nil,
 				assetController.OnlySameOrgAndRole(assetController.UpdateAsset),

chaincode/middleware/units.go

+/**
+ * type.go
+ *
+ * COPYRIGHT: FUNDACIÓN TECNALIA RESEARCH & INNOVATION, 2022.
+ */
+
+package middleware
+
+import (
+	"errors"
+	"strings"
+
+	"git.code.tecnalia.com/ledgerbuilder/sdk/shared"
+	"git.code.tecnalia.com/traceblock/sdk/middleware"
+)
+
+var (
+	ValidateUnits = shared.NewMiddlewareFunction("validate-units", _rejectIfInvalidUnits)
+	errInvalidUnits = errors.New("invalid units provided")
+)
+
+
+func _rejectIfInvalidUnits(stub shared.LedgerBuildrStubInterface, ctl shared.ControllerInterface, req shared.TXRequestInterface, requestAsset shared.LedgerBuildrAsset) (shared.LedgerBuildrAsset, error) {
+	asst, err := middleware.ConvertToTraceableAsset(requestAsset)
+	if err != nil {
+		return nil, err
+	}
+
+	asst.Units = strings.ToLower(asst.Units)
+	if (asst.Units != "kg" && asst.Units != "t") {
+		return nil, errInvalidUnits
+	}
+
+	return asst, nil
+}
\ No newline at end of file