From 96bb84e2a4a20993939e6bf08533f362c2bcc2ea Mon Sep 17 00:00:00 2001
From: Dave Syer <dsyer@pivotal.io>
Date: Wed, 15 Oct 2014 16:58:48 +0100
Subject: [PATCH] Start work on @EnableStickyFilter

---
 .../cloudfoundry/web/EnableStickyFilter.java  | 36 ++++++++++
 .../web/StickyFilterConfiguration.java        | 67 +++++++++++++++++++
 2 files changed, 103 insertions(+)
 create mode 100644 src/main/java/org/springframework/cloud/cloudfoundry/web/EnableStickyFilter.java
 create mode 100644 src/main/java/org/springframework/cloud/cloudfoundry/web/StickyFilterConfiguration.java

diff --git a/src/main/java/org/springframework/cloud/cloudfoundry/web/EnableStickyFilter.java b/src/main/java/org/springframework/cloud/cloudfoundry/web/EnableStickyFilter.java
new file mode 100644
index 0000000..8962629
--- /dev/null
+++ b/src/main/java/org/springframework/cloud/cloudfoundry/web/EnableStickyFilter.java
@@ -0,0 +1,36 @@
+/*
+ * Copyright 2013-2014 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.springframework.cloud.cloudfoundry.web;
+
+import java.lang.annotation.Documented;
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+import org.springframework.context.annotation.Import;
+
+/**
+ * @author Dave Syer
+ *
+ */
+@Target(ElementType.TYPE)
+@Retention(RetentionPolicy.RUNTIME)
+@Documented
+@Import(StickyFilterConfiguration.class)
+public @interface EnableStickyFilter {
+
+}
diff --git a/src/main/java/org/springframework/cloud/cloudfoundry/web/StickyFilterConfiguration.java b/src/main/java/org/springframework/cloud/cloudfoundry/web/StickyFilterConfiguration.java
new file mode 100644
index 0000000..10f5d53
--- /dev/null
+++ b/src/main/java/org/springframework/cloud/cloudfoundry/web/StickyFilterConfiguration.java
@@ -0,0 +1,67 @@
+/*
+ * Copyright 2013-2014 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.springframework.cloud.cloudfoundry.web;
+
+import java.io.IOException;
+import java.util.UUID;
+
+import javax.servlet.FilterChain;
+import javax.servlet.ServletException;
+import javax.servlet.http.Cookie;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.context.embedded.FilterRegistrationBean;
+import org.springframework.cloud.netflix.eureka.EurekaInstanceConfigBean;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.core.Ordered;
+import org.springframework.web.filter.OncePerRequestFilter;
+
+/**
+ * @author Dave Syer
+ *
+ */
+@Configuration
+public class StickyFilterConfiguration {
+	
+	private String cookie = UUID.randomUUID().toString();
+	
+	@Autowired
+	public void init(EurekaInstanceConfigBean eurekaInstance) {
+		eurekaInstance.getMetadataMap().put("cookie", cookie);
+	}
+
+	@Bean
+	public FilterRegistrationBean stickyCloudFoundryFilter() {
+		FilterRegistrationBean filter = new FilterRegistrationBean();
+		filter.setOrder(Ordered.LOWEST_PRECEDENCE);
+		filter.setFilter(new OncePerRequestFilter() {
+			@Override
+			protected void doFilterInternal(HttpServletRequest request,
+					HttpServletResponse response, FilterChain filterChain)
+					throws ServletException, IOException {
+				if (!response.containsHeader("Set-Cookie")) {
+					response.addCookie(new Cookie("JSESSIONID", cookie));
+				}
+				filterChain.doFilter(request, response);
+			}
+		});
+		return filter;
+	}
+
+}