1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17 package org.apache.logging.log4j.nosql.appender.couchdb;
18
19 import java.util.Map;
20 import java.util.concurrent.atomic.AtomicBoolean;
21
22 import org.apache.logging.log4j.core.appender.AppenderLoggingException;
23 import org.apache.logging.log4j.nosql.appender.DefaultNoSqlObject;
24 import org.apache.logging.log4j.nosql.appender.NoSqlConnection;
25 import org.apache.logging.log4j.nosql.appender.NoSqlObject;
26 import org.lightcouch.CouchDbClient;
27 import org.lightcouch.Response;
28
29
30
31
32 public final class CouchDbConnection implements NoSqlConnection<Map<String, Object>, DefaultNoSqlObject> {
33 private final CouchDbClient client;
34 private final AtomicBoolean closed = new AtomicBoolean(false);
35
36 public CouchDbConnection(final CouchDbClient client) {
37 this.client = client;
38 }
39
40 @Override
41 public DefaultNoSqlObject createObject() {
42 return new DefaultNoSqlObject();
43 }
44
45 @Override
46 public DefaultNoSqlObject[] createList(final int length) {
47 return new DefaultNoSqlObject[length];
48 }
49
50 @Override
51 public void insertObject(final NoSqlObject<Map<String, Object>> object) {
52 try {
53 final Response response = this.client.save(object.unwrap());
54 if (response.getError() != null && response.getError().length() > 0) {
55 throw new AppenderLoggingException("Failed to write log event to CouchDB due to error: " +
56 response.getError() + '.');
57 }
58 } catch (final Exception e) {
59 throw new AppenderLoggingException("Failed to write log event to CouchDB due to error: " + e.getMessage(),
60 e);
61 }
62 }
63
64 @Override
65 public void close() {
66 if (this.closed.compareAndSet(false, true)) {
67 this.client.shutdown();
68 }
69 }
70
71 @Override
72 public boolean isClosed() {
73 return this.closed.get();
74 }
75 }