spring-cloud配置文件3

为了解决用户模块与业务模块的耦合，我想到用策略模式来来实现两套获取当前用户的方案。首先我要定义一个获取当前登陆用户的接口，如下：

package com.shangqiao56.tms; import com.shangqiao56.User; public interface UserHolder<T extends User> { T getUser(); }

接着是一个controller的基类，并用getCurrentUser表示获取当前用户

public class TmsUserController{ @Autowired private UserHolder<TmsUser> userHolder; protected TmsUser getCurrentUser(){ return userHolder.getUser(); } }

接着我添加一行配置。

shangqiao56: user-holder: ##测试时配置的固定用户 type: fix # fix|default-sys-user fix: user-name: wangzhiping user-id: 12 stat-id: 15 stat-name: 上海转运站

FixTmsUserHolder 表示从配置中读取一个静态用户。

public class FixTmsUserHolder implements UserHolder<TmsUser> { @Value("${shangqiao56.user-holder.fix.user-id}") private Long userId; @Value("${shangqiao56.user-holder.fix.user-name}") private String userName; @Value("${shangqiao56.user-holder.fix.stat-id}") private Long stationId; @Value("${shangqiao56.user-holder.fix.stat-name}") private String stationName; public TmsUser getUser() { TmsUser user = new TmsUser(); user.setId(userId); user.setName(userName); Station station = new Station(); station.setId(stationId); station.setName(stationName); user.setStation(station); return user; } }

DefaultTmsUserHolder 表示正式环境中从网关获取用户信息。

public class DefaultTmsUserHolder implements UserHolder<TmsUser> { private static final Logger log = LoggerFactory.getLogger(DefaultTmsUserHolder.class); @Autowired private HttpServletRequest request; @Override public TmsUser getUser() { TmsUser tmsUser = new TmsUser(); Long userId = Long.parseLong(request.getHeader("x-user-id")); Long statId = Long.parseLong(request.getHeader("x-station-id")); String userName = getStringFromHeader("x-user-name"); String statName = getStringFromHeader("x-station-name"); Station station = new Station(); station.setId(statId); station.setName(statName); tmsUser.setStation(station); tmsUser.setName(userName); tmsUser.setId(userId); return tmsUser; } private String getStringFromHeader(String key){ try { return URLDecoder.decode(request.getHeader(key),"utf-8"); } catch (UnsupportedEncodingException e) { log.error("UnsupportedEncodingException :{}",e); } return "";//todo } }

剩下的就是如何动态选择UserHolder.

public class TmsUserHolderConfig { @Bean @ConditionalOnProperty( name = {"shangqiao56.user-holder.type"}, havingValue = "fix" ) public UserHolder<TmsUser> fixUserHolder(){ return new FixTmsUserHolder(); } @Bean @ConditionalOnProperty( name = {"shangqiao56.user-holder.type"}, havingValue = "false", matchIfMissing = true) public UserHolder<TmsUser> defaultUserHolder(){ return new DefaultTmsUserHolder(); } }

这里主要要启到选择作用的是ConditionalOnProperty注解。 为了让idea能自动提示，添加 META-INF/spring-configuration-metadata.json文件:

{ "groups":[ { "sourceType": "com.shangqiao56.common.config.SysConfigurationProperty", "name": "shangqiao56.user-holder.fix", "type": "com.shangqiao56.common.config.SysConfigurationProperty" } ], "properties":[ { "name":"shangqiao56.user-holder.fix.user-name" }, { "name":"shangqiao56.user-holder.fix.user-id" }, { "name":"shangqiao56.user-holder.type", "description" : "指定获取用户的类型,开发环境可以指定为fix，生产环境可指定的default-sys-user,每个模块可以根据自己需求扩展，以方便开发" } ] }