org.restcomm.cluster.serializers.GenericJbossMarshaller Maven / Gradle / Ivy
package org.restcomm.cluster.serializers;
/*
* Copyright 2022-2023, Mobius Software LTD. and individual contributors
* by the @authors tag. See the copyright.txt in the distribution for a
* full listing of individual contributors.
*
* This program is free software: you can redistribute it and/or modify
* under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation; either version 3 of
* the License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see > writers=new ConcurrentHashMap>();
private ConcurrentHashMap> externalizers=new ConcurrentHashMap>();
public GenericJbossMarshaller(ClassLoader classLoader) {
super();
AbstractTreeSegment.Externalizer treeExternalizer = new AbstractTreeSegment.Externalizer();
externalizers.put(AbstractTreeSegment.Externalizer.EXTERNALIZER_ID, treeExternalizer);
ClusteredUUIDExternalizer idExternalizer = new ClusteredUUIDExternalizer();
externalizers.put(ClusteredUUIDExternalizer.EXTERNALIZER_ID, idExternalizer);
ExternalizableExternalizer stringAndClusteredIDExternalizer = new ExternalizableExternalizer(StringAndClusteredIDKey.class,StringAndClusteredIDKey.EXTERNALIZER_ID);
externalizers.put(StringAndClusteredIDKey.EXTERNALIZER_ID, stringAndClusteredIDExternalizer);
ExternalizableExternalizer clusteredIDAndStringExternalizer = new ExternalizableExternalizer(ClusteredIDAndStringKey.class,ClusteredIDAndStringKey.EXTERNALIZER_ID);
externalizers.put(ClusteredIDAndStringKey.EXTERNALIZER_ID, clusteredIDAndStringExternalizer);
ExternalizableExternalizer multiStringExternalizer = new ExternalizableExternalizer(MultiStringKey.class,MultiStringKey.EXTERNALIZER_ID);
externalizers.put(MultiStringKey.EXTERNALIZER_ID, multiStringExternalizer);
KnownObjectsWriter writer = new KnownObjectsWriter>(treeExternalizer);
writers.put(AbstractTreeSegment.class.getCanonicalName(), writer);
writers.put(ClusteredIDTreeSegment.class.getCanonicalName(), writer);
writers.put(StringTreeSegment.class.getCanonicalName(), writer);
writers.put(StringAndClusteredIDTreeSegment.class.getCanonicalName(), writer);
writers.put(ClusteredIDAndStringTreeSegment.class.getCanonicalName(), writer);
writers.put(MultiStringTreeSegment.class.getCanonicalName(), writer);
writer = new KnownObjectsWriter(idExternalizer);
writers.put(ClusteredUUID.class.getCanonicalName(),writer);
writer = new KnownObjectsWriter(stringAndClusteredIDExternalizer);
writers.put(StringAndClusteredIDKey.class.getCanonicalName(),writer);
writer = new KnownObjectsWriter(clusteredIDAndStringExternalizer);
writers.put(ClusteredIDAndStringKey.class.getCanonicalName(),writer);
writer = new KnownObjectsWriter(multiStringExternalizer);
writers.put(MultiStringKey.class.getCanonicalName(),writer);
baseCfg.setClassResolver(new DefaultContextClassResolver(classLoader != null ? classLoader : this.getClass().getClassLoader()));
baseCfg.setObjectTable(new ObjectTable() {
@Override
public Writer getObjectWriter(Object object) throws IOException {
return writers.get(object.getClass().getCanonicalName());
}
@Override
public Object readObject(Unmarshaller unmarshaller) throws IOException, ClassNotFoundException {
@SuppressWarnings("rawtypes")
AdvancedExternalizer ext = null;
ext=externalizers.get(unmarshaller.readInt());
if (ext == null)
return null;
return ext.readObject(unmarshaller);
}
});
}
public void registerKnownClass(int id,Class knownClass) {
ExternalizableExternalizer externalizer=new ExternalizableExternalizer(knownClass,FIRST_ID + id);
externalizers.put(FIRST_ID + id, externalizer);
writers.put(knownClass.getCanonicalName(),new KnownObjectsWriter(externalizer));
}
private class KnownObjectsWriter implements Writer {
private AdvancedExternalizer externalizer;
public KnownObjectsWriter(AdvancedExternalizer externalizer) {
this.externalizer = externalizer;
}
@SuppressWarnings("unchecked")
@Override
public void writeObject(Marshaller marshaller, Object object) throws IOException {
marshaller.writeInt(externalizer.getId());
externalizer.writeObject((ObjectOutput) marshaller, (T) object);
}
}
}