/
AsyncHandlerAdd.java
128 lines (113 loc) · 6.17 KB
/
AsyncHandlerAdd.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
/*
* JBoss, Home of Professional Open Source.
* Copyright 2010, Red Hat, Inc., and individual contributors
* as indicated by the @author tags. See the copyright.txt file in the
* distribution for a full listing of individual contributors.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This software 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
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.jboss.as.logging;
import org.jboss.as.controller.BasicOperationResult;
import org.jboss.as.controller.OperationFailedException;
import org.jboss.as.controller.OperationResult;
import org.jboss.as.controller.RuntimeTask;
import org.jboss.as.controller.RuntimeTaskContext;
import static org.jboss.as.controller.descriptions.ModelDescriptionConstants.OP;
import static org.jboss.as.controller.descriptions.ModelDescriptionConstants.OP_ADDR;
import static org.jboss.as.controller.descriptions.ModelDescriptionConstants.REMOVE;
import static org.jboss.as.logging.CommonAttributes.HANDLER_TYPE;
import static org.jboss.as.logging.CommonAttributes.LEVEL;
import static org.jboss.as.logging.CommonAttributes.OVERFLOW_ACTION;
import static org.jboss.as.logging.CommonAttributes.QUEUE_LENGTH;
import static org.jboss.as.logging.CommonAttributes.SUBHANDLERS;
import java.util.ArrayList;
import java.util.List;
import java.util.Locale;
import java.util.logging.Handler;
import java.util.logging.Level;
import org.jboss.as.controller.ModelAddOperationHandler;
import org.jboss.as.controller.OperationContext;
import org.jboss.as.controller.PathAddress;
import org.jboss.as.controller.ResultHandler;
import org.jboss.as.controller.descriptions.DescriptionProvider;
import org.jboss.dmr.ModelNode;
import org.jboss.msc.service.ServiceBuilder;
import org.jboss.msc.service.ServiceController;
import org.jboss.msc.service.ServiceTarget;
import org.jboss.msc.value.InjectedValue;
/**
* @author <a href="mailto:david.lloyd@redhat.com">David M. Lloyd</a>
* @author Emanuel Muckenhuber
*/
class AsyncHandlerAdd implements ModelAddOperationHandler, DescriptionProvider {
static final AsyncHandlerAdd INSTANCE = new AsyncHandlerAdd();
static final String OPERATION_NAME = "add-async-handler";
/** {@inheritDoc} */
@Override
public OperationResult execute(final OperationContext context, final ModelNode operation, final ResultHandler resultHandler) throws OperationFailedException {
final PathAddress address = PathAddress.pathAddress(operation.require(OP_ADDR));
final String name = address.getLastElement().getValue();
final ModelNode compensatingOperation = new ModelNode();
compensatingOperation.get(OP_ADDR).set(operation.require(OP_ADDR));
compensatingOperation.get(OP).set(REMOVE);
final String handlerType = operation.require(HANDLER_TYPE).asString();
final LoggerHandlerType type = LoggerHandlerType.valueOf(handlerType);
if(type != LoggerHandlerType.ASYNC_HANDLER) {
throw new OperationFailedException(new ModelNode().set("invalid operation for handler-type: " + type));
}
final ModelNode subModel = context.getSubModel();
subModel.get(HANDLER_TYPE).set(handlerType);
subModel.get(QUEUE_LENGTH).set(operation.get(QUEUE_LENGTH));
subModel.get(SUBHANDLERS).set(operation.get(SUBHANDLERS));
subModel.get(LEVEL).set(operation.get(LEVEL));
subModel.get(OVERFLOW_ACTION).set(operation.get(OVERFLOW_ACTION));
if (context.getRuntimeContext() != null) {
context.getRuntimeContext().setRuntimeTask(new RuntimeTask() {
public void execute(RuntimeTaskContext context) throws OperationFailedException {
final ServiceTarget serviceTarget = context.getServiceTarget();
try {
final AsyncHandlerService service = new AsyncHandlerService();
final ServiceBuilder<Handler> serviceBuilder = serviceTarget.addService(LogServices.handlerName(name), service);
final List<InjectedValue<Handler>> list = new ArrayList<InjectedValue<Handler>>();
for (final ModelNode handlerName : operation.get(SUBHANDLERS).asList()) {
final InjectedValue<Handler> injectedValue = new InjectedValue<Handler>();
serviceBuilder.addDependency(LogServices.handlerName(handlerName.asString()), Handler.class, injectedValue);
list.add(injectedValue);
}
service.addHandlers(list);
if (operation.hasDefined(QUEUE_LENGTH))
service.setQueueLength(operation.get(QUEUE_LENGTH).asInt());
service.setLevel(Level.parse(operation.get(LEVEL).asString()));
service.setOverflowAction(OverflowAction.valueOf(operation.get(OVERFLOW_ACTION).asString()));
serviceBuilder.setInitialMode(ServiceController.Mode.ACTIVE);
serviceBuilder.install();
} catch (Throwable t) {
throw new OperationFailedException(new ModelNode().set(t.getLocalizedMessage()));
}
resultHandler.handleResultComplete();
}
});
} else {
resultHandler.handleResultComplete();
}
return new BasicOperationResult(compensatingOperation);
}
/** {@inheritDoc} */
@Override
public ModelNode getModelDescription(Locale locale) {
return LoggingSubsystemProviders.getAsyncModelDescription(locale);
}
}