singrdk/base/Applications/SeattleTrafficProxy/SeattleTrafficProxy.sg

///////////////////////////////////////////////////////////////////////////////
//
//  Microsoft Research Singularity
//
//  Copyright (c) Microsoft Corporation.  All rights reserved.
//
//  Note:   A simple proxy module for the Seattle Traffic web service
//

using System;
using System.Diagnostics;
using System.Text;
using System.Collections;
using Microsoft.SingSharp;
using Microsoft.Singularity;
using Microsoft.Singularity.Channels;
using Microsoft.Singularity.SeattleTrafficProxy.Contracts;
using Microsoft.Singularity.Directory;
using Microsoft.Singularity.Xml;
using Microsoft.Singularity.Security;
using System;

using Microsoft.Contracts;
using Microsoft.SingSharp.Reflection;
using Microsoft.Singularity.Applications;
using Microsoft.Singularity.Io;
using Microsoft.Singularity.Configuration;
[assembly: Transform(typeof(ApplicationResourceTransform))]

// the two lines below give this module permission
// to register endpoints on the local machine. 
[assembly: ApplicationPublisherAttribute("singularity.microsoft.com")]
[assembly: AssertPrivilegeAttribute("$register-privilege.localhost")]

namespace Microsoft.Singularity.Applications
{
    [ConsoleCategory(HelpMessage="Mappoint proxy service", DefaultAction=true)]
    internal class Parameters 
    {
        [InputEndpoint("data")]
        public readonly TRef<UnicodePipeContract.Exp:READY> Stdin;

        [OutputEndpoint("data")]
        public readonly TRef<UnicodePipeContract.Imp:READY> Stdout;

        [Endpoint]
        public readonly TRef<DirectoryServiceContract.Imp:Start> nsRef;

        reflective internal Parameters();

        internal int AppMain() {
            return SeattleTrafficProxy.AppMain(this);
        }
    }

    public class SeattleTrafficProxy
    {
        internal static int AppMain(Parameters! config)
        {
            
            DirectoryServiceContract.Imp ds = ((!)config.nsRef).Acquire(); 
            if (ds == null) { 
                throw new Exception("Unable to acquire handle to the Directory Service root"); 
            } 
            ds.RecvSuccess();
            
            
            // Here is the channel we use to communicate with
            // the NameServer
            ServiceProviderContract.Imp! nsImp;
            ServiceProviderContract.Exp! nsExp;
            ServiceProviderContract.NewChannel(out nsImp, out nsExp);

            try {
                ds.SendRegister(Bitter.FromString2(SeattleTrafficProxyContract.ModuleName), nsImp);

                switch receive {
                    case ds.AckRegister() :
                        // All is well.
                        break;

                    case ds.NakRegister(ServiceProviderContract.Imp:Start rejectedEP, error) :
                        // All is very much not well; abort.
                        Console.WriteLine("Failed to register the SeattleTraffic proxy module as " +
                                        SeattleTrafficProxyContract.ModuleName + " error:"  + SdsUtils.ErrorCodeToString(error));
                        delete nsExp;
                        delete rejectedEP;
                        return -1;

                    case ds.ChannelClosed():
                        Console.WriteLine("ds channel closed");
                        delete nsExp;
                        return -1;
                }
            }
            finally {
                delete ds;
            }

            // Here is the set of client channels we service
            ESet<SeattleTrafficProxyContract.Exp:ReadyState> epSet = new ESet<SeattleTrafficProxyContract.Exp:ReadyState>();

            while (true) {
                switch receive {
                    // ------------------------------- Requests for new connections

                    case nsExp.Connect(ServiceContract.Exp:Start! newEp) :
                        {
                            // We expect people top give us SeattleTrafficProxyContract.Exp instances
                            SeattleTrafficProxyContract.Exp newClient = newEp as SeattleTrafficProxyContract.Exp;

                            if (newClient == null) {
                                // Invalid contract type. Fail.
                                nsExp.SendNackConnect(newEp);
                            }
                            else {
                                // Signal ready and start servicing this contract
                                nsExp.SendAckConnect();
                                newClient.SendReady();
                                epSet.Add(newClient);
                            }
                        }
                        break;

                        // ------------------------------- Requests on existing connections
                        //
                        // Don't forget that we're selecting endpoints
                        // from the epSet endpoint-set. In each case that we
                        // receive a message from one of those endpoints, we
                        // need to remember to put the endpoint back into epSet
                        // if we want to keep listening to it.
                        //
                    case ep.GetTraffic() in epSet:
                        {
                            Console.WriteLine("SeattleTrafficProxy - received a GetTraffic request");
                            TrafficInfo[]! in ExHeap trafficInfo = LookupTraffic();
                            ep.SendTraffic(trafficInfo);
                            epSet.Add(ep);
                        }
                        break;
                        
                    case nsExp.ChannelClosed() && epSet.Empty() :
                        // The namespace channel is closed and we have no more client
                        // channels, so pack up and go home.
                        delete nsExp;
                        epSet.Dispose();
                        return -1;

                    case ep.ChannelClosed() in epSet :
                        // Dispose of channels when our clients close them
                        delete ep;
                        break;
                }
            }
            return 0;
        }
        
        //
        // HACK: LookupTraffic - This should leverage the Bayesian prediction work
        // to retrieve traffic bottleneck times.
        //
        private static TrafficInfo[]! in ExHeap LookupTraffic()
        {
		    // HACK: Disable until ready.
		    // 
		    //try
		    //{
			   //using (WebClient webclient = new WebClient())
			   //{
				  //// Download traffic data
				  //// TODO: Only download every five minutes.
				  //webclient.DownloadFile(@"http://www.adapt.msresearch.us/smartphlow/Desktop/tip02.dat", "tip02.dat");
			   //}
			   //// Parse data
			   //byte [] rawTraffic = ParseTipFile("tip02.dat");
    		//
			   //// Generate the results for the client.
			   //// TODO: Add more value to the traffic response.
			   //TrafficInfo[]! in ExHeap trafficInfo = BuildTrafficInfo(rawTraffic);
			   //return trafficInfo;
		    //}
		    //catch (Exception ex) {
		    //}
		    //
    		
            TrafficInfo[]! in ExHeap trafficInfo = new[ExHeap] TrafficInfo[12];
            
            // West side
            
            // 520 WB at Montlake
            trafficInfo[0].minUntilFree = 6;
            trafficInfo[0].latitude = 47.64574f;
            trafficInfo[0].longitude = -122.30687f;
            // 520 EB at Montlake
            trafficInfo[1].minUntilFree = 30;
            trafficInfo[1].latitude = 47.64314f;
            trafficInfo[1].longitude = -122.30634f;
            // 5 SB at Lake Union
            trafficInfo[2].minUntilFree = 12;
            trafficInfo[2].latitude = 47.63600f;
            trafficInfo[2].longitude = -122.32949f;
            // 5 NB at Lake Union
            trafficInfo[3].minUntilFree = 27;
            trafficInfo[3].latitude = 47.63519f;
            trafficInfo[3].longitude = -122.32143f;

            // East side
            
            // 520 WB at bridge
            trafficInfo[4].minUntilFree = 51;
            trafficInfo[4].latitude = 47.64571f;
            trafficInfo[4].longitude = -122.21226f;
            // 520 EB at bridge
            trafficInfo[5].minUntilFree = 0;
            trafficInfo[5].latitude = 47.63983f;
            trafficInfo[5].longitude = -122.20965f;
            // 520 EB at Redmond
            trafficInfo[6].minUntilFree = 22;
            trafficInfo[6].latitude = 47.66538f;
            trafficInfo[6].longitude = -122.11552f;
            // 520 WB at Redmond
            trafficInfo[7].minUntilFree = 0;
            trafficInfo[7].latitude = 47.66986f;
            trafficInfo[7].longitude = -122.13050f;
            
            // 405 N in Bellevue
            trafficInfo[8].minUntilFree = 39;
            trafficInfo[8].latitude = 47.60998f;
            trafficInfo[8].longitude = -122.18467f;
            // 405 S in Bellevue
            trafficInfo[9].minUntilFree = 16;
            trafficInfo[9].latitude = 47.61012f;
            trafficInfo[9].longitude = -122.19992f;
            // 405 N in Kirkland
            trafficInfo[10].minUntilFree = 14;
            trafficInfo[10].latitude = 47.66809f;
            trafficInfo[10].longitude = -122.18236f;
            // 405 S in Kirkland
            trafficInfo[11].minUntilFree = 9;
            trafficInfo[11].latitude = 47.66761f;
            trafficInfo[11].longitude = -122.19861f;
            
            return trafficInfo;
        }
    }
}