Skip to content

vaclav/ScalaAgent

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

29 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

The Agent class was strongly inspired by the agent principle in Clojure. Essentially, an agent wraps a shared mutable state and hides it behind a message-passing interface.
Agents accept messages and process them on behalf of the wrapped state. Typically agents accept functions / commands as messages and ensure the submitted commands are executed against the internal agent's state in a thread-safe manner (sequentially). The submitted functions / commands take the internal state as a parameter and their output becomes the new internal state value.
The code that is submitted to an agent doesn't need to pay attention to threading or synchronization, the agent will provide such guarantees by itself.

See the examples of use for more details.

/**
 * Stores a list of names and allows concurrent threads to manipulate the list.
 */
object AgentSample {
    def main(args: Array[String]) {
        val agent = Agent(List("Joe"))
        agent start()
        agent(List("Dave"))                      //Set the state to a new value
        agent((x: List[String]) => "Alice" :: x) //Pre-pend a value to the internal list
        agent("Susan" :: _)                      //Pre-pend a value to the internal list
        agent get(_.foreach(println _))     //Print the content asynchronously
    }
}

/**
 * Wraps a counter allowing other threads to safely update the counter concurrently
 */
object CounterSample {

    def increment(x: Long) = x + 1
    def decrement(amount : Long)(x: Long) = x - amount

    def main(args: Array[String]) {
        val agent = Agent(0L)    //Wrap a counter by an Agent
        agent start

        agent((x: Long) => x + 100) //Increment the value by 100
        agent{_ + 100}              //Increment the value by 100
        def a = actor {
            agent(_ + 100)              //Increment the value by 100 from within an actor
        }
        a.start

        new Thread(new Runnable() { //Start a new thread to manipulate the counter in parallel
            def run = {
                    agent(increment(_)) //Increment the value by 1 calling the increment function
                }
        }).start

        agent(increment _)           //Send a method that will increment the counter by 1
        agent(decrement(3) _)        //Send a method that will decrement the counter by 3

        println(agent get)               //Print the content
    }
}

/**
 * A thread-safe shopping cart wrapping the internal state represented as a HashMap inside an Agent.
 * All public method calls on the ShoppingCart are transformed into a command execution on the internal agent.
 */
class ShoppingCart {
    val content = Agent(Map[String, Int]())

    def start() { content start}

    def getContent() : Map[String, Int] = { content.get }

    def addItem(product: String) {
        content((x: Map[String, Int]) => {
            x + product -> 1
        })
    }

    def removeItem(product: String) {
        content((x: Map[String, Int]) => {
            x - product
        })
    }

    def clear() {
        content((x: Map[String, Int]) => {
            Map[String, Int]()
        })
    }

    /**
     * Curry the changeQuantity() private method and send it off to the Agent for processing
     */
    def increaseQuantity(product:String, quantity:Int) {
        content(changeQuantity(product, quantity, _:Map[String, Int]))
    }

    /**
     * Manipulates the cart's map directly, since it is never called directly by clients, but always indirectly
     * inside the Agent's thread.
     */
    private def changeQuantity(product:String, quantity:Int, items:Map[String, Int]) = {
        if (items.contains(product)) {
            def currentQuantity = items(product)
            items + product -> (currentQuantity + quantity)
        } else items
    }
}

object ShoppingCartExample {
    def main(args: Array[String]) {
        val cart: ShoppingCart = new ShoppingCart
        cart start

        cart addItem "Budweiser"
        cart addItem "Pilsner"
        cart addItem "Staropramen"
        cart removeItem "Budweiser"

        println(cart getContent)
        cart increaseQuantity("Staropramen", 5)
        println(cart getContent)
        cart clear()
        println(cart getContent)

    }
}

/**
 * Uses a custom copy strategy to avoid returning the original state instance.
 */
object CustomCopyStrategyExample {

    object MyIntCopyStrategy extends CopyStrategy[Int] {
        def apply(x:Int) : Int = {
            def a = x
            return a
        }
    }

    def main(args: Array[String]) {
        val agent = Agent(0, MyIntCopyStrategy)
        agent start

        agent(_ + 10)
        agent(_ + 20)

        println(agent get)
    }
}

/**
 * Shows (after un-commenting) the default error handling mechanics.
 */
object NumberExample {
    def main(args: Array[String]) {
        val agent = Agent(List(10))
        agent start

        agent() = List(1)
        agent() = (x: List[Int]) => 2 :: x
        agent((x: List[Int]) => 3 :: x)
        agent((x: List[Int]) => x.reverse)
//        agent((x: List[Int]) => {
//            if (true) throw new RuntimeException("test")
//            return List()
//        })
        agent((x: List[Int]) => {
            x.map(_ * 2)
        })
//        agent.get((x: List[Int]) => {throw new RuntimeException("read test")})
        agent.get((x: List[Int]) => {println("Value: " + x)})
        println("Result: " + agent())
    }
}

About

An experimental Clojure-like Agent in Scala

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages