Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 1 | package org.onlab.onos.store.serializers; |
| 2 | |
| 3 | import java.net.URI; |
| 4 | import java.util.ArrayList; |
Madan Jampani | 53e44e6 | 2014-10-07 12:39:51 -0700 | [diff] [blame] | 5 | import java.util.Arrays; |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 6 | import java.util.HashMap; |
| 7 | |
| 8 | import org.onlab.onos.cluster.ControllerNode; |
| 9 | import org.onlab.onos.cluster.DefaultControllerNode; |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 10 | import org.onlab.onos.cluster.NodeId; |
Yuta HIGUCHI | 80912e6 | 2014-10-12 00:15:47 -0700 | [diff] [blame] | 11 | import org.onlab.onos.mastership.MastershipTerm; |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 12 | import org.onlab.onos.net.ConnectPoint; |
| 13 | import org.onlab.onos.net.DefaultAnnotations; |
| 14 | import org.onlab.onos.net.DefaultDevice; |
| 15 | import org.onlab.onos.net.DefaultLink; |
| 16 | import org.onlab.onos.net.DefaultPort; |
| 17 | import org.onlab.onos.net.Device; |
| 18 | import org.onlab.onos.net.DeviceId; |
| 19 | import org.onlab.onos.net.Element; |
Yuta HIGUCHI | 5fa3dc0 | 2014-10-15 17:08:13 -0700 | [diff] [blame^] | 20 | import org.onlab.onos.net.HostId; |
| 21 | import org.onlab.onos.net.HostLocation; |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 22 | import org.onlab.onos.net.Link; |
| 23 | import org.onlab.onos.net.LinkKey; |
Yuta HIGUCHI | d40483d | 2014-10-09 15:20:30 -0700 | [diff] [blame] | 24 | import org.onlab.onos.net.MastershipRole; |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 25 | import org.onlab.onos.net.Port; |
| 26 | import org.onlab.onos.net.PortNumber; |
Madan Jampani | 53e44e6 | 2014-10-07 12:39:51 -0700 | [diff] [blame] | 27 | import org.onlab.onos.net.device.DefaultDeviceDescription; |
| 28 | import org.onlab.onos.net.device.DefaultPortDescription; |
Yuta HIGUCHI | 5fa3dc0 | 2014-10-15 17:08:13 -0700 | [diff] [blame^] | 29 | import org.onlab.onos.net.host.DefaultHostDescription; |
| 30 | import org.onlab.onos.net.host.HostDescription; |
Madan Jampani | 2ff0559 | 2014-10-10 15:42:47 -0700 | [diff] [blame] | 31 | import org.onlab.onos.net.link.DefaultLinkDescription; |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 32 | import org.onlab.onos.net.provider.ProviderId; |
Yuta HIGUCHI | 9ee60f6 | 2014-10-09 10:00:01 -0700 | [diff] [blame] | 33 | import org.onlab.onos.store.Timestamp; |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 34 | import org.onlab.packet.IpAddress; |
| 35 | import org.onlab.packet.IpPrefix; |
Yuta HIGUCHI | 5fa3dc0 | 2014-10-15 17:08:13 -0700 | [diff] [blame^] | 36 | import org.onlab.packet.MacAddress; |
| 37 | import org.onlab.packet.VlanId; |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 38 | import org.onlab.util.KryoPool; |
| 39 | |
Yuta HIGUCHI | 47c4088 | 2014-10-10 18:44:37 -0700 | [diff] [blame] | 40 | import com.google.common.collect.ImmutableList; |
| 41 | import com.google.common.collect.ImmutableMap; |
Yuta HIGUCHI | 5fa3dc0 | 2014-10-15 17:08:13 -0700 | [diff] [blame^] | 42 | import com.google.common.collect.ImmutableSet; |
Yuta HIGUCHI | 47c4088 | 2014-10-10 18:44:37 -0700 | [diff] [blame] | 43 | |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 44 | public final class KryoPoolUtil { |
| 45 | |
| 46 | /** |
| 47 | * KryoPool which can serialize ON.lab misc classes. |
| 48 | */ |
| 49 | public static final KryoPool MISC = KryoPool.newBuilder() |
| 50 | .register(IpPrefix.class, new IpPrefixSerializer()) |
| 51 | .register(IpAddress.class, new IpAddressSerializer()) |
Yuta HIGUCHI | 5fa3dc0 | 2014-10-15 17:08:13 -0700 | [diff] [blame^] | 52 | .register(MacAddress.class, new MacAddressSerializer()) |
| 53 | .register(VlanId.class) |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 54 | .build(); |
| 55 | |
| 56 | // TODO: Populate other classes |
| 57 | /** |
| 58 | * KryoPool which can serialize API bundle classes. |
| 59 | */ |
| 60 | public static final KryoPool API = KryoPool.newBuilder() |
| 61 | .register(MISC) |
Yuta HIGUCHI | 47c4088 | 2014-10-10 18:44:37 -0700 | [diff] [blame] | 62 | .register(ImmutableMap.class, new ImmutableMapSerializer()) |
| 63 | .register(ImmutableList.class, new ImmutableListSerializer()) |
Yuta HIGUCHI | 5fa3dc0 | 2014-10-15 17:08:13 -0700 | [diff] [blame^] | 64 | .register(ImmutableSet.class, new ImmutableSetSerializer()) |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 65 | .register( |
| 66 | // |
| 67 | ArrayList.class, |
Madan Jampani | 53e44e6 | 2014-10-07 12:39:51 -0700 | [diff] [blame] | 68 | Arrays.asList().getClass(), |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 69 | HashMap.class, |
| 70 | // |
Yuta HIGUCHI | 47c4088 | 2014-10-10 18:44:37 -0700 | [diff] [blame] | 71 | // |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 72 | ControllerNode.State.class, |
| 73 | Device.Type.class, |
| 74 | DefaultAnnotations.class, |
| 75 | DefaultControllerNode.class, |
| 76 | DefaultDevice.class, |
Madan Jampani | 53e44e6 | 2014-10-07 12:39:51 -0700 | [diff] [blame] | 77 | DefaultDeviceDescription.class, |
Madan Jampani | 2ff0559 | 2014-10-10 15:42:47 -0700 | [diff] [blame] | 78 | DefaultLinkDescription.class, |
Yuta HIGUCHI | d40483d | 2014-10-09 15:20:30 -0700 | [diff] [blame] | 79 | MastershipRole.class, |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 80 | Port.class, |
Madan Jampani | 53e44e6 | 2014-10-07 12:39:51 -0700 | [diff] [blame] | 81 | DefaultPortDescription.class, |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 82 | Element.class, |
Yuta HIGUCHI | 9ee60f6 | 2014-10-09 10:00:01 -0700 | [diff] [blame] | 83 | Link.Type.class, |
Yuta HIGUCHI | 5fa3dc0 | 2014-10-15 17:08:13 -0700 | [diff] [blame^] | 84 | Timestamp.class, |
| 85 | HostId.class, |
| 86 | HostDescription.class, |
| 87 | DefaultHostDescription.class |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 88 | ) |
| 89 | .register(URI.class, new URISerializer()) |
| 90 | .register(NodeId.class, new NodeIdSerializer()) |
| 91 | .register(ProviderId.class, new ProviderIdSerializer()) |
| 92 | .register(DeviceId.class, new DeviceIdSerializer()) |
| 93 | .register(PortNumber.class, new PortNumberSerializer()) |
| 94 | .register(DefaultPort.class, new DefaultPortSerializer()) |
| 95 | .register(LinkKey.class, new LinkKeySerializer()) |
| 96 | .register(ConnectPoint.class, new ConnectPointSerializer()) |
| 97 | .register(DefaultLink.class, new DefaultLinkSerializer()) |
Yuta HIGUCHI | fa891c9 | 2014-10-09 15:21:40 -0700 | [diff] [blame] | 98 | .register(MastershipTerm.class, new MastershipTermSerializer()) |
Yuta HIGUCHI | d40483d | 2014-10-09 15:20:30 -0700 | [diff] [blame] | 99 | .register(MastershipRole.class, new MastershipRoleSerializer()) |
Yuta HIGUCHI | 5fa3dc0 | 2014-10-15 17:08:13 -0700 | [diff] [blame^] | 100 | .register(HostLocation.class, new HostLocationSerializer()) |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 101 | |
| 102 | .build(); |
| 103 | |
| 104 | |
| 105 | // not to be instantiated |
| 106 | private KryoPoolUtil() {} |
| 107 | } |