Exemplo n.º 1
0
        private void DataSender_Elapsed(object sender, ElapsedEventArgs e)
        {
            if (MetadataSender.Enabled)
            {
                //  start the timer again, as we need to make sure we're updating
                DataSender.Start();
                return;
            }

            this.Message = String.Format("Converting {0} \n objects", BucketObjects.Count);

            var convertedObjects = Converter.Serialise(BucketObjects).Select(obj =>
            {
                if (ObjectCache.ContainsKey(obj.Hash))
                {
                    return new SpeckleObjectPlaceholder()
                    {
                        Hash = obj.Hash, DatabaseId = ObjectCache[obj.Hash].DatabaseId
                    }
                }
                ;
                return(obj);
            }).ToList();

            this.Message = String.Format("Creating payloads");

            long totalBucketSize   = 0;
            long currentBucketSize = 0;
            List <PayloadMultipleObjects> objectUpdatePayloads = new List <PayloadMultipleObjects>();
            List <SpeckleObject>          currentBucketObjects = new List <SpeckleObject>();
            List <SpeckleObject>          allObjects           = new List <SpeckleObject>();

            foreach (SpeckleObject convertedObject in convertedObjects)
            {
                long size = Converter.getBytes(convertedObject).Length;
                currentBucketSize += size;
                totalBucketSize   += size;
                currentBucketObjects.Add(convertedObject);

                if (currentBucketSize > 5e5) // restrict max to ~500kb; should it be user config? anyway these functions should go into core. at one point.
                {
                    Debug.WriteLine("Reached payload limit. Making a new one, current  #: " + objectUpdatePayloads.Count);
                    objectUpdatePayloads.Add(new PayloadMultipleObjects()
                    {
                        Objects = currentBucketObjects.ToArray()
                    });
                    currentBucketObjects = new List <SpeckleObject>();
                    currentBucketSize    = 0;
                }
            }

            if (currentBucketObjects.Count > 0)
            {
                objectUpdatePayloads.Add(new PayloadMultipleObjects()
                {
                    Objects = currentBucketObjects.ToArray()
                });
            }

            Debug.WriteLine("Finished, payload object update count is: " + objectUpdatePayloads.Count + " total bucket size is (kb) " + totalBucketSize / 1000);

            if (objectUpdatePayloads.Count > 100)
            {
                this.AddRuntimeMessage(GH_RuntimeMessageLevel.Error, "This is a humongous update, in the range of ~50mb. For now, create more streams instead of just one massive one! Updates will be faster and snappier, and you can combine them back together at the other end easier.");
                return;
            }

            int k = 0;
            List <ResponsePostObjects> responses = new List <ResponsePostObjects>();

            foreach (var payload in objectUpdatePayloads)
            {
                this.Message = String.Format("Sending payload\n{0} / {1}", k++, objectUpdatePayloads.Count);
                responses.Add(mySender.ObjectCreateBulkAsync(payload).GetAwaiter().GetResult());
            }

            this.Message = "Updating stream...";

            // create placeholders for stream update payload
            List <SpeckleObjectPlaceholder> placeholders = new List <SpeckleObjectPlaceholder>();

            foreach (var myResponse in responses)
            {
                foreach (string dbId in myResponse.Objects)
                {
                    placeholders.Add(new SpeckleObjectPlaceholder()
                    {
                        DatabaseId = dbId
                    });
                }
            }

            PayloadStreamUpdate streamUpdatePayload = new PayloadStreamUpdate();

            streamUpdatePayload.Layers  = BucketLayers;
            streamUpdatePayload.Name    = BucketName;
            streamUpdatePayload.Objects = placeholders;

            // set some base properties (will be overwritten)
            var baseProps = new Dictionary <string, object>();

            baseProps["units"]                 = Rhino.RhinoDoc.ActiveDoc.ModelUnitSystem.ToString();
            baseProps["tolerance"]             = Rhino.RhinoDoc.ActiveDoc.ModelAbsoluteTolerance;
            baseProps["angleTolerance"]        = Rhino.RhinoDoc.ActiveDoc.ModelAngleToleranceRadians;
            streamUpdatePayload.BaseProperties = baseProps;

            var response = mySender.StreamUpdate(streamUpdatePayload, mySender.StreamId);

            mySender.BroadcastMessage(new { eventType = "update-global" });

            // put the objects in the cache
            int l = 0;

            foreach (var obj in streamUpdatePayload.Objects)
            {
                obj.DatabaseId = response.Objects[l];
                ObjectCache[convertedObjects[l].Hash] = placeholders[l];
                l++;
            }

            AddRuntimeMessage(GH_RuntimeMessageLevel.Remark, "Data sent at " + DateTime.Now);
            Message = "Data sent\n@" + DateTime.Now.ToString("hh:mm:ss");
        }