-
Notifications
You must be signed in to change notification settings - Fork 1
/
jelh2.py
executable file
·208 lines (192 loc) · 10 KB
/
jelh2.py
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
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
# Any copyright is dedicated to the Public Domain.
# http://creativecommons.org/publicdomain/zero/1.0/
#Jython Elasticsearch Little Helper
#import org.apache.tomcat.jdbc.pool as dbpool
import java.net.InetAddress as InetAddress
import org.elasticsearch.common.transport.InetSocketTransportAddress as InetSocketTransportAddress
#import java.net.InetAddress as InetSocketTransportAddress
import org.elasticsearch.client.transport.TransportClient as TransportClient
#import org.elasticsearch.common.settings.ImmutableSettings as ImmutableSettings
#import com.google.common.collect.ImmutableSet as ImmutableSet
import org.elasticsearch.common.settings.Settings as Settings
import org.elasticsearch.client.transport.NoNodeAvailableException as NoNodeAvailableException
import org.elasticsearch.indices.IndexAlreadyExistsException as IndexAlreadyExistsException
import traceback
import time
import datetime
import com.xhaus.jyson.JysonCodec as json
import threading
import sys
import logging
logger = logging.getLogger(__name__)
class Elasticsearch():
def __init__(self, config):
if isinstance(config, basestring):
self.config = json.loads(config.decode('utf-8'))
else:
self.config = config
self.runtime = {}
#TODO: ugly ugly initialization all around, review, make it sane
clusterName = "elasticsearch"
host = "localhost"
port = 9300
if "host" in self.config:
host = self.config["host"]
else:
host = "localhost"
if "port" in self.config:
port = self.config["port"]
else:
port = "9300"
if "bulkActions" not in self.config:
self.config["bulkActions"] = 1000
if "bulkSize" not in self.config:
self.config["bulkSize"] = 107374182400
if "flushInterval" not in self.config:
self.config["flushInterval"] = 60000
if "concurrentRequests" not in self.config:
self.config["concurrentRequests"] = 1
if "actionRetryTimeout" not in self.config:
self.config["actionRetryTimeout"] = 5
if "type" not in self.config:
self.config["type"] = "logs"
if "indexPrefix" not in self.config:
self.config["indexPrefix"] = "sampleindex"
if "indexSuffix" not in self.config:
self.config["indexSuffix"] = "-%Y.%m.%d"
logger.debug("Initializing elasticsearch output %s: %s", self.config["indexPrefix"], json.dumps("self.config"))
self.config["settings"] = Settings.settingsBuilder();
if "options" not in self.config:
self.config["options"] = {}
if "cluster" in self.config:
self.config["options"]["cluster.name"] = self.config["cluster"]
else:
self.config["options"]["cluster.name"] = "elasticsearch"
else:
if "cluster.name" not in self.config["options"]:
if "cluster" in self.config:
self.config["options"]["cluster.name"] = self.config["cluster"]
else:
self.config["options"]["cluster.name"] = "elasticsearch"
for setting in self.config["options"]:
value = self.config["options"][setting]
logger.info("Setting Elasticsearch options: %s = %s", setting, value)
self.config["settings"].put(setting, value)
self.config["settings"].build()
self.runtime["client"] = TransportClient.builder().settings(self.config["settings"]).build()
if "host" in self.config:
#self.runtime["client"].addTransportAddress(InetSocketTransportAddress(host, port))
self.runtime["client"].addTransportAddress(InetSocketTransportAddress(InetAddress.getByName(host), port))
if "hosts" in self.config:
for hostport in self.config["hosts"]:
host, port = hostport.split(":")
logger.info("Setting Elasticsearch host: %s = %s", host, port)
#self.runtime["client"].addTransportAddress(InetSocketTransportAddress(str(host), int(port)))
self.runtime["client"].addTransportAddress(InetSocketTransportAddress(InetAddress.getByName(str(host)), int(port)))
self.readyBulk()
self.runtime["indices"] = {}
def createIndex(self, indexName):
if indexName not in self.runtime["indices"]:
if self.runtime["client"].admin().indices().prepareExists(indexName).execute().actionGet().exists:
logger.debug("Index \"%s\" already exists", indexName)
self.runtime["indices"][indexName] = time.time()
return False
else:
logger.info("Creating index %s", indexName)
if "index_settings" in self.config:
self.config["indexSettings"] = self.config["index_settings"]
if "type_mapping" in self.config:
self.config["typeMapping"] = self.config["type_mapping"]
try:
if "indexSettings" in self.config:
settingsJsonStr = json.dumps(self.config["indexSettings"])
logger.info("Index settings: %s", settingsJsonStr)
self.runtime["client"].admin().indices().prepareCreate(indexName).setSettings(settingsJsonStr).execute().actionGet()
else:
self.runtime["client"].admin().indices().prepareCreate(indexName).execute().actionGet()
except IndexAlreadyExistsException, ex:
logger.warning(ex)
logger.warning("Index %s already exists, this should be harmless", indexName)
if "typeMapping" in self.config:
mappingJsonStr = json.dumps(self.config["typeMapping"])
logger.info("Setting mapping for %s/%s - %s", indexName, self.config["type"], mappingJsonStr)
self.runtime["client"].admin().indices().preparePutMapping().setIndices(indexName).setType(self.config["type"]).setSource(mappingJsonStr).execute().actionGet()
self.runtime["indices"][indexName] = time.time()
logger.debug("Created index: \"%s\"", indexName)
return True
logger.debug("Index already initialized: \"%s\"", indexName)
return False
def writeDocument(self, data, force):
bulkRequest = self.runtime["bulkRequest"]
client = self.runtime["client"]
if data != None:
indexName = self.getIndexName(data)
if "_id" in data:
_id = data["_id"]
del data["_id"]
bulkRequest.add(client.prepareIndex(indexName, self.config["type"], _id).setSource(json.dumps(data)))
elif "_ID" in data:
_ID = data["_ID"]
del data["_ID"]
bulkRequest.add(client.prepareIndex(indexName, self.config["type"], _ID).setSource(json.dumps(data)))
else:
bulkRequest.add(client.prepareIndex(indexName, self.config["type"]).setSource(json.dumps(data)))
self.runtime["requestsPending"] = self.runtime["requestsPending"] + 1
#TIME TO FLUSH
if (self.runtime["requestsPending"] > 0) and ((self.runtime["requestsPending"] >= self.config["bulkActions"]) or (force == True)):
logger.info("Flushing %d records", self.runtime["requestsPending"])
#TODO: handle failure: org.elasticsearch.client.transport.NoNodeAvailableException
#TODO: use JodaTime instead of jython's datetime/time
bulkReady = False
while not bulkReady:
try:
bulkResponse = bulkRequest.execute().actionGet();
bulkReady = True
except NoNodeAvailableException, ex:
logger.error(ex)
logger.warning("Bad bulk response, sleeping %d seconds before retrying, execution paused", self.config["actionRetryTimeout"])
time.sleep(self.config["actionRetryTimeout"]);
raise
if bulkResponse.hasFailures():
logger.warning("Failures indexing!")
logger.warning(bulkResponse.buildFailureMessage())
self.readyBulk()
return True
def readyBulk(self):
self.runtime["bulkRequest"] = self.runtime["client"].prepareBulk()
self.runtime["requestsPending"] = 0
def getIndexName(self):
return self.config["indexPrefix"]
def getIndexName(self, data):
indexName = self.config["indexPrefix"]
if "indexSuffix" in self.config:
indexSuffix = self.config["indexSuffix"]
if indexSuffix in data:
indexSuffixStr = ""
logger.debug("Index suffix is a fieldname: %s, %s", indexSuffix, data[indexSuffix])
#TODO: deal with the timezone
try:
dateobj = datetime.datetime.strptime(data[indexSuffix], "%Y-%m-%d")
indexSuffixStr = dateobj.strftime("%Y.%m.%d")
except Exception, e:
indexSuffixStr = data[indexSuffix].lower()
indexName = indexName + "-" + indexSuffixStr
else:
logger.debug("Index suffix is will be treated as datetime format: %s", indexSuffix)
indexName = indexName + datetime.datetime.utcnow().strftime(indexSuffix)
logger.debug("Going for index creation: %s", indexName)
#TODO: handle other failures
indexReady = False
while not indexReady:
try:
self.createIndex(indexName)
indexReady = True
except NoNodeAvailableException, ex:
logger.error(ex)
logger.warning("Failed to initialize index, sleeping a %d seconds before retrying, execution paused", self.config["actionRetryTimeout"])
time.sleep(self.config["actionRetryTimeout"])
return indexName
def flush(self):
return self.writeDocument(None, True)
def cleanup(self):
return False