Over a million developers have joined DZone.

Custom Operators for Collections in Java

Learn how to use operator overloading in Java, using it's existing conventions.

· Java Zone

Check out this 8-step guide to see how you can increase your productivity by skipping slow application redeploys and by implementing application profiling, as you code! Brought to you in partnership with ZeroTurnaround.

Operator overloading is available in an number of languages.  Java has very limited operator overloading in it's support for the + operator for String types.

We can draw on the different ways other languages support operators, but can we have an implementation in Java that uses conventions that Java already uses?

Get, Set and Put Operations

A common example of operator overloading for collections is using the array notation a[b] to access the collection itself.  When getting this is straight forward as both List and Map have a get method and this is consistent with the JavaBean getXxx() naming convention.

List<String> text = ...

String s = text[2]; // text.get(2);


Map<String, MyType> map = ...

MyType mt = map["Hello"]; // map.get("Hello")


MyType mt = ...

String xxx = ...

String s = mt[xxx]; // mt.getXxx();

When it comes to setting a value based on an index or key, we have List.set(), Map.put() and setXxx() from JavaBeans. We could go three way to resolve this.

  1. Add a set method to Map.
  2. Use a convention which looks for a set or put method and complains if there is both.
  3. Default to set() but add an annotation which override it to put().
  4. We add a new special method to all collections for setting.

The simplest option to demonstrate is where the compiler chooses either set or put, though this is unlikely to be the best option.

text[2] = "Hi"; // text.set(2, "Hi");

map["Hello"] = "World"; // text.put("Hello", "World");

mt[xxx] = "Updated"; // mt.setXxx("Updated");

Add Operation

The add operations are more interesting as it could be used in combination.

List<Integer> nums =
AtomicInteger ai =

nums += 5; // nums.add(5);

ai += 5; // ai.addAndGet(5);

nums[1] += 5; // is it thread safe?

mt[xxx] += 5; // mt.addXxx(5);

The last example has the problem that a developer could unknowingly perform an unsafe operation on a thread safe collection.  If this was mapped to...

nums.set(1, nums.get(1) + 5)); // not thread safe

This is not thread safe.  Instead, we could map this to a lambda function.

nums.update(1, x -> x + 5); // could be thread safe

This could be made thread safe by the underlying List.

Similarly for Map, you can call compute

map["Hello"] += " !!";

Converts to 

map.compute("Hello", (k, v) -> v + " !!");

Conclusion

It may be possible to add operator support for object types with very little changes to existing code.  You could use existing conventions though you might find that the use of annotations is needed in some cases for more explicit control on how this works.

The Java Zone is brought to you in partnership with ZeroTurnaround. Check out this 8-step guide to see how you can increase your productivity by skipping slow application redeploys and by implementing application profiling, as you code!

Topics:
java

Published at DZone with permission of Peter Lawrey, DZone MVB. See the original article here.

Opinions expressed by DZone contributors are their own.

The best of DZone straight to your inbox.

SEE AN EXAMPLE
Please provide a valid email address.

Thanks for subscribing!

Awesome! Check your inbox to verify your email so you can start receiving the latest in tech news and resources.
Subscribe

{{ parent.title || parent.header.title}}

{{ parent.tldr }}

{{ parent.urlSource.name }}