private void parseFileStore(
      XMLExtendedStreamReader reader, ModelNode cache, List<ModelNode> operations)
      throws XMLStreamException {
    // ModelNode for the store add operation
    ModelNode storeAddress = cache.get(ModelDescriptionConstants.OP_ADDR).clone();
    storeAddress.add(ModelKeys.FILE_STORE, ModelKeys.FILE_STORE_NAME);
    storeAddress.protect();
    ModelNode store = Util.getEmptyOperation(ModelDescriptionConstants.ADD, storeAddress);

    for (int i = 0; i < reader.getAttributeCount(); i++) {
      String value = reader.getAttributeValue(i);
      Attribute attribute = Attribute.forName(reader.getAttributeLocalName(i));
      switch (attribute) {
        case RELATIVE_TO:
          {
            CommonAttributes.RELATIVE_TO.parseAndSetParameter(value, store, reader);
            break;
          }
        case PATH:
          {
            CommonAttributes.PATH.parseAndSetParameter(value, store, reader);
            break;
          }
        default:
          {
            this.parseStoreAttribute(reader, i, attribute, value, store);
          }
      }
    }

    this.parseStoreProperties(reader, store);
    operations.add(store);
  }
Exemplo n.º 2
0
  private void buildCacheStore(
      OperationContext context,
      LoaderConfigurationBuilder builder,
      String containerName,
      ModelNode store,
      String storeKey,
      List<Dependency<?>> dependencies)
      throws OperationFailedException {
    final Properties properties = new TypedProperties();
    if (store.hasDefined(ModelKeys.PROPERTY)) {
      for (Property property : store.get(ModelKeys.PROPERTY).asPropertyList()) {
        // the format of the property elements
        //  "property" => {
        //       "relative-to" => {"value" => "fred"},
        //   }
        String propertyName = property.getName();
        Property complexValue = property.getValue().asProperty();
        String propertyValue = complexValue.getValue().asString();
        properties.setProperty(propertyName, propertyValue);
      }
    }
    builder.withProperties(properties);

    ModelNode resolvedValue = null;
    if (storeKey.equals(ModelKeys.FILE_STORE)) {
      builder.cacheLoader(new FileCacheStore());

      final String path =
          ((resolvedValue = CommonAttributes.PATH.resolveModelAttribute(context, store))
                  .isDefined())
              ? resolvedValue.asString()
              : InfinispanExtension.SUBSYSTEM_NAME + File.separatorChar + containerName;
      final String relativeTo =
          ((resolvedValue = CommonAttributes.RELATIVE_TO.resolveModelAttribute(context, store))
                  .isDefined())
              ? resolvedValue.asString()
              : ServerEnvironment.SERVER_DATA_DIR;
      Injector<PathManager> injector =
          new SimpleInjector<PathManager>() {
            volatile PathManager.Callback.Handle callbackHandle;

            @Override
            public void inject(PathManager value) {
              callbackHandle =
                  value.registerCallback(
                      relativeTo,
                      PathManager.ReloadServerCallback.create(),
                      PathManager.Event.UPDATED,
                      PathManager.Event.REMOVED);
              properties.setProperty("location", value.resolveRelativePathEntry(path, relativeTo));
            }

            @Override
            public void uninject() {
              super.uninject();
              if (callbackHandle != null) {
                callbackHandle.remove();
              }
            }
          };
      dependencies.add(
          new Dependency<PathManager>(
              PathManagerService.SERVICE_NAME, PathManager.class, injector));
      properties.setProperty("fsyncMode", "perWrite");
    } else if (storeKey.equals(ModelKeys.STRING_KEYED_JDBC_STORE)
        || storeKey.equals(ModelKeys.BINARY_KEYED_JDBC_STORE)
        || storeKey.equals(ModelKeys.MIXED_KEYED_JDBC_STORE)) {
      builder.cacheLoader(this.createJDBCStore(properties, context, store));

      final String datasource =
          CommonAttributes.DATA_SOURCE.resolveModelAttribute(context, store).asString();

      dependencies.add(new Dependency<Object>(ServiceName.JBOSS.append("data-source", datasource)));
      properties.setProperty("datasourceJndiLocation", datasource);
      properties.setProperty("connectionFactoryClass", ManagedConnectionFactory.class.getName());
    } else if (storeKey.equals(ModelKeys.REMOTE_STORE)) {
      builder.cacheLoader(new RemoteCacheStore());
      for (ModelNode server : store.require(ModelKeys.REMOTE_SERVERS).asList()) {
        String outboundSocketBinding = server.get(ModelKeys.OUTBOUND_SOCKET_BINDING).asString();
        Injector<OutboundSocketBinding> injector =
            new SimpleInjector<OutboundSocketBinding>() {
              @Override
              public void inject(OutboundSocketBinding value) {
                try {
                  String address =
                      value.getDestinationAddress().getHostAddress()
                          + ":"
                          + value.getDestinationPort();
                  String serverList = properties.getProperty("serverList");
                  properties.setProperty(
                      "serverList", (serverList == null) ? address : serverList + ";" + address);
                } catch (UnknownHostException e) {
                  throw InfinispanMessages.MESSAGES.failedToInjectSocketBinding(e, value);
                }
              }
            };
        dependencies.add(
            new Dependency<OutboundSocketBinding>(
                OutboundSocketBinding.OUTBOUND_SOCKET_BINDING_BASE_SERVICE_NAME.append(
                    outboundSocketBinding),
                OutboundSocketBinding.class,
                injector));
      }
      if (store.hasDefined(ModelKeys.CACHE)) {
        properties.setProperty("remoteCacheName", store.get(ModelKeys.CACHE).asString());
        properties.setProperty("useDefaultRemoteCache", Boolean.toString(false));
      } else {
        properties.setProperty("useDefaultRemoteCache", Boolean.toString(true));
      }
      if (store.hasDefined(ModelKeys.SOCKET_TIMEOUT)) {
        properties.setProperty("soTimeout", store.require(ModelKeys.SOCKET_TIMEOUT).asString());
      }
      if (store.hasDefined(ModelKeys.TCP_NO_DELAY)) {
        properties.setProperty("tcpNoDelay", store.require(ModelKeys.TCP_NO_DELAY).asString());
      }
    } else {
      String className = store.require(ModelKeys.CLASS).asString();
      try {
        CacheLoader loader =
            CacheLoader.class
                .getClassLoader()
                .loadClass(className)
                .asSubclass(CacheLoader.class)
                .newInstance();
        builder.cacheLoader(loader);
      } catch (Exception e) {
        throw new IllegalArgumentException(
            String.format("%s is not a valid cache store", className), e);
      }
    }
  }