fix: Update Example Repo

main.go

@@ -10,16 +10,36 @@ import (
 	"time"
 )
 
+type TaskAttribute struct {
+	SleepTime            *time.Duration `json:"sleepTime,omitempty"`
+	MemoryAllocationSize *int           `json:"memoryAllocationSize,omitempty"`
+}
+
 func main() {
 	containerlib.Run(func(ctx containerlibcontext.Context, task model.Task) error {
-		var unmarshalledData map[string]interface{}
+		var unmarshalledData TaskAttribute
 		err := json.Unmarshal(task.Attributes, &unmarshalledData)
 		if err != nil {
 			logrus.Error(err)
 			return err
 		}
 
-		time.Sleep(15 * time.Second)
+		if unmarshalledData.SleepTime == nil {
+			sleepTime := 15 * time.Second
+			unmarshalledData.SleepTime = &sleepTime
+		}
+
+		if unmarshalledData.MemoryAllocationSize == nil {
+			size := 0
+			unmarshalledData.MemoryAllocationSize = &size
+		}
+
+		fakeAllocation := make([]byte, *unmarshalledData.MemoryAllocationSize)
+		for i := 0; i < *unmarshalledData.MemoryAllocationSize; i++ {
+			fakeAllocation[i] = 0x99
+		}
+
+		time.Sleep(*unmarshalledData.SleepTime)
 		fmt.Println(unmarshalledData)
 
 		ctx.Success()